添加 'tpm.lua'
This commit is contained in:
parent
9dad2ce722
commit
02c357146f
335
tpm.lua
Normal file
335
tpm.lua
Normal file
|
|
@ -0,0 +1,335 @@
|
||||||
|
print("Tiny Package Manager (TPM)")
|
||||||
|
print("Author: Kiritow")
|
||||||
|
|
||||||
|
local component=require("component")
|
||||||
|
local shell=require("shell")
|
||||||
|
local serialization=require("serialization")
|
||||||
|
local uuid=require("uuid")
|
||||||
|
local filesystem=require("filesystem")
|
||||||
|
|
||||||
|
local function showcmd(cmdstr,infostr)
|
||||||
|
local old=component.gpu.setForeground(0xFFFF00)
|
||||||
|
io.write(cmdstr)
|
||||||
|
component.gpu.setForeground(0xFFFFFF)
|
||||||
|
print(" " .. infostr)
|
||||||
|
component.gpu.setForeground(old)
|
||||||
|
end
|
||||||
|
local function showerr(infostr)
|
||||||
|
local old=component.gpu.setForeground(0xFF0000)
|
||||||
|
print(infostr)
|
||||||
|
component.gpu.setForeground(old)
|
||||||
|
end
|
||||||
|
local function cwrite(color,infostr)
|
||||||
|
local old=component.gpu.setForeground(color)
|
||||||
|
io.write(infostr)
|
||||||
|
component.gpu.setForeground(old)
|
||||||
|
end
|
||||||
|
|
||||||
|
local function resolve_github(reponame,fileaddr,branch)
|
||||||
|
if(branch==nil) then branch="master" end
|
||||||
|
return "https://raw.githubusercontent.com/" .. reponame .. "/" .. branch .. "/" .. fileaddr
|
||||||
|
end
|
||||||
|
local function resolve_cngit(reponame,fileaddr,branch)
|
||||||
|
return "http://kiritow.com:3000/" .. reponame .. "/raw/" .. branch .. "/" .. fileaddr
|
||||||
|
end
|
||||||
|
|
||||||
|
-- For Test Purpose only
|
||||||
|
resolve_github=resolve_cngit
|
||||||
|
|
||||||
|
local function beginWith(str,target)
|
||||||
|
local a,b=string.find(str,target)
|
||||||
|
if(a==nil) then return false
|
||||||
|
else return (a==1) end
|
||||||
|
end
|
||||||
|
|
||||||
|
local args,ops=shell.parse(...)
|
||||||
|
local argc=#args
|
||||||
|
|
||||||
|
if(argc<1) then
|
||||||
|
print("Usage:")
|
||||||
|
showcmd("tpm install <package>",
|
||||||
|
"Install package" ..
|
||||||
|
"\n\tInstall via github: github+Author/Repo[/branch]. Branch is 'master' by default" ..
|
||||||
|
"\n\tInstall via http: http+URL"
|
||||||
|
)
|
||||||
|
showcmd("tpm remove <package>","Remove package")
|
||||||
|
showcmd("tpm list [-in] [<package>]",
|
||||||
|
"Search package" ..
|
||||||
|
"\n\t-i Only search installed package" ..
|
||||||
|
"\n\t-n Only search not installed package"
|
||||||
|
)
|
||||||
|
showcmd("tpm update","Update software info")
|
||||||
|
return
|
||||||
|
end
|
||||||
|
|
||||||
|
-- Hardware check
|
||||||
|
if(component.internet==nil) then
|
||||||
|
showerr("No internet device found. tpm requires internet card to function normally.")
|
||||||
|
return
|
||||||
|
end
|
||||||
|
|
||||||
|
-- Functions
|
||||||
|
local function doRealDownload(url)
|
||||||
|
if(component.internet==nil) then
|
||||||
|
error("The downloader requires an Internet card.")
|
||||||
|
end
|
||||||
|
local handle=component.internet.request(url)
|
||||||
|
while true do
|
||||||
|
local ret,err=handle.finishConnect()
|
||||||
|
if(ret==nil) then
|
||||||
|
return false,err
|
||||||
|
elseif(ret==true) then
|
||||||
|
break
|
||||||
|
end
|
||||||
|
--elseif(ret==false)
|
||||||
|
--os.sleep(0.1)
|
||||||
|
end
|
||||||
|
local response_code=handle.response()
|
||||||
|
local ans=""
|
||||||
|
while true do
|
||||||
|
local tmp=handle.read()
|
||||||
|
if(tmp==nil) then break end
|
||||||
|
ans=ans .. tmp
|
||||||
|
end
|
||||||
|
handle.close()
|
||||||
|
return true,ans,response_code
|
||||||
|
end
|
||||||
|
function WriteStringToFile(StringValue,FileName,IsAppend)
|
||||||
|
if(IsAppend==nil) then IsAppend=false end
|
||||||
|
local handle,err
|
||||||
|
if(IsAppend) then
|
||||||
|
handle,err=io.open(FileName,"a")
|
||||||
|
else
|
||||||
|
handle,err=io.open(FileName,"w")
|
||||||
|
end
|
||||||
|
if(handle==nil) then return false,err end
|
||||||
|
|
||||||
|
handle:write(StringValue)
|
||||||
|
handle:close()
|
||||||
|
|
||||||
|
return true
|
||||||
|
end
|
||||||
|
|
||||||
|
-- TPM Main Programs
|
||||||
|
-- Installer
|
||||||
|
local function tpm_set_installed_list(tb)
|
||||||
|
local f=io.open("/etc/tpm/list.bin","w")
|
||||||
|
if(f==nil) then return false,"Unable to open file" end
|
||||||
|
local content=serialization.serialize(tb)
|
||||||
|
f:write(content)
|
||||||
|
f:close()
|
||||||
|
return true
|
||||||
|
end
|
||||||
|
|
||||||
|
local function tpm_get_installed_list()
|
||||||
|
local f=io.open("/etc/tpm/list.bin","r")
|
||||||
|
if(f==nil) then
|
||||||
|
f=io.open("/etc/tpm/list.bin","w")
|
||||||
|
if(f==nil) then
|
||||||
|
return false,"Unable to open file"
|
||||||
|
else
|
||||||
|
f:close()
|
||||||
|
tpm_set_installed_list({})
|
||||||
|
return true,{}
|
||||||
|
end
|
||||||
|
end
|
||||||
|
local content=f:read("a")
|
||||||
|
f:close()
|
||||||
|
return true,serialization.unserialize(content)
|
||||||
|
end
|
||||||
|
|
||||||
|
local function tpm_is_installed(libname)
|
||||||
|
local flg,tb=tpm_get_installed_list()
|
||||||
|
if(not flg) then
|
||||||
|
print("[tpm error] " .. tb)
|
||||||
|
return false
|
||||||
|
else
|
||||||
|
for k,v in pairs(tb) do
|
||||||
|
if(v.libname==libname) then
|
||||||
|
return true
|
||||||
|
end
|
||||||
|
end
|
||||||
|
return false
|
||||||
|
end
|
||||||
|
end
|
||||||
|
local function tpm_mainfest_loader(data)
|
||||||
|
local f,err=load("return " .. data,"Mainfest","t",{})
|
||||||
|
if(f==nil) then
|
||||||
|
return false,err
|
||||||
|
end
|
||||||
|
local tb=f()
|
||||||
|
-- Check returned table
|
||||||
|
if(tb.libname==nil or tb.name==nil or tb.files==nil) then
|
||||||
|
return false
|
||||||
|
else
|
||||||
|
return true,tb
|
||||||
|
end
|
||||||
|
end
|
||||||
|
local function tpm_mainfest_show(mtb)
|
||||||
|
print("Library Name: " .. mtb.name)
|
||||||
|
print("libname: " .. mtb.libname)
|
||||||
|
print("Info: " .. (mtb.info or "Not provided"))
|
||||||
|
print("Author: " .. (mtb.author or "Not provided"))
|
||||||
|
print("Version: " .. (mtb.version or "Not provided"))
|
||||||
|
print("Setup: " .. (mtb.setup or "Not provided"))
|
||||||
|
print("Uninst: " .. (mtb.uninst or "Not provided"))
|
||||||
|
print("Required Directories: ")
|
||||||
|
if(mtb.dirs==nil) then
|
||||||
|
print("None")
|
||||||
|
else
|
||||||
|
for k,v in pairs(mtb.dirs) do
|
||||||
|
print(v)
|
||||||
|
end
|
||||||
|
end
|
||||||
|
print("Files:")
|
||||||
|
for k,v in pairs(mtb.files) do
|
||||||
|
print(k," -> ",v)
|
||||||
|
end
|
||||||
|
print("Dependency: ")
|
||||||
|
if(mtb.depends==nil) then
|
||||||
|
print("None")
|
||||||
|
else
|
||||||
|
for k,v in pairs(mtb.depends) do
|
||||||
|
print(v)
|
||||||
|
end
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
local function tpm_install_github_real(repo,branch)
|
||||||
|
local config_url=resolve_github(repo,"mainfest.txt",branch)
|
||||||
|
local ret,data,code=doRealDownload(config_url)
|
||||||
|
if(not ret) then
|
||||||
|
return false,"Failed to download mainfest from repo " .. repo .. ". Error: " .. data
|
||||||
|
end
|
||||||
|
if(code~=200) then
|
||||||
|
return false,"Got response code " .. code .. " while downloading mainfest from repo " .. repo
|
||||||
|
end
|
||||||
|
local flg,tb=tpm_mainfest_loader(data)
|
||||||
|
if(not flg) then
|
||||||
|
return false,"Invalid mainfest in repo " .. repo .. " Error: " .. tb
|
||||||
|
end
|
||||||
|
|
||||||
|
-- Debug
|
||||||
|
tpm_mainfest_show(tb)
|
||||||
|
|
||||||
|
-- Check
|
||||||
|
if(tpm_is_installed(tb.libname)) then
|
||||||
|
print("Package " .. tb.libname .. " is already installed.")
|
||||||
|
return true
|
||||||
|
end
|
||||||
|
|
||||||
|
-- Depends
|
||||||
|
if(tb.depends~=nil) then
|
||||||
|
print("Analyzing dependency...")
|
||||||
|
local dependsz=#tb.depends
|
||||||
|
local cnt_now=1
|
||||||
|
for k,v in pairs(tb.depends) do
|
||||||
|
print("[" .. cnt_now .. "/" .. dependsz .. "] Checking " .. v .. "...")
|
||||||
|
local ret,msg=tpm_install(v)
|
||||||
|
if(not ret) then
|
||||||
|
return false,"An error occurs while analyzing dependency: " .. v .. ". Error: " .. msg
|
||||||
|
end
|
||||||
|
cnt_now=cnt_now+1
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
-- Make directories
|
||||||
|
if(tb.dirs~=nil) then
|
||||||
|
for k,v in pairs(tb.dirs) do
|
||||||
|
print("Making directory " .. v)
|
||||||
|
filesystem.makeDirectory(v)
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
-- Download
|
||||||
|
for k,v in pairs(tb.files) do
|
||||||
|
io.write("Downloading file " .. k .. "...")
|
||||||
|
local xret,xdata,xcode=doRealDownload(resolve_github(repo,k,branch))
|
||||||
|
if(not xret) then
|
||||||
|
print("[Failed]")
|
||||||
|
return false,"Failed to download " .. k
|
||||||
|
elseif(xcode~=200) then
|
||||||
|
print("[Failed]")
|
||||||
|
return false,"Response code is " .. xcode .. " while downloading " .. k
|
||||||
|
else
|
||||||
|
local xret,xmsg=WriteStringToFile(xdata,v)
|
||||||
|
if(not xret) then
|
||||||
|
print("[Write Failed]")
|
||||||
|
return false,"Failed to write to file " .. v .. " while downloading " .. k
|
||||||
|
end
|
||||||
|
|
||||||
|
-- OK here
|
||||||
|
print("[OK]")
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
-- Run Setup
|
||||||
|
if(tb.setup~=nil) then
|
||||||
|
print("Running setup...")
|
||||||
|
local setup_file=tb.files[tb.setup]
|
||||||
|
local setup_fn,err=loadfile(setup_file)
|
||||||
|
if(setup_fn==nil) then
|
||||||
|
print("Failed to start setup program. Error: " .. err)
|
||||||
|
end
|
||||||
|
local ret,msg=pcall(setup_fn)
|
||||||
|
if(not ret) then
|
||||||
|
printf("Failed to run setup program. Error: " .. msg)
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
-- Update software list
|
||||||
|
print("Updating software info...")
|
||||||
|
local flg,installed_lst=tpm_get_installed_list()
|
||||||
|
table.insert(installed_lst,tb)
|
||||||
|
tpm_set_installed_list(installed_lst)
|
||||||
|
print("Software info updated.")
|
||||||
|
return true
|
||||||
|
end
|
||||||
|
local function tpm_install_github(url)
|
||||||
|
local a=string.find(url,"/")
|
||||||
|
if(a==nil) then
|
||||||
|
return false,"Unknown format of github url"
|
||||||
|
end
|
||||||
|
|
||||||
|
local author=string.sub(url,1,a-1)
|
||||||
|
local reponame=''
|
||||||
|
local b=string.find(url,"/",a+1)
|
||||||
|
|
||||||
|
-- Branch set to master by default
|
||||||
|
local branch="master"
|
||||||
|
if(b~=nil) then
|
||||||
|
reponame=string.sub(url,a+1,b-1)
|
||||||
|
branch=string.sub(url,b+1)
|
||||||
|
else
|
||||||
|
reponame=string.sub(url,a+1)
|
||||||
|
end
|
||||||
|
|
||||||
|
local repo_fullname=author .. "/" .. reponame
|
||||||
|
print("Checking github repo " .. repo_fullname .. " at branch " .. branch .." ...")
|
||||||
|
|
||||||
|
return tpm_install_github_real(repo_fullname,branch)
|
||||||
|
end
|
||||||
|
local function tpm_install(install_url)
|
||||||
|
print("tpm_install: ",install_url)
|
||||||
|
if(beginWith(install_url,"github+")) then
|
||||||
|
return tpm_install_github(string.sub(install_url,string.len("github+")+1))
|
||||||
|
elseif(beginWith(install_url,"http+")) then
|
||||||
|
return tpm_install_http(string.sub(install_url,string.len("http+")+1))
|
||||||
|
else
|
||||||
|
return tpm_install_package(install_url)
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
if(args[1]=="install") then
|
||||||
|
if(argc<2) then
|
||||||
|
showerr("Package name or url is required.")
|
||||||
|
return
|
||||||
|
end
|
||||||
|
for i=2,argc,1 do
|
||||||
|
local ret,msg=tpm_install(args[i])
|
||||||
|
if(not ret) then
|
||||||
|
print("Failed to install task " .. (i-1) .. ". Error: " .. msg)
|
||||||
|
break
|
||||||
|
end
|
||||||
|
end
|
||||||
|
end
|
||||||
Reference in New Issue
Block a user