pinestore-cc

local apiPath = "https://pinestore.cc/api/"

if not fs.exists("installed") then
    fs.makeDir("installed")
end

local width, height = term.getSize()
local renderWindow = window.create(term.current(), 1, 1, width, height)
local oldTerm = term.redirect(renderWindow)
local maxWidth = 70

local function updateTermSize()
    width, height = oldTerm.getSize()
    local newW = math.min(width, maxWidth)
    renderWindow.reposition(1 + math.floor((width - newW)*0.5 + 0.5), 1, newW, height)
    oldTerm.setBackgroundColor(colors.black)
    oldTerm.clear()
end
updateTermSize()

local online = true

local categories = {
    "fun",
    "tools",
    "turtle",
    "audio",
    "other",
}
local selectedCategory = 1
local selectedProject = 1

local projectActionsOpened = false
local projectActionSelected = 1
local searchOpened = false
local searchQuery = ""
local searchResultsOpened = false

local function getAPI(path)
    local res = http.get(apiPath .. path)
    if not res then
        online = false
        return
    end
    local data = res.readAll()
    res.close()
    return textutils.unserialiseJSON(data)
end

function postAPI(path, body)
    local res = http.post(apiPath .. path, textutils.serialiseJSON(body), {["Content-Type"] = "application/json"})
    if not res then
        return
    end
    local data = res.readAll()
    res.close()
    return textutils.unserialiseJSON(data)
end

local installedInfo = {
    projects = {},
}
if fs.exists("installed.json") then
    local h = fs.open("installed.json", "r")
    installedInfo = textutils.unserialiseJSON(h.readAll())
    h.close()
end
local function saveInstalled()
    local encoded = textutils.serialiseJSON(installedInfo)
    local h = fs.open("installed.json", "w")
    h.write(encoded)
    h.close()
end

