From d3800d43342252f1bc2ecd3ddb0f905dfb90042e Mon Sep 17 00:00:00 2001
From: Martin Bielik <mx.bielik@gmail.com>
Date: Fri, 21 Apr 2023 18:16:58 +0200
Subject: move prompt to python

---
 py/chat.py | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

(limited to 'py')

diff --git a/py/chat.py b/py/chat.py
index 491b7c2..93a8664 100644
--- a/py/chat.py
+++ b/py/chat.py
@@ -4,6 +4,7 @@ vim.command(f"py3file {plugin_root}/py/utils.py")
 
 config_options = vim.eval("l:options")
 config_ui = vim.eval("l:ui")
+prompt = vim.eval("l:prompt").strip()
 
 def initialize_chat_window():
     lines = vim.eval('getline(1, "$")')
@@ -21,11 +22,15 @@ def initialize_chat_window():
                 vim.command("normal! i" + key + "=" + value + "\n")
         vim.command("normal! " + ("o" if populates_options else "O"))
         vim.command("normal! i>>> user\n")
-        vim.command("normal! G")
+
+    vim.command("normal! G")
+    vim_break_undo_sequence()
+    vim.command("redraw")
+
+    if prompt:
+        vim.command("normal! a" + prompt)
         vim_break_undo_sequence()
         vim.command("redraw")
-    else:
-        vim_break_undo_sequence()
 
 initialize_chat_window()
 
-- 
cgit v1.2.3