clean up the code

Author: taniki

src/index.ts

@@ -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 = parse(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('');