From 718d1751e687e54108af963a48a350b939beb861 Mon Sep 17 00:00:00 2001
From: Martin Bielik <mx.bielik@gmail.com>
Date: Mon, 27 Feb 2023 19:26:03 +0100
Subject: multiple argument support

---
 vim-ai.vim | 14 ++++++--------
 1 file changed, 6 insertions(+), 8 deletions(-)

diff --git a/vim-ai.vim b/vim-ai.vim
index e6751de..e02936c 100644
--- a/vim-ai.vim
+++ b/vim-ai.vim
@@ -1,18 +1,18 @@
 let g:openaiToken = system("cat ~/.config/openai.token")
 
 function! AIRun(...) range
-  let selection = getline(a:firstline, a:lastline)
+  let prompt = getline(a:firstline, a:lastline)
   if a:0
-    let instruction = a:1 . ":"
-    call insert(selection, instruction, 0)
+    " join arguments and prepend to the prompt
+    let instruction = join(a:000, ", ") . ":"
+    call insert(prompt, instruction, 0)
   endif
 
   let buff_lastline = line('$')
-
-  call writefile(selection, "/tmp/vim-ai.temp")
+  let prompt = join(prompt, "\n")
 
   echo "Completing..."
-  let output = system("cat /tmp/vim-ai.temp | openai complete - -t " . g:openaiToken)
+  let output = system("echo " . shellescape(prompt) . " | openai complete - -t " . g:openaiToken)
   let output = trim(output)
 
   execute a:firstline . ',' . a:lastline . 'd'
@@ -26,5 +26,3 @@ function! AIRun(...) range
 endfunction
 
 command! -range -nargs=? AI <line1>,<line2>call AIRun(<f-args>)
-nnoremap <leader>o :call AIRun()<CR>
-vnoremap <leader>o :call AIRun()<CR>
-- 
cgit v1.2.3