1use std::path::PathBuf;
 2
 3use db::{define_connection, query, sqlez_macros::sql};
 4use gpui::EntityId;
 5use workspace::{ItemId, WorkspaceDb, WorkspaceId};
 6
 7define_connection! {
 8    pub static ref TERMINAL_DB: TerminalDb<WorkspaceDb> =
 9        &[sql!(
10            CREATE TABLE terminals (
11                workspace_id INTEGER,
12                item_id INTEGER UNIQUE,
13                working_directory BLOB,
14                PRIMARY KEY(workspace_id, item_id),
15                FOREIGN KEY(workspace_id) REFERENCES workspaces(workspace_id)
16                ON DELETE CASCADE
17            ) STRICT;
18        ),
19        // Remove the unique constraint on the item_id table
20        // SQLite doesn't have a way of doing this automatically, so
21        // we have to do this silly copying.
22        sql!(
23            CREATE TABLE terminals2 (
24                workspace_id INTEGER,
25                item_id INTEGER,
26                working_directory BLOB,
27                PRIMARY KEY(workspace_id, item_id),
28                FOREIGN KEY(workspace_id) REFERENCES workspaces(workspace_id)
29                ON DELETE CASCADE
30            ) STRICT;
31
32            INSERT INTO terminals2 (workspace_id, item_id, working_directory)
33            SELECT workspace_id, item_id, working_directory FROM terminals;
34
35            DROP TABLE terminals;
36
37            ALTER TABLE terminals2 RENAME TO terminals;
38        )];
39}
40
41impl TerminalDb {
42    query! {
43       pub async fn update_workspace_id(
44            new_id: WorkspaceId,
45            old_id: WorkspaceId,
46            item_id: ItemId
47        ) -> Result<()> {
48            UPDATE terminals
49            SET workspace_id = ?
50            WHERE workspace_id = ? AND item_id = ?
51        }
52    }
53
54    query! {
55        pub async fn save_working_directory(
56            item_id: ItemId,
57            workspace_id: WorkspaceId,
58            working_directory: PathBuf
59        ) -> Result<()> {
60            INSERT OR REPLACE INTO terminals(item_id, workspace_id, working_directory)
61            VALUES (?, ?, ?)
62        }
63    }
64
65    query! {
66        pub fn get_working_directory(item_id: ItemId, workspace_id: WorkspaceId) -> Result<Option<PathBuf>> {
67            SELECT working_directory
68            FROM terminals
69            WHERE item_id = ? AND workspace_id = ?
70        }
71    }
72}