prepare client test

Author: sumneko

.luarc.json

@@ -20,7 +20,9 @@
         "version": "Lua 5.4",
         "path": [
             "script/?.lua",
-            "script/?/init.lua"
+            "script/?/init.lua",
+            "test/?.lua",
+            "test/?/init.lua"
         ],
         "pathStrict": true
     },

script/files.lua

@@ -16,19 +16,24 @@ local encoder  = require 'encoder'
 ---@class files
 local m = {}
 
-m.openMap        = {}
-m.libraryMap     = {}
-m.fileMap        = {}
-m.dllMap         = {}
 m.watchList      = {}
 m.notifyCache    = {}
-m.visible        = {}
 m.assocVersion   = -1
 m.assocMatcher   = nil
-m.globalVersion  = 0
-m.fileCount      = 0
-m.astCount       = 0
-m.astMap         = {} -- setmetatable({}, { __mode = 'v' })
+
+function m.reset()
+    m.openMap        = {}
+    m.libraryMap     = {}
+    m.fileMap        = {}
+    m.dllMap         = {}
+    m.visible        = {}
+    m.globalVersion  = 0
+    m.fileCount      = 0
+    m.astCount       = 0
+    m.astMap         = {}
+end
+
+m.reset()
 
 local fixedUri = {}
 --- 获取文件的真实uri(真实大小写)

script/gc.lua

@@ -0,0 +1,117 @@
+local util = require 'share.utility'
+
+---@class gc
+---@field _list table
+local mt = {}
+mt.__index = mt
+mt.type = 'gc'
+
+mt._max = 10
+
+local function destroyGCObject(obj)
+    local tp = type(obj)
+    if tp == 'function' then
+        xpcall(obj, log.error)
+    elseif tp == 'table' then
+        local remove = obj.remove
+        if type(remove) == 'function' then
+            xpcall(remove, log.error, obj)
+        end
+    end
+end
+
+local function isRemoved(obj)
+    local tp = type(obj)
+    if tp == 'function' then
+        for i = 1, 1000 do
+            local n, v = debug.getupvalue(obj, i)
+            if not n then
+                log.warn('函数式析构器没有 removed 上值！', util.dump(debug.getinfo(obj)))
+                break
+            end
+            if n == 'removed' then
+                if v then
+                    return true
+                end
+                break
+            end
+        end
+    elseif tp == 'table' then
+        if obj._removed then
+            return true
+        end
+    end
+    return false
+end
+
+local function zip(self)
+    local list = self._list
+    local index = 1
+    for i = 1, #list do
+        local obj = list[index]
+        if not obj then
+            break
+        end
+        if isRemoved(obj) then
+            if index == #list then
+                list[#list] = nil
+                break
+            end
+            list[index] = list[#list]
+        else
+            index = index + 1
+        end
+    end
+    self._max = #list * 1.5
+    if self._max < 10 then
+        self._max = 10
+    end
+end
+
+function mt:remove()
+    if self._removed then
+        return
+    end
+    self._removed = true
+    local list = self._list
+    for i = 1, #list do
+        destroyGCObject(list[i])
+    end
+end
+
+--- 标记`obj`在buff移除时自动移除。如果`obj`是个`function`,
+--- 则直接调用；如果`obj`是个`table`，则调用内部的`remove`方法。
+--- 其他情况不做处理
+---@param obj any
+---@return any
+function mt:add(obj)
+    if self._removed then
+        destroyGCObject(obj)
+        return nil
+    end
+    self._list[#self._list+1] = obj
+    if #self._list > self._max then
+        zip(self)
+    end
+    return obj
+end
+
+--- 创建一个gc容器，使用 `gc:add(obj)` 将析构器放入gc容器。
+---
+--- 当gc容器被销毁时，会调用内部的析构器（不保证调用顺序）
+---
+--- 析构器必须是以下格式中的一种：
+--- 1. 一个对象，使用 `obj:remove()` 方法来析构，使用 `obj._removed` 属性来标记已被析构。
+--- 2. 一个析构函数，使用上值 `removed` 来标记已被析构。
+---
+--- ```lua
+--- local gc = ac.gc() -- 创建gc容器
+--- gc:add(obj1)       -- 将obj1放入gc容器
+--- gc:add(obj2)       -- 将obj2放入gc容器
+--- gc:remove()        -- 移除gc容器，同时也会移除obj1与obj2
+--- ```
+return function ()
+    return setmetatable({
+        _list = {},
+    }, mt)
+end

script/workspace/scope.lua

@@ -76,10 +76,14 @@ end
 ---@class scope.manager
 local m = {}
 
----@type scope[]
-m.folders  = {}
-m.override = createScope 'override'
-m.fallback = createScope 'fallback'
+function m.reset()
+    ---@type scope[]
+    m.folders  = {}
+    m.override = createScope 'override'
+    m.fallback = createScope 'fallback'
+end
+
+m.reset()
 
 ---@param uri uri
 ---@return scope

script/workspace/workspace.lua

@@ -16,8 +16,6 @@ local loading    = require 'workspace.loading'
 ---@class workspace
 local m = {}
 m.type = 'workspace'
----@type scope[]
-m.folders = {}
 m.watchList = {}
 
 --- 注册事件
@@ -53,6 +51,12 @@ function m.create(uri)
     m.folders[#m.folders+1] = scp
 end
 
+function m.reset()
+    ---@type scope[]
+    m.folders = {}
+    m.rootUri = nil
+end
+
 function m.getRootUri(uri)
     local scp = m.getScope(uri)
     return scp.uri

test.lua

@@ -99,7 +99,9 @@ local function main()
     require 'bee.platform'.OS = 'macOS'
     testAll()
 
-    test 'full'
+    test 'client'
+
+    --test 'full'
 
     print('测试完成')
 end

test/client/init.lua

@@ -0,0 +1 @@
+require 'client.tests.single-mode'

test/client/lclient.lua

@@ -0,0 +1,61 @@
+local gc   = require 'gc'
+local util = require 'utility'
+
+---@class languageClient
+---@field _inFile  file*
+---@field _outFile file*
+---@field _gc      gc
+local mt = {}
+mt.__index = mt
+
+function mt:__close()
+    self:remove()
+end
+
+function mt:_openIO()
+    local r = math.random(1, 10000)
+    self._inPath  = ('%s/%s.in'):format(LOGPATH, r)
+    self._outPath = ('%s/%s.out'):format(LOGPATH, r)
+    local stdin   = io.open(self._inPath, 'rb')
+    local stdout  = io.open(self._outPath, 'wb')
+    self._inFile  = io.open(self._inPath, 'wb')
+    self._outFile = io.open(self._outPath, 'rb')
+    io.stdin      = stdin
+    io.stdout     = stdout
+    self:gc(function ()
+        stdin:close()
+        stdout:close()
+        self._inFile:close()
+        self._outFile:close()
+    end)
+end
+
+function mt:_flushServer()
+    -- reset scopes
+    local ws    = require 'workspace'
+    local scope = require 'workspace.scope'
+    local files = require 'files'
+    ws.reset()
+    scope.reset()
+    files.reset()
+end
+
+function mt:start()
+    self:_openIO()
+    self:_flushServer()
+end
+
+function mt:gc(obj)
+    return self._gc:add(obj)
+end
+
+function mt:remove()
+    self._gc:remove()
+end
+
+return function ()
+    local self = setmetatable({
+        _gc = gc(),
+    }, mt)
+    return self
+end