1 files changed, 125 insertions, 0 deletions
diff --git a/script/core/command/autoRequire.lua b/script/core/command/autoRequire.lua
new file mode 100644
index 00000000..947e78aa
--- /dev/null
+++ b/script/core/command/autoRequire.lua
@@ -0,0 +1,125 @@
+local files  = require 'files'
+local furi   = require 'file-uri'
+local config = require 'config'
+local rpath  = require 'workspace.require-path'
+local client = require 'client'
+local lang   = require 'language'
+
+local function findInsertOffset(uri)
+    local lines = files.getLines(uri)
+    local text  = files.getText(uri)
+    local fmt   = {
+        pair = false,
+        quot = '"',
+        col  = nil,
+    }
+    for i = 1, #lines do
+        local ln = lines[i]
+        local lnText = text:sub(ln.start, ln.finish)
+        if not lnText:find('require', 1, true) then
+            return ln.start, fmt
+        else
+            local lpPos = lnText:find '%('
+            if lpPos then
+                fmt.pair = true
+            else
+                fmt.pair = false
+            end
+            local quot = lnText:match [=[(['"])]=]
+            fmt.quot = quot or fmt.quot
+            local eqPos = lnText:find '='
+            if eqPos then
+                fmt.col = eqPos
+            end
+        end
+    end
+    return 1, fmt
+end
+
+local function askAutoRequire(visiblePaths)
+    -- TODO: translate
+    local selects = {}
+    local nameMap = {}
+    for _, visible in ipairs(visiblePaths) do
+        local expect = visible.expect
+        local select = lang.script(expect)
+        nameMap[select] = expect
+        selects[#selects+1] = select
+    end
+    local disable = lang.script('禁用自动require')
+    selects[#selects+1] = disable
+
+    local result = client.awaitRequestMessage('Info'
+        , lang.script('在文件顶部添加代码 require 此文件？')
+        , selects
+    )
+    if not result then
+        return
+    end
+    if result == disable then
+        client.setConfig {
+            {
+                key    = 'Lua.completion.autoRequire',
+                action = 'set',
+                value  = false,
+            }
+        }
+        return
+    end
+    return nameMap[result]
+end
+
+local function applyAutoRequire(uri, offset, name, result, fmt)
+    local quotedResult = ('%q'):format(result)
+    if fmt.quot == "'" then
+        quotedResult = ([['%s']]):format(quotedResult:sub(2, -2)
+            :gsub([[']], [[\']])
+            :gsub([[\"]], [["]])
+        )
+    end
+    if fmt.pair then
+        quotedResult = ('(%s)'):format(quotedResult)
+    else
+        quotedResult = (' %s'):format(quotedResult)
+    end
+    local sp = ' '
+    local text = ('local %s'):format(name)
+    if fmt.col and fmt.col > #text then
+        sp = (' '):rep(fmt.col - #text - 1)
+    end
+    text = ('local %s%s= require%s\n'):format(name, sp, quotedResult)
+    client.editText(uri, {
+        {
+            start  = offset,
+            finish = offset - 1,
+            text   = text,
+        }
+    })
+end
+
+return function (data)
+    local uri    = data.uri
+    local target = data.target
+    local name   = data.name
+    local state  = files.getState(uri)
+    if not state then
+        return
+    end
+
+    local offset, fmt = findInsertOffset(uri)
+    local path = furi.decode(target)
+    local visiblePaths = rpath.getVisiblePath(path, config.get 'Lua.runtime.path')
+    if not visiblePaths or #visiblePaths == 0 then
+        return
+    end
+    table.sort(visiblePaths, function (a, b)
+        return #a.expect < #b.expect
+    end)
+
+    local result = askAutoRequire(visiblePaths)
+    if not result then
+        return
+    end
+
+    applyAutoRequire(uri, offset, name, result, fmt)
+end