MakePack

--[[

MakePack by MarcusD
http://www.computercraft.info/forums2/index.php?/topic/23513-

--]]

local argz = {...}

--[HARDCODED VARIABLES]--

local isapi = shell == nil

--lineComp
local lineCompOffs = 0
local lineCompNum = 9

--lineExt
local lineExtOffs = 0
local lineExtOffsEnd = -0

--[HARDCODED VARIABLES]--

if #argz == 0 or argz[1] == "help" and not isapi then
  do
    local str =
    [[
MakePack by MarcusD

Usage:

makepack help -- Displays this message
makepack pack folder to_file -- Packs folder into to_file
makepack unpack file to_folder -- Unpacks file to to_folder
makepack sfx folder to_file -- Creates a SelF-eXtracting archive

Note:

Please don't use this on recursive symlinks, or else... >.<
    ]]
    print(str)
    return nil
  end
end

local function base_convert(n, b)
  if type(n) == "string" then return tonumber(n, b) end
  n = math.floor(n)
  if not b or b == 10 then return tostring(n) end
  local digits = "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ"
  local t = {}
  if n < 0 then
    table.insert(t, "-")
    n = -n
  end
  repeat
    local d = (n % b) + 1
    n = floor(n / b)
    insert(t, 1, digits:sub(d,d))
  until n == 0
  return table.concat(t,"")
end

local function iter_w(whand, from, comp)
  local function wint(x)
    local b4=x%256 x=(x-x%256)/256
    local b3=x%256 x=(x-x%256)/256
    local b2=x%256 x=(x-x%256)/256
    local b1=x%256 x=(x-x%256)/256
    whand.write(b1)
    whand.write(b2)
    whand.write(b3)
    whand.write(b4)
  end

  local function wstr(str)
    whand.write(string.len(str))
    for i in str:gmatch(".") do
      whand.write(string.byte(i))
    end
  end

  local buf = {}
  local strbuf = ""

  if fs.exists(from) and not fs.isDir(from) then
    wstr(fs.getName(from)) -- name
    whand.write(1) -- isfile
    local rhand = fs.open(from, "rb")
    local nao = rhand.read()
    while nao do --because there's no rhand.size and fs.size is inaccurate
      table.insert(buf, string.char(nao))
      nao = rhand.read()
    end
    rhand.close()
    os.queueEvent("keepalive")
    os.pullEvent("keepalive")
    strbuf = comp.encode(table.concat(buf, ""))
    os.queueEvent("keepalive")
    os.pullEvent("keepalive")
    wint(string.len(strbuf)) -- file size
    for asd in strbuf:gmatch(".") do
      whand.write(string.byte(asd))
    end
    os.queueEvent("keepalive")
    os.pullEvent("keepalive")
    buf = {}
    strbuf = ""
    whand.flush()
    return nil
  end

  local function iter(pth, short)
    os.queueEvent("keepalive")
    os.pullEvent("keepalive")
    if not short then short = "/" end
    local tbl = fs.list(pth)
    local cnt = #tbl
    wstr(short) -- name
    whand.write(0) -- isfile
    wint(cnt) -- #entries / if not isfile
    for _, v in pairs(tbl) do
      local pat = fs.combine(pth, v)
      if not isapi then print("Processing " .. pat) end
      if fs.isDir(pat) then
        iter(pat, v)
      else
        wstr(v) -- name
        whand.write(1) -- isfile
        local rhand = fs.open(pat, "rb")
        local nao = rhand.read()
        while nao do --because there's no rhand.size and fs.size is inaccurate
          table.insert(buf, string.char(nao))
          nao = rhand.read()
        end
        rhand.close()
        os.queueEvent("keepalive")
        os.pullEvent("keepalive")
        strbuf = comp.encode(table.concat(buf, ""))
        os.queueEvent("keepalive")
        os.pullEvent("keepalive")
        wint(string.len(strbuf)) -- file size
        for asd in strbuf:gmatch(".") do
          whand.write(string.byte(asd))
        end
        os.queueEvent("keepalive")
        os.pullEvent("keepalive")
        buf = {}
        strbuf = ""
        whand.flush()
      end
    end
  end

  return iter(from)
