fix: handle return statements in init function separately (#794)

Author: Gusarich

CHANGELOG.md

@@ -36,6 +36,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 - Tact AST keeps the original format of integer literals (hex/dec/oct/bin): PR [#771](https://github.com/tact-lang/tact/pull/771)
 - Message opcodes are now checked if they fit into 32 bits: PR [#771](https://github.com/tact-lang/tact/pull/771)
 - Disallow zero binary message opcodes as those are reserved for text messages: PR [#786](https://github.com/tact-lang/tact/pull/786)
+- Return-statements in `init()` function do not cause FunC compilation error anymore: PR [#794](https://github.com/tact-lang/tact/pull/794)
 
 ## [1.4.4] - 2024-08-18
 

src/generator/writers/writeContract.ts

@@ -146,7 +146,11 @@ export function writeInit(
             // Generate statements
             const returns = resolveFuncTypeUnpack(t, funcIdOf("self"), ctx);
             for (const s of init.ast.statements) {
-                writeStatement(s, returns, null, ctx);
+                if (s.kind === "statement_return") {
+                    ctx.append(`return ${returns};`);
+                } else {
+                    writeStatement(s, returns, null, ctx);
+                }
             }
 
             // Return result

src/test/e2e-emulated/contracts/init-return.tact

@@ -0,0 +1,14 @@
+contract Test {
+    a: Int;
+    
+    init() {
+        self.a = 123;
+        return;
+    }
+
+    receive () {}
+
+    get fun a(): Int {
+        return self.a;
+    }
+}

src/test/e2e-emulated/init-return.spec.ts

@@ -0,0 +1,34 @@
+import { toNano } from "@ton/core";
+import { Blockchain, SandboxContract, TreasuryContract } from "@ton/sandbox";
+import { Test } from "./contracts/output/init-return_Test";
+import "@ton/test-utils";
+
+describe("init-return", () => {
+    let blockchain: Blockchain;
+    let treasure: SandboxContract<TreasuryContract>;
+    let contract: SandboxContract<Test>;
+
+    beforeEach(async () => {
+        blockchain = await Blockchain.create();
+        treasure = await blockchain.treasury("treasure");
+
+        contract = blockchain.openContract(await Test.fromInit());
+
+        const deployResult = await contract.send(
+            treasure.getSender(),
+            { value: toNano("10") },
+            null,
+        );
+
+        expect(deployResult.transactions).toHaveTransaction({
+            from: treasure.address,
+            to: contract.address,
+            success: true,
+            deploy: true,
+        });
+    });
+
+    it("should deploy with return statement in init", async () => {
+        expect(await contract.getA()).toEqual(123n);
+    });
+});

tact.config.json

@@ -395,6 +395,11 @@
       "name": "init-of-message",
       "path": "./src/test/e2e-emulated/contracts/init-of-message.tact",
       "output": "./src/test/e2e-emulated/contracts/output"
+    },
+    {
+      "name": "init-return",
+      "path": "./src/test/e2e-emulated/contracts/init-return.tact",
+      "output": "./src/test/e2e-emulated/contracts/output"
     }
   ]
 }