整理代码

1 files changed, 28 insertions, 0 deletions

diff --git a/script-beta/vm/getClass.lua b/script-beta/vm/getClass.lua
new file mode 100644
index 00000000..f1a50568
--- /dev/null
+++ b/script-beta/vm/getClass.lua
@@ -0,0 +1,28 @@
+local vm = require 'vm.vm'
+
+local function getClass(source, deep)
+    if deep > 3 then
+        return nil
+    end
+    local classes = {}
+    vm.eachField(source, function (src)
+        local key = vm.getKeyName(src)
+        local lkey = key:lower()
+        if lkey == 's|type'
+        or lkey == 's|__name'
+        or lkey == 's|name'
+        or lkey == 's|class' then
+            if src.value and src.value.type == 'string' then
+                classes[#classes+1] = src.value[1]
+            end
+        end
+    end)
+    if #classes == 0 then
+        return nil
+    end
+    return vm.mergeTypeViews(table.unpack(classes))
+end
+
+function vm.getClass(source)
+    return getClass(source, 1)
+end