feat(bot): add support for ALLOWED_USERS

Author: EverythingSuckz

README.md

@@ -202,7 +202,9 @@ 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 a user bot. Used for auto adding the bots to `LOG_CHANNEL`.(default: `null`)
+- `USER_SESSION` : A pyrogram session string for a user bot. Used for auto adding the bots to `LOG_CHANNEL`. (default: `null`)
+
+- `ALLOWED_USERS` : A list of user IDs separated by comma (`,`). If this is set, only the users in this list will be able to use the bot. (default: `null`)
 
 <hr>
 

config/config.go

@@ -20,18 +20,36 @@ import (
 
 var ValueOf = &config{}
 
+type allowedUsers []int64
+
+func (au *allowedUsers) Decode(value string) error {
+	if value == "" {
+		return nil
+	}
+	ids := strings.Split(string(value), ",")
+	for _, id := range ids {
+		idInt, err := strconv.ParseInt(id, 10, 64)
+		if err != nil {
+			return err
+		}
+		*au = append(*au, idInt)
+	}
+	return nil
+}
+
 type config struct {
-	ApiID          int32  `envconfig:"API_ID" required:"true"`
-	ApiHash        string `envconfig:"API_HASH" required:"true"`
-	BotToken       string `envconfig:"BOT_TOKEN" required:"true"`
-	LogChannelID   int64  `envconfig:"LOG_CHANNEL" required:"true"`
-	Dev            bool   `envconfig:"DEV" default:"false"`
-	Port           int    `envconfig:"PORT" default:"8080"`
-	Host           string `envconfig:"HOST" default:""`
-	HashLength     int    `envconfig:"HASH_LENGTH" default:"6"`
-	UseSessionFile bool   `envconfig:"USE_SESSION_FILE" default:"true"`
-	UserSession    string `envconfig:"USER_SESSION"`
-	UsePublicIP    bool   `envconfig:"USE_PUBLIC_IP" default:"true"`
+	ApiID          int32        `envconfig:"API_ID" required:"true"`
+	ApiHash        string       `envconfig:"API_HASH" required:"true"`
+	BotToken       string       `envconfig:"BOT_TOKEN" required:"true"`
+	LogChannelID   int64        `envconfig:"LOG_CHANNEL" required:"true"`
+	Dev            bool         `envconfig:"DEV" default:"false"`
+	Port           int          `envconfig:"PORT" default:"8080"`
+	Host           string       `envconfig:"HOST" default:""`
+	HashLength     int          `envconfig:"HASH_LENGTH" default:"6"`
+	UseSessionFile bool         `envconfig:"USE_SESSION_FILE" default:"true"`
+	UserSession    string       `envconfig:"USER_SESSION"`
+	UsePublicIP    bool         `envconfig:"USE_PUBLIC_IP" default:"true"`
+	AllowedUsers   allowedUsers `envconfig:"ALLOWED_USERS"`
 	MultiTokens    []string
 }
 

internal/commands/start.go

@@ -1,6 +1,9 @@
 package commands
 
 import (
+	"EverythingSuckz/fsb/config"
+	"EverythingSuckz/fsb/internal/utils"
+
 	"github.com/celestix/gotgproto/dispatcher"
 	"github.com/celestix/gotgproto/dispatcher/handlers"
 	"github.com/celestix/gotgproto/ext"
@@ -19,6 +22,10 @@ func start(ctx *ext.Context, u *ext.Update) error {
 	if peerChatId.Type != int(storage.TypeUser) {
 		return dispatcher.EndGroups
 	}
+	if len(config.ValueOf.AllowedUsers) != 0 && !utils.Contains(config.ValueOf.AllowedUsers, chatId) {
+		ctx.Reply(u, "You are not allowed to use this bot.", nil)
+		return dispatcher.EndGroups
+	}
 	ctx.Reply(u, "Hi, send me any file to get a direct streamble link to that file.", nil)
 	return dispatcher.EndGroups
 }

internal/commands/stream.go

@@ -46,6 +46,10 @@ func sendLink(ctx *ext.Context, u *ext.Update) error {
 	if peerChatId.Type != int(storage.TypeUser) {
 		return dispatcher.EndGroups
 	}
+	if len(config.ValueOf.AllowedUsers) != 0 && !utils.Contains(config.ValueOf.AllowedUsers, chatId) {
+		ctx.Reply(u, "You are not allowed to use this bot.", nil)
+		return dispatcher.EndGroups
+	}
 	supported, err := supportedMediaFilter(u.EffectiveMessage)
 	if err != nil {
 		return err

internal/utils/helpers.go

@@ -16,6 +16,16 @@ import (
 	"go.uber.org/zap"
 )
 
+// https://stackoverflow.com/a/70802740/15807350
+func Contains[T comparable](s []T, e T) bool {
+	for _, v := range s {
+		if v == e {
+			return true
+		}
+	}
+	return false
+}
+
 func GetTGMessage(ctx context.Context, client *gotgproto.Client, messageID int) (*tg.Message, error) {
 	inputMessageID := tg.InputMessageClass(&tg.InputMessageID{ID: messageID})
 	channel, err := GetLogChannelPeer(ctx, client.API(), client.PeerStorage)