拦截广告用户

src/config/read_config.go

@@ -16,6 +16,8 @@ var RecruitMissing map[string]string
 
 var RecruitTagMap map[string]string
 
+var ADWords []string
+
 func init() {
 	// 设置配置文件的名字
 	viper.SetConfigName("arknights")
@@ -51,6 +53,7 @@ func initData() {
 	recruitTags := viper.GetString("recruit.tags")
 	RecruitMissing = make(map[string]string)
 	RecruitTagMap = make(map[string]string)
+	ADWords = viper.GetStringSlice("ad")
 	for _, missing := range strings.Split(jpMissing, "/") {
 		RecruitMissing[missing] = missing
 	}

src/core/bot/bot.go

@@ -21,15 +21,14 @@ func Serve() {
 	b := bot.Arknights.AddHandle()
 	b.NewProcessor(
 		func(update tgbotapi.Update) bool {
-			if update.Message != nil {
+			if update.Message != nil && update.Message.ViaBot != nil {
 				return update.Message.ViaBot.ID == 273234066 //PostBot Id
 			}
 			return false
 		},
 		func(update tgbotapi.Update) error {
-			msg := tgbotapi.NewDeleteMessage(update.FromChat().ID, update.Message.MessageID)
-			_, err := bot.Arknights.Send(msg)
-			return err
+			update.Message.Delete()
+			return nil
 		},
 	)
 	b.NewMemberProcessor(gatekeeper.NewMemberHandle)

src/plugins/gatekeeper/new_member_handle.go

@@ -1,15 +1,28 @@
 package gatekeeper
 
 import (
+	bot "arknights_bot/config"
 	"arknights_bot/utils"
 	tgbotapi "github.com/ijnkawakaze/telegram-bot-api"
 	"github.com/spf13/viper"
+	"strings"
 )
 
 func NewMemberHandle(update tgbotapi.Update) error {
 	message := update.Message
 	for _, member := range message.NewChatMembers {
 		if member.ID == message.From.ID { // 自己加入群组
+			chat, err := bot.Arknights.GetChat(tgbotapi.ChatInfoConfig{ChatConfig: tgbotapi.ChatConfig{ChatID: member.ID}})
+			if err != nil {
+				return err
+			}
+			for _, word := range bot.ADWords {
+				if strings.Contains(chat.Bio, word) {
+					message.Delete()
+					bot.Arknights.BanChatMember(message.Chat.ID, member.ID)
+					return nil
+				}
+			}
 			go VerifyMember(message)
 			continue
 		}

src/plugins/system/callback_handle.go

@@ -30,14 +30,7 @@ func Report(callBack tgbotapi.Update) error {
 
 	if d[1] == "BAN" {
 		fmt.Printf("在群组 %s 用户 %s 封禁了 %d", callbackQuery.Message.Chat.Title, callbackQuery.From.FullName(), target)
-		banChatMemberConfig := tgbotapi.BanChatMemberConfig{
-			ChatMemberConfig: tgbotapi.ChatMemberConfig{
-				ChatID: chatId,
-				UserID: target,
-			},
-			RevokeMessages: true,
-		}
-		bot.Arknights.Send(banChatMemberConfig)
+		bot.Arknights.BanChatMember(chatId, target)
 		delMsg := tgbotapi.NewDeleteMessage(chatId, targetMessageId)
 		bot.Arknights.Send(delMsg)