fix performance issue

2 files changed, 23 insertions, 12 deletions

diff --git a/changelog.md b/changelog.md
index 590d38ee..13aa482b 100644
--- a/changelog.md
+++ b/changelog.md
@@ -5,6 +5,7 @@
 * `NEW` using `enum (partial)`, it suggests all fields with the same `enum` type rather than just the fields from the current table.
 * `NEW` When using `enum["<key>" or <index>]`, undefined fields will raise an 'undefined' error.
 * `FIX` Renaming files in the directory leads to the auto-correction in "require" adding extra characters.
+* `FIX` Performance issue
 
 ## 3.10.4
 `2024-8-16`
diff --git a/script/core/fix-indent.lua b/script/core/fix-indent.lua
index 0542e4a8..0b4b6985 100644
--- a/script/core/fix-indent.lua
+++ b/script/core/fix-indent.lua
@@ -5,12 +5,16 @@ local lookBackward = require 'core.look-backward'
 local util = require 'utility'
 local client = require 'client'
 
----@param state parser.state
+---@param uri uri
 ---@param change table
-local function removeSpacesAfterEnter(state, change)
+local function removeSpacesAfterEnter(uri, change)
     if not change.text:match '^\r?\n[\t ]+\r?\n$' then
         return false
     end
+    local state = files.getState(uri)
+    if not state then
+        return false
+    end
     local lines = state.originLines or state.lines
     local text  = state.originText  or state.lua
     ---@cast text -?
@@ -76,10 +80,15 @@ local function getBlock(state, pos)
     return block
 end
 
-local function fixWrongIndent(state, change)
+---@param uri uri
+local function fixWrongIndent(uri, change)
     if not change.text:match '^\r?\n[\t ]+$' then
         return false
     end
+    local state = files.getState(uri)
+    if not state then
+        return false
+    end
     local position = guide.positionOf(change.range.start.line, change.range.start.character)
     local row = guide.rowColOf(position)
     local myIndent   = getIndent(state, row + 1)
@@ -114,12 +123,17 @@ local function fixWrongIndent(state, change)
     return edits
 end
 
----@param state parser.state
-local function applyEdits(state, edits)
+---@param uri uri
+local function applyEdits(uri, edits)
     if #edits == 0 then
         return
     end
 
+    local state = files.getState(uri)
+    if not state then
+        return
+    end
+
     local lines = state.originLines or state.lines
 
     local results = {}
@@ -157,17 +171,13 @@ return function (uri, changes)
     if not client.getOption('fixIndents') then
         return
     end
-    local state = files.compileState(uri)
-    if not state then
-        return
-    end
 
     local firstChange = changes[1]
     if firstChange.range then
-        local edits = removeSpacesAfterEnter(state, firstChange)
-                or    fixWrongIndent(state, firstChange)
+        local edits = removeSpacesAfterEnter(uri, firstChange)
+                or    fixWrongIndent(uri, firstChange)
         if edits then
-            applyEdits(state, edits)
+            applyEdits(uri, edits)
         end
     end
 end