136 lines
4.6 KiB
Go
136 lines
4.6 KiB
Go
package v1
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/pkg/errors"
|
|
"google.golang.org/grpc/codes"
|
|
"google.golang.org/grpc/status"
|
|
|
|
"github.com/usememos/memos/internal/webhook"
|
|
v1pb "github.com/usememos/memos/proto/gen/api/v1"
|
|
"github.com/usememos/memos/server/access"
|
|
"github.com/usememos/memos/server/auth"
|
|
"github.com/usememos/memos/store"
|
|
)
|
|
|
|
// DispatchMemoCreatedWebhook dispatches a webhook when a memo is created.
|
|
func (s *APIV1Service) DispatchMemoCreatedWebhook(ctx context.Context, memo *v1pb.Memo) error {
|
|
return s.dispatchMemoRelatedWebhook(ctx, memo, "memos.memo.created")
|
|
}
|
|
|
|
// DispatchMemoUpdatedWebhook dispatches webhook when memo is updated.
|
|
func (s *APIV1Service) DispatchMemoUpdatedWebhook(ctx context.Context, memo *v1pb.Memo) error {
|
|
return s.dispatchMemoRelatedWebhook(ctx, memo, "memos.memo.updated")
|
|
}
|
|
|
|
// DispatchMemoDeletedWebhook dispatches webhook when memo is deleted.
|
|
func (s *APIV1Service) DispatchMemoDeletedWebhook(ctx context.Context, memo *v1pb.Memo) error {
|
|
return s.dispatchMemoRelatedWebhook(ctx, memo, "memos.memo.deleted")
|
|
}
|
|
|
|
// DispatchMemoCommentCreatedWebhook dispatches webhook to the related memo owner when a comment is created.
|
|
func (s *APIV1Service) DispatchMemoCommentCreatedWebhook(
|
|
ctx context.Context,
|
|
comment, relatedMemo *store.Memo,
|
|
relatedMemoCreatorID int32,
|
|
) error {
|
|
receiver, err := s.Store.GetUser(ctx, &store.FindUser{ID: &relatedMemoCreatorID})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if receiver == nil || receiver.RowStatus != store.Normal {
|
|
return nil
|
|
}
|
|
for _, subject := range []*store.Memo{comment, relatedMemo} {
|
|
readContext, err := s.buildMemoReadContextForViewer(ctx, subject, receiver, false, nil)
|
|
if err != nil {
|
|
return errors.Wrap(err, "failed to resolve webhook subject access")
|
|
}
|
|
decision := access.CheckMemoReadContext(readContext)
|
|
if !decision.Allowed() {
|
|
return nil
|
|
}
|
|
}
|
|
// The request context belongs to the comment author, but the webhook belongs
|
|
// to the context memo's author. Build the prepared payload under the receiver's
|
|
// identity before it enters the existing asynchronous delivery queue.
|
|
receiverCtx := auth.SetUserInContext(ctx, receiver, "")
|
|
reactions, err := s.Store.ListReactions(receiverCtx, &store.FindReaction{MemoID: &comment.ID})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
attachments, err := s.Store.ListAttachments(receiverCtx, &store.FindAttachment{MemoID: &comment.ID})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
relations, err := s.loadMemoRelations(receiverCtx, comment)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
commentMessage, err := s.convertMemoFromStore(receiverCtx, comment, reactions, attachments, relations)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
webhooks, err := s.Store.GetUserWebhooks(ctx, relatedMemoCreatorID)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
for _, hook := range webhooks {
|
|
payload, err := convertMemoToWebhookPayload(commentMessage)
|
|
if err != nil {
|
|
return errors.Wrap(err, "failed to convert memo to webhook payload")
|
|
}
|
|
payload.ActivityType = "memos.memo.comment.created"
|
|
payload.URL = hook.Url
|
|
payload.SigningSecret = hook.SigningSecret
|
|
webhook.PostAsync(payload)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (s *APIV1Service) dispatchMemoRelatedWebhook(ctx context.Context, memo *v1pb.Memo, activityType string) error {
|
|
creator, err := ResolveUserByName(ctx, s.Store, memo.Creator)
|
|
if err != nil {
|
|
return status.Errorf(codes.InvalidArgument, "invalid memo creator")
|
|
}
|
|
if creator == nil {
|
|
return status.Errorf(codes.NotFound, "memo creator not found")
|
|
}
|
|
creatorID := creator.ID
|
|
webhooks, err := s.Store.GetUserWebhooks(ctx, creatorID)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
for _, hook := range webhooks {
|
|
payload, err := convertMemoToWebhookPayload(memo)
|
|
if err != nil {
|
|
return errors.Wrap(err, "failed to convert memo to webhook payload")
|
|
}
|
|
payload.ActivityType = activityType
|
|
payload.URL = hook.Url
|
|
payload.SigningSecret = hook.SigningSecret
|
|
webhook.PostAsync(payload)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func convertMemoToWebhookPayload(memo *v1pb.Memo) (*webhook.WebhookRequestPayload, error) {
|
|
return &webhook.WebhookRequestPayload{
|
|
Creator: memo.Creator,
|
|
Memo: memo,
|
|
}, nil
|
|
}
|
|
|
|
func (s *APIV1Service) getMemoContentSnippet(content string) (string, error) {
|
|
// Use goldmark service for snippet generation
|
|
snippet, err := s.MarkdownService.GenerateSnippet([]byte(content), 64)
|
|
if err != nil {
|
|
return "", errors.Wrap(err, "failed to generate snippet")
|
|
}
|
|
return snippet, nil
|
|
}
|
|
|
|
// parseMemoOrderBy parses the order_by field and sets the appropriate ordering in memoFind.
|
|
// Follows AIP-132: supports comma-separated list of fields with optional "desc" suffix.
|
|
// Example: "pinned desc, create_time desc" or "update_time asc".
|