From 4184b61b25f5674430dfa0c412c20c4816b8c90f Mon Sep 17 00:00:00 2001
From: Egor Ivkov <e.ivkov@picodata.io>
Date: Tue, 13 Jun 2023 14:47:23 +0000
Subject: [PATCH] feat: display picodata> in interactive prompt

---
 src/lib.rs | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/src/lib.rs b/src/lib.rs
index 20bd2ff557..e86c99e755 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -295,6 +295,20 @@ fn init_handlers() {
     }
 }
 
+/// Sets interactive prompt to display `picodata>`.
+fn set_console_prompt() {
+    tarantool::exec(
+        r#"
+        local console = require('console')
+
+        console.on_start(function(self)
+            self.prompt = "picodata"
+        end)
+        "#,
+    )
+    .expect("setting prompt should never fail")
+}
+
 #[allow(clippy::enum_variant_names)]
 #[derive(Debug, Serialize, Deserialize)]
 pub enum Entrypoint {
@@ -336,6 +350,7 @@ fn init_common(args: &args::Run, cfg: &tarantool::Cfg) -> (Clusterwide, RaftSpac
     preload_http();
     init_sbroad();
 
+    set_console_prompt();
     init_handlers();
     traft::event::init();
 
-- 
GitLab