From 8a613b73b452d52f13f6bb0ed53a79e3ba856b4c Mon Sep 17 00:00:00 2001
From: Muhammad Abdullah <abd880@mit.edu>
Date: Fri, 24 Feb 2023 02:11:27 -0500
Subject: [PATCH 1/3] Initial Implementation.

<Tab> accepts one word from suggestion.

<S-Tab> accepts all.
---
 autoload/copilot.vim | 54 ++++++++++++++++++++++++++++++++++++++++++--
 plugin/copilot.vim   | 10 +++++---
 2 files changed, 59 insertions(+), 5 deletions(-)

diff --git a/autoload/copilot.vim b/autoload/copilot.vim
index c8fd8eb3..68ccc00b 100644
--- a/autoload/copilot.vim
+++ b/autoload/copilot.vim
@@ -419,7 +419,7 @@ endfunction
 
 function! copilot#IsMapped() abort
   return get(g:, 'copilot_assume_mapped') ||
-        \ hasmapto('copilot#Accept(', 'i')
+        \ hasmapto('copilot#AcceptOne(', 'i')
 endfunction
 let s:is_mapped = copilot#IsMapped()
 
@@ -465,7 +465,57 @@ function! copilot#TextQueuedForInsertion() abort
   endtry
 endfunction
 
