feat: download telegraph images , close #5

types/types.go

@@ -5,6 +5,8 @@ import (
 	"crypto/md5"
 	"encoding/hex"
 	"fmt"
+	"net/url"
+	"strings"
 	"time"
 
 	"github.com/gotd/td/tg"
@@ -41,29 +43,46 @@ type Task struct {
 	Cancel      context.CancelFunc
 	Error       error
 	Status      TaskStatus
-	File        *File
 	StorageName string
 	StoragePath string
 	StartTime   time.Time
 
+	File          *File
 	FileMessageID int
 	FileChatID    int64
+
+	IsTelegraph  bool
+	TelegraphURL string
+
 	// to track the reply message
 	ReplyMessageID int
 	ReplyChatID    int64
-	// to track the user
-	UserID int64
+	UserID         int64
 }
 
 func (t Task) Key() string {
+	if t.IsTelegraph {
+		return hashStr(t.TelegraphURL)
+	}
 	return fmt.Sprintf("%d:%d", t.FileChatID, t.FileMessageID)
 }
 
 func (t Task) String() string {
+	if t.IsTelegraph {
+		return fmt.Sprintf("[telegraph]:%s", t.TelegraphURL)
+	}
 	return fmt.Sprintf("[%d:%d]:%s", t.FileChatID, t.FileMessageID, t.File.FileName)
 }
 
 func (t Task) FileName() string {
+	if t.IsTelegraph {
+		tgphPath := strings.Split(t.TelegraphURL, "/")[len(strings.Split(t.TelegraphURL, "/"))-1]
+		tgphPathUnescaped, err := url.PathUnescape(tgphPath)
+		if err != nil {
+			return tgphPath
+		}
+		return tgphPathUnescaped
+	}
 	return t.File.FileName
 }
 

types/utils.go

@@ -0,0 +1,12 @@
+package types
+
+import (
+	"crypto/md5"
+	"encoding/hex"
+)
+
+func hashStr(s string) string {
+	hash := md5.New()
+	hash.Write([]byte(s))
+	return hex.EncodeToString(hash.Sum(nil))
+}