alfajrimutawadhi/feature/telegram-userid-whitelist (#18)

* fix(whitelist): add middleware for check whitelist

Closes 15

* feat(whitelist): add access whitelist by user id based on config

#15

Closes 15

empty/config.yaml

@@ -1,3 +1,6 @@
 teleterm:
-  telegram_token: "xxxxxxxxx"
+  telegram_token: "xxxxxx"
   shell_executor: "/bin/bash"
+whitelist:
+  - <USER_ID>
+  - <USER_ID>

teleterm/telegram_handler.go

@@ -10,8 +10,8 @@ import (
 	"github.com/alfiankan/teleterm/v2/common"
 	"github.com/alfiankan/teleterm/v2/executor"
 	"github.com/spf13/viper"
-	"gopkg.in/telebot.v3"
 	tele "gopkg.in/telebot.v3"
+	"gopkg.in/telebot.v3/middleware"
 )
 
 const (
@@ -32,7 +32,7 @@ Err :
 
 func createButtonReplay(ctx context.Context, persist Persistence, menu *tele.ReplyMarkup) (teleMenus []tele.Row, ere error) {
 
-	menus := []tele.Row{}
+	var menus []tele.Row
 
 	buttons, err := persist.GetAllButtons(ctx)
 	if err != nil {
@@ -66,6 +66,13 @@ func Start(ctx context.Context, db *sql.DB, telebotToken string) {
 		return
 	}
 
+	configWhitelist := viper.GetIntSlice("whitelist")
+	var whitelist []int64
+	for _, id := range configWhitelist {
+		whitelist = append(whitelist, int64(id))
+	}
+	b.Use(middleware.Whitelist(whitelist...))
+
 	menu := &tele.ReplyMarkup{ResizeKeyboard: true}
 
 	b.Handle("/refresh", func(c tele.Context) error {
@@ -185,8 +192,8 @@ func Start(ctx context.Context, db *sql.DB, telebotToken string) {
 
 	})
 
-	//receive document
-	b.Handle(telebot.OnDocument, func(c tele.Context) error {
+	// receive document
+	b.Handle(tele.OnDocument, func(c tele.Context) error {
 
 		log.Info().Str("state", "upload file").Msg(c.Message().Document.FileID)
 

vendor/modules.txt

@@ -109,6 +109,7 @@ gopkg.in/ini.v1
 # gopkg.in/telebot.v3 v3.0.0
 ## explicit; go 1.13
 gopkg.in/telebot.v3
+gopkg.in/telebot.v3/middleware
 # gopkg.in/yaml.v2 v2.4.0
 ## explicit; go 1.15
 gopkg.in/yaml.v2