local searchResults = {}
local projectsData = getAPI("projects")
local projects = projectsData.projects
if not projects then
    local ps = {}
    for id, project in pairs(installedInfo.projects) do
        ps[#ps+1] = project
    end
    projects = ps
else
    for i = 1, #projects do
        local project = projects[i]
        if installedInfo.projects[tostring(project.id)] then
            installedInfo.projects[tostring(project.id)].downloads = project.downloads
        end
    end
    saveInstalled()
end
for i = #projects, 1, -1 do
    local project = projects[i]
    if not project.install_command or not project.target_file then
        table.remove(projects, i)
    end
end

local categoryProjects = {}
for i = 1, #projects do
    local project = projects[i]
    if not project.category then project.category = "Other" end
    if not categoryProjects[project.category] then
        categoryProjects[project.category] = {}
    end
    categoryProjects[project.category][#categoryProjects[project.category]+1] = project
end
for category, projectList in pairs(categoryProjects) do
    table.sort(projectList, function(a, b)
        return b.downloads < a.downloads
    end)
end

local function installProject(project)
    -- redirect term to old one
    term.redirect(oldTerm)

    -- override fs methods
    local projectPath = "installed/" .. project.id .. "/"
    fs.makeDir(projectPath)
    local oldFSOpen = fs.open
    local oldFSMakeDir = fs.makeDir
    local oldFSExists = fs.exists
    function fs.open(path, mode)
        -- print("open " .. path)
        -- sleep(0.5)
        if path:sub(1, 12) == "rom/programs" then
            return oldFSOpen(path, mode)
        end
        return oldFSOpen(projectPath .. path, mode)
    end
    function fs.makeDir(path)
        return oldFSMakeDir(projectPath .. path)
    end
    function fs.exists(path)
        if path:sub(1, 12) == "rom/programs" then
            return oldFSExists(path)
        end
        return oldFSExists(projectPath .. path)
    end

    -- actually run the install command
    local success, res = xpcall(shell.run, debug.traceback, project.install_command)

    -- return old fs methods
    fs.open = oldFSOpen
    fs.makeDir = oldFSMakeDir
    fs.exists = oldFSExists

    -- use render window again
    oldTerm = term.redirect(renderWindow)

    updateTermSize()

    if success then
        -- set project info to installed
        installedInfo.projects[tostring(project.id)] = project
        saveInstalled()
        postAPI("newdownload", {projectId = project.id})
    else
        error(res)
    end
end

local function startProject(project)
    -- override fs methods
    local projectPath = "installed/" .. project.id .. "/"
    local oldFSOpen = fs.open
    local oldFSMakeDir = fs.makeDir
    local oldFSExists = fs.exists
    local oldFSList = fs.list
    function fs.open(path, mode)
        if path:sub(1, 12) == "rom/programs" then
            return oldFSOpen(path, mode)
        end
        return oldFSOpen(projectPath .. path, mode)
    end
    function fs.makeDir(path)
        return oldFSMakeDir(projectPath .. path)
    end
    function fs.exists(path)
        return oldFSExists(projectPath .. path)
    end
    function fs.list(path)
        return oldFSList(projectPath .. path)
    end

    term.redirect(oldTerm)
    local success, res = xpcall(function()
        local success = shell.run(project.target_file)

        if not success then
            sleep(1)
            term.setTextColor(colors.white)
            print("\nPress any key to continue...")
            os.pullEvent("key")
        end
    end, debug.traceback)

    -- return old fs methods
    fs.open = oldFSOpen
    fs.makeDir = oldFSMakeDir
    fs.exists = oldFSExists
    fs.list = oldFSList

    if not success then
        if res:sub(1, 10) ~= "Terminated" then
            term.setBackgroundColor(colors.black)
            term.setTextColor(colors.red)
            term.clear()
            term.setCursorPos(1, 1)
            print(res)
            term.setTextColor(colors.white)
            sleep(1)
            print("\nPress any key to continue...")
            os.pullEvent("key")
        end
    end

    term.redirect(renderWindow)

    updateTermSize()
end

local function runSearch()
    local searchData = getAPI("search?q=" .. textutils.urlEncode(searchQuery))
    searchResults = searchData.projects
    searchOpened = false
    searchResultsOpened = true
    selectedProject = 1
    searchQuery = ""
end

local function drawCategories()
    term.setCursorPos(1, 2)
    for nr = 1, #categories do
        if nr == selectedCategory then
            term.setTextColor(colors.lime)
        else
            term.setTextColor(colors.green)
        end

        local str = "[" .. nr .. " " .. categories[nr] .. "] "
        local x, y = term.getCursorPos()
        if x + #str - 2 > width then
            print("")
        end
        term.write(str)
    end
end

local function drawProjects(projects)
    print("")
    local startX, startY = term.getCursorPos()
    local width, height = term.getSize()
    local linesAvailable = height - startY - 1
    local startI = math.max(1, selectedProject - linesAvailable + 1)
    for i = startI, math.min(startI + linesAvailable, #projects) do
        if i == selectedProject then
            term.setTextColor(colors.yellow)
            term.write("> ")
        else
            term.write("  ")
        end

        local project = projects[i]
        if installedInfo.projects[tostring(project.id)] then
            term.setTextColor(colors.lightBlue)
        else
            term.setTextColor(colors.white)
        end
        term.write(project.name)

        term.setTextColor(colors.lightGray)
        term.write(" by " .. project.owner_name)

        local showDownloads = true
        local downText = " downloads"
        if project.downloads == 1 then
            downText = " download "
        end
        if width <= 25 then
            showDownloads = false
        elseif width <= 29 then
            downText = ""
        elseif width <= 42 then
            downText = " dls"
            if project.downloads == 1 then
                downText = " dl "
            end
        end
        if showDownloads then
            local downloadsText = project.downloads .. downText
            local x, y = term.getCursorPos()
            term.setCursorPos(width - #downloadsText + 1, y)
            print(downloadsText)
        else
            print("")
        end
    end
end

local function drawProjectOptions()
    print("")

    local projects = categoryProjects[categories[selectedCategory]] or {}
    if searchResultsOpened then
        projects = searchResults
    end
    local project = projects[selectedProject]
    local installed = installedInfo.projects[tostring(project.id)]

    for i = 1, installed and 3 or 2 do
        if i == projectActionSelected then
            term.setTextColor(colors.yellow)
            term.write("> ")
        else
            term.write("  ")
        end

        term.setTextColor(colors.white)
        if installed then
            if i == 1 then
                if project.version > installed.version then
                    print("Update and run")
                else
                    print("Run")
                end
            elseif i == 2 then
                print("Uninstall")
            elseif i == 3 then
                print("Back")
            end
        else
            if i == 1 then
                print("Install")
            elseif i == 2 then
                print("Back")
            end
        end
    end

    print("")

    term.setTextColor(colors.white)
    term.write(project.name)
    term.setTextColor(colors.lightGray)
    print(" by " .. project.owner_name)
    term.setTextColor(colors.gray)
    print("v" .. project.version .. " " .. os.date("%B %d, %Y", math.max(project.date_added, project.date_updated) / 1000))
    print("")
    term.setTextColor(colors.lime)

    local stoppedScroll = false
    local scrollOld = term.scroll
    function term.scroll()
        stoppedScroll = true
    end
    print(project.description)
    term.scroll = scrollOld
    if stoppedScroll then
        local width, height = term.getSize()
        term.setCursorPos(1, height)
        term.clearLine()
    end
end

local function selectProjectAction()
    local projects = categoryProjects[categories[selectedCategory]] or {}
    if searchResultsOpened then
        projects = searchResults
    end
    local project = projects[selectedProject]
    local installed = installedInfo.projects[tostring(project.id)]
    local i = projectActionSelected

    if installed then
        if i == 1 then
            if project.version > installed.version then
                installProject(project) -- run install again to update
            end

            startProject(project)
            projectActionsOpened = false
        elseif i == 2 then
            fs.delete("installed/" .. project.id)
            installedInfo.projects[tostring(project.id)] = nil
            saveInstalled()
            projectActionsOpened = false
        elseif i == 3 then
            projectActionsOpened = false
        end
    else
        if i == 1 then
            installProject(project)
        elseif i == 2 then
            projectActionsOpened = false
        end
    end
end

local function drawSearch()
    term.setTextColor(colors.lightGray)
    print("\n\nSearch query:")
    term.setTextColor(colors.white)
    local lineCount = math.ceil(#searchQuery / width)
    for i = 1, lineCount do
        term.write(searchQuery:sub(1 + width*(i-1), width + width*(i-1)))
        if i < lineCount then
            local x, y = term.getCursorPos()
            term.setCursorPos(1, y+1)
        end
    end
    term.setCursorBlink(true)
end

local function drawSearchResults()
    term.setTextColor(colors.white)
    print("\n\nSearch results:")
    drawProjects(searchResults)
end

local function drawMain()
    term.setBackgroundColor(colors.black)
    term.clear()

    term.setTextColor(colors.yellow)
    term.setCursorPos(1, 1)
    local width, height = term.getSize()
    if width > 37 then
        term.write("PineStore Console v1.0")
    elseif width > 31 then
        term.write("PineStore Console")
    elseif width > 25 then
        term.write("PineStore C.")
    else
        term.write("PStore C.")
    end
    if not online then
        term.setTextColor(colors.orange)
        term.write(" offline mode")
    end
    term.setCursorBlink(false)

    if searchOpened then
        local searchText = "[TAB close]"
        term.setCursorPos(width - #searchText + 1, 1)
        term.write(searchText)

        drawSearch()
    elseif projectActionsOpened then
        drawProjectOptions()
    elseif searchResultsOpened then
        drawSearchResults()
    else
        if online then
            local searchText = "[TAB search]"
            term.setCursorPos(width - #searchText + 1, 1)
            term.write(searchText)
        end

        drawCategories()
        local projects = categoryProjects[categories[selectedCategory]] or {}
        drawProjects(projects)
    end

    renderWindow.setVisible(true)
    renderWindow.setVisible(false)
end

function runStore()
    while true do
        drawMain()

        -- if searchOpened then
        --  local searchText = "[TAB close]"
        --  term.setCursorPos(width - #searchText, 1)
        --  term.write(searchText)

        --  drawSearch()
        -- elseif searchResultsOpened then
        --  drawSearchResults()
        -- elseif projectActionsOpened then
        --  drawProjectOptions()
        -- else

        local event, key, x, y = os.pullEvent()
        if event == "key" then
            if searchOpened then
                if key == keys.tab then
                    searchOpened = false
                elseif key == keys.enter then
                    runSearch()
                elseif key == keys.backspace then
                    searchQuery = searchQuery:sub(1, -2)
                end
            elseif projectActionsOpened then
                if key == keys.up or key == keys.w then
                    projectActionSelected = math.max(1, projectActionSelected - 1)
                elseif key == keys.down or key == keys.s then
                    local projects = categoryProjects[categories[selectedCategory]] or {}
                    if searchResultsOpened then
                        projects = searchResults
                    end
                    local project = projects[selectedProject]
                    local installed = installedInfo.projects[tostring(project.id)]
                    projectActionSelected = math.max(1, math.min(installed and 3 or 2, projectActionSelected + 1))
                elseif key == keys.backspace or key == keys.grave then
                    projectActionsOpened = false
                elseif key == keys.enter or key == keys.space then
                    selectProjectAction()
                end
            elseif searchResultsOpened then
                if key == keys.up or key == keys.w then
                    selectedProject = math.max(1, selectedProject - 1)
                elseif key == keys.down or key == keys.s then
                    selectedProject = math.max(1, math.min(#searchResults, selectedProject + 1))
                elseif key == keys.enter or key == keys.space then
                    if #searchResults > 0 then
                        projectActionsOpened = true
                        projectActionSelected = 1
                    end
                elseif key == keys.backspace or key == keys.grave or key == keys.tab then
                    searchResultsOpened = false
                end
            else
                if key >= keys.one and key <= keys.nine then
                    selectedCategory = math.max(1, math.min(#categories, key - keys.one + 1))
                elseif key == keys.left or key == keys.a then
                    selectedCategory = math.max(1, selectedCategory - 1)
                elseif key == keys.right or key == keys.d then
                    selectedCategory = math.max(1, math.min(#categories, selectedCategory + 1))
                elseif key == keys.up or key == keys.w then
                    selectedProject = math.max(1, selectedProject - 1)
                elseif key == keys.down or key == keys.s then
                    local projects = categoryProjects[categories[selectedCategory]] or {}
                    selectedProject = math.max(1, math.min(#projects, selectedProject + 1))
                elseif key == keys.enter or key == keys.space then
                    projectActionsOpened = true
                    projectActionSelected = 1
                elseif key == keys.tab and online then
                    searchOpened = true
                end
            end
        elseif event == "char" then
            if key ~= "`" then
                searchQuery = searchQuery .. key
            end
        elseif event == "term_resize" then
            updateTermSize()
        end
    end
end

local success, res = xpcall(runStore, debug.traceback)
term.redirect(oldTerm)
if not success then
    if type(res) == "string" and res:sub(1, 10) == "Terminated" then
        return
    end
    term.setBackgroundColor(colors.black)
    term.setTextColor(colors.red)
    term.clear()
    term.setCursorPos(1, 1)
    print(res)
    term.setTextColor(colors.white)
end