E5SubBotForSQLite/handle.go

package main

import (
	"encoding/json"
	"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/"
	bStartContent string = "欢迎使用E5SubBot!"
	bHelpContent  string = `
	命令：
	/my 查看已绑定账户信息
	/bind  绑定新账户
	/unbind 解绑账户
	/export 导出账户信息(JSON格式)
	/help 帮助
	/task 手动执行一次任务(管理员)
	/log 获取最近日志文件(管理员)
    源代码：https://github.com/rainerosion/E5SubBot 
	原作者：https://github.com/iyear/E5SubBot`
)

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
)

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)

	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, "选择一个账户查看具体信息\n\n绑定数: "+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, "信息\n别名："+u.alias+"\nMS_ID(MD5): "+u.msId+"\nclient_id: "+u.clientId+"\nclient_secret: "+u.clientSecret+"\n最近更新时间: "+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, "应用注册： [点击直达]("+MSGetReAppUrl()+")", tb.ModeMarkdown)
	_, err := bot.Send(m.Chat, "请回复client_id+空格+client_secret", &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, "错误的格式")
		return
	}
	logger.Println("client_id: " + tmp[0] + " client_secret: " + tmp[1])
	cid := tmp[0]
	cse := tmp[1]
	bot.Send(m.Chat, "授权账户： [点击直达]("+MSGetAuthUrl(cid)+")", tb.ModeMarkdown)
	_, err := bot.Send(m.Chat, "请回复http://localhost/…… + 空格 + 别名(用于管理)", &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, "选择一个账户将其解绑\n\n当前绑定数: "+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, "解绑失败!")
		return
	}
	logger.Println(u.msId + " UnBind Success")
	bot.Send(c.Message.Chat, "解绑成功!")
	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, "你还没有绑定过账户嗷~")
		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, "获取JSON失败~\n"+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, "写入临时文件失败~\n"+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, "发送/help获取帮助嗷")
			return
		}
	case USBind1:
		{
			if !m.IsReply() {
				bot.Send(m.Chat, "请通过回复方式绑定")
				return
			}
			bBind2(m)
		}
	case USBind2:
		{
			if !m.IsReply() {
				bot.Send(m.Chat, "请通过回复方式绑定")
				return
			}
			if GetBindNum(m.Chat.ID) == BindMaxNum {
				bot.Send(m.Chat, "已经达到最大可绑定数")
				return
			}
			bot.Send(m.Chat, "正在绑定中……")
			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, "绑定成功!")
			}
			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, "您没有权限执行此操作~")
}
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, "您没有权限执行此操作~")
		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, "选择一个日志", &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)
}