feat: rework destructuring to support mapping of fields

tact-lang · Oct 8, 2024 · 6c3ba88 · 6c3ba88
1 parent a0c737b
commit 6c3ba88
Show file tree

Hide file tree

Showing 13 changed files with 456 additions and 141 deletions.
diff --git a/src/generator/writers/writeFunction.ts b/src/generator/writers/writeFunction.ts
@@ -466,8 +466,22 @@ export function writeStatement(
             return;
         }
         case "statement_destruct": {
+            const t = getExpType(ctx.ctx, f.expression);
+            if (t.kind !== "ref") {
+                throwInternalCompilerError(
+                    `invalid destruct expression kind: ${t.kind}`,
+                    f.expression.loc,
+                );
+            }
+            const ty = getType(ctx.ctx, t.name);
+            const ids = ty.fields.map((field) => {
+                const id = f.identifiers.find(
+                    (item) => item.from.text === field.name,
+                );
+                return id ? funcIdOf(id.name) : "_";
+            });
             ctx.append(
-                `var (${f.identifiers.map((n) => funcIdOf(n)).join(", ")}) = ${writeCastedExpression(f.expression, getExpType(ctx.ctx, f.expression), ctx)};`,
+                `var (${ids.join(", ")}) = ${writeCastedExpression(f.expression, t, ctx)};`,
             );
             return;
         }