package main

import (
	"encoding/json"
	"github.com/chai2010/gettext-go"
	"github.com/fsnotify/fsnotify"
	"github.com/spf13/viper"
	tb "gopkg.in/tucnak/telebot.v2"
	"io/ioutil"
	"os"
	"path/filepath"
	"strconv"
	"strings"
	"time"
)

const (
	bLogBasePath string = "./log/"
)

var (
	UserStatus    map[int64]int
	UserCid       map[int64]string
	UserCSecret   map[int64]string
	ErrorTimes    map[string]int //错误次数
	BindMaxNum    int
	ErrMaxTimes   int
	notice        string
	admin         []int64
	bStartContent string
	bHelpContent  string
)

const (
	USNone = iota
	USBind1
	USBind2
)

func init() {
	//read config
	viper.SetConfigName("config")
	viper.AddConfigPath(".")
	err := viper.ReadInConfig()
	CheckErr(err)

	viper.SetDefault("errlimit", 5)
	viper.SetDefault("bindmax", 5)
	viper.SetDefault("bindmax", 5)
	viper.SetDefault("lang", "zh_CN")

	//set language
	gettext.BindLocale(gettext.New("resources", "???", jsonData))
	lang := strings.Trim(viper.GetString("lang"), "")
	gettext.SetLanguage(lang)

	bStartContent = gettext.Gettext("welcome")
	bHelpContent = gettext.Gettext("helpContent")

	BindMaxNum = viper.GetInt("bindmax")
	ErrMaxTimes = viper.GetInt("errlimit")
	notice = viper.GetString("notice")
	admin = GetAdmin()

	viper.WatchConfig()
	viper.OnConfigChange(func(e fsnotify.Event) {
		BindMaxNum = viper.GetInt("bindmax")
		ErrMaxTimes = viper.GetInt("errlimit")
		notice = viper.GetString("notice")
		admin = GetAdmin()
	})

	UserStatus = make(map[int64]int)
	UserCid = make(map[int64]string)
	UserCSecret = make(map[int64]string)
	ErrorTimes = make(map[string]int)
}

func bStart(m *tb.Message) {
	bot.Send(m.Sender, bStartContent)
	bHelp(m)
}

func bMy(m *tb.Message) {
	logger.Println(strconv.FormatInt(m.Chat.ID, 10) + " Start Manager Users")
	data := QueryDataByTG(m.Chat.ID)
	var inlineKeys [][]tb.InlineButton
	for _, u := range data {
		inlineBtn := tb.InlineButton{
			Unique: "my" + u.msId,
			Text:   u.alias,
			Data:   u.msId,
		}
		bot.Handle(&inlineBtn, bMyInlineBtn)
		inlineKeys = append(inlineKeys, []tb.InlineButton{inlineBtn})
	}
	bot.Send(m.Chat, gettext.Gettext("chooseAnAccount")+strconv.Itoa(GetBindNum(m.Chat.ID))+"/"+strconv.Itoa(BindMaxNum), &tb.ReplyMarkup{InlineKeyboard: inlineKeys})
}
func bMyInlineBtn(c *tb.Callback) {
	logger.Println(strconv.FormatInt(c.Message.Chat.ID, 10) + " Get User Info")
	r := QueryDataByMS(c.Data)
	u := r[0]
	bot.Send(c.Message.Chat, gettext.Gettext("accountInformation")+u.alias+"\nMS_ID(MD5): "+u.msId+"\nclient_id: "+u.clientId+"\nclient_secret: "+u.clientSecret+gettext.Gettext("updateTime")+time.Unix(u.uptime, 0).Format("2006-01-02 15:04:05"))
	bot.Respond(c)
}

