forked from CopilotC-Nvim/CopilotChat.nvim
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge branch 'main' of github.com:CopilotC-Nvim/CopilotChat.nvim
- Loading branch information
Showing
8 changed files
with
324 additions
and
6 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,173 @@ | ||
| local actions = require('telescope.actions') | ||
| local action_state = require('telescope.actions.state') | ||
| local telescope_pickers = require('telescope.pickers') | ||
| local finders = require('telescope.finders') | ||
| local themes = require('telescope.themes') | ||
| local conf = require('telescope.config').values | ||
| local utils = require('CopilotChat.utils') | ||
|
|
||
| local help_actions = {} | ||
| local user_prompt_actions = {} | ||
|
|
||
| local function generate_fix_diagnostic_prompt() | ||
| local diagnostic = utils.get_diagnostics() | ||
| local file_name = vim.fn.expand('%:t') | ||
| local line_number = vim.fn.line('.') | ||
| return 'Please assist with fixing the following diagnostic issue in file: "' | ||
| .. file_name | ||
| .. ':' | ||
| .. line_number | ||
| .. '". ' | ||
| .. diagnostic | ||
| end | ||
|
|
||
| local function generate_explain_diagnostic_prompt() | ||
| local diagnostic = utils.get_diagnostics() | ||
| local file_name = vim.fn.expand('%:t') | ||
| local line_number = vim.fn.line('.') | ||
| return 'Please explain the following diagnostic issue in file: "' | ||
| .. file_name | ||
| .. ':' | ||
| .. line_number | ||
| .. '". ' | ||
| .. diagnostic | ||
| end | ||
|
|
||
| --- Help command for telescope picker | ||
| --- This will copy all the lines in the buffer to the unnamed register | ||
| --- Then will send the diagnostic to copilot chat | ||
| ---@param prefix string | ||
| local function diagnostic_help_command(prefix) | ||
| if prefix == nil then | ||
| prefix = '' | ||
| else | ||
| prefix = prefix .. ' ' | ||
| end | ||
|
|
||
| return function(prompt_bufnr, _) | ||
| actions.select_default:replace(function() | ||
| actions.close(prompt_bufnr) | ||
| local selection = action_state.get_selected_entry() | ||
|
|
||
| -- Select all the lines in the buffer to uname register | ||
| vim.cmd('normal! ggVG"*y') | ||
|
|
||
| -- Get value from the help_actions and execute the command | ||
| local value = '' | ||
| for _, action in pairs(help_actions) do | ||
| if action.name == selection[1] then | ||
| value = action.label | ||
| break | ||
| end | ||
| end | ||
|
|
||
| vim.cmd(prefix .. value) | ||
| end) | ||
| return true | ||
| end | ||
| end | ||
|
|
||
| --- Prompt command for telescope picker | ||
| --- This will show all the user prompts in the telescope picker | ||
| --- Then will execute the command selected by the user | ||
| ---@param prefix string | ||
| local function generate_prompt_command(prefix) | ||
| if prefix == nil then | ||
| prefix = '' | ||
| else | ||
| prefix = prefix .. ' ' | ||
| end | ||
|
|
||
| return function(prompt_bufnr, _) | ||
| actions.select_default:replace(function() | ||
| actions.close(prompt_bufnr) | ||
| local selection = action_state.get_selected_entry() | ||
|
|
||
| -- Get value from the prompt_actions and execute the command | ||
| local value = '' | ||
| for _, action in pairs(user_prompt_actions) do | ||
| if action.name == selection[1] then | ||
| value = action.label | ||
| break | ||
| end | ||
| end | ||
|
|
||
| vim.cmd(prefix .. value) | ||
| end) | ||
| return true | ||
| end | ||
| end | ||
|
|
||
| local function show_help_actions() | ||
| help_actions = { | ||
| { | ||
| label = generate_fix_diagnostic_prompt(), | ||
| name = 'Fix diagnostic', | ||
| }, | ||
| { | ||
| label = generate_explain_diagnostic_prompt(), | ||
| name = 'Explain diagnostic', | ||
| }, | ||
| } | ||
|
|
||
| -- Filter all no diagnostics available actions | ||
| help_actions = vim.tbl_filter(function(value) | ||
| return value.label ~= 'No diagnostics available' | ||
| end, help_actions) | ||
|
|
||
| -- Show the menu with telescope pickers | ||
| local opts = themes.get_dropdown({}) | ||
| local action_names = {} | ||
| for _, value in pairs(help_actions) do | ||
| table.insert(action_names, value.name) | ||
| end | ||
| telescope_pickers | ||
| .new(opts, { | ||
| prompt_title = 'Copilot Chat Help Actions', | ||
| finder = finders.new_table({ | ||
| results = action_names, | ||
| }), | ||
| sorter = conf.generic_sorter(opts), | ||
| attach_mappings = diagnostic_help_command('CopilotChat'), | ||
| }) | ||
| :find() | ||
| end | ||
|
|
||
| --- Show prompt actions | ||
| ---@param is_in_visual_mode boolean? | ||
| local function show_prompt_actions(is_in_visual_mode) | ||
| -- Convert user prompts to a table of actions | ||
| user_prompt_actions = {} | ||
|
|
||
| for key, prompt in pairs(vim.g.copilot_chat_user_prompts) do | ||
| table.insert(user_prompt_actions, { name = key, label = prompt }) | ||
| end | ||
|
|
||
| local cmd = 'CopilotChat' | ||
|
|
||
| if is_in_visual_mode then | ||
| cmd = "'<,'>CopilotChatVisual" | ||
| end | ||
|
|
||
| -- Show the menu with telescope pickers | ||
| local opts = themes.get_dropdown({}) | ||
| local action_names = {} | ||
| for _, value in pairs(user_prompt_actions) do | ||
| table.insert(action_names, value.name) | ||
| end | ||
| telescope_pickers | ||
| .new(opts, { | ||
| prompt_title = 'Copilot Chat Actions', | ||
| finder = finders.new_table({ | ||
| results = action_names, | ||
| }), | ||
| sorter = conf.generic_sorter(opts), | ||
| attach_mappings = generate_prompt_command(cmd), | ||
| }) | ||
| :find() | ||
| end | ||
|
|
||
| return { | ||
| show_help_actions = show_help_actions, | ||
| show_prompt_actions = show_prompt_actions, | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.