71 lines
1.4 KiB
Go
71 lines
1.4 KiB
Go
package main
|
|
|
|
import (
|
|
"log"
|
|
"os"
|
|
"strings"
|
|
"./trans"
|
|
"./identify"
|
|
"github.com/deanishe/awgo"
|
|
"regexp"
|
|
)
|
|
|
|
var wf *aw.Workflow
|
|
|
|
func init() {
|
|
wf = aw.New()
|
|
}
|
|
|
|
func main() {
|
|
wf.Run(run)
|
|
}
|
|
|
|
func run() {
|
|
word, fromLang, toLang := parseProgramArgs(os.Args)
|
|
if fromLang == "" && toLang == "" {
|
|
fromLang, toLang = decideWhatLanguagesToUse(word)
|
|
}
|
|
|
|
translations := trans.New(word, fromLang, toLang).Translate()
|
|
for _, translation := range translations {
|
|
wf.NewItem(translation).Arg(translation).Valid(true)
|
|
}
|
|
wf.WarnEmpty("No results", "Try a different query?")
|
|
wf.SendFeedback()
|
|
}
|
|
|
|
func parseProgramArgs(args []string) (string, string, string) {
|
|
if len(args) < 2 {
|
|
log.Fatal("Not enough arguments!")
|
|
}
|
|
word := strings.Join(args[1:], " ")
|
|
fromLang, toLang := "", ""
|
|
|
|
if len(args) > 2 {
|
|
if isTransLanguageSpecifier(args[1]) {
|
|
languages := strings.Split(os.Args[1], ":")
|
|
fromLang, toLang = languages[0], languages[1]
|
|
word = strings.Join(os.Args[2:], " ")
|
|
}
|
|
}
|
|
|
|
return word, fromLang, toLang
|
|
}
|
|
|
|
func isTransLanguageSpecifier(arg string) bool {
|
|
ok, _ := regexp.MatchString("(\\w{2})?:(\\w{2})?", os.Args[1])
|
|
return ok
|
|
}
|
|
|
|
func decideWhatLanguagesToUse(word string) (string, string) {
|
|
fromLang := identify.New(word).Identify()
|
|
toLang := map[string]string{
|
|
"en": "hu",
|
|
"hu": "en",
|
|
}[fromLang]
|
|
if toLang == "" {
|
|
toLang = "en"
|
|
}
|
|
return fromLang, toLang
|
|
}
|