Check in allow/deny that names are defined

1 files changed, 10 insertions, 0 deletions

diff --git a/lib/lace/builtin.lua b/lib/lace/builtin.lua
index e7ecd1b..da728a3 100644
--- a/lib/lace/builtin.lua
+++ b/lib/lace/builtin.lua
@@ -138,6 +138,16 @@ local function _return(compcontext, result, reason, ...)
    local cond = {...}
    if #cond == 0 then
       unconditional_result = result
+   else
+      compcontext._lace.defined = (compcontext._lace.defined or {})
+      for i, dname in ipairs(cond) do
+	 if dname:sub(1,1) == "!" then
+	    dname = dname:sub(2)
+	 end
+	 if not compcontext._lace.defined[dname] then
+	    return err.error("Undefined name used in condition ("..dname..")", {i+2})
+	 end
+      end
    end
    last_result = result