E5BotForSQLite/control.go

package main

import (
	"fmt"
	"github.com/tidwall/gjson"
	tb "gopkg.in/tucnak/telebot.v2"
	"time"
)

//If Successfully return "",else return error information
func BindUser(m *tb.Message) string {
	fmt.Printf("%d Begin Bind\n", m.Chat.ID)
	code := GetURLValue(m.Text, "code")
	fmt.Println(code)
	access, refresh := MSFirGetToken(code)
	if refresh == "" {
		fmt.Printf("%d Bind error:GetRefreshToken\n", m.Chat.ID)
		return "获取RefreshToken失败"
	}

	//token has gotten
	bot.Send(m.Chat, "Token获取成功!")
	info := MSGetUserInfo(access)
	fmt.Printf("TGID:%d Refresh Token: %s\n", m.Chat.ID, refresh)
	if info == "" {
		fmt.Printf("%d Bind error:Getinfo\n", m.Chat.ID)
		return "获取用户信息错误"
	}

	var u MSData
	u.tgId = m.Chat.ID
	u.refreshToken = refresh
	u.msId = gjson.Get(info, "id").String()
	u.uptime = time.Now()
	u.other = ""

	//MS User Is Exist
	if MSUserIsExist(u.tgId, u.msId) {
		fmt.Printf("%d Bind error:MSUserHasExisted\n", m.Chat.ID)
		return "该ID对应的用户已经绑定过了"
	}
	//MS information has gotten
	bot.Send(m.Chat, "MS_ID： "+u.msId+"\nuserPrincipalName： "+gjson.Get(info, "userPrincipalName").String()+"\ndisplayName： "+gjson.Get(info, "displayName").String()+"\n")
	if ok, err := AddData(db, u); !ok {
		fmt.Printf("%d Bind error: %s\n", m.Chat.ID, err)
		return "数据库写入错误"
	}
	fmt.Printf("%d Bind Successfully!\n", m.Chat.ID)
	return ""
}
func GetBindNum(tgId int64) int {
	data := QueryData(db, tgId)
	return len(data)
}

//return true => exist
func MSUserIsExist(tgId int64, msId string) bool {
	data := QueryData(db, tgId)
	var res MSData
	for _, res = range data {
		if res.msId == msId {
			return true
		}
	}
	return false
}