-- File: lua/custom/plugins/chatgpt.lua

return {
  'jackMort/ChatGPT.nvim',
  event = 'VeryLazy',
  config = function()
    vim.keymap.set('n', '<leader>lc', vim.cmd.ChatGPT, { desc = '[C]hatGPT' })
    vim.keymap.set('n', '<leader>lca', vim.cmd.ChatGPTActAs, { desc = 'ChatGPT [A]ct As' })
    vim.keymap.set('n', '<leader>lcc', vim.cmd.ChatGPTCompleteCode, { desc = 'ChatGPT [C]omplete Code' })
    vim.keymap.set('n', '<leader>lce', vim.cmd.ChatGPTEditWithInstructions, { desc = 'ChatGPT [E]dit With Instructions' })
    vim.keymap.set('n', '<leader>lcd', '<cmd>ChatGPTRun docstring<CR>', { desc = 'ChatGPT [D]ocstring' })
    vim.keymap.set('n', '<leader>lco', '<cmd>ChatGPTRun optimize_code<CR>', { desc = 'ChatGPT [o]ptimize code' })
    vim.keymap.set('n', '<leader>lce', '<cmd>ChatGPTRun explain_code<CR>', { desc = 'ChatGPT [e]xplain code' })
    vim.keymap.set('n', '<leader>lcs', '<cmd>ChatGPTRun summarize<CR>', { desc = 'ChatGPT [s]ummarize' })
    vim.keymap.set('n', '<leader>lct', '<cmd>ChatGPTRun translate<CR>', { desc = 'ChatGPT [t]ranslate' })
    require('chatgpt').setup {
      yank_register = '+',
      edit_with_instructions = {
        diff = false,
        keymaps = {
          close = '<C-c>',
          accept = '<C-y>',
          toggle_diff = '<C-d>',
          toggle_settings = '<C-o>',
          toggle_help = '<C-h>',
          cycle_windows = '<Tab>',
          use_output_as_input = '<C-i>',
        },
      },
      chat = {
        welcome_message = 'Welcome Human!',
        loading_text = 'Loading, please wait ...',
        question_sign = '', -- 🙂
        answer_sign = 'ﮧ', -- 🤖
        border_left_sign = '',
        border_right_sign = '',
        max_line_length = 120,
        sessions_window = {
          active_sign = '  ',
          inactive_sign = '  ',
          current_line_sign = '',
          border = {
            style = 'rounded',
            text = {
              top = ' Sessions ',
            },
          },
          win_options = {
            winhighlight = 'Normal:Normal,FloatBorder:FloatBorder',
          },
        },
        keymaps = {
          close = '<C-c>',
          yank_last = '<C-y>',
          yank_last_code = '<C-k>',
          scroll_up = '<C-u>',
          scroll_down = '<C-d>',
          new_session = '<C-n>',
          cycle_windows = '<Tab>',
          cycle_modes = '<C-f>',
          next_message = '<C-j>',
          prev_message = '<C-k>',
          select_session = '<Space>',
          rename_session = 'r',
          delete_session = 'd',
          draft_message = '<C-r>',
          edit_message = 'e',
          delete_message = 'd',
          toggle_settings = '<C-o>',
          toggle_sessions = '<C-p>',
          toggle_help = '<C-h>',
          toggle_message_role = '<C-r>',
          toggle_system_role_open = '<C-s>',
          stop_generating = '<C-x>',
        },
      },
      popup_layout = {
        default = 'center',
        center = {
          width = '80%',
          height = '80%',
        },
        right = {
          width = '30%',
          width_settings_open = '50%',
        },
      },
      popup_window = {
        border = {
          highlight = 'FloatBorder',
          style = 'rounded',
          text = {
            top = ' ChatGPT ',
          },
        },
        win_options = {
          wrap = true,
          linebreak = true,
          foldcolumn = '1',
          winhighlight = 'Normal:Normal,FloatBorder:FloatBorder',
        },
        buf_options = {
          filetype = 'markdown',
        },
      },
      system_window = {
        border = {
          highlight = 'FloatBorder',
          style = 'rounded',
          text = {
            top = ' SYSTEM ',
          },
        },
        win_options = {
          wrap = true,
          linebreak = true,
          foldcolumn = '2',
          winhighlight = 'Normal:Normal,FloatBorder:FloatBorder',
        },
      },
      popup_input = {
        prompt = '  ',
        border = {
          highlight = 'FloatBorder',
          style = 'rounded',
          text = {
            top_align = 'center',
            top = ' Prompt ',
          },
        },
        win_options = {
          winhighlight = 'Normal:Normal,FloatBorder:FloatBorder',
        },
        submit = '<C-Enter>',
        submit_n = '<Enter>',
        max_visible_lines = 20,
      },
      settings_window = {
        setting_sign = '  ',
        border = {
          style = 'rounded',
          text = {
            top = ' Settings ',
          },
        },
        win_options = {
          winhighlight = 'Normal:Normal,FloatBorder:FloatBorder',
        },
      },
      help_window = {
        setting_sign = '  ',
        border = {
          style = 'rounded',
          text = {
            top = ' Help ',
          },
        },
        win_options = {
          winhighlight = 'Normal:Normal,FloatBorder:FloatBorder',
        },
      },
      openai_params = {
        model = 'gpt-4-0125-preview',
        frequency_penalty = 0,
        presence_penalty = 0,
        max_tokens = 1000,
        temperature = 0,
        top_p = 1,
        n = 1,
      },
      openai_edit_params = {
        model = 'gpt-4-0125-preview',
        frequency_penalty = 0,
        presence_penalty = 0,
        temperature = 0,
        top_p = 1,
        n = 1,
      },
      use_openai_functions_for_edits = false,
      actions_paths = {},
      show_quickfixes_cmd = 'Trouble quickfix',
      predefined_chat_gpt_prompts = 'https://raw.githubusercontent.com/f/awesome-chatgpt-prompts/main/prompts.csv',
      highlights = {
        help_key = '@symbol',
        help_description = '@comment',
      },
    }
  end,
  dependencies = {
    'MunifTanjim/nui.nvim',
    'nvim-lua/plenary.nvim',
    'folke/trouble.nvim',
    'nvim-telescope/telescope.nvim',
  },
}