import type { EdgelessModel } from '../../../_common/utils/index.js';
import { groupBy } from '../../../_common/utils/iterable.js';
import type { FrameBlockModel } from '../../../frame-block/index.js';
import type { ImageBlockModel } from '../../../image-block/index.js';
import type { NoteBlockModel } from '../../../note-block/index.js';
import { type CanvasElement } from '../../../surface-block/index.js';
import {
  getCopyElements,
  prepareClipboardData,
} from '../controllers/clipboard.js';
import type { EdgelessPageBlockComponent } from '../edgeless-page-block.js';
import { edgelessElementsBound } from './bound-utils.js';
import { getElementsWithoutGroup } from './group.js';
import { isFrameBlock, isImageBlock, isNoteBlock } from './query.js';

const offset = 10;
export async function duplicate(
  edgeless: EdgelessPageBlockComponent,
  elements: EdgelessModel[],
  select = true
) {
  const { clipboardController } = edgeless;
  const copyElements = getCopyElements(edgeless.surface, elements);
  const totalBound = edgelessElementsBound(copyElements);
  totalBound.x += totalBound.w + offset;

  const data = JSON.parse(
    JSON.stringify(await prepareClipboardData(copyElements, edgeless.std))
  );
  const [canvasElements, blocks] =
    await clipboardController.createElementsFromClipboardData(
      data as Record<string, unknown>[],
      totalBound.center
    );

  const newElements = [...canvasElements, ...blocks];

  edgeless.surface.fitToViewport(totalBound);

  if (select) {
    edgeless.service.selection.set({
      elements: newElements.map(e => e.id),
      editing: false,
    });
  }
}
export const splitElements = (elements: EdgelessModel[]) => {
  const { notes, frames, shapes, images } = groupBy(
    getElementsWithoutGroup(elements),
    element => {
      if (isNoteBlock(element)) {
        return 'notes';
      } else if (isFrameBlock(element)) {
        return 'frames';
      } else if (isImageBlock(element)) {
        return 'images';
      }
      return 'shapes';
    }
  ) as {
    notes: NoteBlockModel[];
    shapes: CanvasElement[];
    frames: FrameBlockModel[];
    images: ImageBlockModel[];
  };

  return {
    notes: notes ?? [],
    shapes: shapes ?? [],
    frames: frames ?? [],
    images: images ?? [],
  };
};