feat partOfSpeech

internal/translate/translator.go

@@ -14,7 +14,7 @@ import (
 )
 
 const (
-	textURL  = "https://translate.googleapis.com/translate_a/single?client=gtx&sl=%s&tl=%s&dt=t&dt=bd&q=%s"
+	textURL  = "https://translate.googleapis.com/translate_a/single?client=gtx&dt=t&dt=bd&sl=%s&tl=%s&q=%s"
 	soundURL = "https://translate.google.com.vn/translate_tts?ie=UTF-8&q=%s&tl=%s&client=tw-ob"
 )
 
@@ -30,7 +30,7 @@ func NewTranslator() *Translator {
 	}
 }
 
-func (t *Translator) Translate(message string) (translated string, err error) {
+func (t *Translator) Translate(message string) (translation string, partOfSpeech string, err error) {
 	var data []interface{}
 
 	urlStr := fmt.Sprintf(
@@ -41,50 +41,52 @@ func (t *Translator) Translate(message string) (translated string, err error) {
 	)
 	res, err := http.Get(urlStr)
 	if err != nil {
-		return "", err
+		return "", "", err
 	}
 	body, err := ioutil.ReadAll(res.Body)
 	if err != nil {
-		return "", err
+		return "", "", err
 	}
 
 	if err = json.Unmarshal(body, &data); err != nil {
-		return "", err
+		return "", "", err
 	}
 
 	if len(data) > 0 {
-		if data[1] == nil {
+		// translation
-			result := data[0]
+		for _, lines := range data[0].([]interface{}) {
-			for _, lines := range result.([]interface{}) {
 			translatedLine := lines.([]interface{})[0]
-				translated += fmt.Sprintf("%v", translatedLine)
+			translation += fmt.Sprintf("%v", translatedLine)
 		}
-			return translated, nil
+
-		} else {
+		// part of speech
-			result := data[1]
+		if data[1] != nil {
-			for _, kinds := range result.([]interface{}) {
+			for _, parts := range data[1].([]interface{}) {
-				translated += fmt.Sprintf("%v\n", kinds.([]interface{})[0])
+				// part of speech
-				for _, words := range kinds.([]interface{})[2].([]interface{}) {
+				partOfSpeech += fmt.Sprintf("[%v]\n", parts.([]interface{})[0])
-					translated += fmt.Sprintf(
+				for _, words := range parts.([]interface{})[2].([]interface{}) {
-						"\t%v:",
+					// dst lang
-						words.([]interface{})[0])
+					partOfSpeech += fmt.Sprintf(
+						"\t%v:", words.([]interface{})[0])
+					// src lang
 					firstWord := true
 					for _, word := range words.([]interface{})[1].([]interface{}) {
 						if firstWord {
-							translated += fmt.Sprintf(" %v", word)
+							partOfSpeech += fmt.Sprintf(" %v", word)
 							firstWord = false
 						} else {
-							translated += fmt.Sprintf(", %v", word)
+							partOfSpeech += fmt.Sprintf(", %v", word)
 						}
 					}
-					translated += "\n"
+					partOfSpeech += "\n"
+				}
+				partOfSpeech += "\n"
 			}
 		}
-			return translated, nil
+		return translation, partOfSpeech, nil
-		}
 	}
 
-	return "", errors.New("Translation not found")
+	return "", "", errors.New("Translation not found")
 }
 
 func (t *Translator) PlaySound(lang string, message string) error {

ui.go

@@ -397,11 +397,12 @@ func translatePageHandler(event *tcell.EventKey) *tcell.EventKey {
 		message := srcInput.GetText()
 		if len(message) > 0 {
 			// Only translate when message exist
-			result, err := translator.Translate(message)
+			translation, partOfSpeech, err := translator.Translate(message)
 			if err != nil {
 				dstOutput.SetText(err.Error())
 			} else {
-				dstOutput.SetText(result)
+				dstOutput.SetText(translation)
+				partOutput.SetText(partOfSpeech, false)
 			}
 		}
 	case tcell.KeyCtrlQ: