init commit
This commit is contained in:
126
server/router/api/v1/activity_service.go
Normal file
126
server/router/api/v1/activity_service.go
Normal file
@@ -0,0 +1,126 @@
|
||||
package v1
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"time"
|
||||
|
||||
"google.golang.org/grpc/codes"
|
||||
"google.golang.org/grpc/status"
|
||||
"google.golang.org/protobuf/types/known/timestamppb"
|
||||
|
||||
v1pb "github.com/usememos/memos/proto/gen/api/v1"
|
||||
storepb "github.com/usememos/memos/proto/gen/store"
|
||||
"github.com/usememos/memos/store"
|
||||
)
|
||||
|
||||
func (s *APIV1Service) ListActivities(ctx context.Context, request *v1pb.ListActivitiesRequest) (*v1pb.ListActivitiesResponse, error) {
|
||||
// Set default page size if not specified
|
||||
pageSize := request.PageSize
|
||||
if pageSize <= 0 || pageSize > 1000 {
|
||||
pageSize = 100
|
||||
}
|
||||
|
||||
// TODO: Implement pagination with page_token and use pageSize for limiting
|
||||
// For now, we'll fetch all activities and the pageSize will be used in future pagination implementation
|
||||
_ = pageSize // Acknowledge pageSize variable to avoid linter warning
|
||||
|
||||
activities, err := s.Store.ListActivities(ctx, &store.FindActivity{})
|
||||
if err != nil {
|
||||
return nil, status.Errorf(codes.Internal, "failed to list activities: %v", err)
|
||||
}
|
||||
|
||||
var activityMessages []*v1pb.Activity
|
||||
for _, activity := range activities {
|
||||
activityMessage, err := s.convertActivityFromStore(ctx, activity)
|
||||
if err != nil {
|
||||
return nil, status.Errorf(codes.Internal, "failed to convert activity from store: %v", err)
|
||||
}
|
||||
activityMessages = append(activityMessages, activityMessage)
|
||||
}
|
||||
|
||||
return &v1pb.ListActivitiesResponse{
|
||||
Activities: activityMessages,
|
||||
// TODO: Implement next_page_token for pagination
|
||||
}, nil
|
||||
}
|
||||
|
||||
func (s *APIV1Service) GetActivity(ctx context.Context, request *v1pb.GetActivityRequest) (*v1pb.Activity, error) {
|
||||
activityID, err := ExtractActivityIDFromName(request.Name)
|
||||
if err != nil {
|
||||
return nil, status.Errorf(codes.InvalidArgument, "invalid activity name: %v", err)
|
||||
}
|
||||
activity, err := s.Store.GetActivity(ctx, &store.FindActivity{
|
||||
ID: &activityID,
|
||||
})
|
||||
if err != nil {
|
||||
return nil, status.Errorf(codes.Internal, "failed to get activity: %v", err)
|
||||
}
|
||||
|
||||
activityMessage, err := s.convertActivityFromStore(ctx, activity)
|
||||
if err != nil {
|
||||
return nil, status.Errorf(codes.Internal, "failed to convert activity from store: %v", err)
|
||||
}
|
||||
return activityMessage, nil
|
||||
}
|
||||
|
||||
func (s *APIV1Service) convertActivityFromStore(ctx context.Context, activity *store.Activity) (*v1pb.Activity, error) {
|
||||
payload, err := s.convertActivityPayloadFromStore(ctx, activity.Payload)
|
||||
if err != nil {
|
||||
return nil, status.Errorf(codes.Internal, "failed to convert activity payload from store: %v", err)
|
||||
}
|
||||
|
||||
// Convert store activity type to proto enum
|
||||
var activityType v1pb.Activity_Type
|
||||
switch activity.Type {
|
||||
case store.ActivityTypeMemoComment:
|
||||
activityType = v1pb.Activity_MEMO_COMMENT
|
||||
default:
|
||||
activityType = v1pb.Activity_TYPE_UNSPECIFIED
|
||||
}
|
||||
|
||||
// Convert store activity level to proto enum
|
||||
var activityLevel v1pb.Activity_Level
|
||||
switch activity.Level {
|
||||
case store.ActivityLevelInfo:
|
||||
activityLevel = v1pb.Activity_INFO
|
||||
default:
|
||||
activityLevel = v1pb.Activity_LEVEL_UNSPECIFIED
|
||||
}
|
||||
|
||||
return &v1pb.Activity{
|
||||
Name: fmt.Sprintf("%s%d", ActivityNamePrefix, activity.ID),
|
||||
Creator: fmt.Sprintf("%s%d", UserNamePrefix, activity.CreatorID),
|
||||
Type: activityType,
|
||||
Level: activityLevel,
|
||||
CreateTime: timestamppb.New(time.Unix(activity.CreatedTs, 0)),
|
||||
Payload: payload,
|
||||
}, nil
|
||||
}
|
||||
|
||||
func (s *APIV1Service) convertActivityPayloadFromStore(ctx context.Context, payload *storepb.ActivityPayload) (*v1pb.ActivityPayload, error) {
|
||||
v2Payload := &v1pb.ActivityPayload{}
|
||||
if payload.MemoComment != nil {
|
||||
memo, err := s.Store.GetMemo(ctx, &store.FindMemo{
|
||||
ID: &payload.MemoComment.MemoId,
|
||||
ExcludeContent: true,
|
||||
})
|
||||
if err != nil {
|
||||
return nil, status.Errorf(codes.Internal, "failed to get memo: %v", err)
|
||||
}
|
||||
relatedMemo, err := s.Store.GetMemo(ctx, &store.FindMemo{
|
||||
ID: &payload.MemoComment.RelatedMemoId,
|
||||
ExcludeContent: true,
|
||||
})
|
||||
if err != nil {
|
||||
return nil, status.Errorf(codes.Internal, "failed to get related memo: %v", err)
|
||||
}
|
||||
v2Payload.Payload = &v1pb.ActivityPayload_MemoComment{
|
||||
MemoComment: &v1pb.ActivityMemoCommentPayload{
|
||||
Memo: fmt.Sprintf("%s%s", MemoNamePrefix, memo.UID),
|
||||
RelatedMemo: fmt.Sprintf("%s%s", MemoNamePrefix, relatedMemo.UID),
|
||||
},
|
||||
}
|
||||
}
|
||||
return v2Payload, nil
|
||||
}
|
||||
Reference in New Issue
Block a user