fix: handle exit signal properly

main.go

@@ -6,6 +6,8 @@ import (
 	"syscall"
 
 	"go.uber.org/zap"
+
+	"git.gensokyo.cafe/kkyy/tgbot_misaka_5882f7/utils"
 )
 
 var logger *zap.SugaredLogger
@@ -26,16 +28,21 @@ func runBot() {
 		logger.Fatalw("Failed to initialize bot", "err", err)
 	}
 
-	go bot.Start()
-	logger.Info("Bot started")
+	botFinCh := utils.WaitFor(bot.Start)
+	logger.Infow("Bot started", "username", bot.Me.Username)
 
 	// listen for shutdown signal
 	sigCh := make(chan os.Signal, 1)
 	signal.Notify(sigCh, os.Interrupt, syscall.SIGTERM)
-	<-sigCh
 
-	logger.Info("Bot shutdown...")
-	bot.Stop()
+	select {
+	case sig := <-sigCh:
+		logger.Infow("Received signal", "signal", sig)
+		logger.Info("Bot shutdown...")
+		bot.Stop()
+	case <-botFinCh:
+		logger.Warn("Bot terminated")
+	}
 }
 
 func main() {