feat: keep track of repo root (#28)

Author: mtth

src/git_draft/bots/openai.py

@@ -109,12 +109,11 @@ def act(self, prompt: str, toolbox: Toolbox) -> Action:
         )
         thread = self._client.beta.threads.create()
 
-        message = self._client.beta.threads.messages.create(
+        self._client.beta.threads.messages.create(
             thread_id=thread.id,
             role="user",
             content=prompt,
         )
-        print(message)
 
         with self._client.beta.threads.runs.stream(
             thread_id=thread.id,

src/git_draft/manager.py

@@ -124,6 +124,7 @@ def _create_branch(self, sync: bool) -> _Branch:
                 sql("add-branch"),
                 {
                     "suffix": suffix,
+                    "repo_path": self._repo.working_dir,
                     "origin_branch": origin_branch,
                     "origin_sha": origin_sha,
                     "sync_sha": sync_sha,

src/git_draft/prompts/update-imports.jinja

@@ -1,2 +1,2 @@
-insert into branches (suffix, origin_branch, origin_sha, sync_sha)
-  values (:suffix, :origin_branch, :origin_sha, :sync_sha);
+insert into branches (suffix, repo_path, origin_branch, origin_sha, sync_sha)
+  values (:suffix, :repo_path, :origin_branch, :origin_sha, :sync_sha);

src/git_draft/queries/add-branch.sql

@@ -1,5 +1,6 @@
 create table if not exists branches (
   suffix text primary key,
+  repo_path text not null,
   created_at timestamp default current_timestamp,
   origin_branch text not null,
   origin_sha text not null,

src/git_draft/queries/create-tables.sql

@@ -14,7 +14,7 @@
 
 
 class Store:
-    _name = "db.v1.sqlite3"
+    _name = "v1.sqlite3"
 
     def __init__(self, conn: sqlite3.Connection) -> None:
         self._connection = conn

src/git_draft/store.py

@@ -6,12 +6,6 @@
 import git_draft.common as sut
 
 
-@pytest.fixture(autouse=True)
-def state_home(monkeypatch, tmp_path) -> None:
-    path = tmp_path / "state"
-    monkeypatch.setenv("XDG_STATE_HOME", str(path))
-
-
 def test_ensure_state_home() -> None:
     path = sut.ensure_state_home()
     assert path.exists()

tests/git_draft/common_test.py

@@ -9,3 +9,9 @@ def repo(tmp_path: Path) -> Iterator[git.Repo]:
     repo = git.Repo.init(str(tmp_path / "repo"), initial_branch="main")
     repo.index.commit("init")
     yield repo
+
+
+@pytest.fixture(autouse=True)
+def state_home(monkeypatch, tmp_path) -> None:
+    path = tmp_path / "state"
+    monkeypatch.setenv("XDG_STATE_HOME", str(path))