import { Text, type Workspace } from '@blocksuite/affine/store';

import type { InitFn } from './utils.js';

export const linked: InitFn = (collection: Workspace, id: string) => {
  const docA =
    collection.getDoc(id)?.getStore({ id }) ??
    collection.createDoc(id).getStore({ id });

  const docBId = 'doc:linked-page';
  const docB = collection.createDoc(docBId).getStore();

  const docCId = 'doc:linked-edgeless';
  const docC = collection.createDoc(docCId).getStore();

  docA.doc.clear();
  docB.doc.clear();
  docC.doc.clear();

  docB.load(() => {
    const rootId = docB.addBlock('affine:page', {
      title: new Text(''),
    });

    docB.addBlock('affine:surface', {}, rootId);

    // Add note block inside root block
    const noteId = docB.addBlock('affine:note', {}, rootId);
    // Add paragraph block inside note block
    docB.addBlock('affine:paragraph', {}, noteId);
  });

  docC.load(() => {
    const rootId = docC.addBlock('affine:page', {
      title: new Text(''),
    });

    docC.addBlock('affine:surface', {}, rootId);

    // Add note block inside root block
    const noteId = docC.addBlock('affine:note', {}, rootId);
    // Add paragraph block inside note block
    docC.addBlock('affine:paragraph', {}, noteId);
  });

  docA.load();
  // Add root block and surface block at root level
  const rootId = docA.addBlock('affine:page', {
    title: new Text('Doc A'),
  });

  docA.addBlock('affine:surface', {}, rootId);

  // Add note block inside root block
  const noteId = docA.addBlock('affine:note', {}, rootId);
  // Add paragraph block inside note block
  docA.addBlock('affine:paragraph', {}, noteId);

  docA.addBlock('affine:embed-linked-doc', { pageId: docBId }, noteId);

  docA.addBlock(
    'affine:embed-linked-doc',
    { pageId: 'doc:deleted-example' },
    noteId
  );

  docA.addBlock('affine:embed-linked-doc', { pageId: docCId }, noteId);

  docA.addBlock(
    'affine:embed-linked-doc',
    { pageId: 'doc:deleted-example-edgeless' },
    noteId
  );

  docA.resetHistory();
  docB.resetHistory();
  docC.resetHistory();
};

linked.id = 'linked';
linked.displayName = 'Linked Doc Editor';
linked.description = 'A demo with linked docs';