Method generateBody

pkg/codegen/lambda.go:108–123

generateBody generates the function body. For expression bodies: - Wraps in { return ... } For block bodies: - If has return type and single expression: adds implicit return - Otherwise: uses body as-is (already has { ... })

()

//   - If has return type and single expression: adds implicit return
//   - Otherwise: uses body as-is (already has { ... })
func (g *LambdaCodeGen) generateBody() {
	if g.expr.IsBlock {
		// Block body - check if we need implicit return
		if g.expr.ReturnType != "" && g.needsImplicitReturn() {
			g.writeBlockWithImplicitReturn()
		} else {
			// Pass through as-is
			g.Write(g.expr.Body)
		}
	} else {
		// Expression body - wrap in { return ... }
		g.Write("{ return ")
		g.Write(g.expr.Body)
		g.Write(" }")
	}
}

// needsImplicitReturn checks if a block body needs an implicit return.
// Returns true if the block contains a single expression without return/if/for/switch.