end

local function errnum(str)
  return tonumber(string.match(str, ":([0-9%.]+):"))
end

local function lineComp() error("thing") end
local comp =[[
--Original location: http://www.wowace.com/addons/libcompress/
--Heavily modified and minified
local type=type;local select=select;local next=next;local loadstring=loadstring;local setmetatable=setmetatable;local assert=assert;local a=table.insert;local b=table.remove;local c=table.concat;local d=table.sort;local e=string.char;local f=string.byte;local g=string.len;local h=string.sub;local unpack=unpack;local pairs=pairs;local i=math.modf;local j=bit.band;local k=bit.bor;local l=bit.bxor;local m=bit.bnot;local n=bit.blshift;local o=bit.brshift;local p={}local function q(r)for s=1,#p do p[s]=nil end;local t;r,t=i(r/255)t=t*255;p[#p+1]=t;while r>0 do r,t=i(r/255)t=t*255;p[#p+1]=t end;if#p==1 and p[1]>0 and p[1]<250 then return e(p[1])else for u=1,#p do p[u]=p[u]+1 end;return e(256-#p,unpack(p))end end;local function v(w,u)u=u or 1;local x=f(w,u,u)if x>249 then local y=0;x=256-x;for z=u+x,u+1,-1 do y=y*255+f(w,z,z)-1 end;return y,x+1 else return x,1 end end;local A={}local function B(C)if type(C)=="string"then local D=256;A={}local E={"\002"}local F=''local G=1;for u=0,255 do A[e(u)]=u end;for u=1,#C do local H=C:sub(u,u)local I=F..H;if A[I]then F=I else A[I]=D;D=D+1;local y=q(A[F])G=G+#y;E[#E+1]=y;F=H end end;if F then local y=q(A[F])G=G+#y;E[#E+1]=y end;if#C+1>G then return c(E)else return e(1)..C end else return nil,"Can only compress strings"end end;local function J(K)if type(K)=="string"then if K:sub(1,1)~="\002"then return nil,"Can only decompress LZW compressed data ("..tostring(K:sub(1,1))..")"end;K=K:sub(2)local D=256;for s in pairs(A)do A[s]=nil end;for u=0,255 do A[u]=e(u)end;local E={}local L=1;local M,s;s,M=v(K,L)L=L+M;E[#E+1]=A[s]local F=A[s]local N;while L<=#K do s,M=v(K,L)L=L+M;N=A[s]or F..F:sub(1,1)E[#E+1]=N;A[D]=F..N:sub(1,1)D=D+1;F=N end;return c(E)else return nil,"Can only uncompress strings"end end;local function O(P,Q,R)if P then P.bcode=Q;P.blength=R;if P.c1 then O(P.c1,k(Q,n(1,R)),R+1)end;if P.c2 then O(P.c2,Q,R+1)end end end;local function S(T,R)local U=0;local V;local W=0;for u=R-1,0,-1 do V=j(T,n(1,u))==0 and 0 or 1;U=n(U,1+V)+V;W=W+V end;return U,R+W end;local X=0;local Y;local Z;local function _(a0,T,R)Y=Y+n(T,Z)Z=R+Z;if Z>32 then return true end;while Z>=8 do X=X+1;a0[X]=e(j(Y,255))Y=o(Y,8)Z=Z-8 end end;local function a1(C)if type(C)~="string"then return nil,"Can only compress strings"end;if#C==0 then return"\001"end;local a2={}local z=0;local a3=g(C)local H;for u=1,a3 do H=f(C,u)a2[H]=(a2[H]or 0)+1 end;local a4={}local a5;local a6={}for a7,a8 in pairs(a2)do a5={symbol=e(a7),weight=a8}a6[a7]=a5;a(a4,a5)end;d(a4,function(x,V)if x.weight<V.weight then return true elseif x.weight>V.weight then return false else return nil end end)local a9=#a4;local aa={}local R,ab,ac,ad,ae,af;local ag;while#a4+#aa>1 do if not next(aa)then ac,ae=next(a4)b(a4,ac)elseif not next(a4)then ad,ae=next(aa)b(aa,ad)else ac,R=next(a4)ad,ab=next(aa)if R.weight<=ab.weight then ae=R;b(a4,ac)else ae=ab;b(aa,ad)end end;if not next(aa)then ac,af=next(a4)b(a4,ac)elseif not next(a4)then ad,af=next(aa)b(aa,ad)else ac,R=next(a4)ad,ab=next(aa)if R.weight<=ab.weight then af=R;b(a4,ac)else af=ab;b(aa,ad)end end;ag={c1=ae,c2=af,weight=ae.weight+af.weight}a(aa,ag)end;if#a4>0 then ac,R=next(a4)a(aa,R)b(a4,ac)end;aa=aa[1]O(aa,0,0)if aa then aa.bcode=0;aa.blength=1 end;Y=0;Z=0;local K={}K[1]="\003"local R=g(C)K[2]=e(j(a9-1,255))K[3]=e(j(R,255))K[4]=e(j(o(R,8),255))K[5]=e(j(o(R,16),255))X=5;for a7,a5 in pairs(a6)do _(K,a7,8)if _(K,S(a5.bcode,a5.blength))then return e(0)..C end;_(K,3,2)end;local ah={}local ai=0;local aj;for u=1,R,200 do aj=R<u+199 and R or u+199;for ak=u,aj do H=f(C,ak)_(K,a6[H].bcode,a6[H].blength)end;ai=ai+1;ah[ai]=c(K,"",1,X)X=0 end;if Z>0 then ai=ai+1;ah[ai]=e(Y)end;local al=c(ah,"",1,ai)if#C+1<=#al then return"\001"..C end;return al end;local am={}setmetatable(am,{__index=function(L,s)local an=n(1,s)rawset(L,s,an)return an end})local ao={}setmetatable(ao,{__index=function(L,s)local an=n(1,s)-1;rawset(L,s,an)return an end})local function ap(aq,ar)if ar>=2 then local V;local as=0;for u=0,ar-1 do V=j(aq,am[u])if not(as==0)and not(V==0)then return j(aq,ao[u-1]),u-1,o(aq,u+1),ar-u-1 end;as=V end end;return nil end;local function at(T,au)local av=0;local V;local W=0;local u=0;while u<au do V=j(T,am[u])if not(V==0)then av=k(av,am[W])u=u+1 end;u=u+1;W=W+1 end;return av,W end;local function aw(K)if not type(C)=="string"then return nil,"Can only uncompress strings"end;local X=#K;local ax=f(K)if ax==1 then return K:sub(2)end;if not(ax==3)then return nil,"Can only decompress Huffman compressed data ("..tostring(ax)..")"end;local ay=f(h(K,2,2))+1;local az=f(h(K,3,3))local c1=f(h(K,4,4))local c2=f(h(K,5,5))local aA=c2*65536+c1*256+az;if aA==0 then return""end;local aq=0;local aB=0;local aC={}setmetatable(aC,{__index=function(L,s)local an={}rawset(L,s,an)return an end})local u=6;local H,aD;local aE=1000;local aF=0;local a7,T,au,aG,aH;local z=0;local aI=0;while z<ay do if u>X then return nil,"Cannot decode map"end;H=f(K,u)aq=k(aq,n(H,aB))aB=aB+8;if aI==0 then a7=j(aq,255)aq=o(aq,8)aB=aB-8;aI=1 else T,au,aG,aH=ap(aq,aB)if T then aq,aB=aG,aH;H,aD=at(T,au)aC[aD][H]=e(a7)aE=aD<aE and aD or aE;aF=aD>aF and aD or aF;z=z+1;aI=0 end end;u=u+1 end;local aJ={}setmetatable(aC,{__index=function(L,s)return aJ end})local C={}local aK={}local a3=0;local aL=0;local aM;local aN=aE;local a7;local aO=0;X=X+1;local aP=200;aP=aP>aA and aA or aP;while true do if aN<=aB then aM=j(aq,ao[aN])a7=aC[aN][aM]if a7 then a3=a3+1;C[a3]=a7;aO=aO+1;if aO>=aP then if aO>=aA then break end;aL=aL+1;aK[aL]=c(C,"",1,a3)a3=0;aP=aP+200;aP=aP>aA and aA or aP end;aq=o(aq,aN)aB=aB-aN;aN=aE else aN=aN+1;if aN>aF then return nil,"Decompression error at "..tostring(u).."/"..tostring(#K)end end else H=f(K,u)aq=aq+n(H or 0,aB)aB=aB+8;if u>X then break end;u=u+1 end end;local aQ={}a(aQ,c(aK,"",1,aL))a(aQ,c(C,"",1,a3))return c(aQ,"")end;local function aR(aS)if type(aS)~="string"then return nil,"Can only handle strings"end;if string.byte(aS)~=1 then return nil,"Can only handle uncompressed data"end;return aS:sub(2)end;local aT={[2]=B,[3]=a1}local aU={[1]=aR,[2]=J,[3]=aw}local function aV(aS)local aW=next(aT)local E=aT[aW](aS)local z;aW=next(aT,aW)while aW do z=aT[aW](aS)if#z<#E then E=z end;aW=next(aT,aW)end;return E end;local function aX(aS)local aY=string.byte(aS)if aU[aY]then return aU[aY](aS)else return nil,"Unknown compression method ("..tostring(aY)..")"end end;return{lzw_c=B,lzw_d=J,huff_c=a1,huff_d=aw,encode=aV,decode=aX}
]]
local ejj
comp, ejj = loadstring(comp)
if not comp then error(ejj) end
comp = comp()

