From 687313e295f299c1e0e410d4d72d96b1f87b681c 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: Wed, 22 Sep 2021 22:07:48 +0800
Subject: update

---
 script/core/command/autoRequire.lua | 23 +++++++++++++----------
 1 file changed, 13 insertions(+), 10 deletions(-)

(limited to 'script')

diff --git a/script/core/command/autoRequire.lua b/script/core/command/autoRequire.lua
index e784a827..d711fb16 100644
--- a/script/core/command/autoRequire.lua
+++ b/script/core/command/autoRequire.lua
@@ -4,19 +4,22 @@ local config = require 'config'
 local rpath  = require 'workspace.require-path'
 local client = require 'client'
 local lang   = require 'language'
+local guide  = require 'parser.guide'
 
-local function findInsertOffset(uri)
+local function findInsertRow(uri)
     local text  = files.getText(uri)
+    local state = files.getState(uri)
+    local lines = state.lines
     local fmt   = {
         pair = false,
         quot = '"',
         col  = nil,
     }
-    for i = 1, #lines do
+    for i = 0, #lines do
         local ln = lines[i]
-        local lnText = text:sub(ln.start, ln.finish)
+        local lnText = text:match('[^\r\n]*', ln)
         if not lnText:find('require', 1, true) then
-            return ln.start, fmt
+            return i, fmt
         else
             local lpPos = lnText:find '%('
             if lpPos then
@@ -32,7 +35,7 @@ local function findInsertOffset(uri)
             end
         end
     end
-    return 1, fmt
+    return 0, fmt
 end
 
 local function askAutoRequire(visiblePaths)
@@ -69,7 +72,7 @@ local function askAutoRequire(visiblePaths)
     return nameMap[result]
 end
 
-local function applyAutoRequire(uri, offset, name, result, fmt)
+local function applyAutoRequire(uri, row, name, result, fmt)
     local quotedResult = ('%q'):format(result)
     if fmt.quot == "'" then
         quotedResult = ([['%s']]):format(quotedResult:sub(2, -2)
@@ -87,11 +90,11 @@ local function applyAutoRequire(uri, offset, name, result, fmt)
     if fmt.col and fmt.col > #text then
         sp = (' '):rep(fmt.col - #text - 1)
     end
-    text = ('\nlocal %s%s= require%s\n'):format(name, sp, quotedResult)
+    text = ('local %s%s= require%s\n'):format(name, sp, quotedResult)
     client.editText(uri, {
         {
-            start  = offset,
-            finish = offset - 1,
+            start  = guide.positionOf(row, 0),
+            finish = guide.positionOf(row, 0),
             text   = text,
         }
     })
@@ -120,6 +123,6 @@ return function (data)
         return
     end
 
-    local offset, fmt = findInsertOffset(uri)
+    local offset, fmt = findInsertRow(uri)
     applyAutoRequire(uri, offset, name, result, fmt)
 end
-- 
cgit v1.2.3