feat: self hosted customization (#100)

* option for restrict uploads only to admins

* feat: allow to customize help text

Author: mr-linch

.env.example

@@ -11,5 +11,7 @@ SFB_REDIS=redis://localhost:6379
 SFB_REDIS_MAX_OPEN_CONNS=5
 SFB_REDIS_MAX_IDLE_CONNS=0
 
+# SFB_IS_USERS_CAN_UPLOAD_FILES=false
+
 SFB_ADDR=:8000
 SFB_SECRET_ID_SALT=-secret-1234-

bot/bot.go

@@ -35,14 +35,25 @@ type Bot struct {
 	adminSrv *service.Admin
 	chatSrv  *service.Chat
 
+	textHelp string
+
 	handler tg.Handler
 }
 
 func (bot *Bot) Self() tgbotapi.User {
 	return bot.client.Self
 }
 
-func New(buildInfo pkg.BuildInfo, client *tgbotapi.BotAPI, state state.Store, authSrv *service.Auth, docSrv *service.File, adminSrv *service.Admin, chatSrv *service.Chat) (*Bot, error) {
+func New(
+	buildInfo pkg.BuildInfo,
+	client *tgbotapi.BotAPI,
+	state state.Store,
+	authSrv *service.Auth,
+	docSrv *service.File,
+	adminSrv *service.Admin,
+	chatSrv *service.Chat,
+	textHelp string,
+) (*Bot, error) {
 
 	bot := &Bot{
 		buildInfo: buildInfo,
@@ -53,6 +64,8 @@ func New(buildInfo pkg.BuildInfo, client *tgbotapi.BotAPI, state state.Store, au
 		fileSrv:  docSrv,
 		adminSrv: adminSrv,
 		chatSrv:  chatSrv,
+
+		textHelp: textHelp,
 	}
 
 	// bot.client.Debug = true
@@ -152,7 +165,7 @@ func (bot *Bot) onUpdate(ctx context.Context, update *tgbotapi.Update) error {
 		}
 
 		if msg.Text == textButtonAbout {
-			answer := bot.newAnswerMsg(msg, textStart)
+			answer := bot.newAnswerMsg(msg, bot.getTextStart())
 			answer.ParseMode = mdv2
 			return bot.send(ctx, answer)
 		}

bot/handler_common.go

@@ -21,6 +21,8 @@ const (
 )
 
 var (
+	textHello = "Привет\\! 👋\n"
+
 	textHelp = dedent.Dedent(`
 		Я помогу тебе поделиться любым медиафайлом \(фото, видео, документы, аудио, голосовые\) с подписчиками твоего канала\. 
 		Отправь любой из перечисленных файлов, а я в ответ дам тебе ссылку\. Желательно указать подпись, чтобы человек не забыл кто ему это пошарил\.
@@ -32,11 +34,27 @@ var (
 		Новости и обновления: @share\_file\_news
 	`)
 
-	textStart = "Привет\\! 👋\n" + textHelp
+	textStart = textHello + textHelp
 )
 
+func (bot *Bot) getTextStart() string {
+	if bot.textHelp != "" {
+		return textHello + bot.textHelp
+	}
+
+	return textStart
+}
+
+func (bot *Bot) getTextHelp() string {
+	if bot.textHelp != "" {
+		return bot.textHelp
+	}
+
+	return textHelp
+}
+
 func (bot *Bot) onHelp(ctx context.Context, msg *tgbotapi.Message) error {
-	answer := bot.newAnswerMsg(msg, textHelp)
+	answer := bot.newAnswerMsg(msg, bot.getTextHelp())
 	answer.ParseMode = mdv2
 	return bot.send(ctx, answer)
 }
@@ -80,7 +98,7 @@ func (bot *Bot) onStart(ctx context.Context, msg *tgbotapi.Message) error {
 		}
 	}
 
-	answer := bot.newAnswerMsg(msg, textStart)
+	answer := bot.newAnswerMsg(msg, bot.getTextStart())
 	return bot.send(ctx, answer)
 }
 

bot/handler_file.go

@@ -280,7 +280,15 @@ func (bot *Bot) onFile(ctx context.Context, msg *tgbotapi.Message) error {
 
 	file, err := bot.fileSrv.AddFile(ctx, user, inputFile)
 
-	if err != nil {
+	switch {
+	case errors.Is(err, service.ErrUsersCantUploadFiles):
+		_ = bot.sendText(ctx,
+			user.ID,
+			"✋ Загрузка файлов доступна только администраторам ботам",
+		)
+
+		return nil
+	case err != nil:
 		_ = bot.sendText(ctx,
 			user.ID,
 			"⚠️ Что-то пошло не так при добавлении файла",
@@ -572,7 +580,7 @@ func (bot *Bot) onFileRestrictionsChatCheck(
 }
 
 func (bot *Bot) onPublicFileHelp(ctx context.Context, cbq *tgbotapi.CallbackQuery) error {
-	answer := tgbotapi.NewMessage(cbq.Message.Chat.ID, textStart)
+	answer := tgbotapi.NewMessage(cbq.Message.Chat.ID, bot.getTextStart())
 	answer.ParseMode = mdv2
 	return bot.send(ctx, answer)
 }

main.go

@@ -54,6 +54,9 @@ type Config struct {
 	SecretIDSalt string `required:"true" split_words:"true"`
 
 	DryRun bool `default:"false" split_words:"true"`
+
+	IsUsersCanUploadFiles bool   `default:"true" split_words:"true"`
+	TextHelp              string `split_words:"true"`
 }
 
 func (cfg Config) getEnv() string {
@@ -251,11 +254,12 @@ func run(ctx context.Context) error {
 	}
 
 	fileSrv := &service.File{
-		File:     pg.File(),
-		Chat:     pg.Chat(),
-		Download: pg.Download(),
-		Telegram: tgClient,
-		Redis:    rdb,
+		File:                  pg.File(),
+		Chat:                  pg.Chat(),
+		Download:              pg.Download(),
+		Telegram:              tgClient,
+		Redis:                 rdb,
+		IsUsersCanUploadFiles: cfg.IsUsersCanUploadFiles,
 	}
 
 	adminSrv := &service.Admin{
@@ -273,7 +277,7 @@ func run(ctx context.Context) error {
 		Download: pg.Download(),
 	}
 
-	tgBot, err := bot.New(buildInfo, tgClient, botState, authSrv, fileSrv, adminSrv, chatSrv)
+	tgBot, err := bot.New(buildInfo, tgClient, botState, authSrv, fileSrv, adminSrv, chatSrv, cfg.TextHelp)
 	if err != nil {
 		return errors.Wrap(err, "init bot")
 	}

service/file.go

@@ -20,6 +20,8 @@ type File struct {
 	Telegram *tgbotapi.BotAPI
 	Redis    redis.UniversalClient
 	Download core.DownloadStore
+
+	IsUsersCanUploadFiles bool
 }
 
 type InputFile struct {
@@ -50,11 +52,19 @@ func (srv *File) newOwnedFile(ctx context.Context, doc *core.File) (*OwnedFile,
 	}, nil
 }
 
+var (
+	ErrUsersCantUploadFiles = errors.New("users can't upload files")
+)
+
 func (srv *File) AddFile(
 	ctx context.Context,
 	user *core.User,
 	in *InputFile,
 ) (*OwnedFile, error) {
+	if !user.IsAdmin && !srv.IsUsersCanUploadFiles {
+		return nil, ErrUsersCantUploadFiles
+	}
+
 	doc := core.NewFile(
 		in.FileID,
 		in.Caption,