func bBind1(m *tb.Message) {
	logger.Println(strconv.FormatInt(m.Chat.ID, 10) + " Start Bind")
	logger.Println("ReApp: " + strconv.FormatInt(m.Chat.ID, 10))
	bot.Send(m.Chat, gettext.Gettext("register")+"("+MSGetReAppUrl()+")", tb.ModeMarkdown)
	_, err := bot.Send(m.Chat, gettext.Gettext("bind1Reply"), &tb.ReplyMarkup{ForceReply: true})
	if err != nil {
		logger.Println(err)
		return
	}
	UserStatus[m.Chat.ID] = USBind1
	UserCid[m.Chat.ID] = m.Text
}
func bBind2(m *tb.Message) {
	logger.Println(strconv.FormatInt(m.Chat.ID, 10) + " Start Bind2")
	logger.Println("Auth: " + strconv.FormatInt(m.Chat.ID, 10))
	tmp := strings.Split(m.Text, " ")
	if len(tmp) != 2 {
		logger.Printf("%d Bind error:Wrong Bind Format\n", m.Chat.ID)
		bot.Send(m.Chat, gettext.Gettext("formatError"))
		return
	}
	logger.Println("client_id: " + tmp[0] + " client_secret: " + tmp[1])
	cid := tmp[0]
	cse := tmp[1]
	bot.Send(m.Chat, gettext.Gettext("signIn")+"("+MSGetAuthUrl(cid)+")", tb.ModeMarkdown)
	_, err := bot.Send(m.Chat, gettext.Gettext("bind2Reply"), &tb.ReplyMarkup{ForceReply: true})
	if err != nil {
		logger.Println(err)
		return
	}
	UserStatus[m.Chat.ID] = USBind2
	UserCid[m.Chat.ID] = cid
	UserCSecret[m.Chat.ID] = cse
}

