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

import createDebug from 'debug';
import type { Group, PrimaryDevice } from '@signalapp/mock-server';
import { StorageState, Proto } from '@signalapp/mock-server';
import { App } from '../playwright';
import { Bootstrap } from '../bootstrap';

export const debug = createDebug('mock:test-storage');

export { App, Bootstrap };

const GROUP_SIZE = 8;

export type InitStorageResultType = Readonly<{
  bootstrap: Bootstrap;
  app: App;
  group: Group;
  members: Array<PrimaryDevice>;
}>;

//
// This function creates an initial storage service state that includes:
//
// - All contacts from contact sync (first contact pinned)
// - A pinned group with GROUP_SIZE members (from the contacts)
// - Account with e164 and profileKey
//
// In addition to above, this function will queue one incoming message in the
// group, and one for the first contact (so that both will appear in the left
// pane).
export async function initStorage(): Promise<InitStorageResultType> {
  // Creates primary device, contacts
  const bootstrap = new Bootstrap();

  await bootstrap.init();

  // Populate storage service
  const { contacts, phone } = bootstrap;

  const [firstContact] = contacts;

  const members = [...contacts].slice(0, GROUP_SIZE);

  const group = await phone.createGroup({
    title: 'Mock Group',
    members: [phone, ...members],
  });

  let state = StorageState.getEmpty();

  state = state.updateAccount({
    profileKey: phone.profileKey.serialize(),
    e164: phone.device.number,
  });

  state = state
    .addGroup(group, {
      whitelisted: true,
    })
    .pinGroup(group);

  for (const contact of contacts) {
    state = state.addContact(contact, {
      identityState: Proto.ContactRecord.IdentityState.VERIFIED,
      whitelisted: true,

      identityKey: contact.publicKey.serialize(),
      profileKey: contact.profileKey.serialize(),
    });
  }

  state = state.pin(firstContact);

  await phone.setStorageState(state);

  // Link new device
  const app = await bootstrap.link();

  const { desktop } = bootstrap;

  // Send a message to the group and the first contact
  const contactSend = contacts[0].sendText(desktop, 'hello from contact', {
    timestamp: bootstrap.getTimestamp(),
    sealed: true,
  });

  const groupSend = members[0].sendText(desktop, 'hello in group', {
    timestamp: bootstrap.getTimestamp(),
    sealed: true,
    group,
  });

  await Promise.all([contactSend, groupSend]);

  return { bootstrap, app, group, members };
}