chore: implement memo service

2025-07-06 21:22:36 +00:00 · 2023-12-10 17:06:12 +08:00
parent add523f8a5
commit 8b13c94b22
7 changed files with 2300 additions and 0 deletions
--- a/api/v2/memo_service.go
+++ b/api/v2/memo_service.go
@ -0,0 +1,187 @@
+package v2
+
+import (
+	"context"
+	"strings"
+	"time"
+
+	"google.golang.org/grpc/codes"
+	"google.golang.org/grpc/status"
+	"google.golang.org/protobuf/types/known/timestamppb"
+
+	apiv2pb "github.com/yourselfhosted/slash/proto/gen/api/v2"
+	storepb "github.com/yourselfhosted/slash/proto/gen/store"
+	"github.com/yourselfhosted/slash/store"
+)
+
+func (s *APIV2Service) ListMemos(ctx context.Context, _ *apiv2pb.ListMemosRequest) (*apiv2pb.ListMemosResponse, error) {
+	find := &store.FindMemo{}
+	memos, err := s.Store.ListMemos(ctx, find)
+	if err != nil {
+		return nil, status.Errorf(codes.Internal, "failed to fetch memo list, err: %v", err)
+	}
+
+	composedMemos := []*apiv2pb.Memo{}
+	for _, memo := range memos {
+		composedMemo, err := s.convertMemoFromStorepb(ctx, memo)
+		if err != nil {
+			return nil, status.Errorf(codes.Internal, "failed to convert memo, err: %v", err)
+		}
+		composedMemos = append(composedMemos, composedMemo)
+	}
+
+	response := &apiv2pb.ListMemosResponse{
+		Memos: composedMemos,
+	}
+	return response, nil
+}
+
+func (s *APIV2Service) GetMemo(ctx context.Context, request *apiv2pb.GetMemoRequest) (*apiv2pb.GetMemoResponse, error) {
+	memo, err := s.Store.GetMemo(ctx, &store.FindMemo{
+		ID: &request.Id,
+	})
+	if err != nil {
+		return nil, status.Errorf(codes.Internal, "failed to get memo by ID: %v", err)
+	}
+	if memo == nil {
+		return nil, status.Errorf(codes.NotFound, "memo not found")
+	}
+
+	composedMemo, err := s.convertMemoFromStorepb(ctx, memo)
+	if err != nil {
+		return nil, status.Errorf(codes.Internal, "failed to convert memo, err: %v", err)
+	}
+	response := &apiv2pb.GetMemoResponse{
+		Memo: composedMemo,
+	}
+	return response, nil
+}
+
+func (s *APIV2Service) CreateMemo(ctx context.Context, request *apiv2pb.CreateMemoRequest) (*apiv2pb.CreateMemoResponse, error) {
+	userID := ctx.Value(userIDContextKey).(int32)
+	memo := &storepb.Memo{
+		CreatorId:  userID,
+		Name:       request.Memo.Name,
+		Title:      request.Memo.Title,
+		Content:    request.Memo.Content,
+		Tags:       request.Memo.Tags,
+		Visibility: storepb.Visibility(request.Memo.Visibility),
+	}
+	memo, err := s.Store.CreateMemo(ctx, memo)
+	if err != nil {
+		return nil, status.Errorf(codes.Internal, "failed to create memo, err: %v", err)
+	}
+
+	composedMemo, err := s.convertMemoFromStorepb(ctx, memo)
+	if err != nil {
+		return nil, status.Errorf(codes.Internal, "failed to convert memo, err: %v", err)
+	}
+	response := &apiv2pb.CreateMemoResponse{
+		Memo: composedMemo,
+	}
+	return response, nil
+}
+
+func (s *APIV2Service) UpdateMemo(ctx context.Context, request *apiv2pb.UpdateMemoRequest) (*apiv2pb.UpdateMemoResponse, error) {
+	if request.UpdateMask == nil || len(request.UpdateMask.Paths) == 0 {
+		return nil, status.Errorf(codes.InvalidArgument, "updateMask is required")
+	}
+
+	userID := ctx.Value(userIDContextKey).(int32)
+	currentUser, err := s.Store.GetUser(ctx, &store.FindUser{
+		ID: &userID,
+	})
+	if err != nil {
+		return nil, status.Errorf(codes.Internal, "failed to get current user, err: %v", err)
+	}
+	memo, err := s.Store.GetMemo(ctx, &store.FindMemo{
+		ID: &request.Memo.Id,
+	})
+	if err != nil {
+		return nil, status.Errorf(codes.Internal, "failed to get memo by ID: %v", err)
+	}
+	if memo == nil {
+		return nil, status.Errorf(codes.NotFound, "memo not found")
+	}
+	if memo.CreatorId != userID && currentUser.Role != store.RoleAdmin {
+		return nil, status.Errorf(codes.PermissionDenied, "Permission denied")
+	}
+
+	update := &store.UpdateMemo{
+		ID: memo.Id,
+	}
+	for _, path := range request.UpdateMask.Paths {
+		switch path {
+		case "name":
+			update.Name = &request.Memo.Name
+		case "title":
+			update.Title = &request.Memo.Title
+		case "content":
+			update.Content = &request.Memo.Content
+		case "tags":
+			tag := strings.Join(request.Memo.Tags, " ")
+			update.Tag = &tag
+		case "visibility":
+			visibility := store.Visibility(request.Memo.Visibility.String())
+			update.Visibility = &visibility
+		}
+	}
+	memo, err = s.Store.UpdateMemo(ctx, update)
+	if err != nil {
+		return nil, status.Errorf(codes.Internal, "failed to update memo, err: %v", err)
+	}
+
+	composedMemo, err := s.convertMemoFromStorepb(ctx, memo)
+	if err != nil {
+		return nil, status.Errorf(codes.Internal, "failed to convert memo, err: %v", err)
+	}
+	response := &apiv2pb.UpdateMemoResponse{
+		Memo: composedMemo,
+	}
+	return response, nil
+}
+
+func (s *APIV2Service) DeleteMemo(ctx context.Context, request *apiv2pb.DeleteMemoRequest) (*apiv2pb.DeleteMemoResponse, error) {
+	userID := ctx.Value(userIDContextKey).(int32)
+	currentUser, err := s.Store.GetUser(ctx, &store.FindUser{
+		ID: &userID,
+	})
+	if err != nil {
+		return nil, status.Errorf(codes.Internal, "failed to get current user, err: %v", err)
+	}
+	memo, err := s.Store.GetMemo(ctx, &store.FindMemo{
+		ID: &request.Id,
+	})
+	if err != nil {
+		return nil, status.Errorf(codes.Internal, "failed to get memo by ID: %v", err)
+	}
+	if memo == nil {
+		return nil, status.Errorf(codes.NotFound, "memo not found")
+	}
+	if memo.CreatorId != userID && currentUser.Role != store.RoleAdmin {
+		return nil, status.Errorf(codes.PermissionDenied, "Permission denied")
+	}
+
+	err = s.Store.DeleteMemo(ctx, &store.DeleteMemo{
+		ID: memo.Id,
+	})
+	if err != nil {
+		return nil, status.Errorf(codes.Internal, "failed to delete memo, err: %v", err)
+	}
+	response := &apiv2pb.DeleteMemoResponse{}
+	return response, nil
+}
+
+func (*APIV2Service) convertMemoFromStorepb(_ context.Context, memo *storepb.Memo) (*apiv2pb.Memo, error) {
+	return &apiv2pb.Memo{
+		Id:          memo.Id,
+		CreatedTime: timestamppb.New(time.Unix(memo.CreatedTs, 0)),
+		UpdatedTime: timestamppb.New(time.Unix(memo.UpdatedTs, 0)),
+		CreatorId:   memo.CreatorId,
+		Name:        memo.Name,
+		Title:       memo.Title,
+		Content:     memo.Content,
+		Tags:        memo.Tags,
+		Visibility:  apiv2pb.Visibility(memo.Visibility),
+	}, nil
+}