fs.lua

local expect = require "cc.expect"
local syscall = require "_syscall"

local fs = {}

function fs.list(path)
    expect(1, path, "string")
    return syscall.list(path)
end

function fs.exists(path)
    expect(1, path, "string")
    return syscall.stat(path) ~= nil
end

function fs.isDir(path)
    expect(1, path, "string")
    local stat = syscall.stat(path)
    return stat ~= nil and stat.type == "directory"
end

function fs.isReadOnly(path)
    expect(1, path, "string")
    local stat = syscall.stat(path)
    while stat == nil do
        path = fs.getDir(path)
        stat = syscall.stat(path)
    end
    local user = syscall.getuser()
    if stat.permissions[user] then return not stat.permissions[user].write
    else return not stat.worldPermissions.write end
end

function fs.getName(path)
    expect(1, path, "string")
    return syscall.combine(path):match "[^/]+$"
end

function fs.getDrive(path)
    expect(1, path, "string")
    local stat = syscall.stat(path)
    return stat and (stat.mountpoint == "/" and "hdd" or stat.mountpoint:gsub("^/", ""))
end

function fs.getSize(path)
    expect(1, path, "string")
    local stat = syscall.stat(path)
    return stat and stat.size
end

function fs.getFreeSpace(path)
    expect(1, path, "string")
    local stat = syscall.stat(path)
    while stat == nil do
        path = fs.getDir(path)
        stat = syscall.stat(path)
    end
    return stat and stat.freeSpace
end

function fs.makeDir(path)
    expect(1, path, "string")
    return syscall.mkdir(path)
end

function fs.move(from, to)
    expect(1, from, "string")
    expect(2, to, "string")
    local fromstat = assert(syscall.stat(from), "No such file or directory")
    local path = to
    local tostat = syscall.stat(path)
    if tostat then error("File exists", 2) end
    while tostat == nil do
        path = fs.getDir(path)
        tostat = syscall.stat(path)
    end
    if fromstat.mountpoint == tostat.mountpoint then
        return syscall.rename(from, to)
    end
    fs.copy(from, to)
    syscall.remove(from)
end

function fs.copy(from, to)
    expect(1, from, "string")
    expect(2, to, "string")
    local stat = syscall.stat(from)
    if stat.type == "directory" then
        local list = syscall.list(from)
        syscall.mkdir(to)
        for _, v in ipairs(list) do fs.copy(syscall.combine(from, v), syscall.combine(to, v)) end
    else
        local fromfile, err = syscall.open(from, "rb")
        if not fromfile then error(err, 2) end
        local tofile, err = syscall.open(to, "wb")
        if not tofile then fromfile.close() error(err, 2) end
        repeat
            local buf = fromfile.read(512)
            if buf then tofile.write(buf) end
        until not buf
        tofile.close()
        fromfile.close()
    end
end

function fs.delete(path)
    expect(1, path, "string")
    return syscall.remove(path)
end

function fs.combine(...)
    return syscall.combine(...)
end

function fs.open(path, mode)
    expect(1, path, "string")
    expect(2, mode, "string")
    return syscall.open(path, mode)
end

local function aux_find(options, pathc, i)
    if i > #pathc then return {} end
    local pathc_regex = pathc[i]:gsub("[%^%$%(%)%%%.%+%-]", "%%%1"):gsub("%*", ".*"):gsub("%?", "."):gsub("%[!", "[^")
    local nextOptions = {}
    for _, opt in ipairs(options) do
        local ok, possible_paths = pcall(syscall.list, opt)
        if ok then
            for _, path in ipairs(possible_paths) do
                if path:match(pathc_regex) then
                    nextOptions[#nextOptions+1] = syscall.combine(opt, path)
                end
            end
        end
    end
    if i + 1 > #pathc then return nextOptions end
    return aux_find(nextOptions, pathc, i + 1)
end

function fs.find(wildcard)
    expect(1, wildcard, "string")
    local parts = {}
    for p in wildcard:gmatch("[^/]+") do parts[#parts+1] = p end
    local retval = aux_find({wildcard:sub(1, 1) == "/" and "/" or "."}, parts, 1)
    table.sort(retval)
    return retval
end

function fs.getDir(path)
    expect(1, path, "string")
    local p = syscall.combine(path):match "^(.*)/[^/]*$" or ""
    if p == "" then
        if path:sub(1, 1) == "/" then return "/"
        else return "." end
    else return p end
end

function fs.complete(partial, path, files, slashes)
    -- TODO
end

function fs.attributes(path)
    expect(1, path, "string")
    local stat = syscall.stat(path)
    if not stat then return nil end
    stat.isDir = stat.type == "directory"
    local user = syscall.getuser()
    if stat.permissions[user] then stat.isReadOnly = not stat.permissions[user].write
    else stat.isReadOnly = not stat.worldPermissions.write end
end

function fs.getCapacity(path)
    expect(1, path, "string")
    local stat = syscall.stat(path)
    while stat == nil do
        path = fs.getDir(path)
        stat = syscall.stat(path)
    end
    return stat and stat.capacity
end

function fs.isDriveRoot(path)
    expect(1, path, "string")
    local stat = syscall.stat(path)
    return stat ~= nil and syscall.combine(path) == stat.mountpoint
end

return fs