From 521feda953a686ad876186366391340f945ca65b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=9C=80=E8=90=8C=E5=B0=8F=E6=B1=90?= <sumneko@hotmail.com>
Date: Mon, 4 Nov 2019 18:22:51 +0800
Subject: =?UTF-8?q?=E6=9B=B4=E6=96=B0=E8=AF=8A=E6=96=AD=E5=AE=9E=E7=8E=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../src/core/diagnostics/undefined-global.lua      | 37 ++++++++++++++++++++--
 .../src/core/diagnostics/unused-function.lua       | 29 +++++++++++++++--
 server-beta/src/core/diagnostics/unused-local.lua  |  2 +-
 3 files changed, 63 insertions(+), 5 deletions(-)

(limited to 'server-beta/src/core')

diff --git a/server-beta/src/core/diagnostics/undefined-global.lua b/server-beta/src/core/diagnostics/undefined-global.lua
index b3d19c21..db7af133 100644
--- a/server-beta/src/core/diagnostics/undefined-global.lua
+++ b/server-beta/src/core/diagnostics/undefined-global.lua
@@ -1,3 +1,36 @@
-return function ()
-    
+local files    = require 'files'
+local guide    = require 'parser.guide'
+local searcher = require 'searcher'
+local define   = require 'proto.define'
+local lang     = require 'language'
+
+return function (uri, callback)
+    local ast = files.getAst(uri)
+    if not ast then
+        return
+    end
+
+    local hasSet = {}
+    searcher.eachGlobal(ast.ast, function (info)
+        local key = info.key
+        if hasSet[key] ~= nil then
+            return
+        end
+        hasSet[key] = false
+        searcher.eachRef(info.source, function (info)
+            if info.mode == 'set' then
+                hasSet[key] = true
+            end
+        end)
+    end)
+    searcher.eachGlobal(ast.ast, function (info)
+        local source = info.source
+        if info.mode == 'get' and not hasSet[info.key] then
+            callback {
+                start   = source.start,
+                finish  = source.finish,
+                message = lang.script('DIAG_UNDEF_GLOBAL', info.key),
+            }
+        end
+    end)
 end
diff --git a/server-beta/src/core/diagnostics/unused-function.lua b/server-beta/src/core/diagnostics/unused-function.lua
index b3d19c21..80123948 100644
--- a/server-beta/src/core/diagnostics/unused-function.lua
+++ b/server-beta/src/core/diagnostics/unused-function.lua
@@ -1,3 +1,28 @@
-return function ()
-    
+local files    = require 'files'
+local guide    = require 'parser.guide'
+local searcher = require 'searcher'
+local define = require 'proto.define'
+local lang   = require 'language'
+
+return function (uri, callback)
+    local ast = files.getAst(uri)
+    if not ast then
+        return
+    end
+    guide.eachSourceType(ast.ast, 'function', function (source)
+        local hasGet
+        searcher.eachRef(source, function (info)
+            if info.mode == 'get' then
+                hasGet = true
+            end
+        end)
+        if not hasGet then
+            callback {
+                start   = source.start,
+                finish  = source.finish,
+                tags    = { define.DiagnosticTag.Unnecessary },
+                message = lang.script.DIAG_UNUSED_FUNCTION,
+            }
+        end
+    end)
 end
diff --git a/server-beta/src/core/diagnostics/unused-local.lua b/server-beta/src/core/diagnostics/unused-local.lua
index 9b97ca88..d110a714 100644
--- a/server-beta/src/core/diagnostics/unused-local.lua
+++ b/server-beta/src/core/diagnostics/unused-local.lua
@@ -8,7 +8,7 @@ local function hasGet(loc)
         return false
     end
     for _, ref in ipairs(loc.ref) do
-        if ref.type == 'get' then
+        if ref.mode == 'get' then
             return true
         end
     end
-- 
cgit v1.2.3