
local custom_highlights = function(colors)
    return {
        CursorLine = { bg = colors.surface0 },
        SignColumnSep = { bg = colors.surface0, fg = colors.surface1 },

        Visual = { bg = colors.surface1 },
        FoldColumn = { fg = colors.surface2 },

        -- Floating windows
        NormalFloat = { fg = colors.text, bg = colors.crust },
        FloatTitle = { fg = colors.base, bg = colors.blue },
        FloatBorder = { fg = colors.surface1, bg = colors.crust },

        -- Window separator
        WinSeparator = { fg = colors.surface0 },

        -- Menus
        Pmenu = { link = "NormalFloat" },
        PmenuSel = { bg = colors.surface2 },
        PmenuSbar = { link = "Pmenu" },
        PmenuThumb = { link = "PmenuSel" },

        -- NoiceMini = { link = "NormalFloat" },
        WhichKeyFloat = { link = "Pmenu" },

        -- indent lines
        IblIndent = { link = "NonText" },

        -- Search matches
        IncSearch = { bg = colors.yellow },

        -- Autocomplete window
        CmpItemAbbr = { fg = colors.overlay2 },
        CmpItemKindText = { fg = colors.text },

        -- CmpKind
        CmpItemKindBladeNav = { fg = "#f55247" },

        -- telescope
        TelescopeNormal = { link = "NormalFloat" },
        TelescopeBorder = { link = "FloatBorder" },
        TelescopeTitle = { link = "FloatTitle" },
        TelescopePromptNormal = { bg = colors.mantle },
        TelescopePromptPrefix = { fg = colors.mauve },
        TelescopePromptBorder = { fg = colors.mantle, bg = colors.mantle },
        TelescopePreviewTitle = { fg = colors.crust, bg = colors.mauve },
        TelescopeResultsNormal = { fg = colors.text, bg = colors.crust },
        TelescopeResultsClass = { link = "CmpItemKindClass" },
        TelescopeResultsInterface = { link = "CmpItemKindInterface" },
        TelescopeResultsField = { link = "CmpItemKindField" },
        TelescopeResultsFunction = { link = "CmpItemKindFunction" },
        TelescopeResultsMethod = { link = "CmpItemKindMethod" },
        TelescopeResultsNumber = { link = "CmpItemKindNumber" },
        TelescopeResultsStruct = { link = "CmpItemKindStruct" },
        TelescopeResultsConstant = { link = "CmpItemKindConstant" },
        TelescopeResultsOperator = { link = "CmpItemKindOperator" },
        TelescopeResultsVariable = { link = "CmpItemKindVariable" },
        TelescopeResultsIdentifier = { link = "CmpItemKindIdentifier" },
        TelescopeMatching = { fg = colors.lavender },
        TelescopeSelection = { fg = colors.text, bg = colors.surface1 },
        TelescopeIndicatorModified = { fg = colors.yellow },
        TelescopeIndicatorReadonly = { fg = colors.red },
        TelescopeIndicatorHidden = { link = "TelescopeResultsComment" },

        -- Snacks picker
        SnacksPickerInput = { link = "TelescopePromptNormal" },
        SnacksPickerInputBorder = { link = "TelescopePromptBorder" },
        SnacksPickerResult = {  link = "TelescopeResultsNormal" },

        -- Ufo
        UfoFoldedEllipsis = { link = "Comment" },

        -- Statusline

        -- Something resets StatusLine highlight background, so use another one for now :)
        -- StatusLine = { fg = colors.text, bg = colors.crust },
        StatusLineLualine = { fg = colors.text, bg = colors.crust },

        StatusLineNormal = { link = "StatusLineLualine" },
        StatusLineSeparator = { fg = colors.rosewater, bg = colors.crust },
        StatusLineInsert = { fg = colors.base, bg = colors.blue },
        StatusLineVisual = { fg = colors.base, bg = colors.mauve },
        StatusLineCommand = { fg = colors.base, bg = colors.yellow },
        StatusLineReplace = { fg = colors.base, bg = colors.maroon },

        -- Notifications
        Notification = { bg = colors.red },

        -- Dashboard
        SnacksDashboardIcon = { fg = colors.lavender },
        SnacksDashboardDesc = { fg = colors.text },
        SnacksDashboardKey = { fg = colors.lavender },

        -- Neotree
        NeoTreeTabActive = { bg = colors.surface2, fg = colors.lavender },
        NeoTreeTabInactive = { bg = colors.crust, fg = colors.surface2 },
        NeoTreeTabSeparatorActive = { fg = colors.base, bg = colors.surface2 },
        NeoTreeTabSeparatorInactive = { fg = colors.base, bg = colors.crust },
        NeoTreeFileIcon = { link = "Normal" },
        NeoTreeModified = { fg = colors.yellow },
        NeoTreeWinSeparator = { link = "WinSeparator" },

        -- NvimTree
        NvimTreeNormal = { bg = colors.mantle },
        NvimTreeWinSeparator = { bg = colors.mantle, fg = colors.surface0 },
        NvimTreeOpenedFile = { fg = colors.lavender },
        NvimTreeSpecialFile = { fg = colors.peach },
        NvimTreeGitStaged = { fg = colors.green },
        NvimTreeGitDirty = { fg = colors.yellow },

        -- BlinkCmp
        BlinkCmpLabel = { link = 'CmpItemAbbr' },
        BlinkCmpLabelMatch = { link = 'CmpItemAbbrMatch' },
        BlinkCmpLabelDeprecated = { link = 'CmpItemAbbrDeprecated' },
        BlinkCmpLabelDetail = { link = 'CmpItemMenu' },
        BlinkCmpLabelDescription = { link = 'CmpItemMenu' },

        BlinkCmpSource = { link = 'CmpItemMenu' },
        BlinkCmpKind = { link = 'CmpItemKind' },
        BlinkCmpKindText = { link = "CmpItemKindText" },
        BlinkCmpKindMethod = { link = "CmpItemKindMethod" },
        BlinkCmpKindFunction = { link = "CmpItemKindFunction" },
        BlinkCmpKindConstructor = { link = "CmpItemKindConstructor" },
        BlinkCmpKindField = { link = "CmpItemKindField" },
        BlinkCmpKindVariable = { link = "CmpItemKindVariable" },
        BlinkCmpKindClass = { link = "CmpItemKindClass" },
        BlinkCmpKindInterface = { link = "CmpItemKindInterface" },
        BlinkCmpKindModule = { link = "CmpItemKindModule" },
        BlinkCmpKindProperty = { link = "CmpItemKindProperty" },
        BlinkCmpKindUnit = { link = "CmpItemKindUnit" },
        BlinkCmpKindValue = { link = "CmpItemKindValue" },
        BlinkCmpKindEnum = { link = "CmpItemKindEnum" },
        BlinkCmpKindKeyword = { link = "CmpItemKindKeyword" },
        BlinkCmpKindSnippet = { link = "CmpItemKindSnippet" },
        BlinkCmpKindColor = { link = "CmpItemKindColor" },
        BlinkCmpKindFile = { link = "CmpItemKindFile" },
        BlinkCmpKindReference = { link = "CmpItemKindReference" },
        BlinkCmpKindFolder = { link = "CmpItemKindFolder" },
        BlinkCmpKindEnumMember = { link = "CmpItemKindEnumMember" },
        BlinkCmpKindConstant = { link = "CmpItemKindConstant" },
        BlinkCmpKindStruct = { link = "CmpItemKindStruct" },
        BlinkCmpKindEvent = { link = "CmpItemKindEvent" },
        BlinkCmpKindOperator = { link = "CmpItemKindOperator" },
        BlinkCmpKindTypeParameter = { link = "CmpItemKindTypeParameter" },

        -- DAP
        DapStopped = { fg = colors.yellow },

        -- Neotest
        NeoTestWinSelect = { fg = colors.mauve },
        NeoTestIndent = { fg = colors.surface1 },
        NeoTestExpandMarker = { link = "NeoTestIndent" },
        NeoTestAdapterName = { fg = colors.mauve },
        NeoTestFile = { fg = colors.lavender },
        NeoTestDir = { fg = colors.blue },
        NeoTestTarget = { fg = colors.blue },
        NeoTestNamespace = { fg = colors.mauve },
        NeoTestPassed = { fg = colors.green },
        NeoTestFailed = { fg = colors.red },
        NeoTestSkipped = { fg = colors.blue },
        NeoTestRunning = { fg = colors.yellow },
        NeoTestWatching = { fg = colors.yellow },
        NeoTestMarked = { fg = colors.orange },

        -- Diagnostics
        DiagnosticUnderlineInfo = { style = { "undercurl" } },
        DiagnosticUnderlineHint = { style = { "undercurl" } },
        DiagnosticUnderlineWarn = { style = { "undercurl" } },
        DiagnosticUnderlineError = { style = { "undercurl" } },

        -- Markdown rendering
        RenderMarkdownCode = { bg = colors.crust },
        RenderMarkdownCodeInline = { fg = colors.rosewater, bg = colors.crust },

        -- todo comments
        TodoCommentsDefault = { fg = colors.flamingo },
        TodoCommentsInfo = { fg = colors.blue },
        TodoCommentsError = { fg = colors.maroon },
        TodoCommentsWarning = { fg = colors.peach },
        TodoCommentsHint = { fg = colors.sapphire },
        TodoCommentsTest = { fg = colors.pink },

        -- Text
        NonText = { fg = colors.surface0 },
        TextMute = { fg = colors.overlay2 },
        Whitespace = { link = "NonText" },

        -- Syntax
        PreProc = { fg = colors.mauve },
        Operator = { fg = colors.flamingo },
        Function = { link = "@text" },
        Delimiter = { link = "Operator" },
        Keyword = { fg = colors.yellow },
        Include = { link = "Keyword" },
        Repeat = { link = "Keyword" },
        Conditional = { link = "Keyword" },
        Type = { fg = colors.blue },
        Label = { link = "String" },
        String = { fg = colors.lavender },
        Character = { link = "String" },
        Exception = { link = "Keyword" },
        Macro = { fg = colors.pink },

        -- Treesitter tokens
        ["@constructor"] = { link = "Function" },
        ["@variable"] = { fg = colors.green },
        ["@variable.builtin"] = { link = "@variable" },
        ["@variable.parameter"] = { link = "@parameter" },
        ["@variable.member"] = { link = "@variable" },
        ["@parameter"] = { link = "@variable" },
        ["@keyword.function"] = { link = "Keyword" },
        ["@keyword.return"] = { link = "Keyword" },
        ["@keyword.operator"] = { link = "Keyword" },
        ["@property"] = { link = "@variable" },
        ["@type.builtin"] = { fg = colors.yellow },
        ["@tag"] = { link = "Keyword" },
        ["@tag.delimiter"] = { link = "@text" },
        ["@module"] = { link = "@text" },
        ["@namespace"] = { link = "@keyword" },
        ["@punctuation.bracket"] = { link = "@punctuation" },
        ["@function.macro"] = { link = "Macro" },

        -- Comments
        ["@punctuation.delimiter.comment"] = { link = "@Comment" },
        ["@comment.error"] = { link = "@Comment" },
        ["@comment.warning"] = { link = "@Comment" },
        ["@comment.todo"] = { link = "@Comment" },
        ["@comment.note.comment"] = { link = "@Comment" },
        ["@number.comment"] = { fg = colors.green },

        -- LSP
        ["@lsp.type.property"] = { link = "@variable" },
        ["@lsp.type.namespace.go"] = { fg = colors.rosewater },

        -- Bash
        ["@variable.parameter.bash"] = { fg = colors.rosewater },

        -- Makefile
        ["@function.make"] = { link = "Keyword" },
        ["@string.special.symbol.make"] = { link = "@variable" },

        -- Markup
        ["@markup.raw"] = { link = "@text" },
        ["@markup.strong"] = { fg = colors.blue },
        ["@markup.italic"] = { fg = colors.green },

        -- PHP
        ["@keyword.import.php"] = { link = "@keyword" },
        ["@function.builtin.php"] = { fg = colors.blue },
        ["@class.name.php"] = { link = "@text" },
        ["@class.inherit.name.php"] = { link = "@text" },
        ["@class.implements.name.php"] = { link = "@text" },
        ["@interface.name.php"] = { link = "@text" },
        ["@namespace.name.php"] = { link = "@text" },
        ["@namespace.alias.name.php"] = { link = "@text" },
        ["@use.trait.name.php"] = { link = "@text" },

        -- Blade
        ["@directive.blade"] = { link = "PreProc" },
    }
end

return {
    "catppuccin/nvim",
    name = "catppuccin",
    lazy = false,
    priority = 1000,
    opts = {
        flavour = "mocha",
        transparent_background = false,
        color_overrides = {
            mocha = {
                base = "#0E1019",
                mantle = "#0D0F17",
                crust = "#0C0D14",
                surface0 = "#1a1c2d",
                surface1 = "#343959",
                surface2 = "#41476F",
                overlay0 = "#3F4256",
                overlay1 = "#5B5F7C",
                overlay2 = "#767BA0",

                text = "#eceef4",
            },
        },
        no_italic = true,
        no_bold = true,
        custom_highlights = custom_highlights,
        integrations = {
            cmp = true,
            treesitter = true,
            neotree = true,
            telescope = {
                enabled = true,
            },
        }
    },
    config = function(_, opts)
        require("catppuccin").setup(opts)
        vim.cmd.colorscheme("catppuccin")
    end,
}
