Augmented assignment operators (#87)

tact-lang · Jan 22, 2024 · db9ac5d · db9ac5d
1 parent 3ca10a8
commit db9ac5d
Show file tree

Hide file tree

Showing 22 changed files with 904 additions and 1,471 deletions.
diff --git a/src/generator/writers/writeFunction.ts b/src/generator/writers/writeFunction.ts
@@ -97,6 +97,11 @@ export function writeStatement(f: ASTStatement, self: string | null, returns: Ty
 
         ctx.append(`${path} = ${writeCastedExpression(f.expression, t, ctx)};`);
         return;
+    } else if (f.kind === 'statement_augmentedassign') {
+        let path = f.path.map((v, i) => (i === 0) ? id(v.name) : v.name).join(`'`);
+        let t = getExpType(ctx.ctx, f.path[f.path.length - 1]);
+        ctx.append(`${path} = ${cast(t, t, `${path} ${f.op} ${writeExpression(f.expression, ctx)}`, ctx)};`);
+        return;
     } else if (f.kind === 'statement_condition') {
         writeCondition(f, self, false, returns, ctx);
         return;