alias支持字面量

2 files changed, 21 insertions, 0 deletions

diff --git a/server/src/emmy/alias.lua b/server/src/emmy/alias.lua
index cec23b78..be47cb4f 100644
--- a/server/src/emmy/alias.lua
+++ b/server/src/emmy/alias.lua
@@ -21,11 +21,22 @@ function mt:bindType(type)
     end
 end
 
+function mt:addEnum(enum)
+    self._enum[#self._enum+1] = enum
+end
+
+function mt:eachEnum(callback)
+    for _, enum in ipairs(self._enum) do
+        callback(enum)
+    end
+end
+
 return function (manager, source)
     local self = setmetatable({
         name = source[1][1],
         source = source.id,
         _manager = manager,
+        _enum = {},
     }, mt)
     return self
 end
diff --git a/server/src/emmy/manager.lua b/server/src/emmy/manager.lua
index 85f25bd1..cce1e0e4 100644
--- a/server/src/emmy/manager.lua
+++ b/server/src/emmy/manager.lua
@@ -128,6 +128,9 @@ function mt:addAlias(source, typeObj)
     aliasObj:bindType(typeObj)
     local list = self:getClass(aliasName)
     list[source.id] = aliasObj
+    for i = 3, #source do
+        aliasObj:addEnum(source[i][1])
+    end
     return aliasObj
 end
 
@@ -135,6 +138,13 @@ function mt:addParam(source, bind)
     local paramObj = newParam(self, source)
     if bind.type == 'emmy.type' then
         paramObj:bindType(bind)
+        self:eachClass(bind:getName(), function (class)
+            if class.type == 'emmy.alias' then
+                class:eachEnum(function (enum)
+                    paramObj:addEnum(enum)
+                end)
+            end
+        end)
     elseif bind.type == 'emmy.generic' then
         paramObj:bindGeneric(bind)
     end