import type { Page } from '@playwright/test';

import { waitNextFrame } from './misc.js';

export async function updateBlockType(
  page: Page,
  flavour: string,
  type?: string
) {
  await page.evaluate(
    ([flavour, type]) => {
      window.host.std.command.exec(
        window.$blocksuite.blocks.note.updateBlockType,
        {
          flavour,
          props: {
            type,
          },
        }
      );
    },
    [flavour, type] as [string, string?]
  );
  await waitNextFrame(page, 400);
}

export async function getBlockIds(page: Page) {
  return page.evaluate(() => {
    return window.host.std.store.getAllModels().map(m => m.id);
  });
}