Пример #1
0
// GetDefinitionYandex get definitions from yandex
func GetDefinitionYandex(word string) (string, error) {
	params := url.Values{}

	params.Set("format", "plain")
	params.Set("lang", "en-vi")
	params.Set("text", word)
	params.Set("key", YANDEX_TRANSLATOR_APIKEY)

	resp, err := http.PostForm(YURL, params)
	if err != nil {
		util.LogError("yandex", 0, err)
		return "", err
	}

	defer resp.Body.Close()

	meJSON, err := ioutil.ReadAll(resp.Body)
	if err != nil {
		util.LogError("yandex", 1, err)
		return "", err
	}

	var result Result

	err = json.Unmarshal(meJSON, &result)
	if err != nil {
		util.LogError("yandex", 2, err)
		return "", err
	}

	definition := strings.Join(result.Text, " ")
	if definition == word {
		return "", errors.New("sameWord")
	}
	return "-> " + definition, nil
}
Пример #2
0
func main() {

	//init redis
	redisClient = redis.NewClient(&redis.Options{
		Addr:     "localhost:6379",
		Password: "",
		DB:       1,
	})

	dict.PrepareIndex()

	bot, err := telebot.NewBot(BOT_TOKEN)
	if err != nil {
		return
	}

	util.LogInit()

	messages := make(chan telebot.Message)
	bot.Listen(messages, 2*time.Second)

	for message := range messages {
		sender_id := message.Chat.ID

		// analytics purpose
		k := "u:" + strconv.Itoa(sender_id)
		redisClient.ZIncrBy("sorted_users", 1, k)

		definition := GetDefinition(message.Text, message.Chat)

		err = bot.SendMessage(message.Chat, definition, &telebot.SendOptions{ParseMode: "Markdown"})
		if err != nil {
			util.LogError("sendMessage", sender_id, err)
		}
	}
}
Пример #3
0
// GetDefinitionFromDb defines a function lookup from db/map
func GetDefinitionFromDb(word string, sender telebot.Chat) string {
	word = strings.Trim(strings.ToLower(word), " ")
	sender_id := sender.ID

	// analytics purpose
	w := "w:" + word
	redisClient.ZIncrBy("sorted_words", 1, w)

	// should get from redis
	desc, err := redisClient.Get(word).Result()
	if err == nil {
		return desc
	} else if err == redis.Nil {
		result, err := dict.Check(word)
		if err != nil {
			if r.MatchString(word) {
				def, err := GetDefinitionYandex(word)
				if err != nil {
					util.LogError("yandex", sender_id, err.Error())
					return fmt.Sprintf(APOLOGY_MSG, GetName(sender), word, GetName(sender), word)
				} else {
					SetWord(word, def)
					return def
				}
			} else {
				return fmt.Sprintf(APOLOGY_MSG, GetName(sender), word, GetName(sender), word)
			}

		} else {
			SetWord(word, result)
			return result
		}
	}
	log.Errorln(err.Error())
	return ""
}