feat(userbot): auto add bots to channel

Author: EverythingSuckz

README.md

@@ -158,7 +158,7 @@ An example of `fsb.env` file:
 API_ID=452525
 API_HASH=esx576f8738x883f3sfzx83
 BOT_TOKEN=55838383:yourbottokenhere
-BIN_CHANNEL=-10045145224562
+LOG_CHANNEL=-10045145224562
 PORT=8080
 HOST=http://yourserverip
 # (if you want to set up multiple bots)
@@ -175,7 +175,7 @@ Before running the bot, you will need to set up the following mandatory variable
 
 - `BOT_TOKEN` : This is the bot token for the Telegram Media Streamer Bot, which can be obtained from [@BotFather](https://telegram.dog/BotFather).
 
-- `BIN_CHANNEL` :  This is the channel ID for the log channel where the bot will forward media messages and store these files to make the generated direct links work. To obtain a channel ID, create a new telegram channel (public or private), post something in the channel, forward the message to [@missrose_bot](https://telegram.dog/MissRose_bot) and **reply the forwarded message** with the /id command. Copy the forwarded channel ID and paste it into the this field.
+- `LOG_CHANNEL` :  This is the channel ID for the log channel where the bot will forward media messages and store these files to make the generated direct links work. To obtain a channel ID, create a new telegram channel (public or private), post something in the channel, forward the message to [@missrose_bot](https://telegram.dog/MissRose_bot) and **reply the forwarded message** with the /id command. Copy the forwarded channel ID and paste it into the this field.
 
 ### Optional Vars
 In addition to the mandatory variables, you can also set the following optional variables:
@@ -188,7 +188,7 @@ In addition to the mandatory variables, you can also set the following optional
 
 - `USE_SESSION_FILE` : Use session files for worker client(s). This speeds up the worker bot startups. (default: `false`)
 
-- `USER_SESSION` : A pyrogram session string for the user bot. (default: `null`)
+- `USER_SESSION` : A pyrogram session string for a user bot. Used for auto adding the bots to `LOG_CHANNEL`.(default: `null`)
 
 <hr>
 
@@ -211,7 +211,7 @@ you may also add as many as bots you want. (max limit is 50)
 `MULTI_TOKEN3`, `MULTI_TOKEN4`, etc.
 
 > **Warning**
-> Don't forget to add all these worker bots to the `BIN_CHANNEL` for the proper functioning
+> Don't forget to add all these worker bots to the `LOG_CHANNEL` for the proper functioning
 
 ## Contributing
 

internal/bot/userbot.go

@@ -2,9 +2,11 @@ package bot
 
 import (
 	"EverythingSuckz/fsb/config"
+	"errors"
 
 	"github.com/celestix/gotgproto"
 	"github.com/celestix/gotgproto/sessionMaker"
+	"github.com/gotd/td/tg"
 	"go.uber.org/zap"
 )
 
@@ -40,5 +42,75 @@ func StartUserBot(l *zap.Logger) {
 	UserBot.log = log
 	UserBot.client = client
 	log.Info("Userbot started", zap.String("username", client.Self.Username), zap.String("FirstName", client.Self.FirstName), zap.String("LastName", client.Self.LastName))
+	if err := UserBot.AddBotsAsAdmins(); err != nil {
+		log.Error("Failed to add bots as admins", zap.Error(err))
+		return
+	}
+}
 
+func (u *UserBotStruct) AddBotsAsAdmins() error {
+	u.log.Info("Preparing to add bots as admins")
+	ctx := u.client.CreateContext()
+	channel := config.ValueOf.LogChannelID
+	channelInfos, err := u.client.API().ChannelsGetChannels(
+		ctx,
+		[]tg.InputChannelClass{
+			&tg.InputChannel{
+				ChannelID: channel,
+			},
+		},
+	)
+	if err != nil {
+		u.log.Error("Failed to get channel info", zap.Error(err))
+		return errors.New("failed to get channel info")
+	}
+	if len(channelInfos.GetChats()) == 0 {
+		return errors.New("no channels found")
+	}
+	inputChannel := channelInfos.GetChats()[0].(*tg.Channel).AsInput()
+	currentAdmins := []int64{}
+	admins, err := u.client.API().ChannelsGetParticipants(ctx, &tg.ChannelsGetParticipantsRequest{
+		Channel: inputChannel,
+		Filter:  &tg.ChannelParticipantsAdmins{},
+		Offset:  0,
+		Limit:   100,
+	})
+	if err != nil {
+		u.log.Error("Failed to get admins", zap.Error(err))
+		return err
+	}
+	for _, admin := range admins.(*tg.ChannelsChannelParticipants).Participants {
+		if user, ok := admin.(*tg.ChannelParticipantAdmin); ok {
+			currentAdmins = append(currentAdmins, user.UserID)
+		}
+	}
+	for _, bot := range Workers.Bots {
+		for _, admin := range currentAdmins {
+			if admin == bot.Self.ID {
+				u.log.Sugar().Infof("Bot @%s is already an admin", bot.Self.Username)
+				continue
+			}
+		}
+		botInfo, err := ctx.ResolveUsername(bot.Self.Username)
+		if err != nil {
+			u.log.Warn(err.Error())
+		}
+		_, err = u.client.API().ChannelsEditAdmin(
+			u.client.CreateContext().Context,
+			&tg.ChannelsEditAdminRequest{
+				Channel: inputChannel,
+				UserID:  botInfo.GetInputUser(),
+				AdminRights: tg.ChatAdminRights{
+					PostMessages: true,
+				},
+				Rank: "admin",
+			},
+		)
+		if err != nil {
+			u.log.Sugar().Warnf("Failed to add @%s as admin", bot.Self.Username)
+			u.log.Warn(err.Error())
+		}
+		u.log.Sugar().Infof("Added @%s as admin", bot.Self.Username)
+	}
+	return nil
 }