Merge pull request #9 from RKatarine/riazanova/change-const-to-let

Замена const-декларации на let для пропсов

package.json

@@ -4,7 +4,7 @@
   "version": "0.0.0",
   "type": "module",
   "scripts": {
-    "test": "uvu ./test -i samples -i fixtures"
+    "test": "uvu -r tsm ./test -i samples -i fixtures"
   },
   "devDependencies": {
     "@types/estraverse": "^5.1.3",

src/compiler/ComponentDeclaration.ts

@@ -4,7 +4,13 @@ import Context from './Context';
 import Scope from './Scope';
 import Patcher from './Patcher';
 import logger from './logger';
-import { type SymbolAnalysisResult, type TemplateSource, isFunctionDeclaration, runSymbolAnalysis } from './analyze';
+import {
+    type SymbolAnalysisResult,
+    type TemplateSource,
+    isFunctionDeclaration,
+    runSymbolAnalysis,
+    isVariableDeclaration
+} from './analyze';
 import parse, { type AST } from '../parser';
 import compileTemplate from './template';
 import compileEventHandler from './template/event';
@@ -277,6 +283,7 @@ export default class ComponentDeclaration {
                     break;
                 case 'prop':
                     chunk = `${symbol} = ${nextPropsArg}.${propName}`;
+                    this.patchVariableDeclaration(patcher, this.scope.declarations.get(symbol))
                     break;
                 case 'rest':
                     // TODO поддержать rect-аргумент
@@ -382,6 +389,18 @@ export default class ComponentDeclaration {
             }
         });
     }
+
+    private patchVariableDeclaration(patcher: Patcher, node?: ESTree.Node) {
+        if (!node || !isVariableDeclaration(node)) {
+            return
+        }
+        if (node.kind === 'const') {
+            patcher.replace({
+                start: node.start,
+                end: node.start + 5
+            }, `let`);
+        }
+    }
 }
 
 /**

src/compiler/analyze.ts

@@ -288,6 +288,10 @@ function handleIdentifier(scope: Scope, node: ESTree.Identifier, parents: ESTree
     }
 }
 
+export function isVariableDeclaration(node: ESTree.Node): node is ESTree.VariableDeclaration {
+    return node.type === 'VariableDeclaration';
+}
+
 function declareOrAssign(node: ESTree.Node): node is ESTree.VariableDeclaration | ESTree.AssignmentExpression | ESTree.UpdateExpression {
     return node.type === 'VariableDeclaration'
         || node.type === 'AssignmentExpression'

test/compile.ts

@@ -44,4 +44,14 @@ test('if1.js', async () => {
     same(ctx.render(), await read('./fixtures/compile/if1.js'));
 });
 
+test('const-to-let.js', async () => {
+    const file = await read('./samples/const-to-let.js');
+    const ctx = new Context(file);
+    const component = ctx.getComponents()[0];
+    const decl = new ComponentDeclaration(ctx, component);
+    decl.compile(ctx.patcher);
+
+    same(ctx.render(), await read('./fixtures/compile/const-to-let.js'));
+});
+
 test.run();

test/fixtures/compile/const-to-let.js

@@ -0,0 +1,34 @@
+import { element, attach, text, createContext, setupContext, finalizeContext } from 'endorphin/internal';
+import { html } from './endorphin';
+
+function ConstToLetComponent(props) {
+    createContext();
+    let { a } = props;
+    let b = props.b;
+
+    setupContext([a, b], ConstToLetComponent_template, 3 /* a | b */);
+    return finalizeContext((nextProps) => { props = nextProps;invalidate(0, a = nextProps.a);invalidate(1, b = nextProps.b) });
+}
+
+
+function ConstToLetComponent_template(ctx, stage, refs) {
+    const { scope, dirty } = ctx;
+    if (stage === 1) {
+        refs.length = 3;
+        refs[0] = element("div");
+        attach(refs[0]);
+        refs[0].appendChild(text("\n        "));
+        refs[1] = text(scope[0]);
+        refs[0].appendChild(refs[1]);
+        refs[0].appendChild(text("\n        "));
+        refs[2] = element("span");
+        refs[0].appendChild(refs[2]);
+        refs[2].innerText = scope[1];
+        refs[0].appendChild(text("\n    "));
+    } else if (stage === 2) {
+        (dirty & 1 /* a */) && (refs[1].nodeValue = scope[0]);
+        (dirty & 2 /* b */) && (refs[2].innerText = scope[1]);
+    } else if (stage === 3) {
+        refs[0].remove();
+    }
+}

test/samples/const-to-let.js

@@ -0,0 +1,11 @@
+import { html } from './endorphin';
+
+function ConstToLetComponent(props) {
+    const { a } = props;
+    const b = props.b;
+
+    return html`<div>
+        ${a}
+        <span> ${b}</span>
+    </div>`;
+}