local choice = argz[1]
if choice == "pack" then
  if #argz < 3 then error("Invalid number of parameters supplied: " .. #argz) end
  local from = argz[2]
  local to = argz[3]
  assert(fs.exists(from), "Source does not exists")
  --assert(fs.isDir(from), "Only directories are supported now")
  assert(not fs.exists(to), "Destination exists")

  local whand = fs.open(to, "wb")

  iter_w(whand, from, comp)

  whand.close()

  if not isapi then print("Successfully created archive!") end
  return true
end
local function lineExt() error("thing") end
local function iter_r(rhand, from, comp)
  local function rint()
    local b1, b2, b3, b4 = rhand.read(), rhand.read(), rhand.read(), rhand.read()
    return b1*16777216 + b2*65536 + b3*256 + b4
  end

  local function rstr()
    local buf = {}
    local cnt = rhand.read()
    for i = 1, cnt do
      table.insert(buf, string.char(rhand.read()))
    end
    return table.concat(buf, "")
  end

  local buf = {}

  local function iter(root, pth)
    os.queueEvent("keepalive")
    os.pullEvent("keepalive")
    local fpat = fs.combine(root, pth)
    local name = rstr()
    local nid = rhand.read()
    if not isapi then print("Extracting " .. (nid == 0 and "dir" or "dat") .. " " .. fs.combine(pth, name)) end
    if nid == 0 then -- dir
      local cnt = rint()
      fs.makeDir(fs.combine(fpat, name))
      local relpth = fs.combine(pth, name)
      for i = 1, cnt do
        iter(root, relpth)
      end
    elseif nid == 1 then -- file
      local len = rint()
      local whand = fs.open(fs.combine(fpat, name), "wb")
      for i = 1, len do
        if i % 256 == 0 then
          os.queueEvent("keepalive")
          os.pullEvent("keepalive")
        end
        table.insert(buf, string.char(rhand.read()))
      end
      os.queueEvent("keepalive")
      os.pullEvent("keepalive")
      for chr in comp.decode(table.concat(buf, "")):gmatch(".") do
        whand.write(string.byte(chr))
      end
      whand.flush()
      whand.close()
      buf = {}
    else
      error("Invalid node type: " .. nid)
    end
  end

  return iter(from, "/")
