123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119 |
- package v1
- import (
- "context"
- "fmt"
- "time"
- "google.golang.org/grpc/codes"
- "google.golang.org/grpc/status"
- "google.golang.org/protobuf/types/known/emptypb"
- "google.golang.org/protobuf/types/known/timestamppb"
- v1pb "github.com/usememos/memos/proto/gen/api/v1"
- "github.com/usememos/memos/store"
- )
- func (s *APIV1Service) ListInboxes(ctx context.Context, _ *v1pb.ListInboxesRequest) (*v1pb.ListInboxesResponse, error) {
- user, err := s.GetCurrentUser(ctx)
- 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 := &v1pb.ListInboxesResponse{
- Inboxes: []*v1pb.Inbox{},
- }
- for _, inbox := range inboxes {
- inboxMessage := convertInboxFromStore(inbox)
- if inboxMessage.Type == v1pb.Inbox_TYPE_UNSPECIFIED {
- continue
- }
- response.Inboxes = append(response.Inboxes, inboxMessage)
- }
- return response, nil
- }
- func (s *APIV1Service) UpdateInbox(ctx context.Context, request *v1pb.UpdateInboxRequest) (*v1pb.Inbox, 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 == v1pb.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 convertInboxFromStore(inbox), nil
- }
- func (s *APIV1Service) DeleteInbox(ctx context.Context, request *v1pb.DeleteInboxRequest) (*emptypb.Empty, 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 &emptypb.Empty{}, nil
- }
- func convertInboxFromStore(inbox *store.Inbox) *v1pb.Inbox {
- return &v1pb.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: v1pb.Inbox_Type(inbox.Message.Type),
- ActivityId: inbox.Message.ActivityId,
- }
- }
- func convertInboxStatusFromStore(status store.InboxStatus) v1pb.Inbox_Status {
- switch status {
- case store.UNREAD:
- return v1pb.Inbox_UNREAD
- case store.ARCHIVED:
- return v1pb.Inbox_ARCHIVED
- default:
- return v1pb.Inbox_STATUS_UNSPECIFIED
- }
- }
- func convertInboxStatusToStore(status v1pb.Inbox_Status) store.InboxStatus {
- switch status {
- case v1pb.Inbox_UNREAD:
- return store.UNREAD
- case v1pb.Inbox_ARCHIVED:
- return store.ARCHIVED
- default:
- return store.UNREAD
- }
- }
|