ALFRED/commands/gender.go

/*
* @Author: Bartuccio Antoine
* @Date:   2018-07-24 14:55:33
* @Last Modified by:   Bartuccio Antoine
* @Last Modified time: 2019-01-04 10:38:32
 */

package commands

import (
	"strings"

	"git.klmp200.net/ALFRED/ALFRED/shared"

	tb "gopkg.in/tucnak/telebot.v2"
)

// SetGender register your gender on ALFRED
func SetGender(m *tb.Message) {
	if m.Sender.Username == "" {
		shared.Bot.Send(m.Chat, "Il faut avoir enregistré un username pour pouvoir utiliser cette fonction")
		return
	}
	split := cleanGender(strings.Split(m.Text, " ")[1:])
	if len(split) == 0 {
		shared.Bot.Send(m.Chat, "Désolé, mais je n'ai pas compris.")
		return
	}
	data := strings.Join(split, " ")
	if data == "" {
		shared.Bot.Send(m.Chat, "Attention, votre genre est vide. Ce n'est pas enregistrable.")
		return
	}
	shared.Users.Set(m.Sender.Username, "gender", data)
	shared.Bot.Send(m.Chat, "Votre genre est enregistré, je vous considère maintenant comme « "+data+" ».")
}

func Gender(m *tb.Message) {
	split := strings.Split(m.Text, " ")
	if len(split) > 1 {
		// Username asked
		username := split[1]
		username = strings.Replace(username, "@", "", 1)
		data, exists := shared.Users.Get(username, "gender")
		if !exists {
			shared.Bot.Send(m.Chat, username+" n'est pas un utilisateur existant ou n'a pas renseigné son genre.")
			return
		}
		shared.Bot.Send(m.Chat, "L'utilisateur "+username+" a pour genre « "+data+" ».")
	} else {
		data, exists := shared.Users.Get(m.Sender.Username, "gender")
		if !exists {
			shared.Bot.Send(m.Chat, "Vous n'avez pas enregistré votre genre, je ne voudrais pas l'assumer.")
		} else {
			shared.Bot.Send(m.Chat, data)
		}
	}
}

func cleanGender(slice []string) []string {
	for i := range slice {
		slice[i] = strings.Replace(slice[i], "\\", "", -1)
		slice[i] = strings.Replace(slice[i], "@", "", -1)
		clean := false
		for !clean {
			clean = true
			if strings.HasPrefix(slice[i], "/") {
				slice[i] = strings.Replace(slice[i], "/", "", 1)
				clean = false
			}
		}
	}
	return slice
}