chore: set admin commands for group in admin db

Author: krau

cmd/version.go

@@ -7,7 +7,7 @@ import (
 )
 
 const (
-	Version string = "0.7.15"
+	Version string = "0.7.16"
 )
 
 var VersionCmd = &cobra.Command{

dao/admin.go

@@ -44,3 +44,16 @@ func DeleteAdminByUserID(ctx context.Context, userID int64) (*mongo.DeleteResult
 func UpdateAdmin(ctx context.Context, admin *model.AdminModel) (*mongo.UpdateResult, error) {
 	return adminCollection.ReplaceOne(ctx, bson.M{"user_id": admin.UserID}, admin)
 }
+
+func GetAdmins(ctx context.Context) ([]model.AdminModel, error) {
+	cursor, err := adminCollection.Find(ctx, bson.M{})
+	if err != nil {
+		return nil, err
+	}
+	var admins []model.AdminModel
+	err = cursor.All(ctx, &admins)
+	if err != nil {
+		return nil, err
+	}
+	return admins, nil
+}

service/user.go

@@ -75,3 +75,31 @@ func CreateOrUpdateAdmin(ctx context.Context, userID int64, permissions []types.
 	_, err = dao.UpdateAdmin(ctx, admin)
 	return err
 }
+
+func GetAdminUserIDs(ctx context.Context) ([]int64, error) {
+	admins, err := dao.GetAdmins(ctx)
+	if err != nil {
+		return nil, err
+	}
+	var userIDs []int64
+	for _, admin := range admins {
+		if admin.UserID > 0 {
+			userIDs = append(userIDs, admin.UserID)
+		}
+	}
+	return userIDs, nil
+}
+
+func GetAdminGroupIDs(ctx context.Context) ([]int64, error) {
+	admins, err := dao.GetAdmins(ctx)
+	if err != nil {
+		return nil, err
+	}
+	var groupIDs []int64
+	for _, admin := range admins {
+		if admin.UserID < 0 {
+			groupIDs = append(groupIDs, admin.UserID)
+		}
+	}
+	return groupIDs, nil
+}

telegram/telegram.go

@@ -2,6 +2,8 @@ package telegram
 
 import (
 	"ManyACG/config"
+	"ManyACG/service"
+	"context"
 	"os"
 
 	. "ManyACG/logger"
@@ -17,35 +19,8 @@ var (
 	GroupChatID   telego.ChatID // 附属群组
 )
 
-func InitBot() {
-	var err error
-	Bot, err = telego.NewBot(
-		config.Cfg.Telegram.Token,
-		telego.WithDefaultLogger(false, true),
-		telego.WithAPIServer(config.Cfg.Telegram.APIURL),
-	)
-	if err != nil {
-		Logger.Fatalf("Error when creating bot: %s", err)
-		os.Exit(1)
-	}
-	if config.Cfg.Telegram.Username != "" {
-		ChannelChatID = telegoutil.Username(config.Cfg.Telegram.Username)
-	} else {
-		ChannelChatID = telegoutil.ID(config.Cfg.Telegram.ChatID)
-	}
-
-	if config.Cfg.Telegram.GroupID != 0 {
-		GroupChatID = telegoutil.ID(config.Cfg.Telegram.GroupID)
-	}
-
-	me, err := Bot.GetMe()
-	if err != nil {
-		Logger.Errorf("Error when getting bot info: %s", err)
-		os.Exit(1)
-	}
-	BotUsername = me.Username
-
-	commonCommands := []telego.BotCommand{
+var (
+	CommonCommands = []telego.BotCommand{
 		{
 			Command:     "start",
 			Description: "开始涩涩",
@@ -76,12 +51,7 @@ func InitBot() {
 		},
 	}
 
-	Bot.SetMyCommands(&telego.SetMyCommandsParams{
-		Commands: commonCommands,
-		Scope:    &telego.BotCommandScopeDefault{Type: telego.ScopeTypeDefault},
-	})
-
-	adminCommands := []telego.BotCommand{
+	AdminCommands = []telego.BotCommand{
 		{
 			Command:     "set_admin",
 			Description: "设置管理员",
@@ -119,12 +89,52 @@ func InitBot() {
 			Description: "处理无哈希的图片",
 		},
 	}
+)
 
-	adminCommands = append(commonCommands, adminCommands...)
+func InitBot() {
+	var err error
+	Bot, err = telego.NewBot(
+		config.Cfg.Telegram.Token,
+		telego.WithDefaultLogger(false, true),
+		telego.WithAPIServer(config.Cfg.Telegram.APIURL),
+	)
+	if err != nil {
+		Logger.Fatalf("Error when creating bot: %s", err)
+		os.Exit(1)
+	}
+	if config.Cfg.Telegram.Username != "" {
+		ChannelChatID = telegoutil.Username(config.Cfg.Telegram.Username)
+	} else {
+		ChannelChatID = telegoutil.ID(config.Cfg.Telegram.ChatID)
+	}
+
+	if config.Cfg.Telegram.GroupID != 0 {
+		GroupChatID = telegoutil.ID(config.Cfg.Telegram.GroupID)
+	}
+
+	me, err := Bot.GetMe()
+	if err != nil {
+		Logger.Errorf("Error when getting bot info: %s", err)
+		os.Exit(1)
+	}
+	BotUsername = me.Username
 
-	for _, adminID := range config.Cfg.Telegram.Admins {
+	Bot.SetMyCommands(&telego.SetMyCommandsParams{
+		Commands: CommonCommands,
+		Scope:    &telego.BotCommandScopeDefault{Type: telego.ScopeTypeDefault},
+	})
+
+	allCommands := append(CommonCommands, AdminCommands...)
+
+	adminUserIDs, err := service.GetAdminUserIDs(context.TODO())
+	if err != nil {
+		Logger.Warnf("Error when getting admin user IDs: %s", err)
+		return
+	}
+
+	for _, adminID := range adminUserIDs {
 		Bot.SetMyCommands(&telego.SetMyCommandsParams{
-			Commands: adminCommands,
+			Commands: allCommands,
 			Scope: &telego.BotCommandScopeChat{
 				Type:   telego.ScopeTypeChat,
 				ChatID: telegoutil.ID(adminID),
@@ -134,12 +144,29 @@ func InitBot() {
 			continue
 		}
 		Bot.SetMyCommands(&telego.SetMyCommandsParams{
-			Commands: adminCommands,
+			Commands: allCommands,
 			Scope: &telego.BotCommandScopeChatMember{
 				Type:   telego.ScopeTypeChat,
 				ChatID: GroupChatID,
 				UserID: adminID,
 			},
 		})
 	}
+
+	adminGroupIDs, err := service.GetAdminGroupIDs(context.TODO())
+	if err != nil {
+		Logger.Warnf("Error when getting admin group IDs: %s", err)
+		return
+	}
+
+	for _, adminID := range adminGroupIDs {
+		Bot.SetMyCommands(&telego.SetMyCommandsParams{
+			Commands: allCommands,
+			Scope: &telego.BotCommandScopeChat{
+				Type:   telego.ScopeTypeChat,
+				ChatID: telegoutil.ID(adminID),
+			},
+		})
+	}
+
 }