//
// Copyright 2023 DXOS.org
//

import { describe, test } from 'vitest';

import { MemorySignalManagerContext, MemorySignalManager } from '@dxos/messaging';
import { Invitation } from '@dxos/protocols/proto/dxos/client/services';
import { openAndClose } from '@dxos/test-utils';

import { createServiceContext, performInvitation } from '../testing';

describe('services/ServiceContext', () => {
  test('new space is synchronized on device invitations', async () => {
    const networkContext = new MemorySignalManagerContext();
    const device1 = await createOpenServiceContext(networkContext);
    await device1.createIdentity();

    const device2 = await createOpenServiceContext(networkContext);
    await Promise.all(performInvitation({ host: device1, guest: device2, options: { kind: Invitation.Kind.DEVICE } }));

    const space1 = await device1.dataSpaceManager!.createSpace();
    await device2.dataSpaceManager!.waitUntilSpaceReady(space1!.key);
    const space2 = await device2.dataSpaceManager!.spaces.get(space1.key);
    await space2!.inner.controlPipeline.state.waitUntilTimeframe(space1.inner.controlPipeline.state.timeframe);
  });

  test('joined space is synchronized on device invitations', async () => {
    const networkContext = new MemorySignalManagerContext();
    const device1 = await createOpenServiceContext(networkContext);
    await device1.createIdentity();

    const device2 = await createOpenServiceContext(networkContext);
    await Promise.all(performInvitation({ host: device1, guest: device2, options: { kind: Invitation.Kind.DEVICE } }));

    const identity2 = await createOpenServiceContext(networkContext);
    await identity2.createIdentity();
    const space1 = await identity2.dataSpaceManager!.createSpace();
    await Promise.all(
      performInvitation({
        host: identity2,
        guest: device1,
        options: { kind: Invitation.Kind.SPACE, spaceKey: space1.key },
      }),
    );

    await device2.dataSpaceManager!.waitUntilSpaceReady(space1!.key);
    const space2 = await device2.dataSpaceManager!.spaces.get(space1.key);
    await space2!.inner.controlPipeline.state.waitUntilTimeframe(space1.inner.controlPipeline.state.timeframe);
  });

  const createOpenServiceContext = async (networkContext: MemorySignalManagerContext) => {
    const serviceContext = await createServiceContext({
      signalManagerFactory: async () => new MemorySignalManager(networkContext),
    });
    await openAndClose(serviceContext);
    return serviceContext;
  };
});