import Header from "@editorjs/header";
import Paragraph from "@editorjs/paragraph";
import CheckList from "@editorjs/checklist";
import Code from "@editorjs/code";
import Delimiter from "@editorjs/delimiter";
import Embed from "@editorjs/embed";
import InlineCode from "@editorjs/inline-code";
import LinkTool from "@editorjs/link";
import List from "@editorjs/list";
import Marker from "@editorjs/marker";
import Quote from "@editorjs/quote";
import Raw from "@editorjs/raw";
import Table from "@editorjs/table";
import Warning from "@editorjs/warning";
import ComponentSelectorTool from "editorjs-component-selector";

type ToolOptions = {
  header?: boolean;
  paragraph?: boolean;
  list?: boolean;
  checklist?: boolean;
  embed?: boolean;
  table?: boolean;
  warning?: boolean;
  code?: boolean;
  link_tool?: boolean;
  raw?: boolean;
  quote?: boolean;
  marker?: boolean;
  delimiter?: boolean;
  inlineCode?: boolean;
  component?: boolean;
};

type ToolConfig = {
  header?: any;
  paragraph?: any;
  list?: any;
  checklist?: any;
  embed?: any;
  table?: any;
  warning?: any;
  code?: any;
  link_tool?: any;
  raw?: any;
  quote?: any;
  marker?: any;
  delimiter?: any;
  inlineCode?: any;
  component?: any;
};

export const customTools = (options: ToolOptions, config: ToolConfig): {} => {
  return {
    ...(options.header
      ? {
          header: {
            class: Header,
            ...(config.header ? config.header : {}),
          },
        }
      : {}),
    paragraph: {
      class: Paragraph,
      ...(config.paragraph ? config.paragraph : {}),
    },
    ...(options.list
      ? {
          list: {
            class: List,
            ...(config.list ? config.list : {}),
          },
        }
      : {}),
    ...(options.checklist
      ? {
          checklist: {
            class: CheckList,
            ...(config.checklist ? config.checklist : {}),
          },
        }
      : {}),
    ...(options.embed
      ? {
          embed: {
            class: Embed,
            ...(config.embed ? config.embed : {}),
          },
        }
      : {}),
    ...(options.table
      ? {
          table: {
            class: Table,
            ...(config.table ? config.table : {}),
          },
        }
      : {}),
    ...(options.warning
      ? {
          warning: {
            class: Warning,
            ...(config.warning ? config.warning : {}),
          },
        }
      : {}),
    ...(options.code
      ? {
          code: {
            class: Code,
            ...(config.code ? config.code : {}),
          },
        }
      : {}),
    ...(options.link_tool
      ? {
          link_tool: {
            class: LinkTool,
            ...(config.link_tool ? config.link_tool : {}),
          },
        }
      : {}),
    ...(options.raw
      ? {
          raw: {
            class: Raw,
            ...(config.raw ? config.raw : {}),
          },
        }
      : {}),
    ...(options.quote
      ? {
          quote: {
            class: Quote,
            ...(config.quote ? config.quote : {}),
          },
        }
      : {}),
    ...(options.marker
      ? {
          marker: {
            class: Marker,
            ...(config.marker ? config.marker : {}),
          },
        }
      : {}),
    ...(options.delimiter
      ? {
          delimiter: {
            class: Delimiter,
            ...(config.delimiter ? config.delimiter : {}),
          },
        }
      : {}),
    ...(options.inlineCode
      ? {
          inlineCode: {
            class: InlineCode,
            ...(config.inlineCode ? config.inlineCode : {}),
          },
        }
      : {}),
    ...(options.component && config.component
      ? {
          component: {
            class: ComponentSelectorTool,
            ...(config.component ? config.component : {}),
          },
        }
      : {}),
  };
};