feat(databases): add current database context with set and auto-set on create

Author: eddietejeda

src/command.rs

@@ -574,6 +574,12 @@ pub enum DatabasesCommands {
         output: String,
     },
 
+    /// Set the current database (used by default when no database is specified)
+    Set {
+        /// Database id or description
+        id_or_description: String,
+    },
+
     /// Delete a managed database and its tables
     Delete {
         /// Database name or connection ID
@@ -610,8 +616,9 @@ pub enum DatabasesCommands {
 pub enum DatabaseTablesCommands {
     /// List tables in a managed database
     List {
-        /// Database name or connection ID
-        database: String,
+        /// Database id or description (defaults to current database)
+        #[arg(long)]
+        database: Option<String>,
 
         /// Filter by schema name
         #[arg(long)]
@@ -624,8 +631,9 @@ pub enum DatabaseTablesCommands {
 
     /// Load a parquet file into a table (creates or replaces the table)
     Load {
-        /// Database name or connection ID
-        database: String,
+        /// Database id or description (defaults to current database)
+        #[arg(long)]
+        database: Option<String>,
 
         /// Table name
         table: String,
@@ -649,8 +657,9 @@ pub enum DatabaseTablesCommands {
 
     /// Delete a table from a managed database
     Delete {
-        /// Database name or connection ID
-        database: String,
+        /// Database id or description (defaults to current database)
+        #[arg(long)]
+        database: Option<String>,
 
         /// Table name
         table: String,

src/config.rs

@@ -101,6 +101,8 @@ pub struct ProfileConfig {
     pub workspaces: Vec<WorkspaceEntry>,
     #[serde(default, skip_serializing_if = "Option::is_none", alias = "session")]
     pub sandbox: Option<String>,
+    #[serde(default, skip_serializing_if = "Option::is_none")]
+    pub current_database: Option<String>,
 }
 
 #[derive(Debug, Deserialize, Serialize)]
@@ -227,6 +229,38 @@ pub fn clear_sandbox(profile: &str) -> Result<(), String> {
     write_config(&config_path, &content)
 }
 
+pub fn save_current_database(profile: &str, database_id: &str) -> Result<(), String> {
+    let config_path = config_path()?;
+
+    let mut config_file: ConfigFile = if config_path.exists() {
+        let content = fs::read_to_string(&config_path)
+            .map_err(|e| format!("error reading config file: {e}"))?;
+        serde_yaml::from_str(&content).map_err(|e| format!("error parsing config file: {e}"))?
+    } else {
+        ConfigFile { profiles: HashMap::new() }
+    };
+
+    config_file
+        .profiles
+        .entry(profile.to_string())
+        .or_default()
+        .current_database = Some(database_id.to_string());
+
+    let content = serde_yaml::to_string(&config_file)
+        .map_err(|e| format!("error serializing config: {e}"))?;
+    write_config(&config_path, &content)
+}
+
+pub fn load_current_database(profile: &str) -> Option<String> {
+    let config_path = config_path().ok()?;
+    if !config_path.exists() {
+        return None;
+    }
+    let content = fs::read_to_string(&config_path).ok()?;
+    let config_file: ConfigFile = serde_yaml::from_str(&content).ok()?;
+    config_file.profiles.get(profile)?.current_database.clone()
+}
+
 pub fn resolve_workspace_id(provided: Option<String>, profile_config: &ProfileConfig) -> Result<String, String> {
     if let Some(id) = provided {
         return Ok(id);

src/databases.rs

@@ -436,6 +436,8 @@ pub fn create(
         }
     };
 
+    let _ = crate::config::save_current_database("default", &result.id);
+
     match format {
         "json" => println!("{}", serde_json::to_string_pretty(&result).unwrap()),
         "yaml" => print!("{}", serde_yaml::to_string(&result).unwrap()),
@@ -450,6 +452,34 @@ pub fn create(
     }
 }
 
+pub fn set(id_or_description: &str, workspace_id: &str) {
+    use crossterm::style::Stylize;
+    let api = ApiClient::new(Some(workspace_id));
+    let db = resolve_database(&api, id_or_description);
+    if let Err(e) = crate::config::save_current_database("default", &db.id) {
+        eprintln!("{}", format!("error saving current database: {e}").red());
+        std::process::exit(1);
+    }
+    println!("{}", format!("Current database set to {}", db.id).green());
+}
+
+fn resolve_current_database(provided: Option<&str>, _workspace_id: &str) -> String {
+    if let Some(id) = provided {
+        return id.to_string();
+    }
+    match crate::config::load_current_database("default") {
+        Some(id) => id,
+        None => {
+            use crossterm::style::Stylize;
+            eprintln!(
+                "{}",
+                "error: no current database set. Use 'hotdata databases set <id>' or pass a database id.".red()
+            );
+            std::process::exit(1);
+        }
+    }
+}
+
 pub fn delete(workspace_id: &str, id_or_description: &str) {
     use crossterm::style::Stylize;
 
@@ -465,9 +495,10 @@ pub fn delete(workspace_id: &str, id_or_description: &str) {
     println!("{}", "Database deleted.".green());
 }
 
-pub fn tables_list(workspace_id: &str, database: &str, schema: Option<&str>, format: &str) {
+pub fn tables_list(workspace_id: &str, database: Option<&str>, schema: Option<&str>, format: &str) {
+    let database = resolve_current_database(database, workspace_id);
     let api = ApiClient::new(Some(workspace_id));
-    let db = resolve_database(&api, database);
+    let db = resolve_database(&api, &database);
     let tables = collect_tables(&api, &db.default_connection_id, schema);
 
     let rows = table_rows(tables);
@@ -502,7 +533,7 @@ pub fn tables_list(workspace_id: &str, database: &str, schema: Option<&str>, for
 
 pub fn tables_load(
     workspace_id: &str,
-    database: &str,
+    database: Option<&str>,
     table: &str,
     schema: Option<&str>,
     file: Option<&str>,
@@ -511,8 +542,9 @@ pub fn tables_load(
 ) {
     use crossterm::style::Stylize;
 
+    let database = resolve_current_database(database, workspace_id);
     let api = ApiClient::new(Some(workspace_id));
-    let db = resolve_database(&api, database);
+    let db = resolve_database(&api, &database);
     let schema = schema_name(schema);
 
     // clap enforces mutual exclusion; only one of these is ever Some.
@@ -564,11 +596,12 @@ pub fn tables_load(
     println!("rows:      {}", result.row_count);
 }
 
-pub fn tables_delete(workspace_id: &str, database: &str, table: &str, schema: Option<&str>) {
+pub fn tables_delete(workspace_id: &str, database: Option<&str>, table: &str, schema: Option<&str>) {
     use crossterm::style::Stylize;
 
+    let database = resolve_current_database(database, workspace_id);
     let api = ApiClient::new(Some(workspace_id));
-    let db = resolve_database(&api, database);
+    let db = resolve_database(&api, &database);
     let schema = schema_name(schema);
 
     let path = managed_table_delete_path(&db.default_connection_id, schema, table);

src/main.rs

@@ -125,11 +125,23 @@ extern "C" fn print_sandbox_footer() {
     );
 }
 
+extern "C" fn print_database_footer() {
+    use crossterm::style::Stylize;
+    if let Some(id) = config::load_current_database("default") {
+        eprintln!(
+            "{}",
+            format!("current database: {id}  use 'hotdata databases set' to change")
+                .dark_grey(),
+        );
+    }
+}
+
 fn main() {
     // Register before `Cli::parse`, since `--help` / `--version` exit
     // from inside the parser. Safety: `atexit` is async-signal-safe;
     // the callback only reads env vars / files and writes to stderr.
     unsafe { atexit(print_sandbox_footer) };
+    unsafe { atexit(print_database_footer) };
 
     dotenvy::dotenv().ok();
     let cli = Cli::parse();
@@ -393,6 +405,9 @@ fn main() {
                             &tables,
                             &output,
                         ),
+                        Some(DatabasesCommands::Set { id_or_description }) => {
+                            databases::set(&id_or_description, &workspace_id)
+                        }
                         Some(DatabasesCommands::Delete { name_or_id }) => {
                             databases::delete(&workspace_id, &name_or_id)
                         }
@@ -405,7 +420,7 @@ fn main() {
                             let (database, schema, table) = parse_db_target(&target);
                             databases::tables_load(
                                 &workspace_id,
-                                &database,
+                                Some(database.as_str()),
                                 &table,
                                 Some(schema.as_str()),
                                 file.as_deref(),
@@ -420,7 +435,7 @@ fn main() {
                                 output,
                             } => databases::tables_list(
                                 &workspace_id,
-                                &database,
+                                database.as_deref(),
                                 schema.as_deref(),
                                 &output,
                             ),
@@ -433,7 +448,7 @@ fn main() {
                                 upload_id,
                             } => databases::tables_load(
                                 &workspace_id,
-                                &database,
+                                database.as_deref(),
                                 &table,
                                 Some(schema.as_str()),
                                 file.as_deref(),
@@ -446,7 +461,7 @@ fn main() {
                                 schema,
                             } => databases::tables_delete(
                                 &workspace_id,
-                                &database,
+                                database.as_deref(),
                                 &table,
                                 Some(schema.as_str()),
                             ),

tests/databases_cli.rs

@@ -52,7 +52,6 @@ fn databases_tables_load_rejects_both_file_and_upload_id_at_parse_time() {
             "databases",
             "tables",
             "load",
-            "mydb",
             "t1",
             "--file",
             "a.parquet",