Oeed

OneOS Installer

Feb 8th, 2014
52,836
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
  1. tArgs = {...}
  2.  
  3. if not term.isColor or not term.isColor() then
  4.     error('OneOS Requires an Advanced (gold) Computer')
  5. end
  6.  
  7. _jstr = [[
  8.     local base = _G
  9.  
  10.     -----------------------------------------------------------------------------
  11.     -- Module declaration
  12.     -----------------------------------------------------------------------------
  13.  
  14.     -- Public functions
  15.  
  16.     -- Private functions
  17.     local decode_scanArray
  18.     local decode_scanComment
  19.     local decode_scanConstant
  20.     local decode_scanNumber
  21.     local decode_scanObject
  22.     local decode_scanString
  23.     local decode_scanWhitespace
  24.     local encodeString
  25.     local isArray
  26.     local isEncodable
  27.  
  28.     -----------------------------------------------------------------------------
  29.     -- PUBLIC FUNCTIONS
  30.     -----------------------------------------------------------------------------
  31.     --- Encodes an arbitrary Lua object / variable.
  32.     -- @param v The Lua object / variable to be JSON encoded.
  33.     -- @return String containing the JSON encoding in internal Lua string format (i.e. not unicode)
  34.     function encode (v)
  35.       -- Handle nil values
  36.       if v==nil then
  37.         return "null"
  38.       end
  39.      
  40.       local vtype = base.type(v)  
  41.  
  42.       -- Handle strings
  43.       if vtype=='string' then    
  44.         return '"' .. encodeString(v) .. '"'      -- Need to handle encoding in string
  45.       end
  46.      
  47.       -- Handle booleans
  48.       if vtype=='number' or vtype=='boolean' then
  49.         return base.tostring(v)
  50.       end
  51.      
  52.       -- Handle tables
  53.       if vtype=='table' then
  54.         local rval = {}
  55.         -- Consider arrays separately
  56.         local bArray, maxCount = isArray(v)
  57.         if bArray then
  58.           for i = 1,maxCount do
  59.             table.insert(rval, encode(v[i]))
  60.           end
  61.         else -- An object, not an array
  62.           for i,j in base.pairs(v) do
  63.             if isEncodable(i) and isEncodable(j) then
  64.               table.insert(rval, '"' .. encodeString(i) .. '":' .. encode(j))
  65.             end
  66.           end
  67.         end
  68.         if bArray then
  69.           return '[' .. table.concat(rval,',') ..']'
  70.         else
  71.           return '{' .. table.concat(rval,',') .. '}'
  72.         end
  73.       end
  74.      
  75.       -- Handle null values
  76.       if vtype=='function' and v==null then
  77.         return 'null'
  78.       end
  79.      
  80.       base.assert(false,'encode attempt to encode unsupported type ' .. vtype .. ':' .. base.tostring(v))
  81.     end
  82.  
  83.  
  84.     --- Decodes a JSON string and returns the decoded value as a Lua data structure / value.
  85.     -- @param s The string to scan.
  86.     -- @param [startPos] Optional starting position where the JSON string is located. Defaults to 1.
  87.     -- @param Lua object, number The object that was scanned, as a Lua table / string / number / boolean or nil,
  88.     -- and the position of the first character after
  89.     -- the scanned JSON object.
  90.     function decode(s, startPos)
  91.       startPos = startPos and startPos or 1
  92.       startPos = decode_scanWhitespace(s,startPos)
  93.       base.assert(startPos<=string.len(s), 'Unterminated JSON encoded object found at position in [' .. s .. ']')
  94.       local curChar = string.sub(s,startPos,startPos)
  95.       -- Object
  96.       if curChar=='{' then
  97.         return decode_scanObject(s,startPos)
  98.       end
  99.       -- Array
  100.       if curChar=='[' then
  101.         return decode_scanArray(s,startPos)
  102.       end
  103.       -- Number
  104.       if string.find("+-0123456789.e", curChar, 1, true) then
  105.         return decode_scanNumber(s,startPos)
  106.       end
  107.       -- String
  108.       if curChar=='"' or curChar=="'" then
  109.         return decode_scanString(s,startPos)
  110.       end
  111.       if string.sub(s,startPos,startPos+1)=='/*' then
  112.         return decode(s, decode_scanComment(s,startPos))
  113.       end
  114.       -- Otherwise, it must be a constant
  115.       return decode_scanConstant(s,startPos)
  116.     end
  117.  
  118.     --- The null function allows one to specify a null value in an associative array (which is otherwise
  119.     -- discarded if you set the value with 'nil' in Lua. Simply set t = { first=json.null }
  120.     function null()
  121.       return null -- so json.null() will also return null ;-)
  122.     end
  123.     -----------------------------------------------------------------------------
  124.     -- Internal, PRIVATE functions.
  125.     -- Following a Python-like convention, I have prefixed all these 'PRIVATE'
  126.     -- functions with an underscore.
  127.     -----------------------------------------------------------------------------
  128.  
  129.     --- Scans an array from JSON into a Lua object
  130.     -- startPos begins at the start of the array.
  131.     -- Returns the array and the next starting position
  132.     -- @param s The string being scanned.
  133.     -- @param startPos The starting position for the scan.
  134.     -- @return table, int The scanned array as a table, and the position of the next character to scan.
  135.     function decode_scanArray(s,startPos)
  136.       local array = {}   -- The return value
  137.       local stringLen = string.len(s)
  138.       base.assert(string.sub(s,startPos,startPos)=='[','decode_scanArray called but array does not start at position ' .. startPos .. ' in string:\n'..s )
  139.       startPos = startPos + 1
  140.       -- Infinite loop for array elements
  141.       repeat
  142.         startPos = decode_scanWhitespace(s,startPos)
  143.         base.assert(startPos<=stringLen,'JSON String ended unexpectedly scanning array.')
  144.         local curChar = string.sub(s,startPos,startPos)
  145.         if (curChar==']') then
  146.           return array, startPos+1
  147.         end
  148.         if (curChar==',') then
  149.           startPos = decode_scanWhitespace(s,startPos+1)
  150.         end
  151.         base.assert(startPos<=stringLen, 'JSON String ended unexpectedly scanning array.')
  152.         object, startPos = decode(s,startPos)
  153.         table.insert(array,object)
  154.       until false
  155.     end
  156.  
  157.     --- Scans a comment and discards the comment.
  158.     -- Returns the position of the next character following the comment.
  159.     -- @param string s The JSON string to scan.
  160.     -- @param int startPos The starting position of the comment
  161.     function decode_scanComment(s, startPos)
  162.       base.assert( string.sub(s,startPos,startPos+1)=='/*', "decode_scanComment called but comment does not start at position " .. startPos)
  163.       local endPos = string.find(s,'*/',startPos+2)
  164.       base.assert(endPos~=nil, "Unterminated comment in string at " .. startPos)
  165.       return endPos+2  
  166.     end
  167.  
  168.     --- Scans for given constants: true, false or null
  169.     -- Returns the appropriate Lua type, and the position of the next character to read.
  170.     -- @param s The string being scanned.
  171.     -- @param startPos The position in the string at which to start scanning.
  172.     -- @return object, int The object (true, false or nil) and the position at which the next character should be
  173.     -- scanned.
  174.     function decode_scanConstant(s, startPos)
  175.       local consts = { ["true"] = true, ["false"] = false, ["null"] = nil }
  176.       local constNames = {"true","false","null"}
  177.  
  178.       for i,k in base.pairs(constNames) do
  179.         --print ("[" .. string.sub(s,startPos, startPos + string.len(k) -1) .."]", k)
  180.         if string.sub(s,startPos, startPos + string.len(k) -1 )==k then
  181.           return consts[k], startPos + string.len(k)
  182.         end
  183.       end
  184.       base.assert(nil, 'Failed to scan constant from string ' .. s .. ' at starting position ' .. startPos)
  185.     end
  186.  
  187.     --- Scans a number from the JSON encoded string.
  188.     -- (in fact, also is able to scan numeric +- eqns, which is not
  189.     -- in the JSON spec.)
  190.     -- Returns the number, and the position of the next character
  191.     -- after the number.
  192.     -- @param s The string being scanned.
  193.     -- @param startPos The position at which to start scanning.
  194.     -- @return number, int The extracted number and the position of the next character to scan.
  195.     function decode_scanNumber(s,startPos)
  196.       local endPos = startPos+1
  197.       local stringLen = string.len(s)
  198.       local acceptableChars = "+-0123456789.e"
  199.       while (string.find(acceptableChars, string.sub(s,endPos,endPos), 1, true)
  200.        and endPos<=stringLen
  201.        ) do
  202.         endPos = endPos + 1
  203.       end
  204.       local stringValue = 'return ' .. string.sub(s,startPos, endPos-1)
  205.       local stringEval = base.loadstring(stringValue)
  206.       base.assert(stringEval, 'Failed to scan number [ ' .. stringValue .. '] in JSON string at position ' .. startPos .. ' : ' .. endPos)
  207.       return stringEval(), endPos
  208.     end
  209.  
  210.     --- Scans a JSON object into a Lua object.
  211.     -- startPos begins at the start of the object.
  212.     -- Returns the object and the next starting position.
  213.     -- @param s The string being scanned.
  214.     -- @param startPos The starting position of the scan.
  215.     -- @return table, int The scanned object as a table and the position of the next character to scan.
  216.     function decode_scanObject(s,startPos)
  217.       local object = {}
  218.       local stringLen = string.len(s)
  219.       local key, value
  220.       base.assert(string.sub(s,startPos,startPos)=='{','decode_scanObject called but object does not start at position ' .. startPos .. ' in string:\n' .. s)
  221.       startPos = startPos + 1
  222.       repeat
  223.         startPos = decode_scanWhitespace(s,startPos)
  224.         base.assert(startPos<=stringLen, 'JSON string ended unexpectedly while scanning object.')
  225.         local curChar = string.sub(s,startPos,startPos)
  226.         if (curChar=='}') then
  227.           return object,startPos+1
  228.         end
  229.         if (curChar==',') then
  230.           startPos = decode_scanWhitespace(s,startPos+1)
  231.         end
  232.         base.assert(startPos<=stringLen, 'JSON string ended unexpectedly scanning object.')
  233.         -- Scan the key
  234.         key, startPos = decode(s,startPos)
  235.         base.assert(startPos<=stringLen, 'JSON string ended unexpectedly searching for value of key ' .. key)
  236.         startPos = decode_scanWhitespace(s,startPos)
  237.         base.assert(startPos<=stringLen, 'JSON string ended unexpectedly searching for value of key ' .. key)
  238.         base.assert(string.sub(s,startPos,startPos)==':','JSON object key-value assignment mal-formed at ' .. startPos)
  239.         startPos = decode_scanWhitespace(s,startPos+1)
  240.         base.assert(startPos<=stringLen, 'JSON string ended unexpectedly searching for value of key ' .. key)
  241.         value, startPos = decode(s,startPos)
  242.         object[key]=value
  243.       until false  -- infinite loop while key-value pairs are found
  244.     end
  245.  
  246.     --- Scans a JSON string from the opening inverted comma or single quote to the
  247.     -- end of the string.
  248.     -- Returns the string extracted as a Lua string,
  249.     -- and the position of the next non-string character
  250.     -- (after the closing inverted comma or single quote).
  251.     -- @param s The string being scanned.
  252.     -- @param startPos The starting position of the scan.
  253.     -- @return string, int The extracted string as a Lua string, and the next character to parse.
  254.     function decode_scanString(s,startPos)
  255.       base.assert(startPos, 'decode_scanString(..) called without start position')
  256.       local startChar = string.sub(s,startPos,startPos)
  257.       base.assert(startChar=="'" or startChar=='"','decode_scanString called for a non-string')
  258.       local escaped = false
  259.       local endPos = startPos + 1
  260.       local bEnded = false
  261.       local stringLen = string.len(s)
  262.       repeat
  263.         local curChar = string.sub(s,endPos,endPos)
  264.         -- Character escaping is only used to escape the string delimiters
  265.         if not escaped then
  266.           if curChar=='\\' then
  267.             escaped = true
  268.           else
  269.             bEnded = curChar==startChar
  270.           end
  271.         else
  272.           -- If we're escaped, we accept the current character come what may
  273.           escaped = false
  274.         end
  275.         endPos = endPos + 1
  276.         base.assert(endPos <= stringLen+1, "String decoding failed: unterminated string at position " .. endPos)
  277.       until bEnded
  278.       local stringValue = 'return ' .. string.sub(s, startPos, endPos-1)
  279.       local stringEval = base.loadstring(stringValue)
  280.       base.assert(stringEval, 'Failed to load string [ ' .. stringValue .. '] in JSON4Lua.decode_scanString at position ' .. startPos .. ' : ' .. endPos)
  281.       return stringEval(), endPos  
  282.     end
  283.  
  284.     --- Scans a JSON string skipping all whitespace from the current start position.
  285.     -- Returns the position of the first non-whitespace character, or nil if the whole end of string is reached.
  286.     -- @param s The string being scanned
  287.     -- @param startPos The starting position where we should begin removing whitespace.
  288.     -- @return int The first position where non-whitespace was encountered, or string.len(s)+1 if the end of string
  289.     -- was reached.
  290.     function decode_scanWhitespace(s,startPos)
  291.       local whitespace=" \n\r\t"
  292.       local stringLen = string.len(s)
  293.       while ( string.find(whitespace, string.sub(s,startPos,startPos), 1, true)  and startPos <= stringLen) do
  294.         startPos = startPos + 1
  295.       end
  296.       return startPos
  297.     end
  298.  
  299.     --- Encodes a string to be JSON-compatible.
  300.     -- This just involves back-quoting inverted commas, back-quotes and newlines, I think ;-)
  301.     -- @param s The string to return as a JSON encoded (i.e. backquoted string)
  302.     -- @return The string appropriately escaped.
  303.     function encodeString(s)
  304.       s = string.gsub(s,'\\','\\\\')
  305.       s = string.gsub(s,'"','\\"')
  306.       s = string.gsub(s,"'","\\'")
  307.       s = string.gsub(s,'\n','\\n')
  308.       s = string.gsub(s,'\t','\\t')
  309.       return s
  310.     end
  311.  
  312.     -- Determines whether the given Lua type is an array or a table / dictionary.
  313.     -- We consider any table an array if it has indexes 1..n for its n items, and no
  314.     -- other data in the table.
  315.     -- I think this method is currently a little 'flaky', but can't think of a good way around it yet...
  316.     -- @param t The table to evaluate as an array
  317.     -- @return boolean, number True if the table can be represented as an array, false otherwise. If true,
  318.     -- the second returned value is the maximum
  319.     -- number of indexed elements in the array.
  320.     function isArray(t)
  321.       -- Next we count all the elements, ensuring that any non-indexed elements are not-encodable
  322.       -- (with the possible exception of 'n')
  323.       local maxIndex = 0
  324.       for k,v in base.pairs(t) do
  325.         if (base.type(k)=='number' and math.floor(k)==k and 1<=k) then   -- k,v is an indexed pair
  326.           if (not isEncodable(v)) then return false end   -- All array elements must be encodable
  327.           maxIndex = math.max(maxIndex,k)
  328.         else
  329.           if (k=='n') then
  330.             if v ~= table.getn(t) then return false end  -- False if n does not hold the number of elements
  331.           else -- Else of (k=='n')
  332.             if isEncodable(v) then return false end
  333.           end  -- End of (k~='n')
  334.         end -- End of k,v not an indexed pair
  335.       end  -- End of loop across all pairs
  336.       return true, maxIndex
  337.     end
  338.  
  339.     --- Determines whether the given Lua object / table / variable can be JSON encoded. The only
  340.     -- types that are JSON encodable are: string, boolean, number, nil, table and json.null.
  341.     -- In this implementation, all other types are ignored.
  342.     -- @param o The object to examine.
  343.     -- @return boolean True if the object should be JSON encoded, false if it should be ignored.
  344.     function isEncodable(o)
  345.       local t = base.type(o)
  346.       return (t=='string' or t=='boolean' or t=='number' or t=='nil' or t=='table') or (t=='function' and o==null)
  347.     end
  348. ]]
  349.  
  350. function loadJSON()
  351.     local sName = 'JSON'
  352.        
  353.     local tEnv = {}
  354.     setmetatable( tEnv, { __index = _G } )
  355.     local fnAPI, err = loadstring(_jstr)
  356.     if fnAPI then
  357.         setfenv( fnAPI, tEnv )
  358.         fnAPI()
  359.     else
  360.         printError( err )
  361.         return false
  362.     end
  363.    
  364.     local tAPI = {}
  365.     for k,v in pairs( tEnv ) do
  366.         tAPI[k] =  v
  367.     end
  368.    
  369.     _G[sName] = tAPI
  370.     return true
  371. end
  372.  
  373. local mainTitle = 'OneOS Installer'
  374. local subTitle = 'Please wait...'
  375.  
  376. function Draw()
  377.     sleep(0)
  378.     term.setBackgroundColour(colours.white)
  379.     term.clear()
  380.     local w, h = term.getSize()
  381.     term.setTextColour(colours.lightBlue)
  382.     term.setCursorPos(math.ceil((w-#mainTitle)/2), 8)
  383.     term.write(mainTitle)
  384.     term.setTextColour(colours.blue)
  385.     term.setCursorPos(math.ceil((w-#subTitle)/2), 10)
  386.     term.write(subTitle)
  387. end
  388.  
  389. tArgs = {...}
  390.  
  391. Settings = {
  392.     InstallPath = '/', --Where the program's installed, don't always asume root (if it's run under something like OneOS)
  393.     Hidden = false, --Whether or not the update is hidden (doesn't write to the screen), useful for background updates
  394.     GitHubUsername = 'oeed', --Your GitHub username as it appears in the URL
  395.     GitHubRepoName = 'OneOS', --The repo name as it appears in the URL
  396.     DownloadReleases = true, --If true it will download the latest release, otherwise it will download the files as they currently appear
  397.     UpdateFunction = nil, --Sent when something happens (file downloaded etc.)
  398.     TotalBytes = 0, --Do not change this value (especially programatically)!
  399.     DownloadedBytes = 0, --Do not change this value (especially programatically)!
  400.     Status = '',
  401.     SecondaryStatus = '',
  402. }
  403.  
  404. loadJSON()
  405.  
  406. function downloadJSON(path)
  407.     local _json = http.get(path)
  408.     if not _json then
  409.         error('Could not download: '..path..' Check your connection.')
  410.     end
  411.     return JSON.decode(_json.readAll())
  412. end
  413.  
  414. if http then
  415.     subTitle = 'HTTP enabled, attempting update...'
  416.     Draw()
  417. else
  418.     subTitle = 'HTTP is required to update.'
  419.     Draw()
  420.     error('')
  421. end
  422.  
  423. subTitle = 'Determining Latest Version'
  424. Draw()
  425. local releases = downloadJSON('https://api.github.com/repos/'..Settings.GitHubUsername..'/'..Settings.GitHubRepoName..'/releases')
  426. local latestReleaseTag = releases[1].tag_name
  427. if not tArgs or #tArgs ~= 1 and tArgs[1] ~= 'beta' then
  428.     for i, v in ipairs(releases) do
  429.         if not v.prerelease then
  430.             latestReleaseTag = v.tag_name
  431.             break
  432.         end
  433.     end
  434. end
  435. subTitle = 'Optaining Latest Version URL'
  436. Draw()
  437. local refs = downloadJSON('https://api.github.com/repos/'..Settings.GitHubUsername..'/'..Settings.GitHubRepoName..'/git/refs/tags/'..latestReleaseTag)
  438. local latestReleaseSha = refs.object.sha
  439.  
  440. subTitle = 'Downloading File Listing'
  441. Draw()
  442.  
  443. local tree = downloadJSON('https://api.github.com/repos/'..Settings.GitHubUsername..'/'..Settings.GitHubRepoName..'/git/trees/'..latestReleaseSha..'?recursive=1').tree
  444.  
  445. local blacklist = {
  446.     '/.gitignore',
  447.     '/README.md',
  448.     '/TODO',
  449.     '/Desktop/.Desktop.settings',
  450.     '/.version'
  451. }
  452.  
  453. function isBlacklisted(path)
  454.     for i, item in ipairs(blacklist) do
  455.         if item == path then
  456.             return true
  457.         end
  458.     end
  459.     return false
  460. end
  461.  
  462. Settings.TotalFiles = 0
  463. Settings.TotalBytes = 0
  464. for i, v in ipairs(tree) do
  465.     if not isBlacklisted(Settings.InstallPath..v.path) and v.size then
  466.         Settings.TotalBytes = Settings.TotalBytes + v.size
  467.         Settings.TotalFiles = Settings.TotalFiles + 1
  468.     end
  469. end
  470.  
  471. Settings.DownloadedBytes = 0
  472. Settings.DownloadedFiles = 0
  473. function downloadBlob(v)
  474.     if isBlacklisted(Settings.InstallPath..v.path) then
  475.         return
  476.     end
  477.     if v.type == 'tree' then
  478.         -- subTitle = 'Making folder: '..'/'..Settings.InstallPath..v.path
  479.         Draw()
  480.         fs.makeDir('/'..Settings.InstallPath..v.path)
  481.     else
  482.         -- subTitle = 'Starting download for: '..Settings.InstallPath..v.path
  483.         Draw()
  484.  
  485.         local tries, f = 0
  486.         repeat
  487.             f = http.get(('https://raw.github.com/'..Settings.GitHubUsername..'/'..Settings.GitHubRepoName..'/'..latestReleaseTag..Settings.InstallPath..v.path):gsub(' ','%%20'))
  488.                 if not f then sleep(5) end
  489.                 tries = tries + 1
  490.         until tries > 5 or f
  491.  
  492.         if not f then
  493.             error('Downloading failed, try again. '..('https://raw.github.com/'..Settings.GitHubUsername..'/'..Settings.GitHubRepoName..'/'..latestReleaseTag..Settings.InstallPath..v.path):gsub(' ','%%20'))
  494.         end
  495.  
  496.         local h = fs.open('/'..Settings.InstallPath..v.path, 'w')
  497.         h.write(f.readAll())
  498.         h.close()
  499.         -- subTitle = 'Downloading: ' .. math.floor(100*(Settings.DownloadedBytes/Settings.TotalBytes))..'%'
  500.         subTitle = 'Downloading: ' .. math.floor(100*(Settings.DownloadedFiles/Settings.TotalFiles))..'%' -- using the number of files over the number of bytes actually appears to be more accurate, the connection takes longer than sending the data
  501.         -- subTitle = '('..math.floor(100*(Settings.DownloadedBytes/Settings.TotalBytes))..'%) Downloaded: '..Settings.InstallPath..v.path
  502.         Draw()
  503.         if v.size then
  504.             Settings.DownloadedBytes = Settings.DownloadedBytes + v.size
  505.             Settings.DownloadedFiles = Settings.DownloadedFiles + 1
  506.         end
  507.     end
  508. end
  509.  
  510. local connectionLimit = 5
  511. local downloads = {}
  512. for i, v in ipairs(tree) do
  513.     local queueNumber = math.ceil(i / connectionLimit)
  514.     if not downloads[queueNumber] then
  515.         downloads[queueNumber] = {}
  516.     end
  517.     table.insert(downloads[queueNumber], function()
  518.         downloadBlob(v)
  519.     end)
  520. end
  521.  
  522. for i, queue in ipairs(downloads) do
  523.     parallel.waitForAll(unpack(queue))
  524. end
  525.  
  526. local h = fs.open('/System/.version', 'w')
  527. h.write(latestReleaseTag)
  528. h.close()
  529.  
  530. mainTitle = 'Installation Complete!'
  531. subTitle = 'Rebooting in 1 second...'
  532. Draw()
  533. sleep(1)
  534. os.reboot()
RAW Paste Data

Adblocker detected! Please consider disabling it...

We've detected AdBlock Plus or some other adblocking software preventing Pastebin.com from fully loading.

We don't have any obnoxious sound, or popup ads, we actively block these annoying types of ads!

Please add Pastebin.com to your ad blocker whitelist or disable your adblocking software.

×