mirror of
https://codeberg.org/aryak/mozhi
synced 2024-09-19 06:35:34 +05:30
107 lines
2.8 KiB
Go
107 lines
2.8 KiB
Go
package cmd
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"strings"
|
|
|
|
"codeberg.org/aryak/libmozhi"
|
|
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
var (
|
|
engine string
|
|
engines string
|
|
query string
|
|
source string
|
|
dest string
|
|
rawjson bool
|
|
)
|
|
|
|
func printEngineResult(result libmozhi.LangOut, printPlaceHolderAndEngineName bool) {
|
|
if printPlaceHolderAndEngineName {
|
|
fmt.Println("-----------------------------------")
|
|
fmt.Println("Engine: " + result.Engine)
|
|
}
|
|
|
|
fmt.Println("Translated Text: " + result.OutputText)
|
|
if source == "auto" {
|
|
fmt.Println("Detected Language: " + result.AutoDetect)
|
|
}
|
|
fmt.Println("Source Language: " + result.SourceLang)
|
|
fmt.Println("Target Language: " + result.TargetLang)
|
|
}
|
|
|
|
func printRaw(data interface{}) {
|
|
j, err := json.Marshal(data)
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
} else {
|
|
fmt.Println(string(j))
|
|
}
|
|
}
|
|
|
|
var translateCmd = &cobra.Command{
|
|
Use: "translate",
|
|
Short: "Translate.",
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
if engine == "all" {
|
|
data := libmozhi.TranslateAll(dest, source, query)
|
|
if rawjson {
|
|
printRaw(data)
|
|
} else {
|
|
for _, result := range data {
|
|
printEngineResult(result, true)
|
|
}
|
|
fmt.Println("-----------------------------------")
|
|
}
|
|
} else if engine == "some" {
|
|
if engines != "" {
|
|
data, err := libmozhi.TranslateSome(strings.Split(engines, ","), dest, source, query)
|
|
if rawjson {
|
|
printRaw(data)
|
|
} else {
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
} else {
|
|
for _, result := range data {
|
|
printEngineResult(result, true)
|
|
}
|
|
fmt.Println("-----------------------------------")
|
|
}
|
|
}
|
|
} else {
|
|
fmt.Println("Please add the --engines / -E argument when using the engine `some`")
|
|
}
|
|
} else {
|
|
data, err := libmozhi.Translate(engine, dest, source, query)
|
|
if rawjson {
|
|
printRaw(data)
|
|
} else {
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
} else {
|
|
printEngineResult(data, false)
|
|
}
|
|
}
|
|
}
|
|
},
|
|
}
|
|
|
|
func init() {
|
|
rootCmd.AddCommand(translateCmd)
|
|
translateCmd.Flags().SortFlags = false
|
|
|
|
translateCmd.Flags().StringVarP(&engine, "engine", "e", "", "[all|some|google|libre|reverso|deepl|yandex|mymemory|duckduckgo]")
|
|
translateCmd.Flags().StringVarP(&engines, "engines", "E", "", "Engines to select. This flag is to be used with the `some` engine only")
|
|
translateCmd.Flags().StringVarP(&source, "source", "s", "", "Source language. Use langlist command to get code for your language")
|
|
translateCmd.Flags().StringVarP(&dest, "dest", "t", "", "Target language. Use langlist command to get code for your language")
|
|
translateCmd.Flags().StringVarP(&query, "query", "q", "", "Text to be translated")
|
|
translateCmd.Flags().BoolVarP(&rawjson, "raw", "r", false, "Return output as json")
|
|
|
|
translateCmd.MarkFlagRequired("source")
|
|
translateCmd.MarkFlagRequired("dest")
|
|
translateCmd.MarkFlagRequired("query")
|
|
}
|