refactor: complete core features

common/cache.go

@@ -0,0 +1,63 @@
+package common
+
+import (
+	"bytes"
+	"encoding/gob"
+	"sync"
+
+	"github.com/coocood/freecache"
+	"github.com/gotd/td/tg"
+	"github.com/krau/SaveAny-Bot/types"
+)
+
+type CommonCache struct {
+	cache *freecache.Cache
+	mu    sync.RWMutex
+}
+
+var Cache *CommonCache
+
+func initCache() {
+	gob.Register(types.File{})
+	gob.Register(tg.InputDocumentFileLocation{})
+	Cache = &CommonCache{cache: freecache.NewCache(10 * 1024 * 1024)}
+}
+
+func GetCache() *CommonCache {
+	return Cache
+}
+
+func (c *CommonCache) Get(key string, value *types.File) error {
+	c.mu.RLock()
+	defer c.mu.RUnlock()
+	data, err := Cache.cache.Get([]byte(key))
+	if err != nil {
+		return err
+	}
+	dec := gob.NewDecoder(bytes.NewReader(data))
+	err = dec.Decode(&value)
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
+func (c *CommonCache) Set(key string, value *types.File, expireSeconds int) error {
+	c.mu.Lock()
+	defer c.mu.Unlock()
+	var buf bytes.Buffer
+	enc := gob.NewEncoder(&buf)
+	err := enc.Encode(value)
+	if err != nil {
+		return err
+	}
+	Cache.cache.Set([]byte(key), buf.Bytes(), expireSeconds)
+	return nil
+}
+
+func (c *CommonCache) Delete(key string) error {
+	c.mu.Lock()
+	defer c.mu.Unlock()
+	Cache.cache.Del([]byte(key))
+	return nil
+}

common/common.go

@@ -2,4 +2,5 @@ package common
 
 func Init() {
 	initClient()
+	initCache()
 }

common/utils.go

@@ -1,35 +0,0 @@
-package common
-
-import (
-	"fmt"
-
-	"github.com/gotd/td/tg"
-	"github.com/krau/SaveAny-Bot/logger"
-	"github.com/krau/SaveAny-Bot/types"
-)
-
-func FileFromMedia(media tg.MessageMediaClass) (*types.File, error) {
-	logger.L.Debug("FileFromMedia")
-	switch media := media.(type) {
-	case *tg.MessageMediaDocument:
-		document, ok := media.Document.AsNotEmpty()
-		if !ok {
-			return nil, fmt.Errorf("unexpected type %T", media)
-		}
-		var fileName string
-		for _, attribute := range document.Attributes {
-			if name, ok := attribute.(*tg.DocumentAttributeFilename); ok {
-				fileName = name.FileName
-				break
-			}
-		}
-		return &types.File{
-			Location: document.AsInputDocumentFileLocation(),
-			FileSize: document.Size,
-			FileName: fileName,
-			MimeType: document.MimeType,
-			ID:       document.ID,
-		}, nil
-	}
-	return nil, fmt.Errorf("unexpected type %T", media)
-}