storybook-onbook-plugin 0.2.0 → 0.2.1

package/dist/component-loc/component-loc.d.ts

@@ -0,0 +1,7 @@
+import type { Plugin } from 'vite';
+type ComponentLocPluginOptions = {
+    include?: RegExp;
+};
+export declare function componentLocPlugin(options?: ComponentLocPluginOptions): Plugin;
+export default componentLocPlugin;
+//# sourceMappingURL=component-loc.d.ts.map

package/dist/component-loc/component-loc.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"component-loc.d.ts","sourceRoot":"","sources":["../../src/component-loc/component-loc.ts"],"names":[],"mappings":"AAMA,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,MAAM,CAAC;AAEnC,KAAK,yBAAyB,GAAG;IAC/B,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB,CAAC;AAEF,wBAAgB,kBAAkB,CAAC,OAAO,GAAE,yBAA8B,GAAG,MAAM,CAgFlF;AAED,eAAe,kBAAkB,CAAC"}

package/dist/component-loc/component-loc.js

@@ -0,0 +1,58 @@
+import path from 'node:path';
+import generateModule from '@babel/generator';
+import { parse } from '@babel/parser';
+import traverseModule from '@babel/traverse';
+import * as t from '@babel/types';
+export function componentLocPlugin(options = {}) {
+    const include = options.include ?? /\.(jsx|tsx)$/;
+    // @babel/traverse and @babel/generator are CommonJS modules with default exports.
+    // When imported in an ESM context, the default export may be on `.default` or directly on the module.
+    // This workaround handles both cases to ensure compatibility across different bundler configurations.
+    const traverse = traverseModule.default ??
+        traverseModule;
+    const generate = generateModule.default ??
+        generateModule;
+    let root;
+    return {
+        name: 'onbook-component-loc',
+        enforce: 'pre',
+        apply: 'serve',
+        configResolved(config) {
+            root = config.root;
+        },
+        transform(code, id) {
+            const filepath = id.split('?', 1)[0];
+            if (!filepath || filepath.includes('node_modules'))
+                return null;
+            if (!include.test(filepath))
+                return null;
+            const ast = parse(code, {
+                sourceType: 'module',
+                plugins: ['jsx', 'typescript'],
+                sourceFilename: filepath,
+            });
+            let mutated = false;
+            // Get relative path from project root
+            const relativePath = path.relative(root, filepath);
+            traverse(ast, {
+                JSXElement(nodePath) {
+                    const opening = nodePath.node.openingElement;
+                    const element = nodePath.node;
+                    if (!opening.loc || !element.loc)
+                        return;
+                    const alreadyTagged = opening.attributes.some((attribute) => t.isJSXAttribute(attribute) &&
+                        attribute.name.name === 'data-component-file');
+                    if (alreadyTagged)
+                        return;
+                    opening.attributes.push(t.jsxAttribute(t.jsxIdentifier('data-component-file'), t.stringLiteral(relativePath)), t.jsxAttribute(t.jsxIdentifier('data-component-start'), t.stringLiteral(`${element.loc.start.line}:${element.loc.start.column}`)), t.jsxAttribute(t.jsxIdentifier('data-component-end'), t.stringLiteral(`${element.loc.end.line}:${element.loc.end.column}`)));
+                    mutated = true;
+                },
+            });
+            if (!mutated)
+                return null;
+            const output = generate(ast, { retainLines: true, sourceMaps: true, sourceFileName: id }, code);
+            return { code: output.code, map: output.map };
+        },
+    };
+}
+export default componentLocPlugin;

package/dist/component-loc/index.d.ts

@@ -0,0 +1,2 @@
+export { componentLocPlugin } from './component-loc.js';
+//# sourceMappingURL=index.d.ts.map

package/dist/component-loc/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/component-loc/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,kBAAkB,EAAE,MAAM,oBAAoB,CAAC"}

package/dist/component-loc/index.js

@@ -0,0 +1 @@
+export { componentLocPlugin } from './component-loc.js';

package/dist/screenshot-service/types.js

@@ -0,0 +1 @@
+export {};

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "storybook-onbook-plugin",
-  "version": "0.2.0",
+  "version": "0.2.1",
   "type": "module",
   "bin": {
     "generate-screenshots": "./dist/cli/generate-screenshots.js"