Compare commits
3 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
cf0fdfa4d0 | ||
|
|
9a68952fac | ||
|
|
b661afed4c |
162
Client/main.lua
162
Client/main.lua
@@ -4,6 +4,15 @@ local httpServer = args[1] or "http://192.168.2.200:8080"
|
||||
local roomId = args[2]
|
||||
local pollInterval = 1
|
||||
local computerID = tostring(os.computerID() or "unknown")
|
||||
-- ========== 加载 JSON ==========
|
||||
local JsonUrl = "https://git.liulikeji.cn/GitHub/json.lua/raw/branch/master/json.lua"
|
||||
local JsonResp = http.get(JsonUrl)
|
||||
if not JsonResp then
|
||||
error("无法下载 Json 框架")
|
||||
end
|
||||
local json = load(JsonResp.readAll())()
|
||||
JsonResp.close()
|
||||
|
||||
|
||||
-- ========== 加载 Basalt ==========
|
||||
local basaltUrl = "https://git.liulikeji.cn/GitHub/Basalt/releases/download/v1.7/basalt.lua"
|
||||
@@ -21,97 +30,65 @@ local function log(msg)
|
||||
--basalt.debug("[FileClient] " .. tostring(msg))
|
||||
end
|
||||
|
||||
function table_to_json(t, indent)
|
||||
indent = indent or 0
|
||||
local spaces = string.rep(" ", indent)
|
||||
local result = {}
|
||||
|
||||
if type(t) ~= "table" then
|
||||
if type(t) == "string" then
|
||||
-- 正确转义所有特殊字符
|
||||
local escaped = t:gsub("[\\\"\b\f\n\r\t]", function(c)
|
||||
local replacements = {
|
||||
['\\'] = '\\\\',
|
||||
['"'] = '\\"',
|
||||
['\b'] = '\\b',
|
||||
['\f'] = '\\f',
|
||||
['\n'] = '\\n',
|
||||
['\r'] = '\\r',
|
||||
['\t'] = '\\t'
|
||||
}
|
||||
return replacements[c]
|
||||
end)
|
||||
return '"' .. escaped .. '"'
|
||||
elseif type(t) == "number" or type(t) == "boolean" then
|
||||
return tostring(t)
|
||||
else
|
||||
return '"' .. tostring(t) .. '"'
|
||||
end
|
||||
|
||||
local function isBinaryFile(path)
|
||||
local extension = string.lower(string.match(path, "%.([^%.%s]+)$") or "")
|
||||
local binaryExtensions = {
|
||||
["dfpwm"] = true,
|
||||
}
|
||||
|
||||
if binaryExtensions[extension] then
|
||||
return true
|
||||
end
|
||||
|
||||
-- 检查是否是数组
|
||||
local is_array = true
|
||||
local max_index = 0
|
||||
local count = 0
|
||||
for k, v in pairs(t) do
|
||||
count = count + 1
|
||||
if type(k) ~= "number" or k <= 0 or math.floor(k) ~= k then
|
||||
is_array = false
|
||||
end
|
||||
if type(k) == "number" and k > max_index then
|
||||
max_index = k
|
||||
end
|
||||
|
||||
-- 对于没有扩展名的文件,检查内容
|
||||
local absPath = path
|
||||
if not fs.exists(absPath) then
|
||||
return false
|
||||
end
|
||||
|
||||
-- 空表当作对象处理
|
||||
if count == 0 then
|
||||
is_array = false
|
||||
|
||||
local ok, handle = pcall(fs.open, absPath, "rb")
|
||||
if not ok or not handle then
|
||||
return false
|
||||
end
|
||||
|
||||
if is_array then
|
||||
-- 处理数组
|
||||
table.insert(result, "[")
|
||||
local items = {}
|
||||
for i = 1, max_index do
|
||||
if t[i] ~= nil then
|
||||
table.insert(items, table_to_json(t[i], indent + 2))
|
||||
else
|
||||
table.insert(items, "null")
|
||||
|
||||
local data = handle.read(math.min(1024, fs.getSize(absPath)))
|
||||
handle.close()
|
||||
|
||||
if not data then
|
||||
return false
|
||||
end
|
||||
|
||||
-- 检查是否存在控制字符(除常见的空白字符外)
|
||||
for i = 1, #data do
|
||||
local b = data:byte(i)
|
||||
-- 控制字符范围是 0-8, 11-12, 14-31, 127
|
||||
if (b >= 0 and b <= 8) or (b == 11) or (b == 12) or (b >= 14 and b <= 31) or (b == 127) then
|
||||
-- 如果控制字符过多(超过5%),则认为是二进制文件
|
||||
local controlCount = 0
|
||||
for j = 1, #data do
|
||||
local byte = data:byte(j)
|
||||
if (byte >= 0 and byte <= 8) or (byte == 11) or (byte == 12) or (byte >= 14 and byte <= 31) or (byte == 127) then
|
||||
controlCount = controlCount + 1
|
||||
end
|
||||
end
|
||||
|
||||
if controlCount / #data > 0.05 then
|
||||
return true
|
||||
end
|
||||
end
|
||||
table.insert(result, table.concat(items, ", "))
|
||||
table.insert(result, "]")
|
||||
else
|
||||
-- 处理对象
|
||||
table.insert(result, "{")
|
||||
local items = {}
|
||||
for k, v in pairs(t) do
|
||||
local key = '"' .. tostring(k) .. '"'
|
||||
local value = table_to_json(v, indent + 2)
|
||||
if indent > 0 then
|
||||
table.insert(items, spaces .. " " .. key .. ": " .. value)
|
||||
else
|
||||
table.insert(items, key .. ":" .. value)
|
||||
end
|
||||
end
|
||||
if indent > 0 then
|
||||
table.insert(result, table.concat(items, ",\n"))
|
||||
table.insert(result, "\n" .. spaces .. "}")
|
||||
else
|
||||
table.insert(result, table.concat(items, ","))
|
||||
table.insert(result, "}")
|
||||
end
|
||||
end
|
||||
|
||||
return table.concat(result, indent > 0 and "\n" .. spaces or "")
|
||||
|
||||
return false
|
||||
end
|
||||
|
||||
|
||||
local function cleanPath(path)
|
||||
return (path:gsub("^computer/", ""):gsub("^computer\\", ""))
|
||||
end
|
||||
|
||||
local function httpPost(path, data)
|
||||
local jsonData = table_to_json(data)
|
||||
local jsonData = json.encode(data)
|
||||
local url = httpServer .. path
|
||||
|
||||
-- 使用长轮询
|
||||
@@ -132,7 +109,7 @@ local function httpPost(path, data)
|
||||
local responseBody = response.readAll()
|
||||
response.close()
|
||||
|
||||
local ok, result = pcall(textutils.unserialiseJSON, responseBody)
|
||||
local ok, result = pcall(json.decode, responseBody)
|
||||
if ok then
|
||||
return result
|
||||
else
|
||||
@@ -141,16 +118,6 @@ local function httpPost(path, data)
|
||||
end
|
||||
|
||||
-- ========== 文件系统操作 ==========
|
||||
local function isLikelyText(data)
|
||||
for i = 1, math.min(#data, 1024) do
|
||||
local b = data:byte(i)
|
||||
if b < 32 and b ~= 9 and b ~= 10 and b ~= 13 then
|
||||
return false
|
||||
end
|
||||
end
|
||||
return true
|
||||
end
|
||||
|
||||
local function getFiles(currentPath, result, prefix)
|
||||
local computerPrefix = "computer_" .. computerID
|
||||
local fullPrefix = currentPath == "" and prefix:sub(1, -2) or prefix .. currentPath
|
||||
@@ -165,15 +132,16 @@ local function getFiles(currentPath, result, prefix)
|
||||
end
|
||||
else
|
||||
local content = "[binary]"
|
||||
local ok, handle = pcall(fs.open, absPath, "rb")
|
||||
if ok and handle then
|
||||
local data = handle.readAll()
|
||||
handle.close()
|
||||
if data and isLikelyText(data) then
|
||||
content = data
|
||||
if not isBinaryFile(absPath) then
|
||||
local ok, handle = pcall(fs.open, absPath, "r")
|
||||
if ok and handle then
|
||||
local data = handle.readAll()
|
||||
handle.close()
|
||||
content = data or ""
|
||||
end
|
||||
end
|
||||
result[fullPrefix] = { isFile = true, content = content }
|
||||
|
||||
result[fullPrefix] = { isFile = true, content = content, isBinary = isBinaryFile(absPath) }
|
||||
end
|
||||
end
|
||||
|
||||
@@ -399,4 +367,4 @@ local function main()
|
||||
end
|
||||
|
||||
-- 启动主逻辑和Basalt事件循环
|
||||
parallel.waitForAll(basalt.autoUpdate, main)
|
||||
parallel.waitForAll(basalt.autoUpdate, main)
|
||||
Reference in New Issue
Block a user