packages/docs-core/lib/Api/Requests/ApiEditComment.ts

   1 import type { HttpHeaders } from '../Types/HttpHeaders'
   2 import type { EditCommentResponse } from '../Types/EditCommentResponse'
   3 import type { ApiResult } from '@proton/docs-shared'
   4 import { DocsApiPrivateRouteBuilder } from '../Routes/DocsApiPrivateRouteBuilder'
   5 import { DocsApiPublicRouteBuilder } from '../Routes/DocsApiPublicRouteBuilder'
   6 import type { RouteExecutor } from '../RouteExecutor'
   7 import type { PublicDocumentEntitlements } from '../../Types/DocumentEntitlements'
   8 import { isPublicDocumentEntitlements } from '../../Types/DocumentEntitlements'
   9 import type { DocumentEntitlements } from '../../Types/DocumentEntitlements'
  10
  11 export type EditCommentDTO = {
  12   threadId: string
  13   commentId: string
  14   encryptedContent: string
  15 }
  16
  17 export class ApiEditComment {
  18   constructor(
  19     private readonly executor: RouteExecutor,
  20     private publicContextHeaders: HttpHeaders | undefined,
  21   ) {}
  22
  23   async execute(
  24     dto: EditCommentDTO,
  25     entitlements: PublicDocumentEntitlements | DocumentEntitlements,
  26   ): Promise<ApiResult<EditCommentResponse>> {
  27     if (isPublicDocumentEntitlements(entitlements)) {
  28       return this.executePublic(dto, entitlements)
  29     } else {
  30       return this.executePrivate(dto, entitlements)
  31     }
  32   }
  33
  34   private async executePrivate(
  35     dto: EditCommentDTO,
  36     entitlements: DocumentEntitlements,
  37   ): Promise<ApiResult<EditCommentResponse>> {
  38     const { volumeId, linkId } = entitlements.nodeMeta
  39     const { threadId, commentId, encryptedContent } = dto
  40     const authorEmail = entitlements.keys.userOwnAddress
  41
  42     const route = new DocsApiPrivateRouteBuilder({ volumeId, linkId }).editComment({
  43       threadId,
  44       commentId,
  45       data: {
  46         Content: encryptedContent,
  47         AuthorEmail: authorEmail,
  48       },
  49     })
  50
  51     return this.executor.execute(route)
  52   }
  53
  54   private async executePublic(
  55     dto: EditCommentDTO,
  56     entitlements: PublicDocumentEntitlements,
  57   ): Promise<ApiResult<EditCommentResponse>> {
  58     if (!this.publicContextHeaders) {
  59       throw new Error('Public context headers not set')
  60     }
  61
  62     const { token, linkId } = entitlements.nodeMeta
  63
  64     const route = new DocsApiPublicRouteBuilder({
  65       token: token,
  66       linkId: linkId,
  67       headers: this.publicContextHeaders,
  68     }).editComment({
  69       threadId: dto.threadId,
  70       commentId: dto.commentId,
  71       data: {
  72         Content: dto.encryptedContent,
  73       },
  74     })
  75
  76     return this.executor.execute(route)
  77   }
  78 }