1 files changed, 517 insertions, 0 deletions
diff --git a/script/workspace.lua b/script/workspace.lua
new file mode 100644
index 00000000..f9b243ad
--- /dev/null
+++ b/script/workspace.lua
@@ -0,0 +1,517 @@
+local fs = require 'bee.filesystem'
+local async = require 'async'
+local config = require 'config'
+local ll = require 'lpeglabel'
+local platform = require 'bee.platform'
+local glob = require 'glob'
+local uric = require 'uri'
+
+local TrueName = {}
+
+local function getFileName(path)
+    local name = path:string()
+    if platform.OS == 'Windows' then
+        local lname = name:lower()
+        TrueName[lname] = name
+        return lname
+    else
+        return name
+    end
+end
+
+local function getTrueName(name)
+    return TrueName[name] or name
+end
+
+local function split(str, sep)
+    local t = {}
+    for s in str:gmatch('[^' .. sep .. ']+') do
+        t[#t+1] = s
+    end
+    return t
+end
+
+local function similarity(a, b)
+    local ta = split(a, '/\\')
+    local tb = split(b, '/\\')
+    for i = 1, #ta do
+        if ta[i] ~= tb[i] then
+            return i - 1
+        end
+    end
+    return #ta
+end
+
+local mt = {}
+mt.__index = mt
+
+function mt:fileNameEq(a, b)
+    if platform.OS == 'Windows' then
+        return a:lower() == b:lower()
+    else
+        return a == b
+    end
+end
+
+function mt:listenLoadFile()
+    self._loadFileRequest = async.run('loadfile', nil, function (filename, mode, buf)
+        local path = fs.path(filename)
+        local name = getFileName(path)
+        local uri = uric.encode(path)
+        self.files[name] = uri
+        if mode == 'workspace' then
+            self.lsp:readText(uri, path, buf, self._currentScanCompiled)
+        elseif mode == 'library' then
+            self.lsp:readLibrary(uri, path, buf, self._currentScanCompiled)
+        else
+            error('Unknown mode:' .. tostring(mode))
+        end
+    end)
+end
+
+function mt:buildScanPattern()
+    local pattern = {}
+
+    -- config.workspace.ignoreDir
+    for path in pairs(config.config.workspace.ignoreDir) do
+        pattern[#pattern+1] = path
+    end
+    -- config.files.exclude
+    for path, ignore in pairs(config.other.exclude) do
+        if ignore then
+            pattern[#pattern+1] = path
+        end
+    end
+    -- config.workspace.ignoreSubmodules
+    if config.config.workspace.ignoreSubmodules then
+        local buf = io.load(self.root / '.gitmodules')
+        if buf then
+            for path in buf:gmatch('path = ([^\r\n]+)') do
+                log.info('忽略子模块：', path)
+                pattern[#pattern+1] = path
+            end
+        end
+    end
+    -- config.workspace.useGitIgnore
+    if config.config.workspace.useGitIgnore then
+        local buf = io.load(self.root / '.gitignore')
+        if buf then
+            for line in buf:gmatch '[^\r\n]+' do
+                pattern[#pattern+1] = line
+            end
+        end
+    end
+    -- config.workspace.library
+    for path in pairs(config.config.workspace.library) do
+        pattern[#pattern+1] = path
+    end
+
+    return pattern
+end
+
+---@param options table
+function mt:buildLibraryRequests(options)
+    local requests = {}
+    for path, pattern in pairs(config.config.workspace.library) do
+        requests[#requests+1] = {
+            mode = 'library',
+            root = fs.absolute(fs.path(path)):string(),
+            pattern = pattern,
+            options = options,
+        }
+    end
+    return table.unpack(requests)
+end
+
+function mt:scanFiles()
+    if self._scanRequest then
+        log.info('Break scan.')
+        self._scanRequest:push('stop')
+        self._scanRequest = nil
+        self._complete = false
+        self:reset()
+    end
+
+    local pattern = self:buildScanPattern()
+    local options = {
+        ignoreCase = platform.OS == 'Windows',
+    }
+
+    self.gitignore = glob.gitignore(pattern, options)
+    self._currentScanCompiled = {}
+    local count = 0
+    self._scanRequest = async.run('scanfiles', {
+        {
+            mode = 'workspace',
+            root = self.root:string(),
+            pattern = pattern,
+            options = options,
+        },
+        self:buildLibraryRequests(options),
+    }, function (mode, ...)
+        if mode == 'ok' then
+            log.info('Scan finish, got', count, 'files.')
+            self._complete = true
+            self._scanRequest = nil
+            self:reset()
+            return true
+        elseif mode == 'log' then
+            log.debug(...)
+        elseif mode == 'workspace' then
+            local path = fs.path(...)
+            if not self:isLuaFile(path) then
+                return
+            end
+            self._loadFileRequest:push(path:string(), 'workspace')
+            count = count + 1
+        elseif mode == 'library' then
+            local path = fs.path(...)
+            if not self:isLuaFile(path) then
+                return
+            end
+            self._loadFileRequest:push(path:string(), 'library')
+            count = count + 1
+        elseif mode == 'stop' then
+            log.info('Scan stoped.')
+            return false
+        end
+    end)
+end
+
+function mt:init(rootUri)
+    self.root = uric.decode(rootUri)
+    self.uri = rootUri
+    if not self.root then
+        return
+    end
+    log.info('Workspace inited, root: ', self.root)
+    log.info('Workspace inited, uri: ', rootUri)
+    local logPath = ROOT / 'log' / (rootUri:gsub('[/:]+', '_') .. '.log')
+    log.info('Log path: ', logPath)
+    log.init(ROOT, logPath)
+
+    self:scanFiles()
+end
+
+function mt:isComplete()
+    return self._complete == true
+end
+
+function mt:isLuaFile(path)
+    local ext = path:extension():string()
+    for k, v in pairs(config.other.associations) do
+        if self:fileNameEq(ext, k:match('[^%*]+$')) then
+            if v == 'lua' then
+                return true
+            else
+                return false
+            end
+        end
+    end
+    if self:fileNameEq(ext, '.lua') then
+        return true
+    end
+    return false
+end
+
+function mt:addFile(path)
+    if not self:isLuaFile(path) then
+        return
+    end
+    local name = getFileName(path)
+    local uri = uric.encode(path)
+    self.files[name] = uri
+    self.lsp:readText(uri, path)
+end
+
+function mt:removeFile(path)
+    local name = getFileName(path)
+    if not self.files[name] then
+        return
+    end
+    self.files[name] = nil
+    local uri = uric.encode(path)
+    self.lsp:removeText(uri)
+end
+
+function mt:findPath(baseUri, searchers)
+    local results = {}
+    local basePath = uric.decode(baseUri)
+    if not basePath then
+        return nil
+    end
+    local baseName = getFileName(basePath)
+    for filename, uri in pairs(self.files) do
+        if filename ~= baseName then
+            for _, searcher in ipairs(searchers) do
+                if filename:sub(-#searcher) == searcher then
+                    local sep = filename:sub(-#searcher-1, -#searcher-1)
+                    if sep == '/' or sep == '\\' then
+                        results[#results+1] = uri
+                    end
+                end
+            end
+        end
+    end
+
+    if #results == 0 then
+        return nil
+    end
+    local uri
+    if #results == 1 then
+        uri = results[1]
+    else
+        table.sort(results, function (a, b)
+            return similarity(a, baseUri) > similarity(b, baseUri)
+        end)
+        uri = results[1]
+    end
+    return uri
+end
+
+function mt:createCompiler(str)
+    local state = {
+        'Main',
+    }
+    local function push(c)
+        if state.Main then
+            state.Main = state.Main * c
+        else
+            state.Main = c
+        end
+    end
+    local count = 0
+    local function code()
+        count = count + 1
+        local name = 'C' .. tostring(count)
+        local nextName = 'C' .. tostring(count + 1)
+        state[name] = ll.P(1) * (#ll.V(nextName) + ll.V(name))
+        return ll.V(name)
+    end
+    local function static(c)
+        count = count + 1
+        local name = 'C' .. tostring(count)
+        local nextName = 'C' .. tostring(count + 1)
+        local catch = #ll.V(nextName)
+        if platform.OS == 'Windows' then
+            for i = #c, 1, -1 do
+                local char = c:sub(i, i)
+                local u = char:upper()
+                local l = char:lower()
+                if u == l then
+                    catch = ll.P(char) * catch
+                else
+                    catch = (ll.P(u) + ll.P(l)) * catch
+                end
+            end
+        else
+            catch = ll.P(c) * catch
+        end
+        state[name] = catch
+        return ll.V(name)
+    end
+    local function eof()
+        count = count + 1
+        local name = 'C' .. tostring(count)
+        state[name] = ll.Cmt(ll.P(1) + ll.Cp(), function (_, _, c)
+            return type(c) == 'number'
+        end)
+        return ll.V(name)
+    end
+    local isFirstCode = true
+    local firstCode
+    local compiler = ll.P {
+        'Result',
+        Result = (ll.V'Code' + ll.V'Static')^1,
+        Code   = ll.P'?' / function ()
+            if isFirstCode then
+                isFirstCode = false
+                push(ll.Cmt(ll.C(code()), function (_, pos, code)
+                    firstCode = code
+                    return pos, code
+                end))
+            else
+                push(ll.Cmt(
+                    ll.C(code()),
+                    function (_, _, me)
+                        return firstCode == me
+                    end
+                ))
+            end
+        end,
+        Static = (1 - ll.P'?')^1 / function (c)
+            push(static(c))
+        end,
+    }
+    compiler:match(str)
+    push(eof())
+    return ll.P(state)
+end
+
+function mt:compileLuaPath()
+    for i, luapath in ipairs(config.config.runtime.path) do
+        self.pathMatcher[i] = self:createCompiler(luapath)
+    end
+end
+
+function mt:convertPathAsRequire(filename, start)
+    local list
+    for _, matcher in ipairs(self.pathMatcher) do
+        local str = matcher:match(filename:sub(start))
+        if str then
+            if not list then
+                list = {}
+            end
+            list[#list+1] = str:gsub('/', '.')
+        end
+    end
+    return list
+end
+
+function mt:matchPath(baseUri, input)
+    local first = input:match '^[^%.]+'
+    if not first then
+        return nil
+    end
+    first = first:gsub('%W', '%%%1')
+    local basePath = uric.decode(baseUri)
+    if not basePath then
+        return nil
+    end
+    local baseName = getFileName(basePath)
+    local rootLen = #self.root:string(basePath)
+    local map = {}
+    for filename in pairs(self.files) do
+        if filename ~= baseName then
+            local trueFilename = getTrueName(filename)
+            local start
+            if platform.OS == 'Windows' then
+                start = filename:find('[/\\]' .. first:lower(), rootLen + 1)
+            else
+                start = trueFilename:find('[/\\]' .. first, rootLen + 1)
+            end
+            if start then
+                local list = self:convertPathAsRequire(trueFilename, start + 1)
+                if list then
+                    for _, str in ipairs(list) do
+                        if #str >= #input and self:fileNameEq(str:sub(1, #input), input) then
+                            if not map[str] then
+                                map[str] = trueFilename
+                            else
+                                local s1 = similarity(trueFilename, baseName)
+                                local s2 = similarity(map[str], baseName)
+                                if s1 > s2 then
+                                    map[str] = trueFilename
+                                elseif s1 == s2 then
+                                    if trueFilename < map[str] then
+                                        map[str] = trueFilename
+                                    end
+                                end
+                            end
+                        end
+                    end
+                end
+            end
+        end
+    end
+
+    local list = {}
+    for str in pairs(map) do
+        list[#list+1] = str
+        map[str] = map[str]:sub(rootLen + 2)
+    end
+    if #list == 0 then
+        return nil
+    end
+    table.sort(list, function (a, b)
+        local sa = similarity(map[a], baseName)
+        local sb = similarity(map[b], baseName)
+        if sa == sb then
+            return a < b
+        else
+            return sa > sb
+        end
+    end)
+    return list, map
+end
+
+function mt:searchPath(baseUri, str)
+    str = getFileName(fs.path(str))
+    if self.searched[baseUri] and self.searched[baseUri][str] then
+        return self.searched[baseUri][str]
+    end
+    str = str:gsub('%.', '/')
+    local searchers = {}
+    for i, luapath in ipairs(config.config.runtime.path) do
+        searchers[i] = luapath:gsub('%?', str)
+    end
+
+    local uri = self:findPath(baseUri, searchers)
+    if uri then
+        if not self.searched[baseUri] then
+            self.searched[baseUri] = {}
+        end
+        self.searched[baseUri][str] = uri
+    end
+    return uri
+end
+
+function mt:loadPath(baseUri, str)
+    local ok, relative = pcall(fs.relative, fs.absolute(self.root / str), self.root)
+    if not ok then
+        return nil
+    end
+    str = getFileName(relative)
+    if self.loaded[str] then
+        return self.loaded[str]
+    end
+
+    local searchers = { str }
+
+    local uri = self:findPath(baseUri, searchers)
+    if uri then
+        self.loaded[str] = uri
+    end
+    return uri
+end
+
+function mt:reset()
+    self.searched = {}
+    self.loaded = {}
+    self.lsp:reCompile()
+end
+
+---@param uri uri
+---@return path
+function mt:relativePathByUri(uri)
+    local path = uric.decode(uri)
+    if not path then
+        return nil
+    end
+    local relate = fs.relative(path, self.root)
+    return relate
+end
+
+---@param uri uri
+---@return path
+function mt:absolutePathByUri(uri)
+    local path = uric.decode(uri)
+    if not path then
+        return nil
+    end
+    return fs.absolute(path)
+end
+
+return function (lsp, name)
+    local workspace = setmetatable({
+        lsp = lsp,
+        name = name,
+        files = {},
+        searched = {},
+        loaded = {},
+        pathMatcher = {}
+    }, mt)
+    workspace:compileLuaPath()
+    workspace:listenLoadFile()
+    return workspace
+end