// Markdown Tokenizer
import { lexer } from "marked";

// Types
import { TDocumentDefinitions } from "pdfmake/interfaces";

// Utils
import { pdfMakeImage } from "./utils/image";
import { pdfMakeParagraph } from "./utils/paragraph";
import { pdfMakeHeading } from "./utils/heading";
import { pdfMakeList } from "./utils/list";
import { pdfMakeBlockquote } from "./utils/blockquote";
import { pdfMakeCodeblock } from "./utils/codeblock";
import { pdfMakeHR } from "./utils/hr";

interface MOptions {
  headingFontSizes: number[];
  headingUnderline?: boolean;
}

export const globalOptions: MOptions = {
  headingFontSizes: [36, 30, 24, 18, 15, 12],
  headingUnderline: true,
};

async function mdpdfmake(
  markdown: string,
  options?: MOptions
): Promise<TDocumentDefinitions> {
  // Set default options
  if (options?.headingFontSizes?.length > 0) {
    // override only added values
    options.headingFontSizes.forEach((size, index) => {
      globalOptions.headingFontSizes[index] = size;
    });
  }

  if (options?.headingUnderline !== undefined) {
    globalOptions.headingUnderline = options.headingUnderline;
  }

  // Tokenize markdown
  const tokens = lexer(markdown);
  const content: any[] = [];

  for (const token of tokens) {
    switch (token.type) {
      case "paragraph":
        await pdfMakeParagraph(token, content);
        break;

      case "heading":
        pdfMakeHeading(token, content);
        break;

      case "list":
        await pdfMakeList(token, content);
        break;

      case "blockquote":
        await pdfMakeBlockquote(token, content);
        break;

      case "image":
        await pdfMakeImage(token, content);
        break;

      case "code":
        await pdfMakeCodeblock(token, content);
        break;

      case "hr":
        await pdfMakeHR(token, content);
        break;

      case "space":
        break;

      case "br":
        content.push({ text: "\n" });
        break;

      default:
        console.warn(`Unhandled token type: ${token.type}`);
    }
  }

  return {
    content: content,
    defaultStyle: {
      font: "Roboto",
    },
  };
}

export { mdpdfmake };