mirror of
https://github.com/usememos/memos.git
synced 2024-11-16 03:34:33 +08:00
116 lines
3.4 KiB
Go
116 lines
3.4 KiB
Go
package v2
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"time"
|
|
|
|
"google.golang.org/grpc/codes"
|
|
"google.golang.org/grpc/status"
|
|
"google.golang.org/protobuf/types/known/timestamppb"
|
|
|
|
apiv2pb "github.com/usememos/memos/proto/gen/api/v2"
|
|
"github.com/usememos/memos/store"
|
|
)
|
|
|
|
func (s *APIV2Service) ListInboxes(ctx context.Context, _ *apiv2pb.ListInboxesRequest) (*apiv2pb.ListInboxesResponse, error) {
|
|
user, err := getCurrentUser(ctx, s.Store)
|
|
if err != nil {
|
|
return nil, status.Errorf(codes.Internal, "failed to get user")
|
|
}
|
|
|
|
inboxes, err := s.Store.ListInboxes(ctx, &store.FindInbox{
|
|
ReceiverID: &user.ID,
|
|
})
|
|
if err != nil {
|
|
return nil, status.Errorf(codes.Internal, "failed to list inbox: %v", err)
|
|
}
|
|
|
|
response := &apiv2pb.ListInboxesResponse{
|
|
Inboxes: []*apiv2pb.Inbox{},
|
|
}
|
|
for _, inbox := range inboxes {
|
|
response.Inboxes = append(response.Inboxes, convertInboxFromStore(inbox))
|
|
}
|
|
|
|
return response, nil
|
|
}
|
|
|
|
func (s *APIV2Service) UpdateInbox(ctx context.Context, request *apiv2pb.UpdateInboxRequest) (*apiv2pb.UpdateInboxResponse, error) {
|
|
if request.UpdateMask == nil || len(request.UpdateMask.Paths) == 0 {
|
|
return nil, status.Errorf(codes.InvalidArgument, "update mask is required")
|
|
}
|
|
|
|
inboxID, err := ExtractInboxIDFromName(request.Inbox.Name)
|
|
if err != nil {
|
|
return nil, status.Errorf(codes.InvalidArgument, "invalid inbox name: %v", err)
|
|
}
|
|
update := &store.UpdateInbox{
|
|
ID: inboxID,
|
|
}
|
|
for _, field := range request.UpdateMask.Paths {
|
|
if field == "status" {
|
|
if request.Inbox.Status == apiv2pb.Inbox_STATUS_UNSPECIFIED {
|
|
return nil, status.Errorf(codes.InvalidArgument, "status is required")
|
|
}
|
|
update.Status = convertInboxStatusToStore(request.Inbox.Status)
|
|
}
|
|
}
|
|
|
|
inbox, err := s.Store.UpdateInbox(ctx, update)
|
|
if err != nil {
|
|
return nil, status.Errorf(codes.Internal, "failed to update inbox: %v", err)
|
|
}
|
|
|
|
return &apiv2pb.UpdateInboxResponse{
|
|
Inbox: convertInboxFromStore(inbox),
|
|
}, nil
|
|
}
|
|
|
|
func (s *APIV2Service) DeleteInbox(ctx context.Context, request *apiv2pb.DeleteInboxRequest) (*apiv2pb.DeleteInboxResponse, error) {
|
|
inboxID, err := ExtractInboxIDFromName(request.Name)
|
|
if err != nil {
|
|
return nil, status.Errorf(codes.InvalidArgument, "invalid inbox name: %v", err)
|
|
}
|
|
|
|
if err := s.Store.DeleteInbox(ctx, &store.DeleteInbox{
|
|
ID: inboxID,
|
|
}); err != nil {
|
|
return nil, status.Errorf(codes.Internal, "failed to update inbox: %v", err)
|
|
}
|
|
return &apiv2pb.DeleteInboxResponse{}, nil
|
|
}
|
|
|
|
func convertInboxFromStore(inbox *store.Inbox) *apiv2pb.Inbox {
|
|
return &apiv2pb.Inbox{
|
|
Name: fmt.Sprintf("%s%d", InboxNamePrefix, inbox.ID),
|
|
Sender: fmt.Sprintf("%s%d", UserNamePrefix, inbox.SenderID),
|
|
Receiver: fmt.Sprintf("%s%d", UserNamePrefix, inbox.ReceiverID),
|
|
Status: convertInboxStatusFromStore(inbox.Status),
|
|
CreateTime: timestamppb.New(time.Unix(inbox.CreatedTs, 0)),
|
|
Type: apiv2pb.Inbox_Type(inbox.Message.Type),
|
|
ActivityId: inbox.Message.ActivityId,
|
|
}
|
|
}
|
|
|
|
func convertInboxStatusFromStore(status store.InboxStatus) apiv2pb.Inbox_Status {
|
|
switch status {
|
|
case store.UNREAD:
|
|
return apiv2pb.Inbox_UNREAD
|
|
case store.ARCHIVED:
|
|
return apiv2pb.Inbox_ARCHIVED
|
|
default:
|
|
return apiv2pb.Inbox_STATUS_UNSPECIFIED
|
|
}
|
|
}
|
|
|
|
func convertInboxStatusToStore(status apiv2pb.Inbox_Status) store.InboxStatus {
|
|
switch status {
|
|
case apiv2pb.Inbox_UNREAD:
|
|
return store.UNREAD
|
|
case apiv2pb.Inbox_ARCHIVED:
|
|
return store.ARCHIVED
|
|
default:
|
|
return store.UNREAD
|
|
}
|
|
}
|