From 041ad5739c5e55493c9feafa92be0b281a0ed2dc Mon Sep 17 00:00:00 2001
From: Martin Bielik <mx.bielik@gmail.com>
Date: Mon, 27 Feb 2023 20:28:37 +0100
Subject: plugin dir

---
 plugin/vim-ai.vim | 28 ++++++++++++++++++++++++++++
 1 file changed, 28 insertions(+)
 create mode 100644 plugin/vim-ai.vim

(limited to 'plugin/vim-ai.vim')

diff --git a/plugin/vim-ai.vim b/plugin/vim-ai.vim
new file mode 100644
index 0000000..e02936c
--- /dev/null
+++ b/plugin/vim-ai.vim
@@ -0,0 +1,28 @@
+let g:openaiToken = system("cat ~/.config/openai.token")
+
+function! AIRun(...) range
+  let prompt = getline(a:firstline, a:lastline)
+  if a:0
+    " join arguments and prepend to the prompt
+    let instruction = join(a:000, ", ") . ":"
+    call insert(prompt, instruction, 0)
+  endif
+
+  let buff_lastline = line('$')
+  let prompt = join(prompt, "\n")
+
+  echo "Completing..."
+  let output = system("echo " . shellescape(prompt) . " | openai complete - -t " . g:openaiToken)
+  let output = trim(output)
+
+  execute a:firstline . ',' . a:lastline . 'd'
+
+  if a:lastline == buff_lastline
+    execute "normal! o" . output . "\<Esc>"
+  else
+    execute "normal! O" . output . "\<Esc>"
+  endif
+
+endfunction
+
+command! -range -nargs=? AI <line1>,<line2>call AIRun(<f-args>)
-- 
cgit v1.2.3