feat: use db

internal/db_channel.go

@@ -0,0 +1,126 @@
+package internal
+
+import (
+	"context"
+	"database/sql"
+	"fmt"
+	"os"
+	"path/filepath"
+	"time"
+
+	"exporter/database/telegram_resource/public/model"
+	"exporter/database/telegram_resource/public/table"
+
+	. "github.com/go-jet/jet/v2/postgres"
+	"github.com/pkg/errors"
+	"github.com/samber/lo"
+)
+
+type DBChannel struct {
+	UUID     int64
+	Username string
+	Title    string
+	Offset   int
+	MinID    int
+}
+
+func NewDBChannel(uuid int64, username, title string) *DBChannel {
+	if uuid == 0 {
+		panic("channel id is required")
+	}
+	return &DBChannel{
+		UUID:     uuid,
+		Username: username,
+		Title:    title,
+	}
+}
+
+func (c *DBChannel) Asset(assetID int64, ext string) string {
+	assetFile := fmt.Sprintf("outputs/%d/%d.%s", c.UUID, assetID, ext)
+
+	// if file dir not exists then create it
+	if _, err := os.Stat(filepath.Dir(assetFile)); os.IsNotExist(err) {
+		if err := os.MkdirAll(filepath.Dir(assetFile), 0o755); err != nil {
+			panic(err)
+		}
+	}
+
+	// if file exists then delete it
+	if _, err := os.Stat(assetFile); err == nil {
+		if err := os.Remove(assetFile); err != nil {
+			panic(err)
+		}
+	}
+
+	return assetFile
+}
+
+func (c *DBChannel) Get(ctx context.Context) error {
+	tbl := table.Channels
+
+	var m *model.Channels
+
+	err := tbl.SELECT(tbl.AllColumns).QueryContext(ctx, db, &m)
+	if err != nil {
+		if errors.Is(err, sql.ErrNoRows) {
+			// create new channel with default value
+			m = &model.Channels{
+				UUID:      c.UUID,
+				Username:  c.Username,
+				Title:     c.Title,
+				MinID:     0,
+				Offset:    0,
+				CreatedAt: lo.ToPtr(time.Now()),
+				UpdatedAt: lo.ToPtr(time.Now()),
+			}
+
+			if _, err := tbl.INSERT(tbl.AllColumns).MODEL(m).ExecContext(ctx, db); err != nil {
+				return errors.Wrap(err, "insert channel")
+			}
+		} else {
+			return errors.Wrap(err, "select channel")
+		}
+	}
+	c.MinID = int(m.MinID)
+	c.Offset = int(m.Offset)
+
+	return nil
+}
+
+func (c *DBChannel) Update(ctx context.Context, offsetID int) error {
+	c.Offset = offsetID
+
+	if c.MinID < offsetID {
+		c.MinID = offsetID
+	}
+
+	tbl := table.Channels
+	_, err := tbl.UPDATE().SET(
+		tbl.Offset.SET(Int(int64(c.Offset))),
+		tbl.MinID.SET(Int(int64(c.MinID))),
+	).ExecContext(ctx, db)
+	if err != nil {
+		return errors.Wrap(err, "update channel")
+	}
+
+	return nil
+}
+
+func (c *DBChannel) SaveMessage(ctx context.Context, msg *ChannelMessage) error {
+	message := &model.ChannelMessages{
+		ChannelID:   c.UUID,
+		UUID:        int64(msg.ID),
+		Content:     lo.ToPtr(msg.Message),
+		Media:       msg.GetMedia(),
+		PublishedAt: msg.PublishAt,
+		CreatedAt:   time.Now(),
+	}
+
+	tbl := table.ChannelMessages
+
+	_, err := tbl.INSERT(tbl.AllColumns).MODEL(message).ExecContext(ctx, db)
+	if err != nil {
+		return errors.Wrap(err, "insert message")
+	}
+	return nil
+}