local installation = ""
local owner, repo = "kristify", "themes"
local tArgs = { ... }

-- Check update
local verPath = fs.combine(installation, "src", "version.txt")
local version = "0.0.0"
if fs.exists(verPath) then
  local file = fs.open(verPath, 'r')
  version = file.readAll()
  file.close()
end

local authenticate = _G._GIT_API_KEY and { Authorization = "Bearer " .. _G._GIT_API_KEY }
local gitAPI = http.get("https://raw.githubusercontent.com/Kristify/Kristify/main/src/version.txt", authenticate)
if gitAPI then
  local newV = gitAPI.readAll()
  if newV ~= version then
    term.setTextColor(colors.orange)
    term.write("Update available for Kristify! ")
    term.setTextColor(colors.lightGray)
    print(version .. " (current) --> " .. newV .. " (latest)")
    term.setTextColor(colors.white)
    print("Run \'kristify.lua -u\' or --update")
    sleep(0.8)
  end
  gitAPI.close()
end

-- Run Kristify normally
if #tArgs == 0 then
  local path = fs.combine(installation, "src", "init.lua")
  if not fs.exists(path) then
    error("Kristify is not installed correctly!")
  end

  if term.isColor() then
    local id = shell.openTab(path)
    multishell.setTitle(id, "Kristify")
    shell.switchTab(id)
  else
    shell.run(path)
  end
end

-- Install theme
if tArgs[1] == "--theme" or tArgs[1] == "-t" then
  -- Show current theme
  if not tArgs[2] or tArgs[2] == "" then
    local name, author = "Unknown", "Herobrine"
    local path = fs.combine(installation, "data", "credits.json")
    if fs.exists(path) then
      local file = fs.open(verPath, 'r')
      local data = file.readAll()
      data = textutils.unserialiseJSON(data) or {}
      name = data.name or "Unknown"
      author = data.author or "Herobrine"
      file.close()
    end

    term.setTextColor(colors.lightGray)
    term.write("Theme: ")
    term.setTextColor(colors.white)
    print(name .. " by " .. author)
  else
    -- Change theme
    local file = http.get("https://raw.githubusercontent.com/" .. owner .. '/' ..
      repo .. "/main/" .. tArgs[2] .. "/credits.json")
    if not file then printError("The given theme doesn't exist!") return end

    local data = file.readAll()
    data = textutils.unserialiseJSON(data)
    if not data then printError("The given theme doesn't exist!") return end
    local name = data.name
    local author = data.author
    file.close()

    print("Installing " .. name .. " by " .. author)

    local function httpError(response, err, errResponse)
      if not response then
        errors = true
        justinWeHaveAProblem("Request to GitHub denied; Reason: \'.." ..
          err .. "..\' (code " .. errResponse.getResponseCode() .. ").")
        return false
      end
      return true
    end

    local function getJSON(response)
      if not response then return {} end
      local tData = response.readAll()
      response.close()
      return textutils.unserialiseJSON(tData)
    end

    local function generateTree(name)
      sURL = "https://api.github.com/repos/" .. owner .. '/' .. repo .. "/contents/" .. name .. "?ref=main"
      local function convertItem(item)
        if item.type == "file" then
          return item.name, item.download_url
        elseif item.type == "dir" then
          return item.name, generateTree(item.url)
        end
      end

      local response, sErr, errResponse = http.get(sURL, authenticate)
      httpError(response, sErr, errResponse)
      local tData = getJSON(response)
      local tTree = {}
      for _, v in pairs(tData) do
        local sName, tItem = convertItem(v)
        -- Filter stuff that is not needed
        if not (sName:sub(1, 1) == '.' or sName:find(".md")) then
          tTree[sName] = tItem
        end
      end
      return tTree
    end

    local function downloadBlob(sURL, sPath)
      local response, sErr, errResponse = http.get(sURL, authenticate)
      if not httpError(response, sErr, errResponse) then
        return false
      end

      local sData = response.readAll()
      response.close()

      local file = fs.open(sPath, 'w')
      file.write(sData)
      file.close()

      return true
    end

    local theme = generateTree(name)
    local function downloadItems(tree, sPath)
      sleep(0.3)
      for name, item in pairs(tree) do
        local nextPath = fs.combine(sPath, name)
        if type(item) == "table" then
          downloadItems(item, nextPath)
        else
          downloadBlob(item, nextPath)
        end
      end
    end

    local path = fs.combine(installation, "data", "pages")
    fs.delete(path)
    downloadItems(theme, path)
  end

elseif tArgs[1] == "--version" or tArgs[1] == "-v" then
  print("Kristify v" .. version)
  term.write("GitHub: Kristify/Kristify made with ")
  term.setTextColor(colors.red)
  print("\003")
  term.setTextColor(colors.white)

elseif tArgs[1] == "--update" or tArgs[1] == "-u" then
  local path = fs.combine(installation, "data")
  if fs.exists(path) then
    fs.delete(".kristify_data_backup")
    fs.copy(path, ".kristify_data_backup")
  end

  -- Run installer
  if not http then
    error("Holdup. How- eh whatever. You need the http API!")
  end

  local authenticate = _G._GIT_API_KEY and { Authorization = "Bearer " .. _G._GIT_API_KEY }
  local response, err, errResp = http.get("https://raw.githubusercontent.com/Kristify/kristify/main/installer.lua",
    authenticate)

  if not response then
    error("Couldn't get the install script! Reason: \'" .. err .. "\' (code " .. errResp.getResponseCode() .. ')')
  end

  local content = response.readAll()
  response.close()

  local path = load(content, "install", 't', _ENV)()

  if fs.exists(".kristify_data_backup") then
    fs.delete(fs.combine(path, "data"))
    fs.copy(".kristify_data_backup", fs.combine(path, "data"))
    fs.delete(".kristify_data_backup")
  end

elseif tArgs[1] == "--storage" or tArgs[1] == "-s" then
  os.queueEvent("kstUpdateProducts")

elseif tArgs[1] == "--exit" or tArgs[1] == "-e" then
  os.queueEvent("kristify:exit")

elseif tArgs[1] == "--nbt" or tArgs[1] == "-n" then
  print("NBT Hash of item #1: ")

  local data = turtle.getItemDetail(1, true)
  assert(data, "No data gotten from slot one")

  print(data.nbt or "No NBT data. Leave the field to `nil` or don't define it.")
elseif tArgs[1] == "--help" or tArgs[1] == "-h" then
  print("Usage: " .. (tArgs[0] or "kristify.lua") .. " [flag:]")
  print("-u", "--update", "Updates Kristify.")
  print("-v", "--version", "Shoes the current version.")
  print("-t [name]", "--theme", "Shows or installs a given theme.")
  print("-s", "--storage", "Updates the storage.")
  print("-e", "--exit", "Stops the shop")
  print("-n", "--nbt", "Gets the NBT hash of the item in slot one")
end