Allow implicit function return types (#2866)

Resolves #2865
wiremod · Nov 19, 2023 · 9add762 · 9add762
1 parent 2fdfd38
commit 9add762
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 2 deletions.
diff --git a/data/expression2/tests/runtime/base/userfunctions/functions.txt b/data/expression2/tests/runtime/base/userfunctions/functions.txt
@@ -111,4 +111,18 @@ try {
 	error("unreachable")
 } catch(Err) {
 	assert(Err == "No such function defined at runtime: undefined()")
-}
+}
+
+function implicit() {
+	return 55
+}
+
+assert( implicit() == 55 )
+
+function implicit_lambda() {
+	return function() {
+		return 5
+	}
+}
+
+assert( implicit_lambda()()[number] == 5 )
diff --git a/lua/entities/gmod_wire_expression2/base/compiler.lua b/lua/entities/gmod_wire_expression2/base/compiler.lua
@@ -825,7 +825,11 @@ local CompileVisitors = {
 			end
 		end)
 
-		self:Assert(fn.ret == return_type, "Function " .. name.value .. " expects to return type (" .. (return_type or "void") .. ") but got type (" .. (fn.ret or "void") .. ")", trace)
+		if return_type then
+			self:Assert(fn.ret == return_type, "Function " .. name.value .. " expects to return type (" .. return_type .. ") but got type (" .. (fn.ret or "void") .. ")", trace)
+		else
+			return_type = fn.ret
+		end
 
 		local sig = name.value .. "(" .. (meta_type and (meta_type .. ":") or "") .. sig .. ")"
 		local fn = fn.op