luadoc的第一步

4 files changed, 65 insertions, 1 deletions

diff --git a/script-beta/vm/eachRef.lua b/script-beta/vm/eachRef.lua
index babfa4d8..429cbb73 100644
--- a/script-beta/vm/eachRef.lua
+++ b/script-beta/vm/eachRef.lua
@@ -5,7 +5,7 @@ local await  = require 'await'
 
 local function getRefs(source, simple)
     local results = {}
-    local lock = vm.lock('eachDef', source)
+    local lock = vm.lock('eachRef', source)
     if not lock then
         return results
     end
diff --git a/script-beta/vm/getDocs.lua b/script-beta/vm/getDocs.lua
new file mode 100644
index 00000000..5c8e6304
--- /dev/null
+++ b/script-beta/vm/getDocs.lua
@@ -0,0 +1,58 @@
+local files   = require 'files'
+local util    = require 'utility'
+local guide   = require 'parser.guide'
+local vm      = require 'vm.vm'
+
+local function getTypesOfFile(uri)
+    local classes = {}
+    local ast = files.getAst(uri)
+    if not ast or not ast.ast.docs then
+        return classes
+    end
+    guide.eachSource(ast.ast.docs, function (src)
+        if src.type == 'doc.type.name'
+        or src.type == 'doc.class.name'
+        or src.type == 'doc.extends.name' then
+            local name = src[1]
+            if name then
+                if not classes[name] then
+                    classes[name] = {}
+                end
+                classes[name][#classes[name]+1] = src
+            end
+        end
+    end)
+    return classes
+end
+
+local function getDocTypes(name)
+    local results = {}
+    for uri in files.eachFile() do
+        local cache = files.getCache(uri)
+        cache.classes = cache.classes or getTypesOfFile(uri)
+        if name == '*' then
+            for _, sources in util.sortPairs(cache.classes) do
+                for _, source in ipairs(sources) do
+                    results[#results+1] = source
+                end
+            end
+        else
+            if cache.classes[name] then
+                for _, source in ipairs(cache.classes[name]) do
+                    results[#results+1] = source
+                end
+            end
+        end
+    end
+    return results
+end
+
+function vm.getDocTypes(name)
+    local cache = vm.getCache('getDocTypes')[name]
+    if cache ~= nil then
+        return cache
+    end
+    cache = getDocTypes(name)
+    vm.getCache('getDocTypes')[name] = cache
+    return cache
+end
diff --git a/script-beta/vm/guideInterface.lua b/script-beta/vm/guideInterface.lua
index a77184f3..4d651882 100644
--- a/script-beta/vm/guideInterface.lua
+++ b/script-beta/vm/guideInterface.lua
@@ -85,6 +85,11 @@ function vm.interface.global(name)
     return vm.getGlobals(name)
 end
 
+function vm.interface.docType(name)
+    await.delay()
+    return vm.getDocTypes(name)
+end
+
 function vm.interface.link(uri)
     await.delay()
     return vm.getLinksTo(uri)
diff --git a/script-beta/vm/init.lua b/script-beta/vm/init.lua
index 7621859f..b9e8e147 100644
--- a/script-beta/vm/init.lua
+++ b/script-beta/vm/init.lua
@@ -1,5 +1,6 @@
 local vm = require 'vm.vm'
 require 'vm.getGlobals'
+require 'vm.getDocs'
 require 'vm.getLibrary'
 require 'vm.getInfer'
 require 'vm.getClass'