1 files changed, 66 insertions, 41 deletions
diff --git a/server/src/method/textDocument/codeAction.lua b/server/src/method/textDocument/codeAction.lua
index e85a8aa3..845b2591 100644
--- a/server/src/method/textDocument/codeAction.lua
+++ b/server/src/method/textDocument/codeAction.lua
@@ -1,49 +1,74 @@
-local function solveDiagnostic(lsp, uri, data, callback)
-    if data.code then
-        callback {
-            title = ('禁用诊断(%s)'):format(data.code),
-            kind = 'quickfix',
-            diagnostics = {data},
-            command = {
-                title = '禁用诊断',
-                command = 'config',
-                arguments = {
-                    {
-                        key = {'diagnostics', 'disable'},
-                        action = 'add',
-                        value = data.code,
-                    }
+local function disableDiagnostic(lsp, uri, data, callback)
+    callback {
+        title = ('禁用诊断(%s)'):format(data.code),
+        kind = 'quickfix',
+        command = {
+            title = '禁用诊断',
+            command = 'config',
+            arguments = {
+                {
+                    key = {'diagnostics', 'disable'},
+                    action = 'add',
+                    value = data.code,
                 }
             }
         }
+    }
+end
+
+local function solveUndefinedGlobal(lsp, uri, data, callback)
+    local vm, lines, text = lsp:getVM(uri)
+    if not vm then
+        return
     end
-    if data.code == 'undefined-global' then
-        local vm, lines, text = lsp:getVM(uri)
-        if not vm then
-            return
-        end
-        local start = lines:position(data.range.start.line + 1, data.range.start.character + 1)
-        local finish = lines:position(data.range['end'].line + 1, data.range['end'].character)
-        local name = text:sub(start, finish)
-        if #name < 0 or name:find('[^%w_]') then
-            return
-        end
-        callback {
-            title = ('标记 `%s` 为已定义的全局变量'):format(name),
-            kind = 'quickfix',
-            diagnostics = {data},
-            command = {
-                title = '标记全局变量',
-                command = 'config',
-                arguments = {
-                    {
-                        key = {'diagnostics', 'globals'},
-                        action = 'add',
-                        value = name,
-                    }
+    local start = lines:position(data.range.start.line + 1, data.range.start.character + 1)
+    local finish = lines:position(data.range['end'].line + 1, data.range['end'].character)
+    local name = text:sub(start, finish)
+    if #name < 0 or name:find('[^%w_]') then
+        return
+    end
+    callback {
+        title = ('标记 `%s` 为已定义的全局变量'):format(name),
+        kind = 'quickfix',
+        command = {
+            title = '标记全局变量',
+            command = 'config',
+            arguments = {
+                {
+                    key = {'diagnostics', 'globals'},
+                    action = 'add',
+                    value = name,
                 }
-            },
-        }
+            }
+        },
+    }
+end
+
+local function solveTrailingSpace(lsp, uri, data, callback)
+    callback {
+        title = '清除所有后置空格',
+        kind = 'quickfix',
+        command = {
+            title = '清除所有后置空格',
+            command = 'removeSpace',
+            arguments = {
+                {
+                    uri = uri,
+                }
+            }
+        },
+    }
+end
+
+local function solveDiagnostic(lsp, uri, data, callback)
+    if data.code then
+        disableDiagnostic(lsp, uri, data, callback)
+    end
+    if data.code == 'undefined-global' then
+        solveUndefinedGlobal(lsp, uri, data, callback)
+    end
+    if data.code == 'trailing-space' then
+        solveTrailingSpace(lsp, uri, data, callback)
     end
 end