@prismai/prism-ladle/packages/ladle/lib/cli/vite-plugin/parse/get-named-exports.js

Enhanced component development environment based on Ladle, with multi-project support

import cloneDeep from "../../deps/lodash.clonedeep.js";
import merge from "lodash.merge";
import {
  getEncodedStoryName,
  storyDelimiter,
  titleToFileId,
  kebabCase,
} from "../naming-utils.js";

/**
 * @param {import('../../../shared/types').ParsedStoriesResult} result
 * @param {any} astPath
 */
const getNamedExports = (
  {
    fileId,
    exportDefaultProps,
    namedExportToMeta,
    namedExportToStoryName,
    storyParams,
    stories,
    entry,
  },
  astPath,
) => {
  /**
   * @param {any} namedExportDeclaration
   * @param {string} namedExport
   * @returns {import('../../../shared/types').StoryEntry} result
   */
  const namedExportToStory = (namedExportDeclaration, namedExport) => {
    if (namedExport.includes("__")) {
      throw new Error(
        `Story named ${namedExport} can't contain "__". It's reserved for internal encoding. Please rename this export.`,
      );
    }

    let storyNamespace = fileId;
    if (exportDefaultProps && exportDefaultProps.title) {
      storyNamespace = titleToFileId(exportDefaultProps.title);
    }
    const storyName = namedExportToStoryName[namedExport]
      ? namedExportToStoryName[namedExport]
      : namedExport;
    const storyId = `${kebabCase(
      storyNamespace,
    )}${storyDelimiter}${storyDelimiter}${kebabCase(storyName)}`;
    // attach default meta to each story
    if (exportDefaultProps && exportDefaultProps.meta) {
      storyParams[storyId] = exportDefaultProps;
    }
    // add and merge story specific meta
    if (namedExportToMeta[namedExport]) {
      storyParams[storyId] = merge(cloneDeep(storyParams[storyId] || {}), {
        meta: namedExportToMeta[namedExport],
      });
    }
    const componentName = getEncodedStoryName(
      kebabCase(storyNamespace),
      kebabCase(storyName),
    );
    const story = {
      storyId,
      componentName,
      namedExport,
      locStart: namedExportDeclaration.loc.start.line,
      locEnd: namedExportDeclaration.loc.end.line,
    };
    return story;
  };

  /**
   * @type {string}
   */
  // Inline exports, such as: export const Story = () => <h1>Export List</h1>;
  if (astPath.node?.declaration?.type) {
    let namedExport = "";
    const namedExportDeclaration = astPath.node?.declaration;
    if (namedExportDeclaration.type === "ClassDeclaration") {
      namedExport = namedExportDeclaration.id.name;
    } else if (namedExportDeclaration.type === "VariableDeclaration") {
      namedExport = namedExportDeclaration.declarations[0].id.name;
    } else if (namedExportDeclaration.type === "FunctionDeclaration") {
      namedExport = namedExportDeclaration.id.name;
    } else {
      throw new Error(
        `Named export in ${entry} must be variable, class or function.`,
      );
    }
    const story = namedExportToStory(namedExportDeclaration, namedExport);
    stories.push(story);
  } else if (astPath.node?.specifiers.length > 0) {
    // It's an export block export, such as: { story, story as storyRenamed };
    astPath.node?.specifiers.forEach(
      /** type * @param {any} specifier */
      (specifier) => {
        const namedExport = specifier.exported.name;
        const story = namedExportToStory(specifier, namedExport);
        stories.push(story);
      },
    );
  }
};

export default getNamedExports;