import { assertExists } from '@blockeditor/global/utils';
import type { Page } from '@blockeditor/store';
import { Job } from '@blockeditor/store';

import { MarkdownAdapter } from '../adapters/index.js';
import { defaultImageProxyMiddleware } from './middlewares.js';
import { createAssetsArchive, download } from './utils.js';

export async function exportPage(page: Page) {
  const job = new Job({ workspace: page.workspace });
  const snapshot = await job.pageToSnapshot(page);

  const adapter = new MarkdownAdapter();

  const markdownResult = await adapter.fromPageSnapshot({
    snapshot,
    assets: job.assetsManager,
  });

  let downloadBlob: Blob;
  const pageTitle = page.meta.title || 'Untitled';
  let name: string;
  const contentBlob = new Blob([markdownResult.file], { type: 'plain/text' });
  if (markdownResult.assetsIds.length > 0) {
    const zip = createAssetsArchive(job.assets, markdownResult.assetsIds);

    zip.file('index.md', contentBlob);

    downloadBlob = await zip.generateAsync({ type: 'blob' });
    name = `${pageTitle}.zip`;
  } else {
    downloadBlob = contentBlob;
    name = `${pageTitle}.md`;
  }
  download(downloadBlob, name);
}

type ImportMarkdownOptions = {
  page: Page;
  markdown: string;
  noteId: string;
};

export async function importMarkdown({
  page,
  markdown,
  noteId,
}: ImportMarkdownOptions) {
  const job = new Job({
    workspace: page.workspace,
    middlewares: [defaultImageProxyMiddleware],
  });
  const adapter = new MarkdownAdapter();
  adapter.applyConfigs(job.adapterConfigs);
  const snapshot = await adapter.toSliceSnapshot({
    file: markdown,
    assets: job.assetsManager,
    pageVersion: page.workspace.meta.pageVersion!,
    workspaceVersion: page.workspace.meta.workspaceVersion!,
    workspaceId: page.workspace.id,
    pageId: page.id,
  });

  assertExists(snapshot, 'import markdown failed, expected to get a snapshot');

  const blocks = snapshot.content.flatMap(x => x.children);

  for (const block of blocks) {
    await job.snapshotToBlock(block, page, noteId);
  }

  return;
}