feat: add medias

2024-12-02 17:48:46 +08:00
parent 2b4cfb1a1e
commit 9e7b35e3c9
17 changed files with 806 additions and 11 deletions
--- a/backend/modules/medias/service.go
+++ b/backend/modules/medias/service.go
@@ -0,0 +1,164 @@
+package medias
+
+import (
+	"context"
+	"database/sql"
+
+	"backend/database/models/qvyun/public/model"
+	"backend/database/models/qvyun/public/table"
+
+	. "github.com/go-jet/jet/v2/postgres"
+	"github.com/pkg/errors"
+	"github.com/samber/lo"
+	"github.com/sirupsen/logrus"
+)
+
+// @provider:except
+type Service struct {
+	db  *sql.DB
+	log *logrus.Entry `inject:"false"`
+}
+
+func (svc *Service) Prepare() error {
+	svc.log = logrus.WithField("module", "medias.service")
+	return nil
+}
+
+// GetByID
+func (svc *Service) GetByID(ctx context.Context, id int64) (*ListItem, error) {
+	log := svc.log.WithField("method", "GetByID")
+
+	tbl := table.Medias
+	stmt := tbl.SELECT(tbl.AllColumns).WHERE(tbl.ID.EQ(Int(id)))
+	log.Debug(stmt.Sql())
+
+	var media ListItem
+	if err := stmt.QueryContext(ctx, svc.db, &media); err != nil {
+		return nil, errors.Wrap(err, "query media by id")
+	}
+
+	return &media, nil
+}
+
+// Decorate List Resources
+func (svc *Service) DecorateListResources(ctx context.Context, tenantId, userId int64, items []ListItem) ([]ListItem, error) {
+	log := svc.log.WithField("method", "DecorateListResources")
+
+	mediaIDs := make([]int64, len(items))
+	for i, item := range items {
+		mediaIDs[i] = item.ID
+	}
+
+	tbl := table.MediaResources
+	stmt := tbl.
+		SELECT(
+			tbl.MediaID,
+			tbl.Type,
+			tbl.Size,
+		).
+		WHERE(tbl.MediaID.IN(lo.Map(mediaIDs, func(item int64, _ int) Expression {
+			return Int(item)
+		})...))
+	log.Debug(stmt.DebugSql())
+
+	var resources []model.MediaResources
+	if err := stmt.QueryContext(ctx, svc.db, &resources); err != nil {
+		return nil, errors.Wrap(err, "query media resources")
+	}
+
+	if len(resources) == 0 {
+		return nil, nil
+	}
+
+	// group resources by media id
+	resourcesMap := make(map[int64][]model.MediaResources)
+	for _, resource := range resources {
+		if _, ok := resourcesMap[resource.MediaID]; !ok {
+			resourcesMap[resource.MediaID] = make([]model.MediaResources, 0)
+		}
+		resourcesMap[resource.MediaID] = append(resourcesMap[resource.MediaID], resource)
+	}
+
+	//  set resources to items
+	for i, item := range items {
+		if resources, ok := resourcesMap[item.ID]; ok {
+			items[i].MediaResources = resources
+		}
+	}
+
+	return items, nil
+}
+
+// List
+func (svc *Service) List(ctx context.Context, tenantId, userId int64, filter *ListFilter) ([]ListItem, error) {
+	log := svc.log.WithField("method", "List")
+
+	tbl := table.Medias
+	stmt := tbl.
+		SELECT(tbl.AllColumns).
+		WHERE(tbl.TenantID.EQ(Int(tenantId))).
+		ORDER_BY(tbl.ID.DESC())
+
+	if filter.Title != nil && *filter.Title != "" {
+		stmt = stmt.WHERE(tbl.Title.LIKE(String("%" + *filter.Title + "%")))
+	}
+
+	if filter.Bought != nil && *filter.Bought {
+		boughtIDs, err := svc.GetUserBoughtMedias(ctx, tenantId, userId)
+		if err != nil {
+			return nil, errors.Wrap(err, "get user bought medias")
+		}
+
+		if len(boughtIDs) > 0 {
+			stmt = stmt.
+				WHERE(tbl.ID.IN(lo.Map(boughtIDs, func(item int64, _ int) Expression {
+					return Int(item)
+				})...))
+		}
+	} else {
+		stmt = stmt.WHERE(tbl.Publish.EQ(Bool(true)))
+	}
+
+	if filter.OffsetID > 0 {
+		if filter.Action == 0 {
+			stmt = stmt.WHERE(tbl.ID.GT(Int(filter.OffsetID)))
+		}
+
+		if filter.Action == 1 {
+			stmt = stmt.WHERE(tbl.ID.LT(Int(filter.OffsetID)))
+			stmt = stmt.LIMIT(10)
+		}
+	} else {
+		stmt = stmt.LIMIT(10)
+	}
+	log.Debug(stmt.DebugSql())
+
+	var dest []ListItem
+	if err := stmt.QueryContext(ctx, svc.db, &dest); err != nil {
+		return nil, errors.Wrap(err, "query medias")
+	}
+
+	return dest, nil
+}
+
+// GetUserBoughtMedias
+func (svc *Service) GetUserBoughtMedias(ctx context.Context, tenant int64, userID int64) ([]int64, error) {
+	log := svc.log.WithField("method", "GetUserBoughtMedias")
+
+	tbl := table.UserMedias
+	stmt := tbl.
+		SELECT(tbl.MediaID).
+		WHERE(
+			tbl.TenantID.EQ(Int(tenant)).AND(
+				tbl.UserID.EQ(Int(userID)),
+			),
+		)
+	log.Debug(stmt.Sql())
+
+	var mediaIDs []int64
+	if err := stmt.QueryContext(ctx, svc.db, &mediaIDs); err != nil {
+		return nil, errors.Wrap(err, "query user bought medias")
+	}
+
+	return mediaIDs, nil
+}