clean up the code

Author: taniki

src/index.ts

@@ -15,13 +15,13 @@ import { options as pugOptions } from './options';
 import { convergeOptions } from './options/converge';
 import type { PugPrinterOptions } from './printer';
 import { PugPrinter } from './printer';
-import parse from './utils/frontmatter/parse';
+import { default as parseFrontmatter } from './utils/frontmatter/parse';
 
 /** Ast path stack entry. */
 interface AstPathStackEntry {
   content: string;
   tokens: Token[];
-  frontmatter: Frontmatter<string>;
+  frontmatter: string;
 }
 
 /** The plugin object that is picked up by prettier. */
@@ -45,8 +45,10 @@ export const plugin: Plugin<AstPathStackEntry> = {
       parse(text, options) {
         logger.debug('[parsers:pug:parse]:', { text });
 
-        const parts = parse(text);
-        const frontmatter = parts.frontMatter;
+        const parts: FrontMatter = parseFrontmatter(text);
+        const frontmatter: string = parts.frontMatter
+          ? parts.frontMatter.value
+          : undefined;
         text = parts.content;
 
         let trimmedAndAlignedContent: string = text.replace(/^\s*\n/, '');

src/printer.ts

@@ -236,12 +236,13 @@ export class PugPrinter {
    * @param content The pug content string.
    * @param tokens The pug token array.
    * @param options Options for the printer.
+   * @param frontmatter The extracted frontmatter part.
    */
   public constructor(
     private readonly content: string,
     private tokens: Token[],
     private readonly options: PugPrinterOptions,
-    private readonly frontmatter: Frontmatter<string>,
+    private readonly frontmatter: string,
   ) {
     this.frontmatter = frontmatter;
     this.indentString = options.pugUseTabs
@@ -392,10 +393,15 @@ export class PugPrinter {
     }
 
     if (this.frontmatter) {
-      const frontmatter: string = await this.frontmatterFormat(
-        this.frontmatter.value,
-      );
-      results.unshift('---\n', frontmatter, '---\n');
+      try {
+        results.unshift(
+          '---\n',
+          await this.frontmatterFormat(this.frontmatter),
+          '---\n',
+        );
+      } catch (error: any) {
+        logger.warn(error);
+      }
     }
 
     return results.join('');