Implementing persistence for the terminal working directory, found an issue with my current data model. :(

crates/terminal/Cargo.toml

@@ -17,6 +17,7 @@ settings = { path = "../settings" }
 theme = { path = "../theme" }
 util = { path = "../util" }
 workspace = { path = "../workspace" }
+db = { path = "../db" }
 alacritty_terminal = { git = "https://github.com/zed-industries/alacritty", rev = "a51dbe25d67e84d6ed4261e640d3954fbdd9be45" }
 procinfo = { git = "https://github.com/zed-industries/wezterm", rev = "5cd757e5f2eb039ed0c6bb6512223e69d5efc64d", default-features = false }
 smallvec = { version = "1.6", features = ["union"] }

crates/terminal/src/persistence.rs

@@ -0,0 +1,61 @@
+use std::path::{Path, PathBuf};
+
+use db::{connection, indoc, sqlez::domain::Domain};
+use util::{iife, ResultExt};
+use workspace::{ItemId, Workspace, WorkspaceId};
+
+use crate::Terminal;
+
+connection!(TERMINAL_CONNECTION: TerminalDb<(Workspace, Terminal)>);
+
+impl Domain for Terminal {
+    fn name() -> &'static str {
+        "terminal"
+    }
+
+    fn migrations() -> &'static [&'static str] {
+        &[indoc! {"
+            CREATE TABLE terminals (
+                item_id INTEGER,
+                workspace_id BLOB,
+                working_directory BLOB,
+                PRIMARY KEY(item_id, workspace_id),
+                FOREIGN KEY(workspace_id) REFERENCES workspaces(workspace_id)
+                    ON DELETE CASCADE
+                    ON UPDATE CASCADE
+            ) STRICT;
+        "}]
+    }
+}
+
+impl TerminalDb {
+    pub fn save_working_directory(
+        &self,
+        item_id: ItemId,
+        workspace_id: &WorkspaceId,
+        working_directory: &Path,
+    ) {
+        iife!({
+            self.exec_bound::<(ItemId, &WorkspaceId, &Path)>(indoc! {"
+                INSERT OR REPLACE INTO terminals(item_id, workspace_id, working_directory) 
+                VALUES (?, ?, ?)  
+            "})?((item_id, workspace_id, working_directory))
+        })
+        .log_err();
+    }
+
+    pub fn get_working_directory(
+        &self,
+        item_id: ItemId,
+        workspace_id: &WorkspaceId,
+    ) -> Option<PathBuf> {
+        iife!({
+            self.select_row_bound::<(ItemId, &WorkspaceId), PathBuf>(indoc! {"
+            SELECT working_directory 
+            FROM terminals 
+            WHERE item_id = ? workspace_id = ?"})?((item_id, workspace_id))
+        })
+        .log_err()
+        .flatten()
+    }
+}

crates/terminal/src/terminal.rs

@@ -1,4 +1,5 @@
 pub mod mappings;
+mod persistence;
 pub mod terminal_container_view;
 pub mod terminal_element;
 pub mod terminal_view;
@@ -32,9 +33,11 @@ use mappings::mouse::{
     alt_scroll, grid_point, mouse_button_report, mouse_moved_report, mouse_side, scroll_report,
 };
 
+use persistence::TERMINAL_CONNECTION;
 use procinfo::LocalProcessInfo;
 use settings::{AlternateScroll, Settings, Shell, TerminalBlink};
 use util::ResultExt;
+use workspace::{ItemId, WorkspaceId};
 
 use std::{
     cmp::min,
@@ -281,6 +284,8 @@ impl TerminalBuilder {
         blink_settings: Option<TerminalBlink>,
         alternate_scroll: &AlternateScroll,
         window_id: usize,
+        item_id: ItemId,
+        workspace_id: WorkspaceId,
     ) -> Result<TerminalBuilder> {
         let pty_config = {
             let alac_shell = shell.clone().and_then(|shell| match shell {
@@ -385,6 +390,8 @@ impl TerminalBuilder {
             last_mouse_position: None,
             next_link_id: 0,
             selection_phase: SelectionPhase::Ended,
+            workspace_id,
+            item_id,
         };
 
         Ok(TerminalBuilder {
@@ -528,6 +535,8 @@ pub struct Terminal {
     scroll_px: f32,
     next_link_id: usize,
     selection_phase: SelectionPhase,
+    item_id: ItemId,
+    workspace_id: WorkspaceId,
 }
 
 impl Terminal {
@@ -567,7 +576,17 @@ impl Terminal {
                 cx.emit(Event::Wakeup);
 
                 if self.update_process_info() {
-                    cx.emit(Event::TitleChanged)
+                    cx.emit(Event::TitleChanged);
+
+                    if let Some(foreground_info) = self.foreground_process_info {
+                        cx.background().spawn(async move {
+                            TERMINAL_CONNECTION.save_working_directory(
+                                self.item_id,
+                                &self.workspace_id,
+                                &foreground_info.cwd,
+                            );
+                        });
+                    }
                 }
             }
             AlacTermEvent::ColorRequest(idx, fun_ptr) => {

crates/terminal/src/terminal_container_view.rs

@@ -1,3 +1,4 @@
+use crate::persistence::TERMINAL_CONNECTION;
 use crate::terminal_view::TerminalView;
 use crate::{Event, Terminal, TerminalBuilder, TerminalError};
 
@@ -13,7 +14,7 @@ use workspace::{
     item::{Item, ItemEvent},
     ToolbarItemLocation, Workspace,
 };
-use workspace::{register_deserializable_item, Pane};
+use workspace::{register_deserializable_item, ItemId, Pane, WorkspaceId};
 
 use project::{LocalWorktree, Project, ProjectPath};
 use settings::{AlternateScroll, Settings, WorkingDirectory};
@@ -89,6 +90,8 @@ impl TerminalContainer {
     pub fn new(
         working_directory: Option<PathBuf>,
         modal: bool,
+        item_id: ItemId,
+        workspace_id: WorkspaceId,
         cx: &mut ViewContext<Self>,
     ) -> Self {
         let settings = cx.global::<Settings>();
@@ -115,6 +118,8 @@ impl TerminalContainer {
             settings.terminal_overrides.blinking.clone(),
             scroll,
             cx.window_id(),
+            item_id,
+            workspace_id,
         ) {
             Ok(terminal) => {
                 let terminal = cx.add_model(|cx| terminal.subscribe(cx));
@@ -386,13 +391,14 @@ impl Item for TerminalContainer {
     fn deserialize(
         _project: ModelHandle<Project>,
         _workspace: WeakViewHandle<Workspace>,
-        _workspace_id: workspace::WorkspaceId,
-        _item_id: workspace::ItemId,
+        workspace_id: workspace::WorkspaceId,
+        item_id: workspace::ItemId,
         cx: &mut ViewContext<Pane>,
     ) -> Task<anyhow::Result<ViewHandle<Self>>> {
-        // TODO: Pull the current working directory out of the DB.
-
-        Task::ready(Ok(cx.add_view(|cx| TerminalContainer::new(None, false, cx))))
+        let working_directory = TERMINAL_CONNECTION.get_working_directory(item_id, &workspace_id);
+        Task::ready(Ok(cx.add_view(|cx| {
+            TerminalContainer::new(working_directory, false, cx)
+        })))
     }
 }