note_class.go

package note_class

import (
	"context"
	"github.com/iimeta/iim-client/internal/dao"
	"github.com/iimeta/iim-client/internal/errors"
	"github.com/iimeta/iim-client/internal/model"
	"github.com/iimeta/iim-client/internal/service"
	"github.com/iimeta/iim-client/utility/logger"
)

type sNoteClass struct{}

func init() {
	service.RegisterNoteClass(New())
}

func New() service.INoteClass {
	return &sNoteClass{}
}

// 分类列表
func (s *sNoteClass) List(ctx context.Context) (*model.ClassListRes, error) {

	noteClassList, err := dao.NoteClass.List(ctx, service.Session().GetUid(ctx))
	if err != nil {
		logger.Error(ctx, err)
		return nil, err
	}

	data, err := dao.NoteClass.GroupCount(ctx, service.Session().GetUid(ctx))
	if err != nil {
		logger.Error(ctx, err)
		return nil, err
	}

	items := make([]*model.NoteClass, 0)
	items = append(items, &model.NoteClass{
		ClassName: "默认分类",
	})

	for _, noteClass := range noteClassList {

		classItem := &model.NoteClass{
			Id:        noteClass.Id,
			ClassName: noteClass.ClassName,
			IsDefault: noteClass.IsDefault,
		}

		if num, ok := data[noteClass.Id]; ok {
			classItem.Count = num
		}

		items = append(items, classItem)
	}

	return &model.ClassListRes{
		Items: items,
		Paginate: &model.Paginate{
			Page:  1,
			Size:  100000,
			Total: len(items),
		},
	}, nil
}

// 添加或修改分类
func (s *sNoteClass) Edit(ctx context.Context, params model.ClassEditReq) (*model.ClassEditRes, error) {

	uid := service.Session().GetUid(ctx)

	if params.ClassId == "" {

		id, err := dao.NoteClass.Create(ctx, uid, params.ClassName)
		if err != nil {
			logger.Error(ctx, err)
			return nil, errors.New("笔记分类创建失败")
		}

		params.ClassId = id

	} else {
		if err := dao.NoteClass.Update(ctx, uid, params.ClassId, params.ClassName); err != nil {
			logger.Error(ctx, err)
			return nil, errors.New("笔记分类编辑失败")
		}
	}

	return &model.ClassEditRes{
		Id: params.ClassId,
	}, nil
}

// 删除分类
func (s *sNoteClass) Delete(ctx context.Context, params model.ClassDeleteReq) error {

	err := dao.NoteClass.Delete(ctx, service.Session().GetUid(ctx), params.ClassId)
	if err != nil {
		logger.Error(ctx, err)
		return err
	}

	return nil
}

// 分类排序
func (s *sNoteClass) Sort(ctx context.Context, params model.ClassSortReq) error {

	err := dao.NoteClass.Sort(ctx, service.Session().GetUid(ctx), params.ClassId, params.SortType)
	if err != nil {
		logger.Error(ctx, err)
		return err
	}

	return nil
}