@ng-shangjc/cli 1.0.5-beta → 1.0.6-beta

package/dist/commands/init.js

@@ -286,34 +286,44 @@ async function setupTailwind() {
         }
     }
     else if (existingContent && hasTailwindImport) {
-        // If file had tailwind import, remove the old theme and keep other styles
+        // If file had tailwind import, preserve everything and just add/update the theme
         const lines = existingContent.split('\n');
-        const afterThemeLines = [];
-        let inThemeBlock = false;
-        let foundTheme = false;
-        for (const line of lines) {
-            if (line.includes('@theme {') || line.includes(':root {') || line.includes('.dark {') || line.includes('@layer base {')) {
-                inThemeBlock = true;
-                foundTheme = true;
+        const preservedLines = [];
+        let skipNextClosingBrace = false;
+        let inThemeSection = false;
+        for (let i = 0; i < lines.length; i++) {
+            const line = lines[i];
+            const trimmedLine = line.trim();
+            // Skip theme-related blocks but preserve everything else
+            if (trimmedLine.startsWith('@theme {') ||
+                trimmedLine.startsWith(':root {') ||
+                trimmedLine.startsWith('.dark {') ||
+                trimmedLine.startsWith('@custom-variant dark')) {
+                inThemeSection = true;
                 continue;
             }
-            if (inThemeBlock && line.includes('}') && !line.includes('@layer')) {
-                inThemeBlock = false;
+            if (inThemeSection && trimmedLine === '}' &&
+                !lines[i + 1]?.trim().startsWith('@layer')) {
+                inThemeSection = false;
                 continue;
             }
-            if (inThemeBlock && line.includes('@layer base {')) {
-                // Skip the old @layer base content
+            if (trimmedLine.startsWith('@layer base {')) {
+                inThemeSection = true;
                 continue;
             }
-            if (inThemeBlock) {
+            if (inThemeSection && trimmedLine === '}' &&
+                lines[i - 1]?.trim().startsWith('@layer base')) {
+                inThemeSection = false;
                 continue;
             }
-            if (foundTheme && line.trim()) {
-                afterThemeLines.push(line);
+            if (inThemeSection) {
+                continue;
             }
+            // Preserve this line
+            preservedLines.push(line);
         }
-        if (afterThemeLines.length > 0) {
-            newContent += '\n\n' + afterThemeLines.join('\n');
+        if (preservedLines.length > 0) {
+            newContent += '\n\n' + preservedLines.join('\n');
         }
     }
     // Write the updated content

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ng-shangjc/cli",
-  "version": "1.0.5-beta",
+  "version": "1.0.6-beta",
   "bin": {
     "ng-shangjc": "./dist/index.js"
   },

package/src/commands/init.ts

@@ -263,36 +263,52 @@ async function setupTailwind() {
       newContent += '\n\n' + existingWithoutTailwind;
     }
   } else if (existingContent && hasTailwindImport) {
-    // If file had tailwind import, remove the old theme and keep other styles
+    // If file had tailwind import, preserve everything and just add/update the theme
     const lines = existingContent.split('\n');
-    const afterThemeLines = [];
-    let inThemeBlock = false;
-    let foundTheme = false;
+    const preservedLines = [];
+    let skipNextClosingBrace = false;
+    let inThemeSection = false;
     
-    for (const line of lines) {
-      if (line.includes('@theme {') || line.includes(':root {') || line.includes('.dark {') || line.includes('@layer base {')) {
-        inThemeBlock = true;
-        foundTheme = true;
+    for (let i = 0; i < lines.length; i++) {
+      const line = lines[i];
+      const trimmedLine = line.trim();
+      
+      // Skip theme-related blocks but preserve everything else
+      if (trimmedLine.startsWith('@theme {') || 
+          trimmedLine.startsWith(':root {') || 
+          trimmedLine.startsWith('.dark {') ||
+          trimmedLine.startsWith('@custom-variant dark')) {
+        inThemeSection = true;
         continue;
       }
-      if (inThemeBlock && line.includes('}') && !line.includes('@layer')) {
-        inThemeBlock = false;
+      
+      if (inThemeSection && trimmedLine === '}' && 
+          !lines[i + 1]?.trim().startsWith('@layer')) {
+        inThemeSection = false;
         continue;
       }
-      if (inThemeBlock && line.includes('@layer base {')) {
-        // Skip the old @layer base content
+      
+      if (trimmedLine.startsWith('@layer base {')) {
+        inThemeSection = true;
         continue;
       }
-      if (inThemeBlock) {
+      
+      if (inThemeSection && trimmedLine === '}' && 
+          lines[i - 1]?.trim().startsWith('@layer base')) {
+        inThemeSection = false;
         continue;
       }
-      if (foundTheme && line.trim()) {
-        afterThemeLines.push(line);
+      
+      if (inThemeSection) {
+        continue;
       }
+      
+      // Preserve this line
+      preservedLines.push(line);
     }
     
-    if (afterThemeLines.length > 0) {
-      newContent += '\n\n' + afterThemeLines.join('\n');
+    if (preservedLines.length > 0) {
+      newContent += '\n\n' + preservedLines.join('\n');
     }
   }