feat(dl): file name template

app/dl/dl.go

@@ -8,7 +8,7 @@ import (
 	"github.com/spf13/viper"
 )
 
-func Run(ctx context.Context, urls, files []string) error {
+func Run(ctx context.Context, template string, urls, files []string) error {
 	c, _, err := tgc.NoLogin()
 	if err != nil {
 		return err
@@ -25,7 +25,11 @@ func Run(ctx context.Context, urls, files []string) error {
 			return err
 		}
 
-		return downloader.New(c.API(), viper.GetInt(consts.FlagPartSize), viper.GetInt(consts.FlagThreads), newIter(c.API(), umsgs, fmsgs)).
+		it, err := newIter(c.API(), template, umsgs, fmsgs)
+		if err != nil {
+			return err
+		}
+		return downloader.New(c.API(), viper.GetInt(consts.FlagPartSize), viper.GetInt(consts.FlagThreads), it).
 			Download(ctx, viper.GetInt(consts.FlagLimit))
 	})
 }

app/dl/iter.go

@@ -1,42 +1,61 @@
 package dl
 
 import (
+	"bytes"
 	"context"
 	"fmt"
 	"github.com/gotd/td/telegram/peers"
 	"github.com/gotd/td/telegram/query"
 	"github.com/gotd/td/tg"
 	"github.com/iyear/tdl/pkg/downloader"
 	"github.com/iyear/tdl/pkg/utils"
+	"text/template"
+	"time"
 )
 
 type iter struct {
-	client  *tg.Client
-	dialogs []*dialog
-	curi    int
-	curj    int
-	manager *peers.Manager
+	client   *tg.Client
+	dialogs  []*dialog
+	curi     int
+	curj     int
+	template *template.Template
+	manager  *peers.Manager
 }
 
 type dialog struct {
 	peer tg.InputPeerClass
 	msgs []int
 }
 
-func newIter(client *tg.Client, items ...[]*dialog) *iter {
+type fileTemplate struct {
+	DialogID     int64
+	MessageID    int
+	MessageDate  int64
+	FileName     string
+	FileSize     string
+	DownloadDate int64
+}
+
+func newIter(client *tg.Client, tmpl string, items ...[]*dialog) (*iter, error) {
+	t, err := template.New("dl").Parse(tmpl)
+	if err != nil {
+		return nil, err
+	}
+
 	mm := make([]*dialog, 0)
 
 	for _, m := range items {
 		mm = append(mm, m...)
 	}
 
 	return &iter{
-		client:  client,
-		dialogs: mm,
-		curi:    0,
-		curj:    -1,
-		manager: peers.Options{}.Build(client), // TODO(iyear): use local cache to speed up
-	}
+		client:   client,
+		dialogs:  mm,
+		curi:     0,
+		curj:     -1,
+		template: t,
+		manager:  peers.Options{}.Build(client), // TODO(iyear): use local cache to speed up
+	}, nil
 }
 
 func (i *iter) Next(ctx context.Context) bool {
@@ -91,7 +110,19 @@ func (i *iter) item(ctx context.Context, peer tg.InputPeerClass, msg int) (*down
 			id, message.ID)
 	}
 
-	media.Name = fmt.Sprintf("%d_%d_%s", id, message.ID, media.Name)
+	buf := bytes.Buffer{}
+	err := i.template.Execute(&buf, &fileTemplate{
+		DialogID:     id,
+		MessageID:    message.ID,
+		MessageDate:  int64(message.Date),
+		FileName:     media.Name,
+		FileSize:     utils.Byte.FormatBinaryBytes(media.Size),
+		DownloadDate: time.Now().Unix(),
+	})
+	if err != nil {
+		return nil, err
+	}
+	media.Name = buf.String()
 
 	return media, nil
 }

cmd/dl/dl.go

@@ -8,18 +8,20 @@ import (
 
 var (
 	urls, files []string
+	template    string
 )
 
 var Cmd = &cobra.Command{
 	Use:     "dl",
 	Aliases: []string{"download"},
 	Short:   "Download anything from Telegram (protected) chat",
 	RunE: func(cmd *cobra.Command, args []string) error {
-		return dl.Run(cmd.Context(), urls, files)
+		return dl.Run(cmd.Context(), template, urls, files)
 	},
 }
 
 func init() {
 	Cmd.Flags().StringSliceVarP(&urls, consts.FlagDlUrl, "u", []string{}, "telegram message links")
-	Cmd.Flags().StringSliceVarP(&files, consts.FlagDlFile, "f", []string{}, "official client export files")
+	Cmd.Flags().StringSliceVarP(&files, consts.FlagDlFile, "f", []string{}, "official client exported files")
+	Cmd.Flags().StringVar(&template, consts.FlagDlTemplate, "{{ .DialogID }}_{{ .MessageID }}_{{ .FileName }}", "download file name template")
 }

pkg/consts/flag.go

@@ -13,4 +13,5 @@ const (
 	FlagLoginDesktop = "desktop"
 	FlagDlUrl        = "url"
 	FlagDlFile       = "file"
+	FlagDlTemplate   = "template"
 )