-- This file is part of Capy64 - https://github.com/Capy64/Capy64
-- Copyright 2023 Alessandro "AlexDevs" Proto
--
-- Licensed under the Apache License, Version 2.0 (the "License").
-- you may not use this file except in compliance with the License.
-- You may obtain a copy of the License at
--
--     http://www.apache.org/licenses/LICENSE-2.0
--
-- Unless required by applicable law or agreed to in writing, software
-- distributed under the License is distributed on an "AS IS" BASIS,
-- WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-- See the License for the specific language governing permissions and
-- limitations under the License.

local term = require("term")
local timer = require("timer")
local gpu = require("gpu")
local fs = require("fs")
local machine = require("machine")
local audio = require("audio")
local http = require("http")
local event = require("event")


local INDEX_URL = "https://raw.github.com/Capy64/CapyOS/deploy/index.json"
local JSON_URL = "https://raw.github.com/Capy64/CapyOS/main/lib/json.lua"

local bootSleep = 2000
local bg = 0x0
local fg = 0xffffff

term.setForeground(fg)
term.setBackground(bg)
term.clear()

term.setSize(51, 19)
gpu.setScale(2)

local w, h = term.getSize()

local function sleep(n)
	local timerId = timer.start(n)
	repeat
		local ev, par = coroutine.yield("timer")
	until par == timerId
end

local function writeCenter(text)
	local w, h = term.getSize()
	local _, y = term.getPos()
	term.setPos(
		(1 + w / 2) - (#text / 2),
		y
	)
	term.write(text)
end

local function drawVendorImage()
	if not fs.exists("/boot/vendor.bmp") then
		return
	end

	local w, h = gpu.getSize()
	local ok, err = pcall(function()
		local buffer<close>, width, height = gpu.loadImage("/boot/vendor.bmp")

		local x, y = 
			math.ceil((w / 2) - (width / 2)),
			math.ceil((h / 2) - (height / 2))

		gpu.drawBuffer(buffer, x, y, width, height)
	end)

	if not ok then
		print("Warning: Could not draw vendor.bmp", err)
	end
end

local nPrint = print
local function print( text )
	local x, y = term.getPos()
	term.write(tostring(text))
	if y == h then
		term.scroll(1)
		term.setPos(1, y)
	else
		term.setPos(1, y + 1)
	end
end

local function printError( text )
	term.setForeground(0xff0000)
	print(text)
	term.setForeground(0xffffff)
end

local function hget(url, headers)
	local requestId = http.requestAsync(url, nil, headers, {binary = true})

	local ev, rId, par, info
	repeat
		ev, rId, par, info = event.pull("http_response", "http_failure")
	until rId == requestId

	if ev == "http_failure" then
		return nil, par
	end
	local content = par:read("a")
	par:close()
	return content, info
end

local function promptKey()
	print("Press any key to continue")
	event.pull("key_down")
end

local function installOS()
	term.clear()
	term.setPos(1, 1)

	print("Installing CapyOS...")

	local jsonLib, par = hget(JSON_URL)
	if not jsonLib then
		printError(par)
		promptKey()
		return
	end

	local json = load(jsonLib)()
	local indexData, par = hget(INDEX_URL)
	if not indexData then
		printError(par)
		promptKey()
		return
	end
	local index = json.decode(indexData)

	for i, v in ipairs(index) do
		local dirname = fs.getDir(v.path) 
		if not fs.exists(dirname) then
			fs.makeDir(dirname)
		end
		print("Downloading " .. v.path)
		local fileContent = hget(v.raw_url)
		local f = fs.open(v.path, "w")
		f:write(fileContent)
		f:close()
		print("Written to " .. v.path)
	end

	flagInstalled()

	print("CapyOS installed!")
	promptKey()
end

local function toggleConsole()
	local status = getConsole()
	setConsole(not status)
end

term.setBlink(false)

local function setupScreen()
	local options = {
		{
			"Open data folder",
			openDataFolder,
		},
		{
			"Toggle console window",
			toggleConsole,
		},
		{
			"Install default OS",
			installOS,
		},
		{
			"Exit setup",
			exit,
		},
		{
			"Shutdown",
			machine.shutdown,
		}
	}

	local selection = 1
	local function redraw()
		term.setForeground(fg)
		term.setBackground(bg)
		term.clear()
		term.setPos(1,2)
		writeCenter("Capy64 Setup")

		term.setPos(1,3)

		for k, v in ipairs(options) do
			local _, y = term.getPos()
			term.setPos(1, y + 1)
			term.clearLine()

			if selection == k then
				writeCenter("[ " .. v[1] .. " ]")
			else
				writeCenter(v[1])
			end
		end
	end

	while true do
		redraw()
		local ev = { coroutine.yield("key_down") }
		if ev[3] == "up" then
			selection = selection - 1
		elseif ev[3] == "down" then
			selection = selection + 1
		elseif ev[3] == "enter" then
			options[selection][2]()
		elseif ev[3] == "escape" then
			exit()
		end

		if selection > #options then
			selection = 1
		elseif selection < 1 then
			selection = #options
		end
	end

end

local function bootScreen()
	drawVendorImage()

	term.setPos(1,2)
	writeCenter("Capy64")
	term.setPos(1,4)
	writeCenter("Powered by Capybaras")

	term.setPos(1, h - 1)
	writeCenter("Press F2 to open setup")

	local timerId = timer.start(bootSleep)
	while true do
		local ev = {coroutine.yield("timer", "key_down")}
		if ev[1] == "timer" and ev[2] == timerId then
			exit()
		elseif ev[1] == "key_down" and ev[3] == "f2" then
			setupScreen()
			break
		end
	end
end

audio.beep(1000, 0.2, 0.2, "square")

if shouldInstallOS() then
	installOS()
end

bootScreen()

term.clear()