require("advancedLua")
local web = require("web")
local component = require("component")
local buffer = require("doubleBuffering")
local filesystem = require("filesystem")
local unicode = require("unicode")
local image = require("image")
local color = require("color")
local MineOSPaths = require("MineOSPaths")

----------------------------------------------------------------------------------------------------------------

local MineOSCore = {}
MineOSCore.localization = {}

----------------------------------------------------------------------------------------------------------------

function MineOSCore.getCurrentScriptDirectory()
	return filesystem.path(getCurrentScript())
end

function MineOSCore.getCurrentApplicationResourcesDirectory() 
	return MineOSCore.getCurrentScriptDirectory() .. "/Resources/"
end

function MineOSCore.getLocalization(pathToLocalizationFolder)
	local localizationFileName = pathToLocalizationFolder .. MineOSCore.properties.language .. ".lang"
	if filesystem.exists(localizationFileName) then
		return table.fromFile(localizationFileName)
	else
		error("Localization file \"" .. localizationFileName .. "\" doesn't exists")
	end
end

function MineOSCore.getCurrentApplicationLocalization()
	return MineOSCore.getLocalization(MineOSCore.getCurrentApplicationResourcesDirectory() .. "Localization/")	
end

-----------------------------------------------------------------------------------------------------------------------------------

function MineOSCore.createShortcut(where, forWhat)
	filesystem.makeDirectory(filesystem.path(where))
	local file = io.open(where, "w")
	file:write(forWhat)
	file:close()
end

function MineOSCore.readShortcut(path)
	local file = io.open(path, "r")
	local data = file:read("*a")
	file:close()
	
	return data
end

-----------------------------------------------------------------------------------------------------------------------------------

function MineOSCore.saveProperties()
	table.toFile(MineOSPaths.properties, MineOSCore.properties, true)
end

function MineOSCore.loadPropeties()
	local saveLater = false

	if filesystem.exists(MineOSPaths.properties) then
		MineOSCore.properties = table.fromFile(MineOSPaths.properties)
	else
		MineOSCore.properties = {}
		saveLater = true
	end

	local defaultValues = {
		language = "Russian",
		transparencyEnabled = true,
		showApplicationIcons = true,
		iconHorizontalSpaceBetween = 1,
		iconVerticalSpaceBetween = 1,
		iconWidth = 12,
		iconHeight = 6,
		showExtension = false,
		wallpaper = MineOSPaths.pictures .. "Raspberry.pic",
		wallpaperMode = 2,
		wallpaperBrightness = 1.0,
		screensaver = "Matrix",
		screensaverDelay = 20,
		timezone = 3,		
		dockColor = 0xFFFFFF,
		menuColor = 0xE1E1E1,
		backgroundColor = 0x1E1E1E,
		dockShortcuts = {
			MineOSPaths.applications .. "AppMarket.app/",
			MineOSPaths.applications .. "MineCode IDE.app/",
			MineOSPaths.applications .. "Finder.app/",
			MineOSPaths.applications .. "Photoshop.app/",
			MineOSPaths.applications .. "Control.app/",
		},
		network = {
			users = {},
			enabled = true,
			signalStrength = 512,
		},
	}

	MineOSCore.associateExtension(".pic", MineOSPaths.applications .. "/Photoshop.app/Main.lua", MineOSPaths.icons .. "/Image.pic", MineOSPaths.extensionAssociations .. "Pic/ContextMenu.lua")
	MineOSCore.associateExtension(".txt", MineOSPaths.editor, MineOSPaths.icons .. "/Text.pic")
	MineOSCore.associateExtension(".cfg", MineOSPaths.editor, MineOSPaths.icons .. "/Config.pic")
	MineOSCore.associateExtension(".3dm", MineOSPaths.applications .. "/3DPrint.app/Main.lua", MineOSPaths.icons .. "/3DModel.pic")

	MineOSCore.associateExtension("script", MineOSPaths.extensionAssociations .. "Lua/Launcher.lua", MineOSPaths.icons .. "/Script.pic", MineOSPaths.extensionAssociations .. "Lua/ContextMenu.lua")
	MineOSCore.associateExtension(".lua", MineOSPaths.extensionAssociations .. "Lua/Launcher.lua", MineOSPaths.icons .. "/Lua.pic", MineOSPaths.extensionAssociations .. "Lua/ContextMenu.lua")
	MineOSCore.associateExtension(".arc", MineOSPaths.extensionAssociations .. "Arc/Launcher.lua", MineOSPaths.icons .. "/Archive.pic")

	for key, value in pairs(defaultValues) do
		if MineOSCore.properties[key] == nil then
			MineOSCore.properties[key] = value
			saveLater = true
		end
	end

	if saveLater then
		MineOSCore.saveProperties()
	end
