268 lines
6.7 KiB
Go
268 lines
6.7 KiB
Go
package bots
|
||
|
||
import (
|
||
"encoding/json"
|
||
"fmt"
|
||
"github.com/iyear/E5SubBot/config"
|
||
"github.com/iyear/E5SubBot/model"
|
||
"github.com/iyear/E5SubBot/task"
|
||
"github.com/iyear/E5SubBot/util"
|
||
"go.uber.org/zap"
|
||
tb "gopkg.in/tucnak/telebot.v2"
|
||
"io/ioutil"
|
||
"os"
|
||
"path/filepath"
|
||
"strconv"
|
||
"strings"
|
||
"time"
|
||
)
|
||
|
||
var (
|
||
UserStatus map[int64]int
|
||
UserClientId map[int64]string
|
||
UserClientSecret map[int64]string
|
||
)
|
||
|
||
const (
|
||
USNone = iota
|
||
USBind1
|
||
USBind2
|
||
)
|
||
|
||
func init() {
|
||
UserStatus = make(map[int64]int)
|
||
UserClientId = make(map[int64]string)
|
||
UserClientSecret = make(map[int64]string)
|
||
}
|
||
|
||
func bStart(m *tb.Message) {
|
||
bot.Send(m.Sender, config.WelcomeContent)
|
||
bHelp(m)
|
||
}
|
||
|
||
func bMy(m *tb.Message) {
|
||
var data []*model.Client
|
||
model.DB.Where("tg_id = ?", m.Chat.ID).Find(&data)
|
||
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,
|
||
fmt.Sprintf("选择一个账户查看具体信息\n\n绑定数: %d/%d", GetBindNum(m.Chat.ID), config.BindMaxNum),
|
||
&tb.ReplyMarkup{InlineKeyboard: inlineKeys})
|
||
}
|
||
func bMyInlineBtn(c *tb.Callback) {
|
||
var u *model.Client
|
||
model.DB.Where("ms_id = ?", c.Data).First(&u)
|
||
bot.Send(c.Message.Chat,
|
||
fmt.Sprintf("信息\n别名：%s\nMS_ID(MD5): %s\nclient_id: %s\nclient_secret: %s\n最近更新时间: %s",
|
||
u.Alias,
|
||
u.MsId,
|
||
u.ClientId,
|
||
u.ClientSecret,
|
||
time.Unix(u.Uptime, 0).Format("2006-01-02 15:04:05")),
|
||
)
|
||
bot.Respond(c)
|
||
}
|
||
|
||
func bBind1(m *tb.Message) {
|
||
bot.Send(m.Chat,
|
||
fmt.Sprintf("应用注册： [点击直达](%s)", model.GetMSRegisterAppUrl()),
|
||
tb.ModeMarkdown)
|
||
bot.Send(m.Chat,
|
||
"请回复client_id(空格)client_secret",
|
||
&tb.ReplyMarkup{ForceReply: true})
|
||
UserStatus[m.Chat.ID] = USBind1
|
||
UserClientId[m.Chat.ID] = m.Text
|
||
}
|
||
func bBind2(m *tb.Message) {
|
||
tmp := strings.Split(m.Text, " ")
|
||
if len(tmp) != 2 {
|
||
bot.Send(m.Chat, "错误的格式")
|
||
return
|
||
}
|
||
ClientId := tmp[0]
|
||
ClientSecret := tmp[1]
|
||
bot.Send(m.Chat, "授权账户： [点击直达]("+model.GetMSAuthUrl(ClientId)+")", tb.ModeMarkdown)
|
||
bot.Send(m.Chat,
|
||
"请回复http://localhost/…… + 空格 + 别名(用于管理)",
|
||
&tb.ReplyMarkup{ForceReply: true})
|
||
UserStatus[m.Chat.ID] = USBind2
|
||
UserClientId[m.Chat.ID] = ClientId
|
||
UserClientSecret[m.Chat.ID] = ClientSecret
|
||
}
|
||
|
||
func bUnBind(m *tb.Message) {
|
||
var data []*model.Client
|
||
model.DB.Where("tg_id = ?", m.Chat.ID).Find(&data)
|
||
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,
|
||
fmt.Sprintf("选择一个账户将其解绑\n\n当前绑定数: %d/%d", GetBindNum(m.Chat.ID), config.BindMaxNum),
|
||
&tb.ReplyMarkup{InlineKeyboard: inlineKeys},
|
||
)
|
||
}
|
||
func bUnBindInlineBtn(c *tb.Callback) {
|
||
if result := model.DB.Where("ms_id = ?", c.Data).Delete(&model.Client{}); result.Error != nil {
|
||
zap.S().Errorw("failed to delete db data",
|
||
"error", result.Error,
|
||
"ms_id", c.Data,
|
||
)
|
||
bot.Send(c.Message.Chat, "解绑失败!")
|
||
return
|
||
}
|
||
bot.Send(c.Message.Chat, "解绑成功!")
|
||
bot.Respond(c)
|
||
}
|
||
func bExport(m *tb.Message) {
|
||
type ClientExport struct {
|
||
Alias string
|
||
ClientId string
|
||
ClientSecret string
|
||
RefreshToken string
|
||
Other string
|
||
}
|
||
var exports []ClientExport
|
||
var data []*model.Client
|
||
model.DB.Where("tg_id = ?", m.Chat.ID).Find(&data)
|
||
if len(data) == 0 {
|
||
bot.Send(m.Chat, "你还没有绑定过账户嗷~")
|
||
return
|
||
}
|
||
for _, u := range data {
|
||
var cExport = ClientExport{
|
||
Alias: u.Alias,
|
||
ClientId: u.ClientId,
|
||
ClientSecret: u.ClientSecret,
|
||
RefreshToken: u.RefreshToken,
|
||
Other: u.Other,
|
||
}
|
||
exports = append(exports, cExport)
|
||
}
|
||
export, err := json.MarshalIndent(exports, "", "\t")
|
||
if err != nil {
|
||
zap.S().Errorw("failed to marshal json",
|
||
"error", err)
|
||
bot.Send(m.Chat, fmt.Sprintf("获取JSON失败!\n\nERROR: %s", err.Error()))
|
||
return
|
||
}
|
||
fileName := fmt.Sprintf("./%d_export_tmp.json", m.Chat.ID)
|
||
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,
|
||
config.HelpContent+"\n"+config.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) == config.BindMaxNum {
|
||
bot.Send(m.Chat, "已经达到最大可绑定数")
|
||
return
|
||
}
|
||
bot.Send(m.Chat, "正在绑定中……")
|
||
err := BindUser(m, UserClientId[m.Chat.ID], UserClientSecret[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 config.Admins {
|
||
if a == m.Chat.ID {
|
||
task.SignTask()
|
||
return
|
||
}
|
||
}
|
||
bot.Send(m.Chat, "只有Bot管理员才有权限执行此操作")
|
||
}
|
||
func bLog(m *tb.Message) {
|
||
flag := 0
|
||
for _, a := range config.Admins {
|
||
if a == m.Chat.ID {
|
||
flag = 1
|
||
}
|
||
}
|
||
if flag == 0 {
|
||
bot.Send(m.Chat, "只有Bot管理员才有权限执行此操作")
|
||
return
|
||
}
|
||
logs := util.GetRecentLogs(config.LogBasePath, 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) {
|
||
logfile := &tb.Document{
|
||
File: tb.FromDisk(config.LogBasePath + c.Data),
|
||
FileName: c.Data,
|
||
MIME: "text/plain",
|
||
}
|
||
bot.Send(c.Message.Chat, logfile)
|
||
bot.Respond(c)
|
||
}
|