ds-01 1.0.3 → 1.0.4

package/dist/commands/add.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"add.d.ts","sourceRoot":"","sources":["../../src/commands/add.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAkDpC,eAAO,MAAM,GAAG,SAgHZ,CAAC"}
+{"version":3,"file":"add.d.ts","sourceRoot":"","sources":["../../src/commands/add.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAkDpC,eAAO,MAAM,GAAG,SAqIZ,CAAC"}

package/dist/commands/add.js

@@ -77,10 +77,23 @@ export const add = new Command()
         if (!fs.existsSync(targetDir)) {
             fs.mkdirSync(targetDir, { recursive: true });
         }
+        // 🛠️ EXTRACT NAMES FOR SMART REWRITING (e.g., ["PremiumHero", "Section", "AnimatedText"])
+        // We strip the extension (.tsx, .ts) to get the raw component name for import matching
+        const bundledFileNames = componentData.files.map((file) => file.name.replace(/\.[^/.]+$/, ""));
         // 5. Inject the pure code files
         for (const file of componentData.files) {
+            // Because the server sends just the basename (e.g., PremiumHero.tsx),
+            // path.join inherently flattens all files into the single targetDir folder.
             const filePath = path.join(targetDir, file.name);
-            fs.writeFileSync(filePath, file.content);
+            let content = file.content;
+            // 🛠️ DYNAMIC IMPORT REWRITER
+            bundledFileNames.forEach((fileName) => {
+                // Finds any relative import pointing to a bundled file (e.g., "../Section", "../../Section")
+                // and rewrites it strictly to a sibling import (e.g., "./Section")
+                const regex = new RegExp(`from\\s+["']\\.[^"']*?\\/${fileName}["']`, "g");
+                content = content.replace(regex, `from "./${fileName}"`);
+            });
+            fs.writeFileSync(filePath, content);
         }
         s.stop(pc.green(`Downloaded ${componentData.files.length} files into ${pc.white(`/${config.componentsPath}/${componentName}`)}`));
         // 6. Auto-install Missing Component Dependencies (e.g., framer-motion)
@@ -98,9 +111,11 @@ export const add = new Command()
             }
         }
         // 7. Clean Success Outro
+        // Find the main file to import (prioritize the one that matches the component name or use the first one)
+        const mainFile = bundledFileNames.find((name) => name.toLowerCase() === componentName.toLowerCase()) || bundledFileNames[0];
         outro(`${pc.white("✔")} ${pc.bold(`Component <${componentName} /> is ready!`)}\n` +
             pc.gray(`Import it: `) +
-            pc.cyan(`import { Section } from "@/${config.componentsPath}/${componentName}/Section"`));
+            pc.cyan(`import { ${mainFile} } from "@/${config.componentsPath}/${componentName}/${mainFile}"`));
     }
     catch (error) {
         s.stop(pc.red("An error occurred during injection."));

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ds-01",
-  "version": "1.0.3",
+  "version": "1.0.4",
   "description": "Make your site best with DS01",
   "type": "module",
   "bin": {

package/src/commands/add.ts

@@ -111,10 +111,28 @@ export const add = new Command()
         fs.mkdirSync(targetDir, { recursive: true });
       }
 
+      // 🛠️ EXTRACT NAMES FOR SMART REWRITING (e.g., ["PremiumHero", "Section", "AnimatedText"])
+      // We strip the extension (.tsx, .ts) to get the raw component name for import matching
+      const bundledFileNames = componentData.files.map((file: any) => 
+        file.name.replace(/\.[^/.]+$/, "") 
+      );
+
       // 5. Inject the pure code files
       for (const file of componentData.files) {
+        // Because the server sends just the basename (e.g., PremiumHero.tsx),
+        // path.join inherently flattens all files into the single targetDir folder.
         const filePath = path.join(targetDir, file.name);
-        fs.writeFileSync(filePath, file.content);
+        let content = file.content;
+
+        // 🛠️ DYNAMIC IMPORT REWRITER
+        bundledFileNames.forEach((fileName: string) => {
+          // Finds any relative import pointing to a bundled file (e.g., "../Section", "../../Section")
+          // and rewrites it strictly to a sibling import (e.g., "./Section")
+          const regex = new RegExp(`from\\s+["']\\.[^"']*?\\/${fileName}["']`, "g");
+          content = content.replace(regex, `from "./${fileName}"`);
+        });
+
+        fs.writeFileSync(filePath, content);
       }
 
       s.stop(
@@ -148,11 +166,14 @@ export const add = new Command()
       }
 
       // 7. Clean Success Outro
+      // Find the main file to import (prioritize the one that matches the component name or use the first one)
+      const mainFile = bundledFileNames.find((name: string) => name.toLowerCase() === componentName.toLowerCase()) || bundledFileNames[0];
+      
       outro(
         `${pc.white("✔")} ${pc.bold(`Component <${componentName} /> is ready!`)}\n` +
           pc.gray(`Import it: `) +
           pc.cyan(
-            `import { Section } from "@/${config.componentsPath}/${componentName}/Section"`
+            `import { ${mainFile} } from "@/${config.componentsPath}/${componentName}/${mainFile}"`
           )
       );
     } catch (error: any) {