smartfm/ts/index.ts

import grayMatter from 'gray-matter';

export type TFrontMatter = 'yaml' | 'json';

export interface ISmartfmContructorOptions {
  fmType: TFrontMatter;
}

/**
 * class smartfm handles frontmatter
 */
export class Smartfm {
  fmType: TFrontMatter;

  constructor(optionsArg: ISmartfmContructorOptions) {
    this.fmType = optionsArg.fmType;
  }

  /**
   * add frontmatter to a string
   */
  stringify(bodyString: string, frontmatterData: Record<string, unknown>): string {
    return grayMatter.stringify(bodyString, frontmatterData);
  }

  /**
   * parse a string that has frontmatter attached, YAML notation
   */
  parse(stringToParse: string): grayMatter.GrayMatterFile<string> {
    return grayMatter(stringToParse);
  }

  /**
   * parse from comments
   */
  parseFromComments(commentStart: string, stringToParse: string): grayMatter.GrayMatterFile<string> {
    const diffFunc = (diffMe: string, diffBy: string): string => diffMe.split(diffBy).join('');
    let lines = stringToParse.split('\n');
    lines = lines.map((line) => {
      return diffFunc(line, commentStart);
    });
    const cleanedString = lines.join('\n');
    return this.parse(cleanedString);
  }
}