Fixed blockless arrowFunctions

Floofies · Floofies · commit 2b7d27dd1c5d · 2019-01-06T19:21:09.000-05:00
diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
 	"name": "hertzscript-compiler",
-	"version": "0.0.2",
+	"version": "0.0.3",
 	"description": "Produces preemptible JavaScript coroutines which conform to the HertzScript specification.",
 	"keywords": [
 		"javascript",
diff --git a/src/BabelPlugin.js b/src/BabelPlugin.js
@@ -159,7 +159,8 @@ function Plugin(babel) {
 		funcExp.type = "FunctionExpression";
 		if (funcExp.body.type !== "BlockStatement") {
 			funcExp.body = t.blockStatement([
-				t.returnStatement(funcExp.body)
+				t.expressionStatement(
+					t.yieldExpression(hzReturnArg(funcExp.body)))
 			]);
 		}
 		return t.callExpression(

Original file line number	Diff line number	Diff line change
`@@ -1,6 +1,6 @@`
`1`	`1`	`{`
`2`	`2`	`"name": "hertzscript-compiler",`
`3`		`- "version": "0.0.2",`
	`3`	`+ "version": "0.0.3",`
`4`	`4`	`"description": "Produces preemptible JavaScript coroutines which conform to the HertzScript specification.",`
`5`	`5`	`"keywords": [`
`6`	`6`	`"javascript",`