package models

import (
	"context"
	"time"

	"quyun/app/requests"
	"quyun/database/schemas/public/model"
	"quyun/database/schemas/public/table"

	. "github.com/go-jet/jet/v2/postgres"
	"github.com/samber/lo"
	"github.com/sirupsen/logrus"
)

type MediaItem struct {
	ID         int64  `json:"id"`
	Name       string `json:"name"`
	UploadTime string `json:"upload_time"`
	FileSize   int64  `json:"file_size"`
	MimeType   string `json:"media_type"`
}

// @provider
type mediasModel struct {
	log *logrus.Entry `inject:"false"`
}

func (m *mediasModel) Prepare() error {
	m.log = logrus.WithField("module", "mediasModel")
	return nil
}

func (m *mediasModel) BuildConditionWithKey(key *string) BoolExpression {
	tbl := table.Medias

	cond := Bool(true)

	if key == nil || *key == "" {
		return cond
	}

	cond = cond.AND(
		tbl.Name.LIKE(String("%" + *key + "%")),
	)

	return cond
}

// countByCond
func (m *mediasModel) countByCondition(ctx context.Context, expr BoolExpression) (int64, error) {
	var cnt struct {
		Cnt int64
	}

	tbl := table.Medias
	stmt := SELECT(COUNT(tbl.ID).AS("cnt")).FROM(tbl).WHERE(expr)
	m.log.Infof("sql: %s", stmt.DebugSql())

	err := stmt.QueryContext(ctx, db, &cnt)
	if err != nil {
		m.log.Errorf("error counting media items: %v", err)
		return 0, err
	}

	return cnt.Cnt, nil
}

func (m *mediasModel) List(ctx context.Context, pagination *requests.Pagination, expr BoolExpression) (*requests.Pager, error) {
	pagination.Format()

	tbl := table.Medias
	stmt := tbl.
		SELECT(tbl.AllColumns).
		WHERE(expr).
		ORDER_BY(tbl.ID.DESC()).
		LIMIT(pagination.Limit).
		OFFSET(pagination.Offset)
	m.log.Infof("sql: %s", stmt.DebugSql())

	var medias []model.Medias
	err := stmt.QueryContext(ctx, db, &medias)
	if err != nil {
		m.log.Errorf("error querying media items: %v", err)
		return nil, err
	}

	count, err := m.countByCondition(ctx, expr)
	if err != nil {
		m.log.Errorf("error getting media count: %v", err)
		return nil, err
	}

	// Convert model.Medias to MediaItem
	mediaItems := lo.Map(medias, func(media model.Medias, _ int) *MediaItem {
		return &MediaItem{
			ID:         media.ID,
			Name:       media.Name,
			UploadTime: media.CreatedAt.Format("2006-01-02 15:04:05"),
			FileSize:   media.Size,
			MimeType:   media.MimeType,
		}
	})

	return &requests.Pager{
		Items:      mediaItems,
		Total:      count,
		Pagination: *pagination,
	}, nil
}

func (m *mediasModel) BatchCreate(ctx context.Context, models []*model.Medias) error {
	stmt := table.Medias.INSERT(table.Medias.MutableColumns).MODELS(models)
	m.log.Infof("sql: %s", stmt.DebugSql())

	if _, err := stmt.ExecContext(ctx, db); err != nil {
		m.log.Errorf("error creating media item: %v", err)
		return err
	}

	m.log.Infof("media item created successfully")
	return nil
}

func (m *mediasModel) Create(ctx context.Context, model *model.Medias) error {
	model.CreatedAt = time.Now()
	stmt := table.Medias.INSERT(table.Medias.MutableColumns).MODEL(model)
	m.log.Infof("sql: %s", stmt.DebugSql())

	if _, err := stmt.ExecContext(ctx, db); err != nil {
		m.log.Errorf("error creating media item: %v", err)
		return err
	}

	m.log.Infof("media item created successfully")
	return nil
}

// GetByIds
func (m *mediasModel) GetByIds(ctx context.Context, ids []int64) ([]*model.Medias, error) {
	if len(ids) == 0 {
		return nil, nil
	}

	condIds := lo.Map(ids, func(id int64, _ int) Expression {
		return Int64(id)
	})

	tbl := table.Medias
	stmt := tbl.
		SELECT(tbl.AllColumns).
		WHERE(tbl.ID.IN(condIds...))
	m.log.Infof("sql: %s", stmt.DebugSql())

	var medias []model.Medias
	err := stmt.QueryContext(ctx, db, &medias)
	if err != nil {
		m.log.Errorf("error querying media items: %v", err)
		return nil, err
	}

	return lo.Map(medias, func(media model.Medias, _ int) *model.Medias {
		return &media
	}), nil
}

// GetByHash
func (m *mediasModel) GetByHash(ctx context.Context, hash string) (*model.Medias, error) {
	tbl := table.Medias
	stmt := tbl.
		SELECT(tbl.AllColumns).
		WHERE(tbl.Hash.EQ(String(hash)))
	m.log.Infof("sql: %s", stmt.DebugSql())

	var media model.Medias
	err := stmt.QueryContext(ctx, db, &media)
	if err != nil {
		m.log.Errorf("error querying media item by hash: %v", err)
		return nil, err
	}

	return &media, nil
}

// Update
func (m *mediasModel) Update(ctx context.Context, hash string, model *model.Medias) error {
	tbl := table.Medias
	stmt := tbl.
		UPDATE(tbl.MutableColumns.Except(tbl.CreatedAt)).
		MODEL(model).
		WHERE(table.Medias.Hash.EQ(String(hash)))
	m.log.Infof("sql: %s", stmt.DebugSql())

	if _, err := stmt.ExecContext(ctx, db); err != nil {
		m.log.Errorf("error updating media item: %v", err)
		return err
	}

	m.log.Infof("media item updated successfully")
	return nil
}

// GetByID
func (m *mediasModel) GetByID(ctx context.Context, id int64) (*model.Medias, error) {
	tbl := table.Medias
	stmt := tbl.
		SELECT(tbl.AllColumns).
		WHERE(tbl.ID.EQ(Int64(id)))
	m.log.Infof("sql: %s", stmt.DebugSql())

	var media model.Medias

	if err := stmt.QueryContext(ctx, db, &media); err != nil {
		m.log.Errorf("error querying media item by ID: %v", err)
		return nil, err
	}

	return &media, nil
}