cleanup

1 files changed, 34 insertions, 6 deletions

diff --git a/script/files.lua b/script/files.lua
index b84295bf..df49b183 100644
--- a/script/files.lua
+++ b/script/files.lua
@@ -278,13 +278,17 @@ function m.getWords(uri)
     end
     local mark = {}
     for word in text:gmatch '([%a_][%w_]+)' do
-        if #word >= 3 and not mark[word] then
-            mark[word] = true
-            local head = word:sub(1, 2)
-            if not words[head] then
-                words[head] = {}
+        if #word >= 3 then
+            if not mark[word] then
+                mark[word] = true
+                local head = word:sub(1, 2)
+                if not words[head] then
+                    words[head] = {}
+                end
+                words[head][word] = true
             end
-            words[head][#words[head]+1] = word
+        else
+            words[word] = true
         end
     end
     return words
@@ -303,6 +307,30 @@ function m.getWordsOfHead(uri, head)
     return words[head]
 end
 
+---@param uri uri
+---@param word string
+---@return boolean
+function m.hasWord(uri, word)
+    uri = getUriKey(uri)
+    local file = m.fileMap[uri]
+    if not file then
+        return false
+    end
+    local words = m.getWords(uri)
+    if not words then
+        return false
+    end
+    if #word <= 2 then
+        return words[word] == true
+    end
+    local head = word:sub(1, 2)
+    local headWords = words[head]
+    if not headWords then
+        return false
+    end
+    return headWords[word] == true
+end
+
 --- 获取文件版本
 function m.getVersion(uri)
     uri = getUriKey(uri)