目录直接相关的 API

internal/domain/error.go

@@ -31,3 +31,5 @@ var ErrUserNotFound = errgo.MsgNoTrace(ErrNotFound, "user not found")
 var ErrSubjectNotCollected = errgo.MsgNoTrace(ErrNotFound, "subject is not collected by user")
 
 var ErrInput = errors.New("input not valid")
+
+var ErrExists = errors.New("item already exists")

internal/domain/index.go

@@ -23,11 +23,23 @@ import (
 
 type IndexRepo interface {
 	Get(ctx context.Context, id model.IndexID) (model.Index, error)
+	New(ctx context.Context, i *model.Index) error
+	Update(ctx context.Context, id model.IndexID, title string, desc string) error
+	Delete(ctx context.Context, id model.IndexID) error
 
 	CountSubjects(ctx context.Context, id model.IndexID, subjectType model.SubjectType) (int64, error)
 	ListSubjects(
 		ctx context.Context, id model.IndexID, subjectType model.SubjectType, limit, offset int,
 	) ([]IndexSubject, error)
+	AddIndexSubject(
+		ctx context.Context, id model.IndexID, subjectID model.SubjectID, sort uint32, comment string,
+	) (*IndexSubject, error)
+	UpdateIndexSubject(
+		ctx context.Context, id model.IndexID, subjectID model.SubjectID, sort uint32, comment string,
+	) error
+	DeleteIndexSubject(
+		ctx context.Context, id model.IndexID, subjectID model.SubjectID,
+	) error
 }
 
 type IndexSubject struct {

internal/index/mysql_repository.go

@@ -20,8 +20,10 @@ import (
 	"time"
 
 	"go.uber.org/zap"
+	"gorm.io/gen"
 	"gorm.io/gorm"
 
+	"github.com/bangumi/server/internal/dal/dao"
 	"github.com/bangumi/server/internal/dal/query"
 	"github.com/bangumi/server/internal/domain"
 	"github.com/bangumi/server/internal/model"
@@ -38,7 +40,7 @@ type mysqlRepo struct {
 	log *zap.Logger
 }
 
-func (r mysqlRepo) isNsfw(ctx context.Context, id uint32) (bool, error) {
+func (r mysqlRepo) isNsfw(ctx context.Context, id model.IndexID) (bool, error) {
 	i, err := r.q.IndexSubject.WithContext(ctx).
 		Join(r.q.Subject, r.q.IndexSubject.Sid.EqCol(r.q.Subject.ID)).
 		Where(r.q.IndexSubject.Rid.Eq(id), r.q.Subject.Nsfw.Is(true)).Count()
@@ -50,7 +52,7 @@ func (r mysqlRepo) isNsfw(ctx context.Context, id uint32) (bool, error) {
 	return i != 0, nil
 }
 
-func (r mysqlRepo) Get(ctx context.Context, id uint32) (model.Index, error) {
+func (r mysqlRepo) Get(ctx context.Context, id model.IndexID) (model.Index, error) {
 	i, err := r.q.Index.WithContext(ctx).
 		Where(r.q.Index.ID.Eq(id), r.q.Index.Ban.Is(false)).First()
 	if err != nil {
@@ -66,22 +68,45 @@ func (r mysqlRepo) Get(ctx context.Context, id uint32) (model.Index, error) {
 		return model.Index{}, err
 	}
 
-	return model.Index{
-		CreatedAt:   time.Unix(int64(i.Dateline), 0),
-		Title:       i.Title,
-		Description: i.Desc,
-		CreatorID:   i.CreatorID,
-		Total:       i.SubjectTotal,
-		ID:          id,
-		Comments:    i.Replies,
-		Collects:    i.Collects,
-		Ban:         i.Ban,
-		NSFW:        nsfw,
-	}, nil
+	ret := daoToModel(i)
+	ret.NSFW = nsfw
+	return *ret, nil
+}
+
+func (r mysqlRepo) New(ctx context.Context, i *model.Index) error {
+	dao := modelToDAO(i)
+	if err := r.q.Index.WithContext(ctx).Create(dao); err != nil {
+		return errgo.Wrap(err, "failed to create index in db")
+	}
+	i.ID = dao.ID
+	return nil
+}
+
+func (r mysqlRepo) Update(ctx context.Context, id model.IndexID, title string, desc string) error {
+	query := r.q.Index.WithContext(ctx)
+	result, err := query.Where(r.q.Index.ID.Eq(id)).Updates(dao.Index{
+		Title: title,
+		Desc:  desc,
+	})
+	return r.WrapResult(result, err, "failed to update index info")
+}
+
+func (r mysqlRepo) Delete(ctx context.Context, id model.IndexID) error {
+	return r.q.Transaction(func(tx *query.Query) error {
+		result, err := tx.Index.WithContext(ctx).Where(tx.Index.ID.Eq(id)).Delete()
+		if err = r.WrapResult(result, err, "failed to delete index"); err != nil {
+			return err
+		}
+		result, err = tx.IndexSubject.WithContext(ctx).Where(tx.IndexSubject.Rid.Eq(id)).Delete()
+		if err = r.WrapResult(result, err, "failed to delete subjects in the index"); err != nil {
+			return err
+		}
+		return nil
+	})
 }
 
 func (r mysqlRepo) CountSubjects(
-	ctx context.Context, id uint32, subjectType model.SubjectType,
+	ctx context.Context, id model.IndexID, subjectType model.SubjectType,
 ) (int64, error) {
 	q := r.q.IndexSubject.WithContext(ctx).Where(r.q.IndexSubject.Rid.Eq(id))
 	if subjectType != 0 {
@@ -98,7 +123,7 @@ func (r mysqlRepo) CountSubjects(
 
 func (r mysqlRepo) ListSubjects(
 	ctx context.Context,
-	id uint32,
+	id model.IndexID,
 	subjectType model.SubjectType,
 	limit, offset int,
 ) ([]domain.IndexSubject, error) {
@@ -130,3 +155,160 @@ func (r mysqlRepo) ListSubjects(
 
 	return results, nil
 }
+
+func (r mysqlRepo) AddIndexSubject(
+	ctx context.Context, id model.IndexID,
+	subjectID model.SubjectID, sort uint32, comment string,
+) (*domain.IndexSubject, error) {
+	var err error
+
+	index, err := r.Get(ctx, id)
+	if err != nil {
+		return nil, err
+	}
+
+	subjectDO, err := r.q.Subject.WithContext(ctx).
+		Where(r.q.Subject.ID.Eq(subjectID)).
+		First()
+
+	if err != nil {
+		if errors.Is(err, gorm.ErrRecordNotFound) {
+			return nil, domain.ErrSubjectNotFound
+		}
+		return nil, errgo.Wrap(err, "dal")
+	}
+
+	now := time.Now()
+
+	indexSubject, err := r.q.IndexSubject.WithContext(ctx).
+		Where(r.q.IndexSubject.Rid.Eq(id), r.q.IndexSubject.Sid.Eq(uint32(subjectID))).First()
+
+	if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
+		return nil, errgo.Wrap(err, "dal")
+	}
+
+	if indexSubject != nil && indexSubject.ID != 0 {
+		return nil, domain.ErrExists
+	}
+
+	err = r.addSubjectToIndex(ctx, index, &dao.IndexSubject{
+		Comment:  comment,
+		Order:    sort,
+		Dateline: uint32(now.Unix()),
+		Type:     subjectDO.TypeID,
+		Rid:      id,
+		Sid:      uint32(subjectID),
+	})
+
+	if err != nil {
+		return nil, errgo.Wrap(err, "dal")
+	}
+
+	subject, err := subject.ConvertDao(subjectDO)
+	if err != nil {
+		return nil, errgo.Wrap(err, "subject.ConvertDao")
+	}
+
+	return &domain.IndexSubject{
+		AddedAt: now,
+		Comment: comment,
+		Subject: subject,
+	}, nil
+}
+
+func (r mysqlRepo) addSubjectToIndex(ctx context.Context, index model.Index, subject *dao.IndexSubject) error {
+	return r.q.Transaction(func(tx *query.Query) error {
+		err := r.q.IndexSubject.WithContext(ctx).Create(subject)
+		if err != nil {
+			return errgo.Wrap(err, "failed to create subject in index")
+		}
+
+		result, err := r.q.Index.WithContext(ctx).
+			Where(r.q.Index.ID.Eq(index.ID)).
+			Updates(dao.Index{
+				Lasttouch:    uint32(time.Now().Unix()),
+				SubjectTotal: index.Total + 1,
+			})
+
+		return r.WrapResult(result, err, "failed to update index info")
+	})
+}
+
+func (r mysqlRepo) UpdateIndexSubject(
+	ctx context.Context, id model.IndexID, subjectID model.SubjectID, sort uint32, comment string,
+) error {
+	q := r.q.IndexSubject
+	result, err := q.WithContext(ctx).
+		Where(q.Rid.Eq(id), q.Sid.Eq(uint32(subjectID))).
+		Updates(dao.IndexSubject{
+			Comment: comment,
+			Order:   sort,
+		})
+	return r.WrapResult(result, err, "failed to update index subject")
+}
+
+func (r mysqlRepo) DeleteIndexSubject(
+	ctx context.Context, id model.IndexID, subjectID model.SubjectID,
+) error {
+	return r.q.Transaction(func(tx *query.Query) error {
+		index, err := r.Get(ctx, id)
+		if err != nil {
+			return err
+		}
+		result, err := r.q.IndexSubject.WithContext(ctx).
+			Where(r.q.IndexSubject.Rid.Eq(id), r.q.IndexSubject.Sid.Eq(uint32(subjectID))).
+			Delete()
+		if err = r.WrapResult(result, err, "failed to delete index subject"); err != nil {
+			return err
+		}
+		result, err = r.q.Index.WithContext(ctx).Where(r.q.Index.ID.Eq(id)).Updates(dao.Index{
+			SubjectTotal: index.Total - 1,
+		})
+		if err = r.WrapResult(result, err, "failed to update index info"); err != nil {
+			return err
+		}
+		return nil
+	})
+}
+
+func (r mysqlRepo) WrapResult(result gen.ResultInfo, err error, msg string) error {
+	if err != nil {
+		if errors.Is(err, gorm.ErrRecordNotFound) {
+			return domain.ErrNotFound
+		}
+		return errgo.Wrap(err, msg)
+	}
+	if result.Error != nil {
+		return result.Error
+	}
+	return nil
+}
+
+func daoToModel(index *dao.Index) *model.Index {
+	return &model.Index{
+		ID:          index.ID,
+		Title:       index.Title,
+		Description: index.Desc,
+		CreatorID:   index.CreatorID,
+		Total:       index.SubjectTotal,
+		Comments:    index.Replies,
+		Collects:    index.Collects,
+		Ban:         index.Ban,
+		NSFW:        false, // check nsfw outside of this function
+		CreatedAt:   time.Unix(int64(index.Dateline), 0),
+		UpdateAt:    time.Unix(int64(index.Lasttouch), 0),
+	}
+}
+
+func modelToDAO(index *model.Index) *dao.Index {
+	return &dao.Index{
+		ID:        index.ID,
+		Type:      0,
+		Title:     index.Title,
+		Desc:      index.Description,
+		CreatorID: index.CreatorID,
+		Ban:       index.Ban,
+		Dateline:  int32(index.CreatedAt.Unix()),
+		Lasttouch: uint32(index.UpdateAt.Unix()),
+	}
+}