standardnotes-app-web/packages/snjs/mocha/lib/VaultsContext.js

import { AppContext } from './AppContext.js'
import * as Collaboration from './Collaboration.js'

export class VaultsContext extends AppContext {
  constructor(params) {
    super(params)
  }

  async changeVaultName(vault, nameAndDesc) {
    await this.vaults.changeVaultMetadata(vault, {
      name: nameAndDesc.name,
      description: nameAndDesc.description,
      iconString: 'icon-vault',
    })
  }

  getKeyPair() {
    const result = this.application.dependencies.get(TYPES.GetKeyPairs).execute()

    return result.getValue().encryption
  }

  getSigningKeyPair() {
    const result = this.application.dependencies.get(TYPES.GetKeyPairs).execute()

    return result.getValue().signing
  }

  async changePassword(password) {
    const promise = this.resolveWhenAsyncFunctionCompletes(this.sharedVaults._handleKeyPairChange, 'execute')

    await super.changePassword(password)

    await this.awaitPromiseOrThrow(promise, undefined, 'Waiting for keypair change message to process')
  }

  async syncAndAwaitNotificationsProcessing() {
    await this.sleep(1, 'Waiting for notifications to propagate')

    const promise = this.resolveWhenAsyncFunctionCompletes(this.notifications, 'handleReceivedNotifications')

    await this.sync()

    await this.awaitPromiseOrDoNothing(
      promise,
      1,
      'Waiting for notifications timed out. Notifications might have been processed in previous sync.'
    )

    if (this.notifications['handleReceivedNotifications'].restore) {
      this.notifications['handleReceivedNotifications'].restore()
    }
  }

  async syncAndAwaitMessageProcessing() {
    const promise = this.resolveWhenAsyncFunctionCompletes(this.asymmetric, 'handleRemoteReceivedAsymmetricMessages')

    await this.sync()

    await this.awaitPromiseOrThrow(promise, undefined, 'Waiting for messages to process')
  }

  async syncAndAwaitInviteProcessing() {
    const promise = this.resolveWhenAsyncFunctionCompletes(this.vaultInvites, 'processInboundInvites')

    await this.sync()

    await this.awaitPromiseOrThrow(promise, undefined, 'Waiting for invites to process')
  }

  async syncAndAwaitInviteAndMessageProcessing() {
    const invitePromise = this.resolveWhenAsyncFunctionCompletes(this.vaultInvites, 'processInboundInvites')
    const messagePromise = this.resolveWhenAsyncFunctionCompletes(
      this.asymmetric,
      'handleRemoteReceivedAsymmetricMessages',
    )

    await this.sync()

    await Promise.all([
      this.awaitPromiseOrThrow(invitePromise, undefined, 'Waiting for invites to process'),
      this.awaitPromiseOrThrow(messagePromise, undefined, 'Waiting for messages to process'),
    ])
  }

  /**
   * Keep refresh token from expiring due to long bouts of inactivity for contact context
   * while main context changes password. Tests have a refresh token age of 10s typically, and changing password
   * on CI environment may be time consuming.
   */
  async forceRefreshSession() {
    await this.application.http.refreshSession()
  }

  async createSharedPasswordVault(password) {
    const privateVault = await this.vaults.createUserInputtedPasswordVault({
      name: 'Our Vault',
      userInputtedPassword: password,
      storagePreference: KeySystemRootKeyStorageMode.Ephemeral,
    })

    const note = await this.createSyncedNote('foo', 'bar')

    await this.vaults.moveItemToVault(privateVault, note)

    const sharedVault = await this.sharedVaults.convertVaultToSharedVault(privateVault)

    const { thirdPartyContext, deinitThirdPartyContext } = await Collaboration.inviteNewPartyToSharedVault(
      this,
      sharedVault,
    )

    await Collaboration.acceptAllInvites(thirdPartyContext)

    const contactVaultOrError = thirdPartyContext.vaults.getVault({ keySystemIdentifier: sharedVault.systemIdentifier })
    if (contactVaultOrError.isFailed()) {
      throw new Error(contactVaultOrError.getError())
    }
    const contactVault = contactVaultOrError.getValue()

    return { contactVault, sharedVault, thirdPartyContext, deinitThirdPartyContext }
  }
}