feat: support kick

models/user.go

@@ -0,0 +1,52 @@
+package models
+
+import (
+	"tg-mc/utils/database"
+
+	"github.com/sirupsen/logrus"
+	"gorm.io/gorm"
+)
+
+type User struct {
+	gorm.Model
+	TGID   int64
+	MCName string
+	Status int // 0: pending, 1: normal, 2: banned
+}
+
+func init() {
+	if err := database.GetDB().AutoMigrate(&User{}); err != nil {
+		logrus.Panic(err)
+	}
+}
+
+func (u *User) TableName() string {
+	return "users"
+}
+
+func GetUserByTGID(tgID int64) (user User, err error) {
+	err = database.GetDB().Where(
+		&User{TGID: tgID},
+	).First(&user).Error
+	return
+}
+
+func GetUserByMCName(mcName string) (user User, err error) {
+	err = database.GetDB().Where(
+		&User{MCName: mcName},
+	).First(&user).Error
+	return
+}
+
+func CreateUserIfNotExist(tgID int64, u *User) (err error) {
+	err = database.GetDB().Where(
+		&User{TGID: tgID},
+	).FirstOrCreate(&u).Error
+	return
+}
+
+func (u *User) Delete(tgID int64) error {
+	return database.GetDB().Where(
+		&User{TGID: tgID},
+	).Delete(&u).Error
+}