func bUnBind(m *tb.Message) {
	logger.Println(strconv.FormatInt(m.Chat.ID, 10) + " Start Unbind")
	data := QueryDataByTG(m.Chat.ID)
	var inlineKeys [][]tb.InlineButton
	for _, u := range data {
		inlineBtn := tb.InlineButton{
			Unique: "unbind" + u.msId,
			Text:   u.alias,
			Data:   u.msId,
		}
		bot.Handle(&inlineBtn, bUnBindInlineBtn)
		inlineKeys = append(inlineKeys, []tb.InlineButton{inlineBtn})
	}
	bot.Send(m.Chat, gettext.Gettext("unBind")+strconv.Itoa(GetBindNum(m.Chat.ID))+"/"+strconv.Itoa(BindMaxNum), &tb.ReplyMarkup{InlineKeyboard: inlineKeys})
}
func bUnBindInlineBtn(c *tb.Callback) {
	logger.Println(strconv.FormatInt(c.Message.Chat.ID, 10) + " Unbind: " + c.Data)
	r := QueryDataByMS(c.Data)
	u := r[0]
	if ok, _ := DelData(u.msId); !ok {
		logger.Println(u.msId + " UnBind ERROR")
		bot.Send(c.Message.Chat, gettext.Gettext("unBindError"))
		return
	}
	logger.Println(u.msId + " UnBind Success")
	bot.Send(c.Message.Chat, gettext.Gettext("unBindSuccess"))
	bot.Respond(c)
}
func bExport(m *tb.Message) {
	logger.Println(strconv.FormatInt(m.Chat.ID, 10) + " Start Export")
	type MsMiniData struct {
		Alias        string
		ClientId     string
		ClientSecret string
		RefreshToken string
		Other        string
	}
	var MsMini []MsMiniData
	data := QueryDataByTG(m.Chat.ID)
	if len(data) == 0 {
		bot.Send(m.Chat, gettext.Gettext("unbound"))
		return
	}
	for _, u := range data {
		var ms MsMiniData
		ms.RefreshToken = u.refreshToken
		ms.Alias = u.alias
		ms.ClientId = u.clientId
		ms.ClientSecret = u.clientSecret
		ms.Other = u.other
		MsMini = append(MsMini, ms)
	}
	//MarshalIndent是为json+美化,/t表缩进
	export, err := json.MarshalIndent(MsMini, "", "\t")
	if err != nil {
		logger.Println(err)
		bot.Send(m.Chat, gettext.Gettext("json")+err.Error())
		return
	}
	//fmt.Println(string(export))
	fileName := "./" + strconv.FormatInt(m.Chat.ID, 10) + "_export_tmp.json"
	if err = ioutil.WriteFile(fileName, export, 0644); err != nil {
		logger.Println(err)
		bot.Send(m.Chat, gettext.Gettext("temporary")+err.Error())
		return
	}
	exportFile := &tb.Document{File: tb.FromDisk(fileName), FileName: strconv.FormatInt(m.Chat.ID, 10) + ".json", MIME: "text/plain"}
	_, err = bot.Send(m.Chat, exportFile)
	if err != nil {
		logger.Println(err)
		return
	}
	//不遗留本地文件
	if exportFile.InCloud() == true && os.Remove(fileName) == nil {
		logger.Println(fileName + " Has Removed")
	} else {
		logger.Println(fileName + " Removed ERROR")
	}
}
func bHelp(m *tb.Message) {
	bot.Send(m.Sender, bHelpContent+"\n"+notice, &tb.SendOptions{DisableWebPagePreview: false})
}
func bOnText(m *tb.Message) {
	switch UserStatus[m.Chat.ID] {
	case USNone:
		{
			bot.Send(m.Chat, gettext.Gettext("getHelp"))
			return
		}
	case USBind1:
		{
			if !m.IsReply() {
				bot.Send(m.Chat, gettext.Gettext("replyBind"))
				return
			}
			bBind2(m)
		}
	case USBind2:
		{
			if !m.IsReply() {
				bot.Send(m.Chat, gettext.Gettext("replyBind"))
				return
			}
			if GetBindNum(m.Chat.ID) == BindMaxNum {
				bot.Send(m.Chat, gettext.Gettext("maximum"))
				return
			}
			bot.Send(m.Chat, gettext.Gettext("binding"))
			err := BindUser(m, UserCid[m.Chat.ID], UserCSecret[m.Chat.ID])
			if err != nil {
				bot.Send(m.Chat, err.Error())
			} else {
				bot.Send(m.Chat, gettext.Gettext("bindSuccess"))
			}
			UserStatus[m.Chat.ID] = USNone
		}
	}
}
func bTask(m *tb.Message) {
	logger.Println(strconv.FormatInt(m.Chat.ID, 10) + " Start SignTask")
	for _, a := range admin {
		if a == m.Chat.ID {
			SignTask()
			return
		}
	}
	bot.Send(m.Chat, gettext.Gettext("noPermission"))
}
func bLog(m *tb.Message) {
	logger.Println(strconv.FormatInt(m.Chat.ID, 10) + " Start Get Logs")
	flag := 0
	for _, a := range admin {
		if a == m.Chat.ID {
			flag = 1
		}
	}
	if flag == 0 {
		bot.Send(m.Chat, gettext.Gettext("noPermission"))
		return
	}
	logs := GetRecentLogs(bLogBasePath, 5)
	var inlineKeys [][]tb.InlineButton
	for _, log := range logs {
		inlineBtn := tb.InlineButton{
			Unique: "log" + strings.Replace(strings.TrimSuffix(filepath.Base(log), ".log"), "-", "", -1),
			Text:   filepath.Base(log),
			Data:   filepath.Base(log),
		}
		bot.Handle(&inlineBtn, bLogsInlineBtn)
		inlineKeys = append(inlineKeys, []tb.InlineButton{inlineBtn})
	}
	_, err := bot.Send(m.Chat, gettext.Gettext("logs"), &tb.ReplyMarkup{InlineKeyboard: inlineKeys})
	if err != nil {
		logger.Println(err)
	}
}
func bLogsInlineBtn(c *tb.Callback) {
	logger.Println(strconv.FormatInt(c.Message.Chat.ID, 10) + " Get Logs: " + c.Data)
	//fmt.Println(c.Data)
	//logger.Println(bLogBasePath + c.Data + ".log")
	logfile := &tb.Document{File: tb.FromDisk(bLogBasePath + c.Data), FileName: c.Data, MIME: "text/plain"}
	_, err := bot.Send(c.Message.Chat, logfile)
	if err != nil {
		logger.Println(err)
		return
	}
	bot.Respond(c)
}