From 33884979c7c4cfaf049e354ae528a9bf919bd11b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jind=C5=99ich=20Moravec?= <xmorave2@fi.muni.cz>
Date: Sun, 28 Jan 2024 16:45:28 +0100
Subject: [PATCH] feat: add command complete for create table and index

---
 server/src/main.rs          | 10 ++++++++--
 server/src/proto_wrapper.rs |  4 ++++
 2 files changed, 12 insertions(+), 2 deletions(-)

diff --git a/server/src/main.rs b/server/src/main.rs
index 8ce6785..ea69c48 100644
--- a/server/src/main.rs
+++ b/server/src/main.rs
@@ -210,8 +210,14 @@ async fn handle_query<W>(writer: &mut W, state: &SharedDbState, query: String, t
                 }
                 false
             },
-            Response::TableCreated => true,
-            Response::IndexCreated => true,
+            Response::TableCreated => {
+                writer.write_command_complete(CompleteStatus::CreateTable).await?;
+                true
+            },
+            Response::IndexCreated => {
+                writer.write_command_complete(CompleteStatus::CreateIndex).await?;
+                true
+            },
         }
     };
 
diff --git a/server/src/proto_wrapper.rs b/server/src/proto_wrapper.rs
index 3415255..b7680d1 100644
--- a/server/src/proto_wrapper.rs
+++ b/server/src/proto_wrapper.rs
@@ -13,6 +13,8 @@ pub enum CompleteStatus {
     },
     Delete(usize),
     Select(usize),
+    CreateTable,
+    CreateIndex,
 }
 
 impl CompleteStatus {
@@ -21,6 +23,8 @@ impl CompleteStatus {
             CompleteStatus::Insert { oid, rows } => format!("INSERT {} {}", oid, rows),
             CompleteStatus::Delete(rows) => format!("DELETE {}", rows),
             CompleteStatus::Select(rows) => format!("SELECT {}", rows),
+            CompleteStatus::CreateTable => "CREATE TABLE".to_string(),
+            CompleteStatus::CreateIndex => "CREATE INDEX".to_string(),
         }
     }
 }