import type { ServerContext } from '@affine/sdk/server';
import { getLinkPreview } from 'link-preview-js';

type MetaData = {
  title?: string;
  description?: string;
  icon?: string;
  image?: string;
  [x: string]: string | string[] | undefined;
};

export interface PreviewType {
  url: string;
  title: string;
  siteName: string | undefined;
  description: string | undefined;
  mediaType: string;
  contentType: string | undefined;
  images: string[];
  videos: {
    url: string | undefined;
    secureUrl: string | null | undefined;
    type: string | null | undefined;
    width: string | undefined;
    height: string | undefined;
  }[];
  favicons: string[];
}

export const entry = (context: ServerContext) => {
  context.registerCommand(
    'com.blocksuite.bookmark-block.get-bookmark-data-by-link',
    async (url: string): Promise<MetaData> => {
      const previewData = (await getLinkPreview(url, {
        timeout: 6000,
        headers: {
          'user-agent': 'googlebot',
        },
        followRedirects: 'follow',
      }).catch(() => {
        return {
          title: '',
          siteName: '',
          description: '',
          images: [],
          videos: [],
          contentType: `text/html`,
          favicons: [],
        };
      })) as PreviewType;

      return {
        title: previewData.title,
        description: previewData.description,
        icon: previewData.favicons[0],
        image: previewData.images[0],
      };
    }
  );
  return () => {
    context.unregisterCommand(
      'com.blocksuite.bookmark-block.get-bookmark-data-by-link'
    );
  };
};