Add paths from env variables, support string telegram channel names

Author: Vilsol

README.md

@@ -15,7 +15,8 @@ Flags:
       --interval int         How often to output transcoding status (default 5)
       --keep-old             Keep old version of video if transcoded version is larger (default true)
       --log string           The log level to output (default "info")
+      --nice                 Whether to lower the priority of ffmpeg process (default true)
       --stderr               Whether to output ffmpeg stderr stream
       --tg-bot-key string    Telegram Bot API Key
-      --tg-chat-id int       Telegram Bot Chat ID
+      --tg-chat-id string    Telegram Bot Chat ID
 ```

cmd/root.go

@@ -1,7 +1,6 @@
 package cmd
 
 import (
-	"errors"
 	"github.com/Vilsol/transcoder-go/config"
 	"github.com/Vilsol/transcoder-go/models"
 	"github.com/Vilsol/transcoder-go/notifications"
@@ -49,16 +48,18 @@ var rootCmd = &cobra.Command{
 		config.InitializeConfig()
 		notifications.InitializeNotifications()
 	},
-	Args: func(cmd *cobra.Command, args []string) error {
-		if len(args) < 1 {
-			return errors.New("must supply at least a single path")
-		}
-
-		return nil
-	},
 	Run: func(cmd *cobra.Command, args []string) {
 		fileList := make([]string, 0)
 
+		if len(args) == 0 {
+			args = viper.GetStringSlice("paths")
+		}
+
+		if len(args) == 0 {
+			log.Fatalf("You must supply at least a single path via CLI argument or PATHS env variable")
+			return
+		}
+
 		for _, arg := range args {
 			files, err := filepath.Glob(arg)
 
@@ -71,6 +72,10 @@ var rootCmd = &cobra.Command{
 			fileList = append(fileList, files...)
 		}
 
+		if len(fileList) == 0 {
+			log.Error("Specified paths did not match any files")
+		}
+
 		for _, fileName := range fileList {
 			if terminated {
 				return
@@ -215,7 +220,7 @@ func init() {
 	rootCmd.PersistentFlags().Bool("nice", true, "Whether to lower the priority of ffmpeg process")
 
 	rootCmd.PersistentFlags().String("tg-bot-key", "", "Telegram Bot API Key")
-	rootCmd.PersistentFlags().Int64("tg-chat-id", 0, "Telegram Bot Chat ID")
+	rootCmd.PersistentFlags().String("tg-chat-id", "", "Telegram Bot Chat ID")
 
 	_ = viper.BindPFlag("flags", rootCmd.PersistentFlags().Lookup("flags"))
 	_ = viper.BindPFlag("extensions", rootCmd.PersistentFlags().Lookup("extensions"))

notifications/telegram.go

@@ -7,14 +7,15 @@ import (
 	"github.com/go-telegram-bot-api/telegram-bot-api"
 	log "github.com/sirupsen/logrus"
 	"github.com/spf13/viper"
+	"strconv"
 	"time"
 )
 
 var tgBot *tgbotapi.BotAPI
 
 func init() {
 	initialize = append(initialize, func() {
-		if viper.GetString("tg-bot-key") != "" && viper.GetInt64("tg-chat-id") != 0 {
+		if viper.GetString("tg-bot-key") != "" && viper.GetString("tg-chat-id") != "" {
 			var err error
 			tgBot, err = tgbotapi.NewBotAPI(viper.GetString("tg-bot-key"))
 
@@ -28,8 +29,24 @@ func init() {
 
 			lastMessage := int64(0)
 
+			chatIdStr := viper.GetString("tg-chat-id")
+
+			chatId, err := strconv.ParseInt(chatIdStr, 10, 64)
+
+			if err != nil {
+				chat, err := tgBot.GetChat(tgbotapi.ChatConfig{
+					SuperGroupUsername: chatIdStr,
+				})
+
+				if err != nil {
+					log.Fatalf("Chat not found: %s", chatIdStr)
+				}
+
+				chatId = chat.ID
+			}
+
 			start = append(start, func(data *models.NotificationData) {
-				message := tgbotapi.NewMessage(viper.GetInt64("tg-chat-id"), generateTelegramMessageText(data, nil))
+				message := tgbotapi.NewMessage(chatId, generateTelegramMessageText(data, nil))
 				message.ParseMode = tgbotapi.ModeMarkdown
 				send, err := tgBot.Send(message)
 
@@ -49,7 +66,7 @@ func init() {
 				}
 
 				if currentMessage != nil {
-					message := tgbotapi.NewEditMessageText(viper.GetInt64("tg-chat-id"), currentMessage.MessageID, generateTelegramMessageText(data, nil))
+					message := tgbotapi.NewEditMessageText(chatId, currentMessage.MessageID, generateTelegramMessageText(data, nil))
 					message.ParseMode = tgbotapi.ModeMarkdown
 					_, err := tgBot.Send(message)
 
@@ -63,7 +80,7 @@ func init() {
 
 			end = append(end, func(data *models.NotificationData, result models.Result) {
 				if currentMessage != nil {
-					message := tgbotapi.NewEditMessageText(viper.GetInt64("tg-chat-id"), currentMessage.MessageID, generateTelegramMessageText(data, &result))
+					message := tgbotapi.NewEditMessageText(chatId, currentMessage.MessageID, generateTelegramMessageText(data, &result))
 					message.ParseMode = tgbotapi.ModeMarkdown
 					_, err := tgBot.Send(message)