local renderer = require("neo-tree.ui.renderer")
local buf_manager = require("codemp.buffers")
---@module 'nui.tree'

local M = {}

---@param workspace string workspace name
---@param path string buffer relative path
---@return NuiTree.Node
local function new_item(workspace, path)
	return {
		id = string.format("codemp://%s/item/%s", workspace, path),
		name = path,
		type = "buffer",
		extra = {},
		children = {},
	}
end

---@param workspace string workspace name
---@param username string user display name
---@return NuiTree.Node
local function new_user(workspace, username)
	return {
		id = string.format("codemp://%s@%s", username, workspace),
		name = username,
		type = "user",
		extra = {},
		children = {},
	}
end

---@param name string workspace name
---@param owned boolean true if this workspace is owned by us
---@param expanded? boolean if node should be pre-expanded
---@return NuiTree.Node
local function new_workspace(name, owned, expanded)
	local owned_ext = ""
	if owned then owned_ext = "_owned" end
	return {
		id = "codemp://" .. name .. "/workspace" .. owned_ext,
		name = name,
		type = "workspace",
		extra = {
			owned = owned,
		},
		children = {},
	}
end


---@param key string
---@param value string
---@return NuiTree.Node
local function new_entry(key, value)
	return {
		id = "codemp-entry-" .. key .. "-" .. value,
		name = key .. ": " .. value,
		type = "entry",
		extra = {},
	}
end

local function new_root(name)
	return {
		id = "codemp-tree-" .. name,
		name = name,
		type = "root",
		expanded = true,
		expand = true,
		extra = {},
		children = {}
	}
end

local function new_button(name)
	return {
		id = "codemp-button-" .. name,
		name = name,
		type = "button",
		extra = {},
		children = {}
	}
end

local counter = 0;

---@return NuiTree.Node
local function spacer()
	counter = counter + 1
	return {
		id = "codemp-ws-spacer-" .. counter,
		name = "",
		type = "spacer",
	}
end

local last_state = "N/A"

---@param tree NuiTree
local function expand(tree)
	---@param node? NuiTree.Node
	local function process(node)
		local id = nil
		if node ~= nil then id = node:get_id() end
		for _, node in ipairs(tree:get_nodes(id)) do
			node:expand()
			if node:has_children() then
				process(node)
			end
		end
	end

	process()
end


M.update_state = function(state)
	---@type NuiTree.Node[]
	local root = {
		{
			id = "codemp",
			name = "codemp",
			type = "title",
			extra = {},
		}
	}

	state.force_open_folders = true
	state.default_expanded_nodes = {}

	if CODEMP.workspace ~= nil then
		local ws_section = new_root("#" .. CODEMP.workspace:id())
		table.insert(state.default_expanded_nodes, ws_section.id)
		for i, path in ipairs(CODEMP.workspace:search_buffers()) do
			table.insert(ws_section.children, new_item(CODEMP.workspace:id(), path))
		end

		local usr_section = new_root("users")
		table.insert(state.default_expanded_nodes, usr_section.id)
		for user, buffer in pairs(buf_manager.users) do
			table.insert(usr_section.children, new_user(CODEMP.workspace:id(), user))
		end
		table.insert(ws_section.children, spacer())
		table.insert(ws_section.children, usr_section)

		table.insert(root, spacer())
		table.insert(root, ws_section)
	end

	if CODEMP.client ~= nil then
		local ws_section = new_root("workspaces")
		table.insert(state.default_expanded_nodes, ws_section.id)
		for _, ws in ipairs(CODEMP.available) do
			table.insert(ws_section.children, new_workspace(ws.name, ws.owned))
		end
		table.insert(root, spacer())
		table.insert(root, ws_section)

		local status_section = new_root("client")
		table.insert(state.default_expanded_nodes, status_section.id)
		table.insert(status_section.children, new_entry("id", CODEMP.client:current_user().id))
		table.insert(status_section.children, new_entry("name", CODEMP.client:current_user().name))

		table.insert(root, spacer())
		table.insert(root, status_section)
	end

	if CODEMP.client == nil then
		table.insert(root, spacer())
		table.insert(root, new_button("[connect]"))
	end

	renderer.show_nodes(root, state)
end

return M