134 lines
4.5 KiB
Lua
134 lines
4.5 KiB
Lua
-- handles file and directory processing
|
|
local fs = require("@lune/fs")
|
|
local stdio = require("@lune/stdio")
|
|
local process = require("@lune/process")
|
|
local core = require("./core")
|
|
local types = require("./types")
|
|
|
|
export type FileProc = {
|
|
collectFiles: (opts: types.opts) -> {string},
|
|
processFiles: (filesToProcess: {string}, opts: types.opts) -> {string},
|
|
}
|
|
|
|
local fileproc = {} :: FileProc
|
|
|
|
--[[
|
|
Collects files to process based on command line arguments or directory mode.
|
|
|
|
### Example usage
|
|
|
|
```lua
|
|
local cli = require("./cli")
|
|
local fileproc = require("./fileproc")
|
|
|
|
local opts = cli.parseArgs()
|
|
local filesToProcess = fileproc.collectFiles(opts)
|
|
print("Files to process:", table.concat(filesToProcess, ", "))
|
|
```
|
|
|
|
@param opts Options table of type `types.opts`.
|
|
@return A list of file paths to process.
|
|
]]--
|
|
function fileproc.collectFiles(opts: types.opts): {string}
|
|
assert(typeof(opts) == "table", "Expected opts to be of type 'table'")
|
|
local filesWithWorkspace = {}
|
|
|
|
local function processDir(dirPath)
|
|
for _, file in pairs(fs.readDir(dirPath)) do
|
|
local fullPath = dirPath .. "/" .. file
|
|
if fs.isDir(fullPath) then
|
|
processDir(fullPath)
|
|
elseif fs.isFile(fullPath) then
|
|
if opts.forceBinaryRead or core.isValidModelFile(file) then
|
|
table.insert(filesWithWorkspace, fullPath)
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
if opts.directoryMode then
|
|
if not opts.directoryPath or not fs.isDir(opts.directoryPath) then
|
|
stdio.write(stdio.color("red"))
|
|
stdio.write(`Error: Directory {opts.directoryPath} does not exist.\n`)
|
|
stdio.write(stdio.color("reset"))
|
|
process.exit(1)
|
|
end
|
|
|
|
processDir(opts.directoryPath)
|
|
|
|
if #filesWithWorkspace == 0 then
|
|
stdio.write(stdio.color("yellow"))
|
|
stdio.write(`Warning: No files found in directory {opts.directoryPath}.\n`)
|
|
stdio.write(stdio.color("reset"))
|
|
process.exit(0)
|
|
end
|
|
end
|
|
|
|
if #filesWithWorkspace == 0 then
|
|
stdio.write(stdio.color("red"))
|
|
stdio.write("Error: No files to process.\n")
|
|
stdio.write(stdio.color("reset"))
|
|
process.exit(1)
|
|
end
|
|
|
|
return filesWithWorkspace
|
|
end
|
|
|
|
|
|
--[[
|
|
Processes each file and checks for workspace instances.
|
|
|
|
### Example usage
|
|
|
|
```lua
|
|
local cli = require("./cli")
|
|
local fileproc = require("./fileproc")
|
|
|
|
local opts = cli.parseArgs()
|
|
local filesToProcess = fileproc.collectFiles(opts)
|
|
local filesWithWorkspace = fileproc.processFiles(filesToProcess, opts)
|
|
print("Files containing Workspace instances:", table.concat(filesWithWorkspace, ", "))
|
|
```
|
|
|
|
@param filesToProcess A list of file paths to process.
|
|
@param opts Options table of type `types.opts`.
|
|
@return A list of file paths that contain Workspace instances.
|
|
]]--
|
|
function fileproc.processFiles(filesToProcess: {string}, opts: types.opts): {string}
|
|
assert(typeof(filesToProcess) == "table", "Expected filesToProcess to be of type 'table'")
|
|
assert(typeof(opts) == "table", "Expected opts to be of type 'table'")
|
|
local filesWithWorkspace = {}
|
|
for _, filePath in pairs(filesToProcess) do
|
|
if fs.isFile(filePath) then
|
|
if opts.forceBinaryRead or opts.directoryMode or core.isValidModelFile(filePath) then
|
|
local success, result = pcall(function()
|
|
local fileContents = fs.readFile(filePath)
|
|
return core.fileContainsWorkspace(fileContents, opts)
|
|
end)
|
|
if success then
|
|
if result then
|
|
table.insert(filesWithWorkspace, filePath)
|
|
end
|
|
print(core.formatResult(result, filePath))
|
|
else
|
|
stdio.write(stdio.color("red"))
|
|
stdio.write(`Error processing {filePath}: {result}\n`)
|
|
stdio.write(stdio.color("reset"))
|
|
end
|
|
else
|
|
stdio.write(stdio.color("yellow"))
|
|
stdio.write(`Warning: {filePath} is not a .rbxm or .rbxmx file, skipping.\n`)
|
|
stdio.write(stdio.color("reset"))
|
|
end
|
|
else
|
|
stdio.write(stdio.color("red"))
|
|
stdio.write(`Error: File {filePath} does not exist.\n`)
|
|
stdio.write(stdio.color("reset"))
|
|
end
|
|
end
|
|
return filesWithWorkspace
|
|
end
|
|
|
|
|
|
return fileproc
|