949bcf3208
Signed-off-by: Izuru Yakumo <yakumo.izuru@chaotic.ninja> git-svn-id: file:///srv/svn/repo/suwako/trunk@29 0b558ee1-521d-8b46-a41b-40029c97c055
97 lines
2.0 KiB
Go
97 lines
2.0 KiB
Go
// $TheSupernovaDuo: suwako,v 1.5.4 2024/01/14 21:21:50 yakumo_izuru Exp $
|
|
// Command line client for SimplyTranslate, a privacy friendly frontend to other translation engines
|
|
package main
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"log"
|
|
"net/http"
|
|
"net/url"
|
|
"os"
|
|
|
|
"github.com/integrii/flaggy"
|
|
"gopkg.in/ini.v1"
|
|
"marisa.chaotic.ninja/suwako"
|
|
)
|
|
|
|
var conf struct {
|
|
engine string
|
|
instance string
|
|
}
|
|
|
|
var (
|
|
input string
|
|
source string = "auto"
|
|
target string
|
|
)
|
|
|
|
type Translate struct {
|
|
Output string `json:"translated_text"`
|
|
}
|
|
|
|
func errCheck(err error) {
|
|
if err != nil {
|
|
log.Println("Something happened :(")
|
|
log.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func iniLoad(file string) error {
|
|
cfg, err := ini.Load(file)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
conf.engine = cfg.Section("simplytranslate").Key("engine").String()
|
|
conf.instance = cfg.Section("simplytranslate").Key("instance").String()
|
|
|
|
return nil
|
|
}
|
|
|
|
func flagParse() {
|
|
flaggy.SetName("suwako")
|
|
flaggy.SetDescription("Command line client for SimplyTranslate")
|
|
flaggy.SetVersion(suwako.FullVersion())
|
|
|
|
flaggy.String(&source, "f", "from", "Source language")
|
|
flaggy.String(&target, "t", "to", "Target language")
|
|
flaggy.AddPositionalValue(&input, "input", 1, true, "Text to translate")
|
|
|
|
flaggy.Parse()
|
|
}
|
|
|
|
func main() {
|
|
// Flag parsing
|
|
flagParse()
|
|
|
|
// Load configuration file
|
|
home, err := os.UserHomeDir()
|
|
errCheck(err)
|
|
cfgfile := home + "/.suwako/suwako.conf"
|
|
iniLoad(cfgfile)
|
|
|
|
// Verify command-line inputs
|
|
if len(target) == 0 {
|
|
log.Fatal("No target language")
|
|
}
|
|
|
|
// Map variable to struct
|
|
var translate Translate
|
|
|
|
// Encode input just in case
|
|
var encInput = url.PathEscape(input)
|
|
|
|
// Construct the final path to query
|
|
var queryURL = conf.instance + "?engine=" + conf.engine + "&from=" + source + "&to=" + target + "&text=" + encInput
|
|
|
|
// Shoot danmaku to path
|
|
resp, err := http.Get(queryURL)
|
|
errCheck(err)
|
|
defer resp.Body.Close()
|
|
|
|
// Decode JSON response, discard everything else, print to standard output
|
|
_ = json.NewDecoder(resp.Body).Decode(&translate)
|
|
errCheck(err)
|
|
fmt.Printf("%v\n",translate.Output)
|
|
}
|