utils.ts

import * as ts from "typescript";
import { DocConfig, ExtendedDocConfig } from "./types";
import * as prettier from "prettier";

export const formatCamelCaseToTitle = (str: string) => {
  // Convert camelCase to Title Case with spaces
  return str
    .replace(/([A-Z])/g, " $1")
    .replace(/^./, (str) => str.toUpperCase());
};

const listProperties = (node: ts.ObjectLiteralExpression) => {
  const properties = {};

  node.properties.forEach((property) => {
    if (ts.isPropertyAssignment(property)) {
      properties[property.name.getText()] = property.initializer;
    }
  });

  return properties;
};

const extractTemplates = (node) => {
  const templates = {};

  const components = listProperties(
    listProperties(node)["components"] as ts.ObjectLiteralExpression
  );

  Object.entries(components).forEach(([componentName, value]) => {
    templates[componentName] = {};

    let examples = {};

    try {
      examples = listProperties(
        listProperties(value as ts.ObjectLiteralExpression)[
          "examples"
        ] as ts.ObjectLiteralExpression
      );
    } catch (e) {}

    Object.entries(examples).forEach(([exampleName, value]) => {
      const template = listProperties(value as ts.ObjectLiteralExpression)[
        "template"
      ];

      templates[componentName][exampleName] = template.getText();
    });
  });

  return templates;
};

export const getTemplateContents = (filePath: string) => {
  // Load the file contents to the typescript ast
  const sourceFile = ts.createSourceFile(
    filePath,
    ts.sys.readFile(filePath) || "",
    ts.ScriptTarget.Latest,
    true
  );

  let templates = {};

  sourceFile.forEachChild((node) => {
    if (ts.isVariableStatement(node)) {
      node.declarationList.declarations.forEach((declaration) => {
        if (declaration.name.getText() === "__docConfig") {
          templates = extractTemplates(declaration.initializer);
        }
      });
    }
  });

  return templates;
};

export const mergeTemplateInfo = (
  docInfo: DocConfig,
  templates: any
): ExtendedDocConfig => {
  Object.entries(docInfo.components).forEach(([componentName, value]) => {
    Object.entries(value.examples || {}).forEach(([exampleName, example]) => {
      // @ts-ignore
      example.templateString = templates[componentName][exampleName];
    });
  });

  return docInfo as ExtendedDocConfig;
};

export const formatSnippet = (snippet: string) => {
  return prettier.format(snippet, {
    parser: "typescript",
  });
};

export const safePropType = (str: string) => {
  // Replace all " by ' to avoid conflicts with markdown
  return str.replace(/"/g, "'");
};