end

-----------------------------------------------------------------------------------------------------------------------------------

function MineOSCore.associateExtensionLauncher(extension, pathToLauncher)
	MineOSCore.properties.extensionAssociations = MineOSCore.properties.extensionAssociations or {}
	MineOSCore.properties.extensionAssociations[extension] = MineOSCore.properties.extensionAssociations[extension] or {}
	MineOSCore.properties.extensionAssociations[extension].launcher = pathToLauncher
end

function MineOSCore.associateExtensionIcon(extension, pathToIcon)
	MineOSCore.properties.extensionAssociations[extension] = MineOSCore.properties.extensionAssociations[extension] or {}
	MineOSCore.properties.extensionAssociations[extension].icon = pathToIcon
end

function MineOSCore.associateExtensionContextMenu(extension, pathToContextMenu)
	MineOSCore.properties.extensionAssociations[extension] = MineOSCore.properties.extensionAssociations[extension] or {}
	MineOSCore.properties.extensionAssociations[extension].contextMenu = pathToContextMenu
end

function MineOSCore.associateExtension(extension, pathToLauncher, pathToIcon, pathToContextMenu)
	MineOSCore.associateExtensionLauncher(extension, pathToLauncher)
	MineOSCore.associateExtensionIcon(extension, pathToIcon)
	MineOSCore.associateExtensionContextMenu(extension, pathToContextMenu)
end

function MineOSCore.associationsExtensionAutomatically()
	local path, extension = MineOSPaths.extensionAssociations
	for file in filesystem.list(path) do
		if filesystem.isDirectory(path .. file) then
			extension = "." .. unicode.sub(file, 1, -2)

			if filesystem.exists(path .. file .. "ContextMenu.lua") then
				MineOSCore.associateExtensionContextMenu(extension, path .. file .. "Context menu.lua")
			end

			if filesystem.exists(path .. file .. "Launcher.lua") then
				MineOSCore.associateExtensionLauncher(extension, path .. file .. "Launcher.lua")
			end
		end
	end
end

-----------------------------------------------------------------------------------------------------------------------------------

--Функция парсинга Lua-сообщения об ошибке. Конвертирует из строки в массив.
function MineOSCore.parseErrorMessage(error, indentationWidth)
	local parsedError = {}

	--Замена /r/n и табсов
	error = string.gsub(error, "\r\n", "\n")
	error = string.gsub(error, "	", string.rep(" ", indentationWidth or 4))

	--Удаление энтеров
	local searchFrom, starting, ending = 1
	for i = 1, unicode.len(error) do
		starting, ending = string.find(error, "\n", searchFrom)
		if starting then
			table.insert(parsedError, unicode.sub(error, searchFrom, starting - 1))
			searchFrom = ending + 1
		else
			break
		end
	end

	--На всякий случай, если сообщение об ошибке без энтеров вообще, т.е. однострочное
	if #parsedError == 0 then table.insert(parsedError, error) end

	return parsedError
end

function MineOSCore.call(method, ...)
	local args = {...}
	local function launchMethod()
		method(table.unpack(args))
	end

	local function tracebackMethod(xpcallTraceback)
		local traceback, info, firstMatch = tostring(xpcallTraceback) .. "\n" .. debug.traceback()
		for runLevel = 0, math.huge do
			info = debug.getinfo(runLevel)
			if info then
				if (info.what == "main" or info.what == "Lua") and info.source ~= "=machine" then
					if firstMatch then
						return {
							path = info.source:sub(2, -1),
							line = info.currentline,
							traceback = traceback
						}
					else
						firstMatch = true
					end
				end
			else
				error("Failed to get debug info for runlevel " .. runLevel)
			end
		end
	end
	
	local xpcallSuccess, xpcallReason = xpcall(launchMethod, tracebackMethod)
	if type(xpcallReason) == "string" or type(xpcallReason) == "nil" then xpcallReason = {path = "/lib/MineOSCore.lua", line = 1, traceback = "MineOSCore fatal error: " .. tostring(xpcallReason)} end
	if not xpcallSuccess and not xpcallReason.traceback:match("^table") and not xpcallReason.traceback:match("interrupted") then
		return false, xpcallReason.path, xpcallReason.line, xpcallReason.traceback
	end

	return true
