quyun/backend/app/model/medias.go

package model

import (
	"context"

	"quyun/app/requests"
	"quyun/database/fields"
	"quyun/database/table"

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

var mediasUpdateExcludeColumns = []Column{
	table.Medias.CreatedAt,
}

func (m *Medias) Like(key *string) Cond {
	return func(cond BoolExpression) BoolExpression {
		tbl := table.Medias
		if key == nil || *key == "" {
			return cond
		}

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

		return cond
	}
}

func (m *Medias) List(ctx context.Context, pagination *requests.Pagination, conds ...Cond) (*requests.Pager, error) {
	pagination.Format()

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

	var medias []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.Count(ctx, conds...)
	if err != nil {
		m.log().Errorf("error getting media count: %v", err)
		return nil, err
	}

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

// GetByIds
func (m *Medias) GetByIds(ctx context.Context, ids []int64) ([]*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 []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 Medias, _ int) *Medias {
		return &media
	}), nil
}

// GetByHash
func (m *Medias) GetByHash(ctx context.Context, hash string) (*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 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
}

// UpdateMetas
func (m *Medias) UpdateMetas(ctx context.Context, id int64, metas fields.MediaMetas) error {
	meta := fields.ToJson(metas)

	tbl := table.Medias
	stmt := tbl.
		UPDATE(tbl.Metas).
		SET(meta).
		WHERE(tbl.ID.EQ(Int64(id)))
	m.log().Infof("sql: %s", stmt.DebugSql())

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

	m.log().Infof("media (%d) metas updated successfully", id)
	return nil
}

// GetRelationMedias
func (m *Medias) GetRelations(ctx context.Context, hash string) ([]*Medias, error) {
	tbl := table.Medias
	stmt := tbl.
		SELECT(tbl.AllColumns).
		WHERE(
			RawBool("metas->>'parent_hash' = ?", RawArgs{"?": hash}),
		)
	m.log().Infof("sql: %s", stmt.DebugSql())

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

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

func (m *Medias) Update(ctx context.Context) error {
	stmt := table.Medias.UPDATE(table.Medias.MutableColumns.Except(table.Medias.CreatedAt)).SET(m).WHERE(table.Medias.ID.EQ(Int(m.ID))).RETURNING(table.Medias.AllColumns)
	m.log().WithField("func", "Update").Info(stmt.DebugSql())

	if err := stmt.QueryContext(ctx, db, m); err != nil {
		m.log().WithField("func", "Update").Errorf("error updating Medias item: %v", err)
		return err
	}

	m.log().WithField("func", "Update").Infof("Medias item updated successfully")
	return nil
}