Save names of defines for checking later

1 files changed, 4 insertions, 0 deletions

diff --git a/lib/lace/builtin.lua b/lib/lace/builtin.lua
index 2785389..e7ecd1b 100644
--- a/lib/lace/builtin.lua
+++ b/lib/lace/builtin.lua
@@ -318,6 +318,10 @@ function builtin.define(compcontext, define, name, controltype, ...)
       return false, msg
    end
 
+   -- Note the define name for checking by other control types
+   compcontext._lace.defined = (compcontext._lace.defined or {})
+   compcontext._lace.defined[name] = true
+
    -- Successfully created a control table, return a rule for it
    local rule = {
       fn = _do_define,