mirror of
https://github.com/CCLeonOS/LeonOS.git
synced 2026-03-03 15:17:01 +00:00
45 lines
822 B
Lua
45 lines
822 B
Lua
|
|
-- wget
|
||
|
|
|
||
|
|
if not package.loaded.http then
|
||
|
|
error("HTTP is not enabled in the ComputerCraft configuration", 0)
|
||
|
|
end
|
||
|
|
|
||
|
|
local http = require("http")
|
||
|
|
|
||
|
|
local args = {...}
|
||
|
|
|
||
|
|
if #args == 0 then
|
||
|
|
io.stderr:write([[Usage:
|
||
|
|
wget <url> [filename]
|
||
|
|
wget run <url>
|
||
|
|
]])
|
||
|
|
return
|
||
|
|
end
|
||
|
|
|
||
|
|
local function get(url)
|
||
|
|
local handle, err = http.get(url, nil, true)
|
||
|
|
if not handle then
|
||
|
|
error(err, 0)
|
||
|
|
end
|
||
|
|
|
||
|
|
local data = handle.readAll()
|
||
|
|
handle.close()
|
||
|
|
|
||
|
|
return data
|
||
|
|
end
|
||
|
|
|
||
|
|
if args[1] == "run" then
|
||
|
|
local data = get(args[2])
|
||
|
|
assert(load(data, "=<wget-run>", "t", _G))()
|
||
|
|
else
|
||
|
|
local filename = args[2] or (args[1]:match("[^/]+$")) or
|
||
|
|
error("could not determine file name", 0)
|
||
|
|
local data = get(args[1])
|
||
|
|
local handle, err = io.open(filename, "w")
|
||
|
|
if not handle then
|
||
|
|
error(err, 0)
|
||
|
|
end
|
||
|
|
handle:write(data)
|
||
|
|
handle:close()
|
||
|
|
end
|