feat: announce command list
This commit is contained in:
		
							parent
							
								
									37dffe56ce
								
							
						
					
					
						commit
						440d36b291
					
				
							
								
								
									
										10
									
								
								cfg.go
								
								
								
								
							
							
						
						
									
										10
									
								
								cfg.go
								
								
								
								
							| 
						 | 
				
			
			@ -10,8 +10,9 @@ import (
 | 
			
		|||
)
 | 
			
		||||
 | 
			
		||||
type Config struct {
 | 
			
		||||
	AdminUIDs  map[int64]struct{}
 | 
			
		||||
	TGBotToken string
 | 
			
		||||
	AdminUIDs          map[int64]struct{}
 | 
			
		||||
	TGBotToken         string
 | 
			
		||||
	TGAnnounceCommands bool
 | 
			
		||||
 | 
			
		||||
	WatchedInterface       string
 | 
			
		||||
	MonthlyTrafficLimitGiB int
 | 
			
		||||
| 
						 | 
				
			
			@ -42,6 +43,11 @@ func LoadCfg() error {
 | 
			
		|||
		cfg.AdminUIDs[uid] = struct{}{}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	announceCmdsEnv := os.Getenv("TG_ANNOUNCE_CMDS")
 | 
			
		||||
	if !lo.Contains([]string{"", "no", "false", "0"}, strings.ToLower(announceCmdsEnv)) {
 | 
			
		||||
		cfg.TGAnnounceCommands = true
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	cfg.WatchedInterface = "eth0"
 | 
			
		||||
	if iface := os.Getenv("TG_WATCHED_INTERFACE"); iface != "" {
 | 
			
		||||
		cfg.WatchedInterface = iface
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
							
								
								
									
										14
									
								
								main.go
								
								
								
								
							
							
						
						
									
										14
									
								
								main.go
								
								
								
								
							| 
						 | 
				
			
			@ -6,6 +6,7 @@ import (
 | 
			
		|||
	"syscall"
 | 
			
		||||
 | 
			
		||||
	"go.uber.org/zap"
 | 
			
		||||
	tele "gopkg.in/telebot.v3"
 | 
			
		||||
 | 
			
		||||
	"git.gensokyo.cafe/kkyy/tgbot_misaka_5882f7/utils"
 | 
			
		||||
)
 | 
			
		||||
| 
						 | 
				
			
			@ -28,6 +29,19 @@ func runBot() {
 | 
			
		|||
		logger.Fatalw("Failed to initialize bot", "err", err)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Announce commands
 | 
			
		||||
	if config.TGAnnounceCommands {
 | 
			
		||||
		logger.Info("Announcing commands...")
 | 
			
		||||
 | 
			
		||||
		if err = bot.SetCommands([]tele.Command{
 | 
			
		||||
			{Text: "traffic", Description: "Show traffic usage."},
 | 
			
		||||
			{Text: "dig", Description: "Diggy diggy dig."},
 | 
			
		||||
			{Text: "year_progress", Description: "Time doesn't wait."},
 | 
			
		||||
		}); err != nil {
 | 
			
		||||
			logger.Fatalw("Failed to announce commands", "err", err)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	botFinCh := utils.WaitFor(bot.Start)
 | 
			
		||||
	logger.Infow("Bot started", "username", bot.Me.Username)
 | 
			
		||||
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
		Loading…
	
		Reference in New Issue