end

function MineOSCore.safeLaunch(path, ...)
	path = path:gsub("/+", "/") 
	MineOSCore.lastLaunchPath = path

	local oldResolutionWidth, oldResolutionHeight = buffer.getResolution()
	local finalSuccess, finalPath, finalLine, finalTraceback = true
	
	if filesystem.exists(path) then
		local loadSuccess, loadReason = loadfile("/" .. path)
		if loadSuccess then
			local success, path, line, traceback = MineOSCore.call(loadSuccess, ...)
			if not success then
				finalSuccess, finalPath, finalLine, finalTraceback = false, path, line, traceback
			end
		else
			local match = string.match(loadReason, ":(%d+)%:")
			finalSuccess, finalPath, finalLine, finalTraceback = false, path, tonumber(match) or 1, loadReason
		end
	else
		require("GUI").error("Failed to safely launch file that doesn't exists: \"" .. path .. "\"")
	end

	component.screen.setPrecise(false)
	buffer.setResolution(oldResolutionWidth, oldResolutionHeight)

	return finalSuccess, finalPath, finalLine, finalTraceback
end

-----------------------------------------------------------------------------------------------------------------------------------

function MineOSCore.downloadApplication(application, language, createShortcut)
    if application.type == "Application" then
		filesystem.remove(application.path .. ".app")

		web.download(application.url, application.path .. ".app/Main.lua")
		web.download(application.icon, application.path .. ".app/Resources/Icon.pic")

		if application.resources then
			for i = 1, #application.resources do
				web.download(application.resources[i].url, application.path .. ".app/Resources/" .. application.resources[i].path)
			end
		end

		if application.about then
			web.download(application.about .. language .. ".txt", application.path .. ".app/Resources/About/" .. language .. ".txt")
		end 

		if application.createShortcut or createShortcut then
			MineOSCore.createShortcut(MineOSPaths.desktop .. filesystem.name(application.path) .. ".lnk", application.path .. ".app/")
		end
	else
		web.download(application.url, application.path)
	end
end

-----------------------------------------------------------------------------------------------------------------------------------

function MineOSCore.loadImageFromURL(url)
	local bytes = web.request(url)
	bytes = { string.byte(bytes, 1, #bytes) }

	local signature = string.char(bytes[1], bytes[2], bytes[3], bytes[4])
	if signature == "OCIF" then
		local encodingMethod = bytes[5]
		if encodingMethod == 6 then
			local width, height = bytes[6], bytes[7]
			local picture = {width, height}

			local i = 8
			while i <= #bytes do
				local alphaSize = bytes[i]
				i = i + 1

				for a = 1, alphaSize do
					local alpha = bytes[i] / 255
					local symbolSize = bit32.bor(bit32.lshift(bytes[i + 1], 8), bytes[i + 2])
					i = i + 3

					for s = 1, symbolSize do
						local utf8CharSize = 1
						for j = 1, 7 do
							if bit32.band(bit32.rshift(bytes[i], 8 - j), 0x1) == 0x0 then
								utf8CharSize = j
								break
							end
						end   

						local symbol
						if utf8CharSize == 1 then
							symbol = string.char(bytes[i])
							i = i + 1
						else
							symbol = string.char(table.unpack(bytes, i, i + utf8CharSize - 2))
							i = i + utf8CharSize - 1
						end

						local backgroundSize = bytes[i]
						i = i + 1
						
						for b = 1, backgroundSize do
							local background = color.to24Bit(bytes[i])
							local foregroundSize = bytes[i + 1]
							i = i + 2

							for f = 1, foregroundSize do
								local foreground = color.to24Bit(bytes[i])
								local ySize = bytes[i + 1]
								i = i + 2

								for ys = 1, ySize do
									local y = bytes[i]
									local xSize = bytes[i + 1]
									i = i + 2

									for xs = 1, xSize do
										image.set(picture, bytes[i], y, background, foreground, alpha, symbol)
										i = i + 1
									end
								end
							end
						end
					end
				end
			end

			return picture
		else
			return false, "Unsupported encoding method: " .. encodingMethod
		end
	else
		return false, "Unsupported signature: " .. signature
	end
end

-----------------------------------------------------------------------------------------------------------------------------------

MineOSCore.loadPropeties()

-----------------------------------------------------------------------------------------------------------------------------------

return MineOSCore