generic param

1 files changed, 16 insertions, 1 deletions

diff --git a/script/core/infer.lua b/script/core/infer.lua
index 2578aac4..ce89a6c9 100644
--- a/script/core/infer.lua
+++ b/script/core/infer.lua
@@ -327,7 +327,11 @@ local function getDocName(doc)
     if doc.type == 'doc.class.name'
     or doc.type == 'doc.type.name' then
         local name = doc[1] or '?'
-        return name
+        if doc.typeGeneric then
+            return '<' .. name .. '>'
+        else
+            return name
+        end
     end
     if doc.type == 'doc.type.array' then
         local nodeName = getDocName(doc.node) or '?'
@@ -478,6 +482,17 @@ function m.searchInfers(source, field)
             searchInfer(def, infers)
         end
     end
+    if source.docParam then
+        local docType = source.docParam.extends
+        if docType.type == 'doc.type' then
+            for _, def in ipairs(docType.types) do
+                if def.typeGeneric and not mark[def] then
+                    mark[def] = true
+                    searchInfer(def, infers)
+                end
+            end
+        end
+    end
     cleanInfers(infers)
     return infers
 end