E5BotForSQLite/bots/handle.go

package bots

import (
	"encoding/json"
	"github.com/fsnotify/fsnotify"
	"github.com/iyear/E5SubBot/core"
	"github.com/iyear/E5SubBot/util"
	"github.com/spf13/viper"
	"go.uber.org/zap"
	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 帮助
	源码及使用方法：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()
	util.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) {
	data := core.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) {
	r := core.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) {
	bot.Send(m.Chat, "应用注册： [点击直达]("+core.GetMSRegisterAppUrl()+")", tb.ModeMarkdown)
	bot.Send(m.Chat, "请回复client_id+空格+client_secret", &tb.ReplyMarkup{ForceReply: true})
	UserStatus[m.Chat.ID] = USBind1
	UserCid[m.Chat.ID] = m.Text
}
func bBind2(m *tb.Message) {
	tmp := strings.Split(m.Text, " ")
	if len(tmp) != 2 {
		bot.Send(m.Chat, "错误的格式")
		return
	}
	cid := tmp[0]
	cse := tmp[1]
	bot.Send(m.Chat, "授权账户： [点击直达]("+core.GetMSAuthUrl(cid)+")", tb.ModeMarkdown)
	_, err := bot.Send(m.Chat, "请回复http://localhost/…… + 空格 + 别名(用于管理)", &tb.ReplyMarkup{ForceReply: true})
	if err != nil {
		return
	}
	UserStatus[m.Chat.ID] = USBind2
	UserCid[m.Chat.ID] = cid
	UserCSecret[m.Chat.ID] = cse
}

func bUnBind(m *tb.Message) {
	data := core.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) {
	r := core.QueryDataByMS(c.Data)
	u := r[0]
	if ok, err := core.DelData(u.MsId); !ok {
		zap.S().Errorw("failed to delete db data","error",err,"ms_id",u.MsId)
		bot.Send(c.Message.Chat, "解绑失败!")
		return
	}
	bot.Send(c.Message.Chat, "解绑成功!")
	bot.Respond(c)
}
func bExport(m *tb.Message) {
	type MsMiniData struct {
		Alias        string
		ClientId     string
		ClientSecret string
		RefreshToken string
		Other        string
	}
	var MsMini []MsMiniData
	data := core.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 {
		zap.S().Errorw("failed to marshal json","error",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 {
		zap.S().Errorw("failed to write file","error",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"}
	bot.Send(m.Chat, exportFile)
	//不遗留本地文件
	if exportFile.InCloud() != true || os.Remove(fileName) != nil {
		zap.S().Errorw("failed to export files")
	}
}
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) {
	for _, a := range admin {
		if a == m.Chat.ID {
			SignTask()
			return
		}
	}
	bot.Send(m.Chat, "您没有权限执行此操作~")
}
func bLog(m *tb.Message) {
	flag := 0
	for _, a := range admin {
		if a == m.Chat.ID {
			flag = 1
		}
	}
	if flag == 0 {
		bot.Send(m.Chat, "您没有权限执行此操作~")
		return
	}
	logs := util.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})
	}
	bot.Send(m.Chat, "选择一个日志", &tb.ReplyMarkup{InlineKeyboard: inlineKeys})
}
func bLogsInlineBtn(c *tb.Callback) {
	//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 {
		return
	}
	bot.Respond(c)
}