140 lines
4.0 KiB
Lua
140 lines
4.0 KiB
Lua
local arg = arg or {...}
|
|
|
|
local SRC_DIR = arg[1] or 'src'
|
|
local OUT_DIR = arg[2] or 'docs'
|
|
|
|
local BasaltDoc = require('tools/BasaltDoc')
|
|
|
|
local fileSystem
|
|
|
|
if fs then
|
|
fileSystem = {
|
|
list = fs.list,
|
|
combine = fs.combine,
|
|
isDir = fs.isDir,
|
|
exists = fs.exists,
|
|
makeDir = fs.makeDir,
|
|
open = function(path, mode) return fs.open(path, mode) end,
|
|
getDir = fs.getDir,
|
|
readAll = function(file) return file.readAll() end,
|
|
write = function(file, data) file.write(data) end,
|
|
close = function(file) file.close() end
|
|
}
|
|
else
|
|
local function executeCommand(cmd)
|
|
local handle = io.popen(cmd)
|
|
local result = handle:read("*a")
|
|
local success, _, code = handle:close()
|
|
return result, success, code
|
|
end
|
|
|
|
local function pathExists(path)
|
|
local result, success = executeCommand("test -e '" .. path .. "' && echo 'exists' || echo 'not_exists'")
|
|
return success and result:match("exists")
|
|
end
|
|
|
|
local function isDirectory(path)
|
|
local result, success = executeCommand("test -d '" .. path .. "' && echo 'dir' || echo 'not_dir'")
|
|
return success and result:match("dir")
|
|
end
|
|
|
|
local function makeDirectory(path)
|
|
local _, success = executeCommand("mkdir -p '" .. path .. "'")
|
|
return success
|
|
end
|
|
|
|
local function listDirectory(dir)
|
|
local result, success = executeCommand("ls -1 '" .. dir .. "' 2>/dev/null || true")
|
|
if not success then
|
|
return {}
|
|
end
|
|
|
|
local items = {}
|
|
for item in result:gmatch("[^\r\n]+") do
|
|
if item ~= "" then
|
|
table.insert(items, item)
|
|
end
|
|
end
|
|
return items
|
|
end
|
|
|
|
local function combinePath(a, b)
|
|
if a:sub(-1) == "/" then
|
|
return a .. b
|
|
else
|
|
return a .. "/" .. b
|
|
end
|
|
end
|
|
|
|
local function getDirectory(path)
|
|
return path:match("(.+)/[^/]*$") or ""
|
|
end
|
|
|
|
fileSystem = {
|
|
list = listDirectory,
|
|
combine = combinePath,
|
|
isDir = isDirectory,
|
|
exists = pathExists,
|
|
makeDir = makeDirectory,
|
|
open = io.open,
|
|
getDir = getDirectory,
|
|
readAll = function(file) return file:read("*all") end,
|
|
write = function(file, data) file:write(data) end,
|
|
close = function(file) file:close() end
|
|
}
|
|
end
|
|
|
|
if not fileSystem.exists(OUT_DIR) then
|
|
fileSystem.makeDir(OUT_DIR)
|
|
end
|
|
|
|
local function getLuaFiles(dir)
|
|
local files = {}
|
|
local list = fileSystem.list(dir)
|
|
for _, item in ipairs(list) do
|
|
local path = fileSystem.combine(dir, item)
|
|
if fileSystem.isDir(path) then
|
|
local subFiles = getLuaFiles(path)
|
|
for _, subFile in ipairs(subFiles) do
|
|
table.insert(files, subFile)
|
|
end
|
|
elseif item:match("%.lua$") then
|
|
table.insert(files, path)
|
|
end
|
|
end
|
|
return files
|
|
end
|
|
|
|
local luaFiles = getLuaFiles(SRC_DIR)
|
|
|
|
for _, filePath in ipairs(luaFiles) do
|
|
local file = fileSystem.open(filePath, "r")
|
|
if file then
|
|
local content = fileSystem.readAll(file)
|
|
fileSystem.close(file)
|
|
|
|
local ast = BasaltDoc.parse(content)
|
|
local markdown = BasaltDoc.generateMarkdown(ast)
|
|
|
|
local relativePath = filePath:gsub("^" .. SRC_DIR .. "/", ""):gsub("%.lua$", ".md")
|
|
local outPath = fileSystem.combine(OUT_DIR, relativePath)
|
|
|
|
local outDir = fileSystem.getDir(outPath)
|
|
if outDir and outDir ~= "" and not fileSystem.exists(outDir) then
|
|
fileSystem.makeDir(outDir)
|
|
end
|
|
|
|
local outFile = fileSystem.open(outPath, "w")
|
|
if outFile then
|
|
fileSystem.write(outFile, table.concat(markdown, "\n"))
|
|
fileSystem.close(outFile)
|
|
print("Generated: " .. outPath)
|
|
else
|
|
print("Error writing: " .. outPath)
|
|
end
|
|
else
|
|
print("Error reading: " .. filePath)
|
|
end
|
|
end
|
|
|
|
print("Documentation generation complete.") |