close #300

1 files changed, 16 insertions, 6 deletions

diff --git a/script/core/diagnostics/unused-function.lua b/script/core/diagnostics/unused-function.lua
index a68b8dcc..e6ae9386 100644
--- a/script/core/diagnostics/unused-function.lua
+++ b/script/core/diagnostics/unused-function.lua
@@ -4,6 +4,7 @@ local vm      = require 'vm'
 local define  = require 'proto.define'
 local lang    = require 'language'
 local await   = require 'await'
+local client  = require 'provider.client'
 
 local function isToBeClosed(source)
     if not source.attrs then
@@ -44,12 +45,21 @@ return function (uri, callback)
             end
         end
         if not hasGet then
-            callback {
-                start   = source.start,
-                finish  = source.finish,
-                tags    = { define.DiagnosticTag.Unnecessary },
-                message = lang.script.DIAG_UNUSED_FUNCTION,
-            }
+            if client.isVSCode() then
+                callback {
+                    start   = source.start,
+                    finish  = source.finish,
+                    tags    = { define.DiagnosticTag.Unnecessary },
+                    message = lang.script.DIAG_UNUSED_FUNCTION,
+                }
+            else
+                callback {
+                    start   = source.keyword[1],
+                    finish  = source.keyword[2],
+                    tags    = { define.DiagnosticTag.Unnecessary },
+                    message = lang.script.DIAG_UNUSED_FUNCTION,
+                }
+            end
         end
     end)
 end