ether-code 0.6.6 → 0.6.7

package/generators/html-generator.js

@@ -1023,32 +1023,52 @@ class HTMLGenerator {
     const lowerCode = code.toLowerCase()
     const hasEtherKeywords = etherKeywords.some(kw => lowerCode.includes(kw))
     
-    if (!hasEtherKeywords) {
-      return code
-    }
-    
-    if (tagType === 'script' && this.parser && this.jsGenerator) {
-      try {
-        const ast = this.parser.parse(code, 'js')
-        const generated = this.jsGenerator.generate(ast)
-        return generated || code
-      } catch (e) {
-        return code
+    if (tagType === 'script') {
+      if (hasEtherKeywords && this.parser && this.jsGenerator) {
+        try {
+          const ast = this.parser.parse(code, 'js')
+          const generated = this.jsGenerator.generate(ast)
+          return generated || code
+        } catch (e) {
+          return this.addSemicolons(code)
+        }
       }
+      return this.addSemicolons(code)
     }
     
-    if (tagType === 'style' && this.parser && this.cssGenerator) {
-      try {
-        const ast = this.parser.parse(code, 'css')
-        const generated = this.cssGenerator.generate(ast)
-        return generated || code
-      } catch (e) {
-        return code
+    if (tagType === 'style') {
+      if (hasEtherKeywords && this.parser && this.cssGenerator) {
+        try {
+          const ast = this.parser.parse(code, 'css')
+          const generated = this.cssGenerator.generate(ast)
+          return generated || code
+        } catch (e) {
+          return code
+        }
       }
+      return code
     }
     
     return code
   }
+
+  addSemicolons(code) {
+    const lines = code.split('\n')
+    const result = lines.map(line => {
+      const trimmed = line.trim()
+      if (!trimmed) return line
+      const lastChar = trimmed.slice(-1)
+      const noSemicolonEndings = [';', '{', '}', ',', ':', '(', '[']
+      if (noSemicolonEndings.includes(lastChar)) {
+        return line
+      }
+      if (trimmed.startsWith('//') || trimmed.startsWith('/*') || trimmed.startsWith('*')) {
+        return line
+      }
+      return line + ';'
+    })
+    return result.join('\n')
+  }
 }
 
 module.exports = {