pankod/refine

import Handlebars from "handlebars";
import {
  readFileSync,
  readdirSync,
  createFileSync,
  writeFileSync,
  unlinkSync,
} from "fs-extra";
import { getComponentNameByResource } from "@utils/resource";

export const compile = (filePath: string, params: any): string => {
  const content = readFileSync(filePath);

  Handlebars.registerHelper("ifIn", (elem, list, options) => {
    if (elem.includes(list)) {
      return options.fn(Handlebars);
    }
    return options.inverse(Handlebars);
  });

  Handlebars.registerHelper("formatInferencerComponent", (string) => {
    if (!string) {
      return;
    }

    switch (string) {
      case "chakra-ui":
        return "ChakraUI";

      default:
        return string.charAt(0).toUpperCase() + string.slice(1);
    }
  });

  Handlebars.registerHelper("capitalize", (string) => {
    if (!string) {
      return;
    }

    return string.charAt(0).toUpperCase() + string.slice(1);
  });

  Handlebars.registerHelper("getComponentNameByResource", (string) => {
    if (!string) {
      return;
    }

    return getComponentNameByResource(string);
  });

  const template = Handlebars.compile(content.toString());
  return template(params);
};

/**
 * compile all hbs files under the specified directory. recursively
 */
export const compileDir = (dirPath: string, params: any) => {
  const files = readdirSync(dirPath, { recursive: true });

  files.forEach((file: string | Buffer) => {
    // the target file should be a handlebars file
    if (typeof file !== "string" || !file.endsWith(".hbs")) return;

    const templateFilePath = `${dirPath}/${file}`;
    // create file
    const compiledFilePath = `${dirPath}/${file.replace(".hbs", "")}`;
    createFileSync(compiledFilePath);

    // write compiled file
    writeFileSync(compiledFilePath, compile(templateFilePath, params));

    // delete template file (*.hbs)
    unlinkSync(templateFilePath);
  });
};