228 lines
6.5 KiB
Go
228 lines
6.5 KiB
Go
package bots
|
||
|
||
import (
|
||
"errors"
|
||
"fmt"
|
||
"github.com/spf13/viper"
|
||
"github.com/tidwall/gjson"
|
||
tb "gopkg.in/tucnak/telebot.v2"
|
||
"main/core"
|
||
"main/logger"
|
||
"main/outlook"
|
||
"main/util"
|
||
"strconv"
|
||
"strings"
|
||
"time"
|
||
)
|
||
|
||
var SignOk map[int64]int
|
||
|
||
//If Successfully return "",else return error information
|
||
func BindUser(m *tb.Message, cid, cse string) error {
|
||
logger.Println("%d Begin Bind\n", m.Chat.ID)
|
||
tmp := strings.Split(m.Text, " ")
|
||
if len(tmp) != 2 {
|
||
logger.Println("%d Bind error:Wrong Bind Format\n", m.Chat.ID)
|
||
return errors.New("绑定格式错误")
|
||
}
|
||
logger.Println("Alias: " + tmp[1])
|
||
Alias := tmp[1]
|
||
code := util.GetURLValue(tmp[0], "code")
|
||
//fmt.Println(code)
|
||
access, refresh, err := MSFirGetToken(code, cid, cse)
|
||
if err != nil {
|
||
logger.Println("%d Bind error:GetRefreshToken %s \n", m.Chat.ID, err.Error())
|
||
return err
|
||
}
|
||
|
||
//token has gotten
|
||
bot.Send(m.Chat, "Token获取成功!")
|
||
info, err := outlook.GetUserInfo(access)
|
||
//fmt.Println("TgId:%d Refresh Token: %s\n", m.Chat.ID, refresh)
|
||
if err != nil {
|
||
logger.Println("%d Bind error:Getinfo %s \n", m.Chat.ID, err.Error())
|
||
return err
|
||
}
|
||
|
||
var u core.Client
|
||
u.TgId = m.Chat.ID
|
||
u.RefreshToken = refresh
|
||
//TG的Data传递最高64bytes,一些MsId超过了报错BUTTON_DATA_INVALID (0)，采取md5
|
||
u.MsId = util.Get16MD5Encode(gjson.Get(info, "id").String())
|
||
u.Uptime = time.Now().Unix()
|
||
logger.Println(u.Uptime)
|
||
u.Alias = Alias
|
||
u.ClientId = cid
|
||
u.ClientSecret = cse
|
||
u.Other = ""
|
||
//MS User Is Exist
|
||
if MSAppIsExist(u.TgId, u.ClientId) {
|
||
logger.Println("%d Bind error:MSUserHasExisted\n", m.Chat.ID)
|
||
return errors.New("该应用已经绑定过了，无需重复绑定")
|
||
}
|
||
//MS information has gotten
|
||
bot.Send(m.Chat, "MS_ID(MD5)： "+u.MsId+"\nuserPrincipalName： "+gjson.Get(info, "userPrincipalName").String()+"\ndisplayName： "+gjson.Get(info, "displayName").String()+"\n")
|
||
if ok, err := core.AddData(u); !ok {
|
||
logger.Println("%d Bind error: %s\n", m.Chat.ID, err)
|
||
return err
|
||
}
|
||
logger.Println("%d Bind Successfully!\n", m.Chat.ID)
|
||
return nil
|
||
}
|
||
|
||
//get bind num
|
||
func GetBindNum(TgId int64) int {
|
||
data := core.QueryDataByTG(TgId)
|
||
return len(data)
|
||
}
|
||
|
||
//return true => exist
|
||
func MSAppIsExist(TgId int64, ClientId string) bool {
|
||
data := core.QueryDataByTG(TgId)
|
||
var res core.Client
|
||
for _, res = range data {
|
||
if res.ClientId == ClientId {
|
||
return true
|
||
}
|
||
}
|
||
return false
|
||
}
|
||
|
||
//SignTask
|
||
func SignTask() {
|
||
var (
|
||
SignOk map[int64]int
|
||
SignErr []string
|
||
UnbindUser []string
|
||
num, signOk int
|
||
)
|
||
SignOk = make(map[int64]int)
|
||
fmt.Println("----Task Begin----")
|
||
fmt.Println("Time:" + time.Now().Format("2006-01-02 15:04:05"))
|
||
data := core.QueryDataAll()
|
||
num = len(data)
|
||
fmt.Println("Start Sign")
|
||
//签到任务
|
||
for _, u := range data {
|
||
pre := "您的账户: " + u.Alias + "\n在任务执行时出现了错误!\n错误:"
|
||
chat, err := bot.ChatByID(strconv.FormatInt(u.TgId, 10))
|
||
if err != nil {
|
||
logger.Println(err)
|
||
continue
|
||
}
|
||
//生成解绑按钮
|
||
var inlineKeys [][]tb.InlineButton
|
||
UnBindBtn := tb.InlineButton{Unique: "un" + u.MsId, Text: "点击解绑该账户", Data: u.MsId}
|
||
bot.Handle(&UnBindBtn, bUnBindInlineBtn)
|
||
inlineKeys = append(inlineKeys, []tb.InlineButton{UnBindBtn})
|
||
tmpBtn := &tb.ReplyMarkup{InlineKeyboard: inlineKeys}
|
||
|
||
se := u.MsId + " ( @" + chat.Username + " )"
|
||
access, newRefreshToken, err := outlook.GetToken(u.RefreshToken, u.ClientId, u.ClientSecret)
|
||
|
||
if err != nil {
|
||
logger.Println(u.MsId+" ", err)
|
||
bot.Send(chat, pre+gjson.Get(err.Error(), "error").String(), tmpBtn)
|
||
SignErr = append(SignErr, se)
|
||
ErrorTimes[u.MsId]++
|
||
continue
|
||
}
|
||
if ok, err := outlook.GetOutLookMails(access); !ok {
|
||
logger.Println(u.MsId+" ", err)
|
||
bot.Send(chat, pre+gjson.Get(err.Error(), "error").String(), tmpBtn)
|
||
ErrorTimes[u.MsId]++
|
||
SignErr = append(SignErr, se)
|
||
continue
|
||
}
|
||
u.Uptime = time.Now().Unix()
|
||
u.RefreshToken = newRefreshToken
|
||
if ok, err := core.UpdateData(u); !ok {
|
||
logger.Println(u.MsId+" ", err)
|
||
bot.Send(chat, pre+err.Error(), tmpBtn)
|
||
SignErr = append(SignErr, se)
|
||
ErrorTimes[u.MsId]++
|
||
continue
|
||
}
|
||
fmt.Println(u.MsId + " Sign OK!")
|
||
SignOk[u.TgId]++
|
||
signOk++
|
||
}
|
||
fmt.Println("Sign End,Start Send")
|
||
var isSend map[int64]bool
|
||
isSend = make(map[int64]bool)
|
||
//用户任务反馈
|
||
for _, u := range data {
|
||
chat, err := bot.ChatByID(strconv.FormatInt(u.TgId, 10))
|
||
if err != nil {
|
||
logger.Println("Send Result ERROR: ", err)
|
||
continue
|
||
}
|
||
//错误上限账户清退
|
||
if ErrorTimes[u.MsId] == ErrMaxTimes {
|
||
logger.Println(u.MsId + " Error Limit")
|
||
if ok, err := core.DelData(u.MsId); !ok {
|
||
logger.Println(err)
|
||
} else {
|
||
UnbindUser = append(UnbindUser, u.MsId+" ( @"+chat.Username+" )")
|
||
_, err = bot.Send(chat, "您的账户因达到错误上限而被自动解绑\n后会有期!\n\n别名: "+u.Alias+"\nclient_id: "+u.ClientId+"\nclient_secret: "+u.ClientSecret)
|
||
if err != nil {
|
||
logger.Println(err)
|
||
}
|
||
}
|
||
|
||
}
|
||
if !isSend[u.TgId] {
|
||
//静默发送，过多消息很烦
|
||
_, err = bot.Send(chat, "任务反馈\n时间: "+time.Now().Format("2006-01-02 15:04:05")+"\n结果: "+strconv.Itoa(SignOk[u.TgId])+"/"+strconv.Itoa(GetBindNum(u.TgId)), &tb.SendOptions{DisableNotification: true})
|
||
if err != nil {
|
||
logger.Println(err)
|
||
}
|
||
isSend[u.TgId] = true
|
||
}
|
||
}
|
||
//管理员任务反馈
|
||
var ErrUserStr string
|
||
var UnbindUserStr string
|
||
for _, eu := range SignErr {
|
||
ErrUserStr = ErrUserStr + eu + "\n"
|
||
}
|
||
for _, ubu := range UnbindUser {
|
||
UnbindUserStr = UnbindUserStr + ubu + "\n"
|
||
}
|
||
for _, a := range admin {
|
||
chat, err := bot.ChatByID(strconv.FormatInt(a, 10))
|
||
if err != nil {
|
||
logger.Println(err)
|
||
continue
|
||
}
|
||
bot.Send(chat, "任务反馈(管理员)\n完成时间: "+time.Now().Format("2006-01-02 15:04:05")+"\n结果: "+strconv.Itoa(signOk)+"/"+strconv.Itoa(num)+"\n错误账户:\n"+ErrUserStr+"\n清退账户:\n"+UnbindUserStr)
|
||
}
|
||
fmt.Println("----Task End----")
|
||
}
|
||
func GetAdmin() []int64 {
|
||
var result []int64
|
||
admins := strings.Split(viper.GetString("admin"), ",")
|
||
for _, v := range admins {
|
||
id, _ := strconv.ParseInt(v, 10, 64)
|
||
result = append(result, id)
|
||
}
|
||
return result
|
||
}
|
||
|
||
//func InitLogger() {
|
||
// if !util.PathExists(bLogBasePath) {
|
||
// os.Mkdir(bLogBasePath, 0773)
|
||
// }
|
||
//
|
||
// path := bLogBasePath + time.Now().Format("2006-01-02") + ".log"
|
||
// f, err := os.OpenFile(path, os.O_WRONLY|os.O_CREATE|os.O_APPEND, 0773)
|
||
// if err != nil {
|
||
// logger.Println(err)
|
||
// }
|
||
// writers := []io.Writer{
|
||
// f,
|
||
// os.Stdout}
|
||
// faoWriter := io.MultiWriter(writers...)
|
||
// //logger = log.New(faoWriter, "【E5Sub】", log.Ldate|log.Ltime|log.Lshortfile)
|
||
//}
|