local state = require("gitlab.state")

local colors = state.settings.colors

-- Set icons into global vim variables for syntax matching
local expanders = state.settings.discussion_tree.expanders
vim.g.gitlab_discussion_tree_expander_open = expanders.expanded
vim.g.gitlab_discussion_tree_expander_closed = expanders.collapsed
vim.g.gitlab_discussion_tree_draft = ""
vim.g.gitlab_discussion_tree_resolved = "✓"
vim.g.gitlab_discussion_tree_unresolved = "-"

local discussion = colors.discussion_tree

local function get_colors_for_group(group)
  local normal_fg = vim.fn.synIDattr(vim.fn.synIDtrans(vim.fn.hlID(group)), "fg")
  local normal_bg = vim.fn.synIDattr(vim.fn.synIDtrans(vim.fn.hlID(group)), "bg")
  return { fg = normal_fg, bg = normal_bg }
end

vim.api.nvim_create_autocmd("VimEnter", {
  callback = function()
    vim.api.nvim_set_hl(0, "GitlabUsername", get_colors_for_group(discussion.username))
    vim.api.nvim_set_hl(0, "GitlabMention", get_colors_for_group(discussion.mention))
    vim.api.nvim_set_hl(0, "GitlabDate", get_colors_for_group(discussion.date))
    vim.api.nvim_set_hl(0, "GitlabExpander", get_colors_for_group(discussion.expander))
    vim.api.nvim_set_hl(0, "GitlabDirectory", get_colors_for_group(discussion.directory))
    vim.api.nvim_set_hl(0, "GitlabDirectoryIcon", get_colors_for_group(discussion.directory_icon))
    vim.api.nvim_set_hl(0, "GitlabFileName", get_colors_for_group(discussion.file_name))
    vim.api.nvim_set_hl(0, "GitlabResolved", get_colors_for_group(discussion.resolved))
    vim.api.nvim_set_hl(0, "GitlabUnresolved", get_colors_for_group(discussion.unresolved))
    vim.api.nvim_set_hl(0, "GitlabDraft", get_colors_for_group(discussion.draft))
  end,
})