mui docs layout

codemod.js

// Press ctrl+space for code completion
export default function transformer(file, api, options = {}) {
  const j = api.jscodeshift;
  const root = j(file.source);

  const { layout = "AppFrame" } = options;

  let isDocsPage = false;
  root.find(j.ExportDefaultDeclaration).forEach((path) => {
    isDocsPage = path.node.declaration.params.length > 0;
  });

  if (isDocsPage) {
    const layoutDeclaration = j.importDeclaration([j.importDefaultSpecifier(j.identifier(layout))], j.literal(`docs/src/modules/components/layout/${layout}`), "value");
    root.find(j.ImportDeclaration).at(-1).insertAfter(layoutDeclaration);

    const body = root.find(j.Program).get("body");
    body.value.push("\n");
    j(body.get(body.value.length - 1)).insertAfter(
      j.expressionStatement(j.assignmentExpression("=", j.memberExpression(j.identifier("Page"), j.identifier("Layout")), j.identifier(layout)))
    );
  }

  return root.toSource();
}