packages/docs-core/lib/UseCase/EncryptMessage.ts

   1 import type { UseCaseInterface } from '../Domain/UseCase/UseCaseInterface'
   2 import type { Result } from '@proton/docs-shared'
   3 import type { EncryptionService } from '../Services/Encryption/EncryptionService'
   4 import type { EncryptionContext } from '../Services/Encryption/EncryptionContext'
   5 import type { DocumentKeys } from '@proton/drive-store'
   6 import { GetAssociatedEncryptionDataForRealtimeMessage } from './GetAdditionalEncryptionData'
   7 import type { AnonymousEncryptionMetadata, EncryptionMetadata } from '../Types/EncryptionMetadata'
   8 import { isPrivateDocumentKeys, type PublicDocumentKeys } from '../Types/DocumentEntitlements'
   9
  10 /**
  11  * Encrypts a message directed towards the RTS.
  12  */
  13 export class EncryptMessage implements UseCaseInterface<Uint8Array> {
  14   constructor(private encryption: EncryptionService<EncryptionContext.RealtimeMessage>) {}
  15
  16   async execute(
  17     update: Uint8Array,
  18     metadata: EncryptionMetadata | AnonymousEncryptionMetadata,
  19     keys: DocumentKeys | PublicDocumentKeys,
  20   ): Promise<Result<Uint8Array>> {
  21     const aad = GetAssociatedEncryptionDataForRealtimeMessage(metadata)
  22
  23     if (isPrivateDocumentKeys(keys)) {
  24       const result = await this.encryption.signAndEncryptData(
  25         update,
  26         aad,
  27         keys.documentContentKey,
  28         keys.userAddressPrivateKey,
  29       )
  30
  31       return result
  32     } else {
  33       const result = await this.encryption.encryptAnonymousData(update, aad, keys.documentContentKey)
  34       return result
  35     }
  36   }
  37 }