AFFiNE-Mirror/blocksuite/affine/blocks/bookmark/src/commands/insert-link-by-quick-search.ts

import { insertEmbedIframeWithUrlCommand } from '@blocksuite/affine-block-embed';
import {
  type InsertedLinkType,
  insertEmbedLinkedDocCommand,
  insertEmbedSyncedDocCommand,
  type LinkableFlavour,
} from '@blocksuite/affine-block-embed-doc';
import {
  DocModeProvider,
  EditorSettingProvider,
  QuickSearchProvider,
} from '@blocksuite/affine-shared/services';
import type { Command } from '@blocksuite/std';

import { insertBookmarkCommand } from './insert-bookmark';

export const insertLinkByQuickSearchCommand: Command<
  {},
  { insertedLinkType: Promise<InsertedLinkType> }
> = (ctx, next) => {
  const { std } = ctx;
  const quickSearchService = std.getOptional(QuickSearchProvider);
  if (!quickSearchService) {
    return;
  }

  const insertedLinkType: Promise<InsertedLinkType> = quickSearchService
    .openQuickSearch()
    .then(result => {
      if (!result) return null;

      // add linked doc
      if ('docId' in result) {
        const editorMode = std.get(DocModeProvider).getEditorMode();
        const editorSettings = std.get(EditorSettingProvider);
        let flavour: LinkableFlavour = 'affine:embed-linked-doc';
        if (editorMode === 'edgeless') {
          flavour =
            editorSettings.setting$.value.docCanvasPreferView ?? flavour;
        }

        const insertCommand =
          flavour === 'affine:embed-linked-doc'
            ? insertEmbedLinkedDocCommand
            : insertEmbedSyncedDocCommand;

        std.command.exec(insertCommand, {
          docId: result.docId,
          params: result.params,
        });

        return {
          flavour,
        };
      }

      // add normal link;
      if ('externalUrl' in result) {
        // try to insert embed iframe block first
        const [success, { flavour }] = std.command
          .chain()
          .try(chain => [
            chain.pipe(insertEmbedIframeWithUrlCommand, {
              url: result.externalUrl,
            }),
            chain.pipe(insertBookmarkCommand, { url: result.externalUrl }),
          ])
          .run();
        if (!success || !flavour) return null;
        return {
          flavour: flavour as LinkableFlavour,
        };
      }

      return null;
    });

  next({ insertedLinkType });
};