0.29.1原版
This commit is contained in:
@@ -0,0 +1,74 @@
|
||||
package mcp
|
||||
|
||||
import (
|
||||
"context"
|
||||
|
||||
"github.com/pkg/errors"
|
||||
|
||||
v1pb "github.com/usememos/memos/proto/gen/api/v1"
|
||||
apiv1 "github.com/usememos/memos/server/router/api/v1"
|
||||
"github.com/usememos/memos/store"
|
||||
)
|
||||
|
||||
func visibilityToProto(visibility store.Visibility) v1pb.Visibility {
|
||||
switch visibility {
|
||||
case store.Protected:
|
||||
return v1pb.Visibility_PROTECTED
|
||||
case store.Public:
|
||||
return v1pb.Visibility_PUBLIC
|
||||
default:
|
||||
return v1pb.Visibility_PRIVATE
|
||||
}
|
||||
}
|
||||
|
||||
func rowStatusToProto(rowStatus store.RowStatus) v1pb.State {
|
||||
switch rowStatus {
|
||||
case store.Archived:
|
||||
return v1pb.State_ARCHIVED
|
||||
default:
|
||||
return v1pb.State_NORMAL
|
||||
}
|
||||
}
|
||||
|
||||
func (s *MCPService) loadMemoJSONByName(ctx context.Context, name string) (memoJSON, error) {
|
||||
uid, err := parseMemoUID(name)
|
||||
if err != nil {
|
||||
return memoJSON{}, err
|
||||
}
|
||||
memo, err := s.store.GetMemo(ctx, &store.FindMemo{UID: &uid})
|
||||
if err != nil {
|
||||
return memoJSON{}, errors.Wrap(err, "failed to get memo")
|
||||
}
|
||||
if memo == nil {
|
||||
return memoJSON{}, errors.New("memo not found")
|
||||
}
|
||||
return storeMemoToJSONWithStore(ctx, s.store, memo)
|
||||
}
|
||||
|
||||
func (s *MCPService) loadReactionJSONByID(ctx context.Context, reactionID int32) (reactionJSON, error) {
|
||||
reaction, err := s.store.GetReaction(ctx, &store.FindReaction{ID: &reactionID})
|
||||
if err != nil {
|
||||
return reactionJSON{}, errors.Wrap(err, "failed to get reaction")
|
||||
}
|
||||
if reaction == nil {
|
||||
return reactionJSON{}, errors.New("reaction not found")
|
||||
}
|
||||
creator, err := lookupUsername(ctx, s.store, reaction.CreatorID)
|
||||
if err != nil {
|
||||
return reactionJSON{}, errors.Wrap(err, "failed to resolve reaction creator")
|
||||
}
|
||||
return reactionJSON{
|
||||
ID: reaction.ID,
|
||||
Creator: creator,
|
||||
ReactionType: reaction.ReactionType,
|
||||
CreateTime: reaction.CreatedTs,
|
||||
}, nil
|
||||
}
|
||||
|
||||
func (s *MCPService) loadReactionJSONByName(ctx context.Context, name string) (reactionJSON, error) {
|
||||
_, reactionID, err := apiv1.ExtractMemoReactionIDFromName(name)
|
||||
if err != nil {
|
||||
return reactionJSON{}, err
|
||||
}
|
||||
return s.loadReactionJSONByID(ctx, reactionID)
|
||||
}
|
||||
Reference in New Issue
Block a user