-function! copilot#Accept(...) abort
+let s:sugg_buffer = {'sugg': '', 'count': 0, 'text': '', 'valid': v:false}
+function! copilot#AcceptOne(...) abort
+
+  if !s:sugg_buffer.valid
+    let s = copilot#GetDisplayedSuggestion()
+    if !empty(s)
+      " split the suggestion into words, keep the trailing space
+      let s:sugg_buffer = {'sugg': s, 'count': 0, 'text': split(s.text, '[\t ]\zs') , 'valid': v:true}
+      " if the text is whitespace, invalidate the buffer
+      if (len(s:sugg_buffer.text)==0) || !(s:sugg_buffer.text[0] =~# '\S')
+        let s:sugg_buffer.valid = v:false
+      endif
+    endif
+  endif
+
+  if s:sugg_buffer.valid
+    let s:suggestion_text = s:sugg_buffer.text[s:sugg_buffer.count]
+    let s:sugg_buffer.count += 1
+    if s:sugg_buffer.count >= len(s:sugg_buffer.text)
+      let s:sugg_buffer.valid = v:false
+      unlet! b:_copilot
+      call copilot#Request('notifyAccepted', {'uuid': s:sugg_buffer.sugg.uuid}) 
+      unlet! s:uuid
+      call s:ClearPreview()
+    endif
+    if s:sugg_buffer.valid && s:sugg_buffer.count == 1
+      return repeat("\<Left>\<Del>", s:sugg_buffer.sugg.outdentSize) . repeat("\<Del>", s:sugg_buffer.sugg.deleteSize) .
+            \ "\<C-R>\<C-O>=copilot#TextQueuedForInsertion()\<CR>"
+    endif
+    return "\<C-R>\<C-O>=copilot#TextQueuedForInsertion()\<CR>" 
+  endif
+  
+  let default = get(g:, 'copilot_tab_fallback', pumvisible() ? "\<C-N>" : "\t")
+  if !a:0
+    return default
+  elseif type(a:1) == v:t_string
+    return a:1
+  elseif type(a:1) == v:t_func
+    try
+      return call(a:1, [])
+    catch
+      call copilot#logger#Exception()
+      return default
+    endtry
+  else
+    return default
+  endif
+endfunction
+
+
+function! copilot#AcceptAll(...) abort
   let s = copilot#GetDisplayedSuggestion()
   if !empty(s.text)
     unlet! b:_copilot
diff --git a/plugin/copilot.vim b/plugin/copilot.vim
index 54407ddd..4d4a9b95 100644
--- a/plugin/copilot.vim
+++ b/plugin/copilot.vim
@@ -20,13 +20,15 @@ function! s:ColorScheme() abort
   hi def link CopilotAnnotation Normal
 endfunction
 
+
 function! s:MapTab() abort
   if get(g:, 'copilot_no_tab_map') || get(g:, 'copilot_no_maps')
     return
   endif
   let tab_map = maparg('<Tab>', 'i', 0, 1)
   if !has_key(tab_map, 'rhs')
-    imap <script><silent><nowait><expr> <Tab> copilot#Accept()
+    imap <script><silent><nowait><expr> <Tab> copilot#AcceptOne()
+    imap <script><silent><nowait><expr> <S-Tab> copilot#AcceptAll()
   elseif tab_map.rhs !~# 'copilot'
     if tab_map.expr
       let tab_fallback = '{ -> ' . tab_map.rhs . ' }'
@@ -35,9 +37,11 @@ function! s:MapTab() abort
     endif
     let tab_fallback = substitute(tab_fallback, '<SID>', '<SNR>' . get(tab_map, 'sid') . '_', 'g')
     if get(tab_map, 'noremap') || get(tab_map, 'script') || mapcheck('<Left>', 'i') || mapcheck('<Del>', 'i')
-      exe 'imap <script><silent><nowait><expr> <Tab> copilot#Accept(' . tab_fallback . ')'
+      exe 'imap <script><silent><nowait><expr> <Tab> copilot#AcceptOne(' . tab_fallback . ')'
+      exe 'imap <script><silent><nowait><expr> <S-Tab> copilot#AcceptAll(' . tab_fallback . ')'
     else
-      exe 'imap <silent><nowait><expr>         <Tab> copilot#Accept(' . tab_fallback . ')'
+      exe 'imap <silent><nowait><expr>         <Tab> copilot#AcceptOne(' . tab_fallback . ')'
+      exe 'imap <silent><nowait><expr>         <S-Tab> copilot#AcceptAll(' . tab_fallback . ')'
     endif
   endif
 endfunction

From 5f797cc11e05420777a2f194aeb0fb9381cd40e8 Mon Sep 17 00:00:00 2001
From: Muhammad Abdullah <abd880@mit.edu>
Date: Fri, 24 Feb 2023 12:27:10 -0500
Subject: [PATCH 2/3] inserting character clears buffer.

---
 autoload/copilot.vim | 23 ++++++++++++++++++++++-
 plugin/copilot.vim   | 29 +++++++++++++++++++++++++----
 2 files changed, 47 insertions(+), 5 deletions(-)

diff --git a/autoload/copilot.vim b/autoload/copilot.vim
index cff81963..4459c079 100644
--- a/autoload/copilot.vim
+++ b/autoload/copilot.vim
@@ -488,7 +488,7 @@ function! copilot#AcceptOne(...) abort
     let s = copilot#GetDisplayedSuggestion()
     if !empty(s)
       " split the suggestion into words, keep the trailing space
-      let s:sugg_buffer = {'sugg': s, 'count': 0, 'text': split(s.text, '[\t ]\zs') , 'valid': v:true}
+      let s:sugg_buffer = {'sugg': s, 'count': 0, 'text': split(s.text, '\w\W\+\zs'), 'valid': v:true}
       " if the text is whitespace, invalidate the buffer
       if (len(s:sugg_buffer.text)==0) || !(s:sugg_buffer.text[0] =~# '\S')
         let s:sugg_buffer.valid = v:false
@@ -559,6 +559,27 @@ function! copilot#AcceptAll(...) abort
   endif
 endfunction
 
+function! copilot#clearBuffer(...) abort
+  let s:sugg_buffer.valid = v:false
+  if a:0
+    return a:1
+  endif
+endfunction
+
+function! copilot#OnInsertCharPre() abort
+  " If the user has pressed a key that is not a completion key, clear the
+  " suggestion buffer.
+  if !s:is_mapped || !s:dest || !copilot#Enabled()
+    return
+  endif
+  let key = v:char
+  if key ==# "\t"
+    return
+  endif
+  call copilot#clearBuffer()
+  return key
+endfunction
+
 function! s:BrowserCallback(into, code) abort
   let a:into.code = a:code
 endfunction
diff --git a/plugin/copilot.vim b/plugin/copilot.vim
index 4d4a9b95..e4b8d997 100644
--- a/plugin/copilot.vim
+++ b/plugin/copilot.vim
@@ -20,12 +20,16 @@ function! s:ColorScheme() abort
   hi def link CopilotAnnotation Normal
 endfunction
 
-
-function! s:MapTab() abort
+" Map <Tab> and <S-Tab> to accept suggestions. 
+" If <Tab> is already mapped, use it as a fallback.
+" Map <CR> to clear the suggestion buffer.
+" If <CR> is already mapped, use it as a fallback.
+function! s:MapShortcuts() abort
   if get(g:, 'copilot_no_tab_map') || get(g:, 'copilot_no_maps')
     return
   endif
   let tab_map = maparg('<Tab>', 'i', 0, 1)
+  let cr_map = maparg('<CR>', 'i', 0, 1)
   if !has_key(tab_map, 'rhs')
     imap <script><silent><nowait><expr> <Tab> copilot#AcceptOne()
     imap <script><silent><nowait><expr> <S-Tab> copilot#AcceptAll()
@@ -44,6 +48,22 @@ function! s:MapTab() abort
       exe 'imap <silent><nowait><expr>         <S-Tab> copilot#AcceptAll(' . tab_fallback . ')'
     endif
   endif
+
+  if !has_key(cr_map, 'rhs')
+    imap <script><silent><nowait><expr> <CR> copilot#clearBuffer('<CR>')
+  elseif cr_map.rhs !~# 'copilot'
+    if cr_map.expr
+      let cr_fallback = '{ -> ' . cr_map.rhs . ' }'
+    else
+      let cr_fallback = substitute(json_encode(cr_map.rhs), '<', '\\<', 'g')
+    endif
+    let cr_fallback = substitute(cr_fallback, '<SID>', '<SNR>' . get(cr_map, 'sid') . '_', 'g')
+    if get(cr_map, 'noremap') || get(cr_map, 'script') || mapcheck('<Left>', 'i') || mapcheck('<Del>', 'i')
+      exe 'imap <script><silent><nowait><expr> <CR> copilot#clearBuffer(' . cr_fallback . ')'
+    else
+      exe 'imap <silent><nowait><expr>         <CR> copilot#clearBuffer(' . cr_fallback . ')'
+    endif
+  endif
 endfunction
 
 function! s:Event(type) abort
@@ -63,12 +83,13 @@ augroup github_copilot
   autocmd CursorMovedI         * call s:Event('CursorMovedI')
   autocmd CompleteChanged      * call s:Event('CompleteChanged')
   autocmd ColorScheme,VimEnter * call s:ColorScheme()
-  autocmd VimEnter             * call s:MapTab()
+  autocmd VimEnter             * call s:MapShortcuts()
+  autocmd InsertCharPre        * call s:Event('InsertCharPre')
   autocmd BufReadCmd copilot://* setlocal buftype=nofile bufhidden=wipe nobuflisted readonly nomodifiable
 augroup END
 
 call s:ColorScheme()
-call s:MapTab()
+call s:MapShortcuts()
 if !get(g:, 'copilot_no_maps')
   imap <Plug>(copilot-dismiss)     <Cmd>call copilot#Dismiss()<CR>
   if empty(mapcheck('<C-]>', 'i'))

From 45c2143c7a895d83d28f56023d8daf9d591cb68b Mon Sep 17 00:00:00 2001
From: Abdullah <83081043+abdullah8a0@users.noreply.github.com>
Date: Mon, 13 Mar 2023 16:49:31 -0400
Subject: [PATCH 3/3] Update README.md

---
 README.md | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/README.md b/README.md
index 79944f99..127b2884 100644
--- a/README.md
+++ b/README.md
@@ -30,22 +30,22 @@ Terms](https://docs.github.com/en/site-policy/github-terms/github-terms-for-addi
 
     * Vim, Linux/macOS:
 
-          git clone https://github.com/github/copilot.vim.git \
+          git clone https://github.com/abdullah8a0/copilot.vim.git \
             ~/.vim/pack/github/start/copilot.vim
 
     * Neovim, Linux/macOS:
 
-          git clone https://github.com/github/copilot.vim.git \
+          git clone https://github.com/abdullah8a0/copilot.vim.git \
             ~/.config/nvim/pack/github/start/copilot.vim
 
     * Vim, Windows (PowerShell command):
 
-          git clone https://github.com/github/copilot.vim.git `
+          git clone https://github.com/abdullah8a0/copilot.vim.git `
             $HOME/vimfiles/pack/github/start/copilot.vim
 
     * Neovim, Windows (PowerShell command):
 
-          git clone https://github.com/github/copilot.vim.git `
+          git clone https://github.com/abdullah8a0/copilot.vim.git `
             $HOME/AppData/Local/nvim/pack/github/start/copilot.vim
 
 4.  Start Neovim and invoke `:Copilot setup`.