Add database 'schema' command to retrieve the database schema

Author: flyingsilverfin

Cargo.lock

@@ -46,8 +46,8 @@ features = {}
 
 	[dependencies.typedb-driver]
 		features = []
+		rev = "02edb1b7c4bcd3f5bc822634d667653964e2ab92"
 		git = "https://github.com/typedb/typedb-driver"
-		tag = "3.2.0-rc0"
 		default-features = false
 
 	[dependencies.futures]

Cargo.toml

@@ -15,7 +15,7 @@ def typedb_driver():
     git_repository(
         name = "typedb_driver",
         remote = "https://github.com/typedb/typedb-driver",
-        tag = "3.2.0-rc0",  # sync-marker: do not remove this comment, this is used for sync-dependencies by @typedb_driver
+        commit = "02edb1b7c4bcd3f5bc822634d667653964e2ab92",  # sync-marker: do not remove this comment, this is used for sync-dependencies by @typedb_driver
     )
 
 def typeql():

dependencies/typedb/repositories.bzl

@@ -27,9 +27,9 @@ use crate::{
     cli::Args,
     completions::{database_name_completer_fn, file_completer},
     operations::{
-        database_create, database_delete, database_list, transaction_close, transaction_commit, transaction_query,
-        transaction_read, transaction_rollback, transaction_schema, transaction_source, transaction_write, user_create,
-        user_delete, user_list, user_update_password,
+        database_create, database_delete, database_list, database_schema, transaction_close, transaction_commit,
+        transaction_query, transaction_read, transaction_rollback, transaction_schema, transaction_source,
+        transaction_write, user_create, user_delete, user_list, user_update_password,
     },
     repl::{
         command::{get_word, parse_one_query, CommandInput, CommandLeaf, Subcommand},
@@ -255,6 +255,12 @@ fn entry_repl(driver: Arc<TypeDBDriver>, runtime: BackgroundRuntime) -> Repl<Con
             "Delete the database with the given name.",
             CommandInput::new("db", get_word, None, Some(database_name_completer_fn(driver.clone(), runtime.clone()))),
             database_delete,
+        ))
+        .add(CommandLeaf::new_with_input(
+            "schema",
+            "Retrieve the TypeQL representation of a database's schema.",
+            CommandInput::new("db", get_word, None, Some(database_name_completer_fn(driver.clone(), runtime.clone()))),
+            database_schema,
         ));
 
     let user_commands = Subcommand::new("user")

src/main.rs

@@ -59,6 +59,17 @@ pub(crate) fn database_delete(context: &mut ConsoleContext, input: &[String]) ->
     Ok(())
 }
 
+pub(crate) fn database_schema(context: &mut ConsoleContext, input: &[String]) -> CommandResult {
+    let driver = context.driver.clone();
+    let db_name = input[0].clone();
+    let schema = context
+        .background_runtime
+        .run(async move { driver.databases().get(db_name).await?.schema().await })
+        .map_err(|err| Box::new(err) as Box<dyn Error + Send>)?;
+    println!("{}", schema);
+    Ok(())
+}
+
 pub(crate) fn user_list(context: &mut ConsoleContext, _input: &[String]) -> CommandResult {
     let driver = context.driver.clone();
     let users = context