end
if choice == "unpack" then
  if #argz < 3 then error(argz[4] and "No destination given" or ("Invalid number of parameters supplied: " .. #argz)) end
  local from = argz[2]
  local to = argz[3]
  assert(fs.exists(from), "Source does not exists")
  assert(not fs.isDir(from), "Not a valid makeself file")
  assert(not fs.exists(to), "Destination exists")
  local rhand = fs.open(from, "rb")
  if argz[4] == true then while rhand.read() ~= 0 do end end
  iter_r(rhand, to, comp)
  rhand.close()
  if not isapi then print("Successfully extracted archive!") end
  return true
end
local function lineExtEnd() error("thing") end
if choice == "sfx" then
  if #argz < 3 then error("Invalid number of parameters supplied: " .. #argz) end
  local from = argz[2]
  local to = argz[3]
  assert(fs.exists(from), "Source does not exists")
  --assert(fs.isDir(from), "Only directories are supported now")
  assert(not fs.exists(to), "Destination exists")

  local disfeil = ""
  if shell then
    disfeil = shell.getRunningProgram()
  else
    if argz[4] then
      disfeil = argz[4]
    else
      print("Please supply the location to the makepack executable")
      disfeil = read()
    end
  end
  assert(not fs.isDir(disfeil), "The supplied file is a directory")
  assert(fs.exists(disfeil), "File not exists: " .. disfeil)

  local rhand = fs.open(disfeil, "r")
  local whand = fs.open(to, "w")

  whand.writeLine("local argz = {...} argz[3] = argz[1] argz[2] = shell.getRunningProgram() argz[4] = true local choice=\"unpack\"")

  local num1, num2, num3, num2Diff, num3Diff

  local asd, errloc = pcall(lineComp)
  num1 = errnum(errloc)
  _, errloc = pcall(lineExt)
  num2 = errnum(errloc)
  _, errloc = pcall(lineExtEnd)
  num3 = errnum(errloc)

  for i = 1, num1 do
    rhand.readLine()
  end

  for i = 1, lineCompOffs do
    rhand.readLine()
  end

  for i = 1, lineCompNum do
    whand.writeLine(rhand.readLine())
  end

  num2Diff = num2 - (num1 + lineCompOffs + lineCompNum)

  for i = 1, num2Diff do
    rhand.readLine()
  end

  for i = 1, lineExtOffs do
    rhand.readLine()
  end

  num3Diff = num3 - lineExtOffs + lineExtOffsEnd - num2 - 1

  for i = 1, num3Diff do
    whand.writeLine(rhand.readLine())
  end
  whand.write("--[==[\000")

  whand.flush()
  rhand.close()
  whand.close()

  whand = fs.open(to, "ab")

  iter_w(whand, from, comp)

  whand.write(string.byte("]"))
  whand.write(string.byte("="))
  whand.write(string.byte("="))
  whand.write(string.byte("]"))

  whand.flush()
  whand.close()

  print("SelF-eXtracting archive created")
  return true
elseif choice == "api" then
  isapi = true
  return
  {
    compress = function(wat, where)
      assert(fs.exists(wat), "Source does not exists")
      --assert(fs.isDir(wat), "Only directories are supported now")
      assert(not fs.exists(where), "Destination exists")
      local whand = fs.open(where, "wb")
      iter_w(whand, wat, comp)
      whand.flush()
      whand.close()
      return true
    end,

    decompress = function(wat, where, hazsfx)
      assert(fs.exists(wat), "Source does not exists")
      assert(not fs.isDir(wat), "Not a valid makeself file")
      assert(not fs.exists(where), "Destination exists")
      local rhand = fs.open(wat, "rb")
      if hazsfx then while rhand.read() ~= 0 do end end
      iter_r(rhand, where, comp)
      rhand.close()
      return true
    end
  }
else
  error("Invalid option: " .. tostring(choice))
end