feat: gateway

services/gateway/auth.go

@@ -0,0 +1,86 @@
+package gateway
+
+import (
+	"fmt"
+	"sync"
+	"tg-mc/conf"
+	"tg-mc/defs"
+	"tg-mc/models"
+
+	tgbotapi "github.com/go-telegram-bot-api/telegram-bot-api/v5"
+)
+
+type Auth interface {
+	IsAuthed(u models.User) bool
+	RequestAuth(u models.User, req *LoginRequest)
+	Reject(u models.User)
+	SetAuth(u models.User)
+}
+
+type Authcator struct {
+	UserMap *sync.Map
+}
+
+var authcator *Authcator
+
+func GetAuthcator() Auth {
+	if authcator == nil {
+		authcator = &Authcator{
+			UserMap: &sync.Map{},
+		}
+	}
+	return authcator
+}
+
+func (a *Authcator) getUserLoginReq(u models.User) *LoginRequest {
+	chAny, ok := a.UserMap.Load(u.MCName)
+	if !ok {
+		return nil
+	}
+
+	loginReq, ok := chAny.(*LoginRequest)
+	if !ok {
+		return nil
+	}
+
+	return loginReq
+}
+
+func (a *Authcator) IsAuthed(u models.User) bool {
+	loginReq := a.getUserLoginReq(u)
+	if loginReq == nil {
+		return false
+	}
+
+	return <-loginReq.Resolve
+}
+
+func (a *Authcator) RequestAuth(u models.User, req *LoginRequest) {
+	a.UserMap.Store(u.MCName, req)
+	m := tgbotapi.NewMessage(u.TGID, fmt.Sprintf("MC用户：%v 尝试登录，请选择操作", u.MCName))
+
+	m.ReplyMarkup = tgbotapi.NewInlineKeyboardMarkup(
+		tgbotapi.NewInlineKeyboardRow(
+			tgbotapi.NewInlineKeyboardButtonData("批准", defs.NewApproveCommand(u.MCName).ToJSON()),
+			tgbotapi.NewInlineKeyboardButtonData("拒绝", defs.NewRejectCommand(u.MCName).ToJSON())),
+	)
+	conf.Bot.Send(m)
+}
+
+func (a *Authcator) Reject(u models.User) {
+	loginReq := a.getUserLoginReq(u)
+	if loginReq == nil {
+		return
+	}
+
+	loginReq.Resolve <- false
+}
+
+func (a *Authcator) SetAuth(u models.User) {
+	loginReq := a.getUserLoginReq(u)
+	if loginReq == nil {
+		return
+	}
+
+	loginReq.Resolve <- true
+}