diagnostic group

2 files changed, 21 insertions, 0 deletions

diff --git a/script/proto/define.lua b/script/proto/define.lua
index 93907847..c73203f5 100644
--- a/script/proto/define.lua
+++ b/script/proto/define.lua
@@ -10,6 +10,12 @@ m.DiagnosticSeverity = {
     Hint        = 4,
 }
 
+m.DiagnosticFileStatus = {
+    Any        = 1,
+    Opened     = 2,
+    None       = 3,
+}
+
 --- 诊断类型与默认等级
 m.DiagnosticDefaultSeverity = diag.getDefaultSeverity()
 
diff --git a/script/proto/diagnostic.lua b/script/proto/diagnostic.lua
index cba6a493..d8bac443 100644
--- a/script/proto/diagnostic.lua
+++ b/script/proto/diagnostic.lua
@@ -1,3 +1,5 @@
+local util = require 'utility'
+
 ---@class proto.diagnostic
 local m = {}
 
@@ -217,4 +219,17 @@ function m.getGroupStatus()
     return group
 end
 
+---@param name string
+---@return string[]
+m.getGroups = util.catchReturn(function (name)
+    local groups = {}
+    for groupName, nameMap in pairs(m.diagnosticGroups) do
+        if nameMap[name] then
+            groups[#groups+1] = groupName
+        end
+    end
+    table.sort(groups)
+    return groups
+end)
+
 return m