Initial public release of Warp.

Repo-Sync-Origin: warpdotdev/warp-internal@12af1d983b

app/src/persistence/commands.rs

@@ -0,0 +1,69 @@
+use anyhow::Result;
+use diesel::{sqlite::SqliteConnection, ExpressionMethods, QueryDsl, RunQueryDsl};
+
+use crate::terminal::event::UserBlockCompleted;
+
+/// Returns the command that was run right after `command`
+/// in the same session, if any.
+pub fn get_next_command(
+    conn: &mut SqliteConnection,
+    command: &super::model::Command,
+) -> Result<super::model::Command> {
+    let next_command = super::schema::commands::dsl::commands
+        .filter(super::schema::commands::columns::id.gt(command.id))
+        .filter(super::schema::commands::columns::session_id.eq(&command.session_id))
+        // Skip any empty blocks
+        .filter(super::schema::commands::columns::command.ne(""))
+        .order(super::schema::commands::columns::id.asc())
+        .limit(1)
+        .first::<super::model::Command>(conn)?;
+    Ok(next_command)
+}
+
+/// Returns the commands that were run right before `command`
+/// in the same session, if any. They are ordered from oldest to newest.
+pub fn get_previous_commands(
+    conn: &mut SqliteConnection,
+    command: &super::model::Command,
+    num_commands: usize,
+) -> Result<Vec<super::model::Command>> {
+    let previous_commands = super::schema::commands::dsl::commands
+        .filter(super::schema::commands::columns::id.lt(command.id))
+        .filter(super::schema::commands::columns::session_id.eq(&command.session_id))
+        // Skip any empty blocks
+        .filter(super::schema::commands::columns::command.ne(""))
+        .order(super::schema::commands::columns::id.desc())
+        .limit(num_commands as i64)
+        .load::<super::model::Command>(conn)?;
+    Ok(previous_commands.into_iter().rev().collect())
+}
+
+/// Gets the last num_commands times the same command was run in a similar context
+/// (same pwd, exit code, shell, hostname), from newest to oldest.
+pub fn get_same_commands_from_history(
+    conn: &mut SqliteConnection,
+    completed_block: &UserBlockCompleted,
+    num_commands: usize,
+) -> Result<Vec<super::model::Command>> {
+    let shell_host = completed_block.serialized_block.shell_host.as_ref();
+    let commands = super::schema::commands::dsl::commands
+        .filter(super::schema::commands::columns::command.eq(&completed_block.command))
+        .filter(super::schema::commands::columns::pwd.eq(&completed_block.serialized_block.pwd))
+        .filter(
+            super::schema::commands::columns::exit_code
+                .eq(completed_block.serialized_block.exit_code.value()),
+        )
+        .filter(
+            super::schema::commands::columns::shell
+                .eq(shell_host.map(|host| host.shell_type.name())),
+        )
+        .filter(
+            super::schema::commands::columns::hostname.eq(shell_host.map(|host| &host.hostname)),
+        )
+        // Get newest to oldest commands.
+        .order(super::schema::commands::columns::id.desc())
+        .limit(num_commands as i64)
+        .load::<super::model::Command>(conn)?;
+
+    Ok(commands)
+}