-- LevelOS

term.clear()
term.setCursorPos(1,1)

if not fs.exists("LevelOS") then
	--Let the bios know the system is broken
	fs.delete("/real_startup.lua")
	os.reboot()
	return
end
for k,v in pairs(colors) do
	if type(v) == "number" then
		term.setPaletteColor(v,term.native().getPaletteColor(v))
	end
end

if lOS then
	error("LevelOS is already running, silly!")
end

_G.lOS = {}
lOS.path = shell.getRunningProgram()


local chunkSize = 32768
if jit then
	chunkSize = 4092
end

local function encodeUTF8(asciiText)
    local utf8Text = ""
    
    for i=1, #asciiText, chunkSize do
        utf8Text = utf8Text..utf8.char(string.byte(asciiText, i, math.min(i + (chunkSize-1), #asciiText)))
    end
    
    return utf8Text
end

local function decodeUTF8(utf8Text)
    local asciiText = ""
    
    local ok, t = pcall(function() 
        return string.char(utf8.codepoint(utf8Text, 1, #utf8Text)) 
    end)
    
    if ok then
        asciiText = t
    else
        local ok2, err = pcall(function()
            for _, codepoint in utf8.codes(utf8Text) do
                if codepoint < 256 then
                    asciiText = asciiText .. string.char(codepoint)
                else
                    asciiText = asciiText .. "?"
                end
            end
        end)

        if not ok2 then
            _G.debugCurrentText = asciiText
            _G.debugTextInput = utf8Text

            error(err, 2)
        end
    end
    
    return asciiText
end


local function encodeAll(...)
    local tbl = table.pack(...)
    for k,v in pairs(tbl) do
        if type(v) == "string" and not utf8.len(v) then
            tbl[k] = encodeUTF8(v)
        end
    end
    
    return table.unpack(tbl, 1, tbl.n)
end
 
local function decodeAll(...)
    local tbl = table.pack(...)
    for k,v in pairs(tbl) do
        if type(v) == "string" then
            tbl[k] = decodeUTF8(v)
        end
    end
    
    return table.unpack(tbl, 1, tbl.n)
end

lOS.utf8 = {
	decode = decodeUTF8,
	encode = encodeUTF8,
	decodeAll = decodeAll,
	encodeAll = encodeAll,
}

local function extractVersion(str)
	local version = str:match("ComputerCraft (%d+%.%d+%.%d+)")
	return version
end

local function isVersionAbove(version1, version2)
	local function splitVersion(version)
		local parts = {}
		for part in version:gmatch("(%d+)") do
			table.insert(parts, tonumber(part))
		end
		return parts
	end

	local v1Parts = splitVersion(version1)
	local v2Parts = splitVersion(version2)

	for i = 1, math.max(#v1Parts, #v2Parts) do
		local v1 = v1Parts[i] or 0
		local v2 = v2Parts[i] or 0
		if v1 > v2 then
			return true
		elseif v1 < v2 then
			return false
		end
	end

	return true -- They are equal if all parts are equal
end

if isVersionAbove(extractVersion(_HOST), "1.109") then
    local fopen = fs.open
     
    function fs.open(path, mode)
        local f = fopen(path, mode)
        if not f then return nil end
        
        local customHandle = {}
        
        for k,v in pairs(f) do
            if mode:find("b") then
                customHandle[k] = function(...) return v(...) end
            else
                customHandle[k] = function(...) return decodeAll(v(encodeAll(...))) end
            end
        end
        
        return customHandle
    end
end


--enable mouse if no color
if not rtype then
	_G.rtype = type
	_G.type = function(obj)
		local mt = getmetatable(obj)
		if rtype(mt) == "table" and mt.__type then
			if rtype(mt.__type) == "string" then
				return mt.__type
			elseif rtype(mt.__type) == "function" then
				return mt.__type(obj)
			end
		else
			return rtype(obj)
		end
	end
end

function _G.pairs(t)
    local mt = getmetatable(t)
    if mt and type(mt.__pairs) == "function" then
        return mt.__pairs(t)
    else
        return next, t, nil
    end
end

if not hardreboot then
	_G.hardreboot = os.reboot
end

if fs.combine("a","b","c") == fs.combine("a","b") then
	local ocombine = fs.combine
	function fs.combine(path,...)
		local parts = {...}
		for p=1,#parts do
			path = ocombine(path,parts[p])
		end
		return path
	end
end

local w,h = term.getSize()
if h < 19 then return end
local newwin = false

local therealOGterm = term.current()
local function tokenise( ... )
	local sLine = table.concat( { ... }, " " )
	local tWords = {}
	local bQuoted = false
	for match in string.gmatch( sLine .. "\"", "(.-)\"" ) do
		if bQuoted then
			table.insert( tWords, match )
		else
			for m in string.gmatch( match, "[^ \t]+" ) do
				table.insert( tWords, m )
			end
		end
		bQuoted = not bQuoted
	end
	return tWords
end

if fs.exists("AppData") == false then
	fs.makeDir("AppData")
end

if fs.exists("User") == false then
	fs.makeDir("User")
	fs.makeDir("User/Documents")
	fs.makeDir("User/Images")
	fs.makeDir("User/Scripts")
	fs.makeDir("User/Downloads")
end

if not fs.exists("bigfont") then
	--Let the bios know the system is broken
	fs.delete("/real_startup.lua")
	os.reboot()
	return
end

local to_colors, to_blit = {}, {}
for i = 1, 16 do
	to_blit[2^(i-1)] = ("0123456789abcdef"):sub(i, i)
	to_colors[("0123456789abcdef"):sub(i, i)] = 2^(i-1)
end
 
 
 
local function toColor(theblit)
	return to_colors[theblit] or nil
end
 
 
 
local function toBlit(thecolor)
	return to_blit[thecolor] or nil
end

local function render(spr,x,y)
	local format = "lImg"
	if format == "lImg" then
		local sW,sH = #spr[1][1],#spr
		local w,h = term.getSize()
		for l=1,#spr do
			if not y then
				term.setCursorPos(math.ceil(w/2)-math.floor(sW/2),(math.ceil(h/2)-math.floor(sH/2)+(l-1))+x)
			else
				term.setCursorPos(x,y+(l-1))
			end
			local bl = {}
			bl[1] = spr[l][1]
			bl[2] = string.gsub(spr[l][2],"T",toBlit(term.getBackgroundColor()))
			bl[3] = string.gsub(spr[l][3],"T",toBlit(term.getBackgroundColor()))
			term.blit(unpack(bl))
		end
	elseif format == "nfp" or format == "nfg" then
		local b,e = string.find(spr,"\n")
		local sW,sH
		local w,h = term.getSize()
		local lines,sW = getLines(spr)
		sH = #lines
		for l=1,sH do
			term.setCursorPos(math.ceil(w/2)-math.floor(sW/2),math.ceil(h/2)-math.floor(sH/2)+(l-1))
			term.blit(string.rep(" ",#lines[l]),lines[l],lines[l])
		end
	end
end


local function fread(file)
	local f = fs.open(file,"r")
	local o = f.readAll()
	f.close()
	return o
end

local function fwrite(file,content)
	local f = fs.open(file,"w")
	f.write(content)
	f.close()
	return true
end

local loadingico = textutils.unserialize(fread("LevelOS/assets/loading.limg"))


_G.bigfont = loadfile("bigfont",_ENV)()


local progress = 0

local function centerText(text,customY,customLen) -- i tried to put indentation but pastebin is being stupid for some reason
	local x,y = term.getSize()
	local x2,y2 = term.getCursorPos()
	if customY then y2 = customY end
	local len = customLen or text:len()
	term.setCursorPos((math.ceil(x / 2) - math.floor(len / 2)), y2)
	term.write(text)
	term.setCursorPos(x2,y2+1)
end

local doUpdate
local bootText = "Initializing"
local dots = 1
local frame = 1

local function bootscreen()
	term.setBackgroundColor(colors.black)
	term.setTextColor(colors.cyan)
	term.clear()
	local w,h = term.getSize()
	if h > 19 then
		if fs.exists("LevelOS/assets/logo_christmas.limg") and os.date("%m") == "12" then
			render(textutils.unserialize(fread("LevelOS/assets/logo_christmas.limg"))[1],math.ceil(w/2)-4,h/2-8)
		elseif fs.exists("LevelOS/assets/logo_pride.limg") then
			render(textutils.unserialize(fread("LevelOS/assets/logo_pride.limg"))[1],math.ceil(w/2)-4,h/2-8)
		else
			bigfont.writeOn(term.current(),2,"B",nil,h/2-8)
		end
	else
		if fs.exists("LevelOS/assets/logo_christmas.limg") and os.date("%m") == "12" then
			render(textutils.unserialize(fread("LevelOS/assets/logo_christmas.limg"))[1],math.ceil(w/2)-4,h/2-5)
		elseif fs.exists("LevelOS/assets/logo_pride.limg") then
			render(textutils.unserialize(fread("LevelOS/assets/logo_pride.limg"))[1],math.ceil(w/2)-4,h/2-5)
		else
			bigfont.writeOn(term.current(),2,"B",nil,h/2-5)
		end
	end
	while true do
		if doUpdate then
			term.setBackgroundColor(colors.blue)
			term.setTextColor(colors.white)
			term.clear()
			local w,h = term.getSize()
			term.setCursorPos(1,math.ceil(h/2))
			centerText("Getting ready for updates")
			centerText("Do not turn off your computer")
			local init = false
			while true do
				render(loadingico[frame],-5)
				if progress > 0 then
					if not init then
						term.setBackgroundColor(colors.blue)
						term.setTextColor(colors.white)
						term.clear()
						render(loadingico[frame],-5)
						init = true
					end
					term.setCursorPos(1,math.ceil(h/2))
					centerText("Working on updates")
					centerText(math.floor(progress + 0.5).."% complete")
					centerText("Do not turn off your computer")
				end
				frame = frame+1
				if frame > #loadingico then
					frame = 1
				end
				os.sleep(0.1)
			end
		end
		if h > 19 then
			render(loadingico[frame],10)
			if bootText then
				centerText("          "..bootText..string.rep(".",dots).."          ",h/2+14,#("          "..bootText.."          "))
			end
		else
			render(loadingico[frame],7)
		end
		os.pullEvent()
	end
end

local function loadIco()
	while true do
		frame = frame+1
		if frame > #loadingico then
			frame = 1
		end
		dots = dots+0.5
		if dots > 3 then dots = 0 end
		os.sleep(0.1)
	end
end

local function update()
	os.sleep(2)
	doUpdate = true
	os.sleep(5)
	download("startup.lua")
	fs.delete("real_startup.lua")
	fs.move("tmp/real_startup.lua", "real_startup.lua")
	fs.move("LevelOS/startup/lUtils.lua", "tmp/LevelOS/startup/lUtils.lua")
	os.sleep(1)
	os.reboot()
end


parallel.waitForAny(update,bootscreen,loadIco)