// Copyright 2022 Signal Messenger, LLC
// SPDX-License-Identifier: AGPL-3.0-only

import { assert } from 'chai';

import type { Group } from '@signalapp/mock-server';

import * as durations from '../../util/durations/index.std.ts';
import type { App, Bootstrap } from './fixtures.node.ts';
import { initStorage, debug } from './fixtures.node.ts';

describe('storage service', function (this: Mocha.Suite) {
  this.timeout(durations.MINUTE);

  let bootstrap: Bootstrap;
  let app: App;
  let group: Group;

  afterEach(async function (this: Mocha.Context) {
    if (!bootstrap) {
      return;
    }

    await bootstrap.maybeSaveLogs(this.currentTest, app);
    await app.close();
    await bootstrap.teardown();
  });

  for (const useLegacyStorageEncryption of [false, true]) {
    const suffix = `useLegacyStorageEncryption=${useLegacyStorageEncryption}`;
    // oxlint-disable-next-line no-loop-func
    it(`should pin/unpin groups ${suffix}`, async () => {
      ({ bootstrap, app, group } = await initStorage({
        useLegacyStorageEncryption,
      }));

      const { phone, desktop, contacts } = bootstrap;

      const window = await app.getWindow();

      const leftPane = window.locator('#LeftPane');
      const conversationStack = window.locator('.Inbox__conversation-stack');

      debug('Verifying that the group is pinned on startup');
      await leftPane.locator(`[data-testid="${group.id}"]`).waitFor();

      debug('Unpinning group via storage service');
      {
        const state = await phone.expectStorageState('initial state');
        const newState = state.unpinGroup(group);

        await phone.setStorageState(newState);
        await phone.sendFetchStorage({
          timestamp: bootstrap.getTimestamp(),
        });

        await app.waitForManifestVersion(newState.version);
      }

      debug('Pinning group in the app');
      {
        const state = await phone.expectStorageState('consistency check');

        const convo = leftPane.getByTestId(group.id);
        await convo.click();

        const moreButton = conversationStack.getByRole('button', {
          name: 'More Info',
        });
        await moreButton.click();

        const pinButton = window.getByRole('menuitem', {
          name: 'Pin chat',
        });
        await pinButton.click();

        const newState = await phone.waitForStorageState({
          after: state,
        });
        assert.isTrue(newState.isGroupPinned(group), 'group not pinned');

        // AccountRecord
        const { added, removed } = newState.diff(state);
        assert.strictEqual(added.length, 1, 'only one record must be added');
        assert.strictEqual(
          removed.length,
          1,
          'only one record must be removed'
        );
      }

      debug('Pinning > 4 conversations');
      {
        // We already have one group and first contact pinned so we need three
        // more.
        const toPin = contacts.slice(1, 4);

        // To do that we need them to appear in the left pane, though.
        for (const [i, contact] of toPin.entries()) {
          const isLast = i === toPin.length - 1;

          debug('sending a message to contact=%d', i);
          // oxlint-disable-next-line no-await-in-loop
          await contact.sendText(desktop, 'Hello!', {
            timestamp: bootstrap.getTimestamp(),
          });

          // oxlint-disable-next-line no-await-in-loop
          const state = await phone.expectStorageState('consistency check');

          debug('pinning contact=%d', i);
          const convo = leftPane.locator(
            `[data-testid="${contact.device.aci}"]`
          );
          // oxlint-disable-next-line no-await-in-loop
          await convo.click();

          const moreButton = conversationStack.getByRole('button', {
            name: 'More Info',
          });
          // oxlint-disable-next-line no-await-in-loop
          await moreButton.click();

          const pinButton = window.getByRole('menuitem', {
            name: 'Pin chat',
          });
          // oxlint-disable-next-line no-await-in-loop
          await pinButton.click();

          if (isLast) {
            // Storage state shouldn't be updated because we failed to pin
            // oxlint-disable-next-line no-await-in-loop
            await window
              .locator('.Toast >> "You can only pin up to 4 chats"')
              .waitFor();
            break;
          }

          debug('verifying storage state change contact=%d', i);
          // oxlint-disable-next-line no-await-in-loop
          const newState = await phone.waitForStorageState({
            after: state,
          });
          assert.isTrue(newState.isPinned(contact), 'contact not pinned');

          // AccountRecord
          const { added, removed } = newState.diff(state);
          assert.strictEqual(added.length, 1, 'only one record must be added');
          assert.strictEqual(
            removed.length,
            1,
            'only one record must be removed'
          );
        }
      }

      debug('Verifying the final manifest version');
      const finalState = await phone.expectStorageState('consistency check');

      assert.strictEqual(finalState.version, 5n);
    });
  }
});
