50 lines
1.3 KiB
Lua
50 lines
1.3 KiB
Lua
local vcomponent = require "vcomponent"
|
|
local component = require "component"
|
|
local diskpart = require "diskpart"
|
|
local event = require "event"
|
|
|
|
local listeners = {}
|
|
|
|
local function attachDrive(_,a,t)
|
|
if t ~= "drive" then return end
|
|
for k,v in ipairs(diskpart.getPartitions(a)) do
|
|
if v[4] > 0 then
|
|
vcomponent.register(string.format("%s/%i",a,k), "partition", diskpart.proxyPartition(a, k))
|
|
local w,l = pcall(require,string.format("fs.%s",v[2]))
|
|
if w and l and l.mount then
|
|
local w, f = pcall(l.mount, string.format("%s/%i",a,k))
|
|
if w then
|
|
vcomponent.register(string.format("%s/%i/%s",a,k,v[2]), f.type or "filesystem", f)
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
local function detachDrive(_,a,t)
|
|
if t ~= "drive" then return end
|
|
for ca,t in component.list() do
|
|
if (t == "partition" or t == "filesystem") and ca:sub(1,a:len()) == a then
|
|
vcomponent.unregister(ca)
|
|
end
|
|
end
|
|
end
|
|
|
|
function start()
|
|
listeners[#listeners+1] = event.listen("component_added", attachDrive)
|
|
listeners[#listeners+1] = event.listen("component_removed", detachDrive)
|
|
for k,v in component.list("drive",true) do
|
|
attachDrive(nil,k,v)
|
|
end
|
|
end
|
|
|
|
function stop()
|
|
for k,v in ipairs(listeners) do
|
|
event.cancel(v)
|
|
listeners[k] = nil
|
|
end
|
|
for k,v in component.list("drive",true) do
|
|
detachDrive(nil,k,"drive")
|
|
end
|
|
end
|