portacode 0.3.23__tar.gz → 0.3.25__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.


This version of portacode might be problematic. Click here for more details.

Files changed (84) hide show
  1. {portacode-0.3.23 → portacode-0.3.25}/PKG-INFO +13 -2
  2. {portacode-0.3.23 → portacode-0.3.25}/docker-compose.yaml +29 -0
  3. {portacode-0.3.23 → portacode-0.3.25}/portacode/_version.py +2 -2
  4. {portacode-0.3.23 → portacode-0.3.25}/portacode/connection/handlers/WEBSOCKET_PROTOCOL.md +39 -12
  5. {portacode-0.3.23 → portacode-0.3.25}/portacode/connection/handlers/project_state/git_manager.py +322 -24
  6. {portacode-0.3.23 → portacode-0.3.25}/portacode/connection/handlers/project_state/handlers.py +114 -30
  7. {portacode-0.3.23 → portacode-0.3.25}/portacode/connection/handlers/session.py +28 -1
  8. {portacode-0.3.23 → portacode-0.3.25}/portacode/connection/handlers/tab_factory.py +1 -1
  9. {portacode-0.3.23 → portacode-0.3.25}/portacode/connection/terminal.py +13 -4
  10. {portacode-0.3.23 → portacode-0.3.25}/portacode.egg-info/PKG-INFO +13 -2
  11. {portacode-0.3.23 → portacode-0.3.25}/test.sh +0 -0
  12. {portacode-0.3.23 → portacode-0.3.25}/.claude/agents/communication-manager.md +0 -0
  13. {portacode-0.3.23 → portacode-0.3.25}/.claude/settings.local.json +0 -0
  14. {portacode-0.3.23 → portacode-0.3.25}/.gitignore +0 -0
  15. {portacode-0.3.23 → portacode-0.3.25}/.gitmodules +0 -0
  16. {portacode-0.3.23 → portacode-0.3.25}/LICENSE +0 -0
  17. {portacode-0.3.23 → portacode-0.3.25}/MANIFEST.in +0 -0
  18. {portacode-0.3.23 → portacode-0.3.25}/Makefile +0 -0
  19. {portacode-0.3.23 → portacode-0.3.25}/README.md +0 -0
  20. {portacode-0.3.23 → portacode-0.3.25}/backup.sh +0 -0
  21. {portacode-0.3.23 → portacode-0.3.25}/connect.py +0 -0
  22. {portacode-0.3.23 → portacode-0.3.25}/connect.sh +0 -0
  23. {portacode-0.3.23 → portacode-0.3.25}/portacode/README.md +0 -0
  24. {portacode-0.3.23 → portacode-0.3.25}/portacode/__init__.py +0 -0
  25. {portacode-0.3.23 → portacode-0.3.25}/portacode/__main__.py +0 -0
  26. {portacode-0.3.23 → portacode-0.3.25}/portacode/cli.py +0 -0
  27. {portacode-0.3.23 → portacode-0.3.25}/portacode/connection/README.md +0 -0
  28. {portacode-0.3.23 → portacode-0.3.25}/portacode/connection/__init__.py +0 -0
  29. {portacode-0.3.23 → portacode-0.3.25}/portacode/connection/client.py +0 -0
  30. {portacode-0.3.23 → portacode-0.3.25}/portacode/connection/handlers/README.md +0 -0
  31. {portacode-0.3.23 → portacode-0.3.25}/portacode/connection/handlers/__init__.py +0 -0
  32. {portacode-0.3.23 → portacode-0.3.25}/portacode/connection/handlers/base.py +0 -0
  33. {portacode-0.3.23 → portacode-0.3.25}/portacode/connection/handlers/chunked_content.py +0 -0
  34. {portacode-0.3.23 → portacode-0.3.25}/portacode/connection/handlers/file_handlers.py +0 -0
  35. {portacode-0.3.23 → portacode-0.3.25}/portacode/connection/handlers/project_aware_file_handlers.py +0 -0
  36. {portacode-0.3.23 → portacode-0.3.25}/portacode/connection/handlers/project_state/README.md +0 -0
  37. {portacode-0.3.23 → portacode-0.3.25}/portacode/connection/handlers/project_state/__init__.py +0 -0
  38. {portacode-0.3.23 → portacode-0.3.25}/portacode/connection/handlers/project_state/file_system_watcher.py +0 -0
  39. {portacode-0.3.23 → portacode-0.3.25}/portacode/connection/handlers/project_state/manager.py +0 -0
  40. {portacode-0.3.23 → portacode-0.3.25}/portacode/connection/handlers/project_state/models.py +0 -0
  41. {portacode-0.3.23 → portacode-0.3.25}/portacode/connection/handlers/project_state/utils.py +0 -0
  42. {portacode-0.3.23 → portacode-0.3.25}/portacode/connection/handlers/project_state_handlers.py +0 -0
  43. {portacode-0.3.23 → portacode-0.3.25}/portacode/connection/handlers/registry.py +0 -0
  44. {portacode-0.3.23 → portacode-0.3.25}/portacode/connection/handlers/system_handlers.py +0 -0
  45. {portacode-0.3.23 → portacode-0.3.25}/portacode/connection/handlers/terminal_handlers.py +0 -0
  46. {portacode-0.3.23 → portacode-0.3.25}/portacode/connection/multiplex.py +0 -0
  47. {portacode-0.3.23 → portacode-0.3.25}/portacode/data.py +0 -0
  48. {portacode-0.3.23 → portacode-0.3.25}/portacode/keypair.py +0 -0
  49. {portacode-0.3.23 → portacode-0.3.25}/portacode/logging_categories.py +0 -0
  50. {portacode-0.3.23 → portacode-0.3.25}/portacode/service.py +0 -0
  51. {portacode-0.3.23 → portacode-0.3.25}/portacode.egg-info/SOURCES.txt +0 -0
  52. {portacode-0.3.23 → portacode-0.3.25}/portacode.egg-info/dependency_links.txt +0 -0
  53. {portacode-0.3.23 → portacode-0.3.25}/portacode.egg-info/entry_points.txt +0 -0
  54. {portacode-0.3.23 → portacode-0.3.25}/portacode.egg-info/requires.txt +0 -0
  55. {portacode-0.3.23 → portacode-0.3.25}/portacode.egg-info/top_level.txt +0 -0
  56. {portacode-0.3.23 → portacode-0.3.25}/pyproject.toml +0 -0
  57. {portacode-0.3.23 → portacode-0.3.25}/restore.sh +0 -0
  58. {portacode-0.3.23 → portacode-0.3.25}/run_tests.py +0 -0
  59. {portacode-0.3.23 → portacode-0.3.25}/setup.cfg +0 -0
  60. {portacode-0.3.23 → portacode-0.3.25}/setup.py +0 -0
  61. {portacode-0.3.23 → portacode-0.3.25}/test_modules/README.md +0 -0
  62. {portacode-0.3.23 → portacode-0.3.25}/test_modules/__init__.py +0 -0
  63. {portacode-0.3.23 → portacode-0.3.25}/test_modules/test_device_online.py +0 -0
  64. {portacode-0.3.23 → portacode-0.3.25}/test_modules/test_file_operations.py +0 -0
  65. {portacode-0.3.23 → portacode-0.3.25}/test_modules/test_git_status_ui.py +0 -0
  66. {portacode-0.3.23 → portacode-0.3.25}/test_modules/test_login_flow.py +0 -0
  67. {portacode-0.3.23 → portacode-0.3.25}/test_modules/test_navigate_testing_folder.py +0 -0
  68. {portacode-0.3.23 → portacode-0.3.25}/test_modules/test_terminal_buffer_performance.py +0 -0
  69. {portacode-0.3.23 → portacode-0.3.25}/test_modules/test_terminal_interaction.py +0 -0
  70. {portacode-0.3.23 → portacode-0.3.25}/test_modules/test_terminal_loading_race_condition.py +0 -0
  71. {portacode-0.3.23 → portacode-0.3.25}/test_modules/test_terminal_start.py +0 -0
  72. {portacode-0.3.23 → portacode-0.3.25}/testing_framework/.env.example +0 -0
  73. {portacode-0.3.23 → portacode-0.3.25}/testing_framework/README.md +0 -0
  74. {portacode-0.3.23 → portacode-0.3.25}/testing_framework/__init__.py +0 -0
  75. {portacode-0.3.23 → portacode-0.3.25}/testing_framework/cli.py +0 -0
  76. {portacode-0.3.23 → portacode-0.3.25}/testing_framework/core/__init__.py +0 -0
  77. {portacode-0.3.23 → portacode-0.3.25}/testing_framework/core/base_test.py +0 -0
  78. {portacode-0.3.23 → portacode-0.3.25}/testing_framework/core/cli_manager.py +0 -0
  79. {portacode-0.3.23 → portacode-0.3.25}/testing_framework/core/hierarchical_runner.py +0 -0
  80. {portacode-0.3.23 → portacode-0.3.25}/testing_framework/core/playwright_manager.py +0 -0
  81. {portacode-0.3.23 → portacode-0.3.25}/testing_framework/core/runner.py +0 -0
  82. {portacode-0.3.23 → portacode-0.3.25}/testing_framework/core/shared_cli_manager.py +0 -0
  83. {portacode-0.3.23 → portacode-0.3.25}/testing_framework/core/test_discovery.py +0 -0
  84. {portacode-0.3.23 → portacode-0.3.25}/testing_framework/requirements.txt +0 -0
@@ -1,6 +1,6 @@
1
- Metadata-Version: 2.1
1
+ Metadata-Version: 2.4
2
2
  Name: portacode
3
- Version: 0.3.23
3
+ Version: 0.3.25
4
4
  Summary: Portacode CLI client and SDK
5
5
  Home-page: https://github.com/portacode/portacode
6
6
  Author: Meena Erian
@@ -27,6 +27,17 @@ Provides-Extra: dev
27
27
  Requires-Dist: black; extra == "dev"
28
28
  Requires-Dist: flake8; extra == "dev"
29
29
  Requires-Dist: pytest; extra == "dev"
30
+ Dynamic: author
31
+ Dynamic: author-email
32
+ Dynamic: classifier
33
+ Dynamic: description
34
+ Dynamic: description-content-type
35
+ Dynamic: home-page
36
+ Dynamic: license-file
37
+ Dynamic: provides-extra
38
+ Dynamic: requires-dist
39
+ Dynamic: requires-python
40
+ Dynamic: summary
30
41
 
31
42
  # Portacode
32
43
 
@@ -32,6 +32,35 @@ services:
32
32
  volumes:
33
33
  - ./server/portacode_django:/app
34
34
 
35
+ unibot-worker:
36
+ build:
37
+ context: ./server/portacode_django
38
+ dockerfile: Dockerfile
39
+ container_name: portacode-unibot-worker
40
+ restart: unless-stopped
41
+ command: ["python", "manage.py", "run_worker"]
42
+ env_file:
43
+ - main.env
44
+ depends_on:
45
+ db:
46
+ condition: service_healthy
47
+ volumes:
48
+ - ./server/portacode_django:/app
49
+
50
+ imap-worker:
51
+ build:
52
+ context: ./server/portacode_django
53
+ dockerfile: Dockerfile
54
+ container_name: portacode-imap-worker
55
+ restart: unless-stopped
56
+ command: ["python", "manage.py", "start_imap_listeners"]
57
+ env_file:
58
+ - main.env
59
+ depends_on:
60
+ db:
61
+ condition: service_healthy
62
+ volumes:
63
+ - ./server/portacode_django:/app
35
64
 
36
65
  volumes:
37
66
  pgdata:
@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
28
28
  commit_id: COMMIT_ID
29
29
  __commit_id__: COMMIT_ID
30
30
 
31
- __version__ = version = '0.3.23'
32
- __version_tuple__ = version_tuple = (0, 3, 23)
31
+ __version__ = version = '0.3.25'
32
+ __version_tuple__ = version_tuple = (0, 3, 25)
33
33
 
34
34
  __commit_id__ = commit_id = None
@@ -458,12 +458,19 @@ Requests the content for a specific diff tab identified by its diff parameters.
458
458
 
459
459
  ### `project_state_git_stage`
460
460
 
461
- Stages a file for commit in the project's git repository. Handled by [`project_state_git_stage`](./project_state_handlers.py).
461
+ Stages file(s) for commit in the project's git repository. Supports both single file and bulk operations. Handled by [`project_state_git_stage`](./project_state_handlers.py).
462
462
 
463
463
  **Payload Fields:**
464
464
 
465
465
  * `project_id` (string, mandatory): The project ID from the initialized project state.
466
- * `file_path` (string, mandatory): The absolute path to the file to stage.
466
+ * `file_path` (string, optional): The absolute path to a single file to stage. Used for backward compatibility.
467
+ * `file_paths` (array of strings, optional): Array of absolute paths to files to stage. Used for bulk operations.
468
+ * `stage_all` (boolean, optional): If true, stages all unstaged changes in the repository. Takes precedence over file_path/file_paths.
469
+
470
+ **Operation Modes:**
471
+ - Single file: Provide `file_path`
472
+ - Bulk operation: Provide `file_paths` array
473
+ - Stage all: Set `stage_all` to true
467
474
 
468
475
  **Responses:**
469
476
 
@@ -472,12 +479,19 @@ Stages a file for commit in the project's git repository. Handled by [`project_s
472
479
 
473
480
  ### `project_state_git_unstage`
474
481
 
475
- Unstages a file (removes from staging area) in the project's git repository. Handled by [`project_state_git_unstage`](./project_state_handlers.py).
482
+ Unstages file(s) (removes from staging area) in the project's git repository. Supports both single file and bulk operations. Handled by [`project_state_git_unstage`](./project_state_handlers.py).
476
483
 
477
484
  **Payload Fields:**
478
485
 
479
486
  * `project_id` (string, mandatory): The project ID from the initialized project state.
480
- * `file_path` (string, mandatory): The absolute path to the file to unstage.
487
+ * `file_path` (string, optional): The absolute path to a single file to unstage. Used for backward compatibility.
488
+ * `file_paths` (array of strings, optional): Array of absolute paths to files to unstage. Used for bulk operations.
489
+ * `unstage_all` (boolean, optional): If true, unstages all staged changes in the repository. Takes precedence over file_path/file_paths.
490
+
491
+ **Operation Modes:**
492
+ - Single file: Provide `file_path`
493
+ - Bulk operation: Provide `file_paths` array
494
+ - Unstage all: Set `unstage_all` to true
481
495
 
482
496
  **Responses:**
483
497
 
@@ -486,12 +500,19 @@ Unstages a file (removes from staging area) in the project's git repository. Han
486
500
 
487
501
  ### `project_state_git_revert`
488
502
 
489
- Reverts a file to its HEAD version, discarding local changes in the project's git repository. Handled by [`project_state_git_revert`](./project_state_handlers.py).
503
+ Reverts file(s) to their HEAD version, discarding local changes in the project's git repository. Supports both single file and bulk operations. Handled by [`project_state_git_revert`](./project_state_handlers.py).
490
504
 
491
505
  **Payload Fields:**
492
506
 
493
507
  * `project_id` (string, mandatory): The project ID from the initialized project state.
494
- * `file_path` (string, mandatory): The absolute path to the file to revert.
508
+ * `file_path` (string, optional): The absolute path to a single file to revert. Used for backward compatibility.
509
+ * `file_paths` (array of strings, optional): Array of absolute paths to files to revert. Used for bulk operations.
510
+ * `revert_all` (boolean, optional): If true, reverts all unstaged changes in the repository. Takes precedence over file_path/file_paths.
511
+
512
+ **Operation Modes:**
513
+ - Single file: Provide `file_path`
514
+ - Bulk operation: Provide `file_paths` array
515
+ - Revert all: Set `revert_all` to true
495
516
 
496
517
  **Responses:**
497
518
 
@@ -1019,34 +1040,40 @@ Returns the requested content for a specific diff tab, sent in response to a [`p
1019
1040
 
1020
1041
  ### <a name="project_state_git_stage_response"></a>`project_state_git_stage_response`
1021
1042
 
1022
- Confirms the result of a git stage operation.
1043
+ Confirms the result of a git stage operation. Supports responses for both single file and bulk operations.
1023
1044
 
1024
1045
  **Event Fields:**
1025
1046
 
1026
1047
  * `project_id` (string, mandatory): The project ID the operation was performed on.
1027
- * `file_path` (string, mandatory): The path to the file that was staged.
1048
+ * `file_path` (string, optional): The path to the file that was staged (for single file operations).
1049
+ * `file_paths` (array of strings, optional): Array of paths to files that were staged (for bulk operations).
1050
+ * `stage_all` (boolean, optional): Present if the operation was a "stage all" operation.
1028
1051
  * `success` (boolean, mandatory): Whether the stage operation was successful.
1029
1052
  * `error` (string, optional): Error message if the operation failed.
1030
1053
 
1031
1054
  ### <a name="project_state_git_unstage_response"></a>`project_state_git_unstage_response`
1032
1055
 
1033
- Confirms the result of a git unstage operation.
1056
+ Confirms the result of a git unstage operation. Supports responses for both single file and bulk operations.
1034
1057
 
1035
1058
  **Event Fields:**
1036
1059
 
1037
1060
  * `project_id` (string, mandatory): The project ID the operation was performed on.
1038
- * `file_path` (string, mandatory): The path to the file that was unstaged.
1061
+ * `file_path` (string, optional): The path to the file that was unstaged (for single file operations).
1062
+ * `file_paths` (array of strings, optional): Array of paths to files that were unstaged (for bulk operations).
1063
+ * `unstage_all` (boolean, optional): Present if the operation was an "unstage all" operation.
1039
1064
  * `success` (boolean, mandatory): Whether the unstage operation was successful.
1040
1065
  * `error` (string, optional): Error message if the operation failed.
1041
1066
 
1042
1067
  ### <a name="project_state_git_revert_response"></a>`project_state_git_revert_response`
1043
1068
 
1044
- Confirms the result of a git revert operation.
1069
+ Confirms the result of a git revert operation. Supports responses for both single file and bulk operations.
1045
1070
 
1046
1071
  **Event Fields:**
1047
1072
 
1048
1073
  * `project_id` (string, mandatory): The project ID the operation was performed on.
1049
- * `file_path` (string, mandatory): The path to the file that was reverted.
1074
+ * `file_path` (string, optional): The path to the file that was reverted (for single file operations).
1075
+ * `file_paths` (array of strings, optional): Array of paths to files that were reverted (for bulk operations).
1076
+ * `revert_all` (boolean, optional): Present if the operation was a "revert all" operation.
1050
1077
  * `success` (boolean, mandatory): Whether the revert operation was successful.
1051
1078
  * `error` (string, optional): Error message if the operation failed.
1052
1079
 
@@ -181,22 +181,71 @@ class GitManager:
181
181
  if is_ignored:
182
182
  return {"is_tracked": False, "status": "ignored", "is_ignored": True, "is_staged": False}
183
183
 
184
- # For directories, only report status if they contain tracked or untracked files
184
+ # For directories, aggregate status from contained files
185
185
  if os.path.isdir(file_path):
186
- # Check if directory contains any untracked files using path.startswith()
187
- # This handles cross-platform path separators correctly
188
- has_untracked = any(
189
- os.path.commonpath([f, rel_path]) == rel_path and f != rel_path
190
- for f in self.repo.untracked_files
191
- )
186
+ # Normalize the relative path for cross-platform compatibility
187
+ rel_path_normalized = rel_path.replace('\\', '/')
188
+
189
+ # Check for untracked files in this directory
190
+ has_untracked = False
191
+ for untracked_file in self.repo.untracked_files:
192
+ untracked_normalized = untracked_file.replace('\\', '/')
193
+ if untracked_normalized.startswith(rel_path_normalized + '/') or untracked_normalized == rel_path_normalized:
194
+ has_untracked = True
195
+ break
196
+
197
+ # Check for modified files in this directory using git status
198
+ has_modified = False
199
+ has_deleted = False
200
+ try:
201
+ # Get status for files in this directory
202
+ status_output = self.repo.git.status(rel_path, porcelain=True)
203
+ if status_output.strip():
204
+ for line in status_output.strip().split('\n'):
205
+ if len(line) >= 2:
206
+ # When filtering git status by path, GitPython strips the leading space
207
+ # So format is either "XY filename" or " XY filename"
208
+ if line.startswith(' '):
209
+ # Full status format: " XY filename"
210
+ index_status = line[0]
211
+ worktree_status = line[1]
212
+ file_path_from_status = line[3:] if len(line) > 3 else ""
213
+ else:
214
+ # Path-filtered format: "XY filename" (leading space stripped)
215
+ # Two possible formats:
216
+ # 1. Regular files: "M filename" (index + worktree + space + filename)
217
+ # 2. Submodules: "M filename" (index + space + filename)
218
+ index_status = line[0] if len(line) > 0 else ' '
219
+ worktree_status = line[1] if len(line) > 1 else ' '
220
+
221
+ # Detect format by checking if position 2 is a space
222
+ if len(line) > 2 and line[2] == ' ':
223
+ # Regular file format: "M filename"
224
+ file_path_from_status = line[3:] if len(line) > 3 else ""
225
+ else:
226
+ # Submodule format: "M filename"
227
+ file_path_from_status = line[2:] if len(line) > 2 else ""
228
+
229
+ # Check if this file is within our directory
230
+ file_normalized = file_path_from_status.replace('\\', '/')
231
+ if (file_normalized.startswith(rel_path_normalized + '/') or
232
+ file_normalized == rel_path_normalized):
233
+ if index_status in ['M', 'A', 'R', 'C'] or worktree_status in ['M', 'A', 'R', 'C']:
234
+ has_modified = True
235
+ elif index_status == 'D' or worktree_status == 'D':
236
+ has_deleted = True
237
+ except Exception as e:
238
+ logger.debug("Error checking directory git status for %s: %s", rel_path, e)
239
+
240
+ # Priority order: untracked > modified/deleted > clean
192
241
  if has_untracked:
193
242
  return {"is_tracked": False, "status": "untracked", "is_ignored": False, "is_staged": is_staged}
194
-
195
- # Check if directory is dirty - GitPython handles path normalization
196
- if self.repo.is_dirty(path=rel_path):
243
+ elif has_deleted:
244
+ return {"is_tracked": True, "status": "deleted", "is_ignored": False, "is_staged": is_staged}
245
+ elif has_modified:
197
246
  return {"is_tracked": True, "status": "modified", "is_ignored": False, "is_staged": is_staged}
198
247
 
199
- # Check if directory has tracked files - let GitPython handle paths
248
+ # Check if directory has tracked files to determine if it should show as clean
200
249
  try:
201
250
  tracked_files = self.repo.git.ls_files(rel_path)
202
251
  is_tracked = bool(tracked_files.strip())
@@ -1134,6 +1183,41 @@ class GitManager:
1134
1183
  logger.error("Error getting staged content for %s: %s", file_path, e)
1135
1184
  return None
1136
1185
 
1186
+ def _is_submodule(self, file_path: str) -> bool:
1187
+ """Check if the given path is a submodule."""
1188
+ if not self.is_git_repo or not self.repo:
1189
+ return False
1190
+
1191
+ try:
1192
+ # Convert to relative path from repo root
1193
+ rel_path = os.path.relpath(file_path, self.repo.working_dir)
1194
+
1195
+ # Check if this path is listed in .gitmodules
1196
+ gitmodules_path = os.path.join(self.repo.working_dir, '.gitmodules')
1197
+ if os.path.exists(gitmodules_path):
1198
+ try:
1199
+ with open(gitmodules_path, 'r') as f:
1200
+ content = f.read()
1201
+ # Simple check - look for path = rel_path in .gitmodules
1202
+ for line in content.splitlines():
1203
+ if line.strip().startswith('path ='):
1204
+ submodule_path = line.split('=', 1)[1].strip()
1205
+ if submodule_path == rel_path:
1206
+ return True
1207
+ except Exception as e:
1208
+ logger.warning("Error reading .gitmodules: %s", e)
1209
+
1210
+ # Alternative check: see if the path has a .git file (submodule indicator)
1211
+ git_path = os.path.join(file_path, '.git')
1212
+ if os.path.isfile(git_path): # Submodules have .git as a file, not directory
1213
+ return True
1214
+
1215
+ return False
1216
+
1217
+ except Exception as e:
1218
+ logger.warning("Error checking if %s is submodule: %s", file_path, e)
1219
+ return False
1220
+
1137
1221
  def stage_file(self, file_path: str) -> bool:
1138
1222
  """Stage a file for commit."""
1139
1223
  if not self.is_git_repo or not self.repo:
@@ -1143,8 +1227,15 @@ class GitManager:
1143
1227
  # Convert to relative path from repo root
1144
1228
  rel_path = os.path.relpath(file_path, self.repo.working_dir)
1145
1229
 
1146
- # Stage the file
1147
- self.repo.index.add([rel_path])
1230
+ # Check if this is a submodule
1231
+ if self._is_submodule(file_path):
1232
+ logger.info("Detected submodule, using git add command directly: %s", rel_path)
1233
+ # For submodules, use git add directly to stage only the submodule reference
1234
+ self.repo.git.add(rel_path)
1235
+ else:
1236
+ # For regular files, use the index method
1237
+ self.repo.index.add([rel_path])
1238
+
1148
1239
  logger.info("Successfully staged file: %s", rel_path)
1149
1240
  return True
1150
1241
 
@@ -1161,19 +1252,25 @@ class GitManager:
1161
1252
  # Convert to relative path from repo root
1162
1253
  rel_path = os.path.relpath(file_path, self.repo.working_dir)
1163
1254
 
1164
- # Check if repository has any commits (HEAD exists)
1165
- try:
1166
- self.repo.head.commit
1167
- has_head = True
1168
- except Exception:
1169
- has_head = False
1170
-
1171
- if has_head:
1172
- # Reset the file from HEAD (unstage) - for repos with commits
1255
+ # Check if this is a submodule
1256
+ if self._is_submodule(file_path):
1257
+ logger.info("Detected submodule, using git restore for unstaging: %s", rel_path)
1258
+ # For submodules, always use git restore --staged (works with submodules)
1173
1259
  self.repo.git.restore('--staged', rel_path)
1174
1260
  else:
1175
- # For repositories with no commits, use git rm --cached to unstage
1176
- self.repo.git.rm('--cached', rel_path)
1261
+ # Check if repository has any commits (HEAD exists)
1262
+ try:
1263
+ self.repo.head.commit
1264
+ has_head = True
1265
+ except Exception:
1266
+ has_head = False
1267
+
1268
+ if has_head:
1269
+ # Reset the file from HEAD (unstage) - for repos with commits
1270
+ self.repo.git.restore('--staged', rel_path)
1271
+ else:
1272
+ # For repositories with no commits, use git rm --cached to unstage
1273
+ self.repo.git.rm('--cached', rel_path)
1177
1274
 
1178
1275
  logger.info("Successfully unstaged file: %s", rel_path)
1179
1276
  return True
@@ -1217,6 +1314,207 @@ class GitManager:
1217
1314
  logger.error("Error reverting file %s: %s", file_path, e)
1218
1315
  raise RuntimeError(f"Failed to revert file: {e}")
1219
1316
 
1317
+ def stage_files(self, file_paths: List[str]) -> bool:
1318
+ """Stage multiple files for commit in one atomic operation."""
1319
+ if not self.is_git_repo or not self.repo:
1320
+ raise RuntimeError("Not a git repository")
1321
+
1322
+ if not file_paths:
1323
+ logger.info("No files provided for staging")
1324
+ return True
1325
+
1326
+ try:
1327
+ # Convert all paths to relative paths from repo root
1328
+ rel_paths = []
1329
+ submodule_paths = []
1330
+
1331
+ for file_path in file_paths:
1332
+ rel_path = os.path.relpath(file_path, self.repo.working_dir)
1333
+ if self._is_submodule(file_path):
1334
+ submodule_paths.append(rel_path)
1335
+ else:
1336
+ rel_paths.append(rel_path)
1337
+
1338
+ # Stage submodules using git add directly
1339
+ if submodule_paths:
1340
+ logger.info("Staging submodules using git add directly: %s", submodule_paths)
1341
+ for submodule_path in submodule_paths:
1342
+ self.repo.git.add(submodule_path)
1343
+
1344
+ # Stage regular files using index.add for efficiency
1345
+ if rel_paths:
1346
+ logger.info("Staging regular files: %s", rel_paths)
1347
+ self.repo.index.add(rel_paths)
1348
+
1349
+ logger.info("Successfully staged %d files (%d submodules, %d regular)",
1350
+ len(file_paths), len(submodule_paths), len(rel_paths))
1351
+ return True
1352
+
1353
+ except Exception as e:
1354
+ logger.error("Error staging files %s: %s", file_paths, e)
1355
+ raise RuntimeError(f"Failed to stage files: {e}")
1356
+
1357
+ def unstage_files(self, file_paths: List[str]) -> bool:
1358
+ """Unstage multiple files in one atomic operation."""
1359
+ if not self.is_git_repo or not self.repo:
1360
+ raise RuntimeError("Not a git repository")
1361
+
1362
+ if not file_paths:
1363
+ logger.info("No files provided for unstaging")
1364
+ return True
1365
+
1366
+ try:
1367
+ # Convert all paths to relative paths from repo root
1368
+ rel_paths = []
1369
+ submodule_paths = []
1370
+
1371
+ for file_path in file_paths:
1372
+ rel_path = os.path.relpath(file_path, self.repo.working_dir)
1373
+ if self._is_submodule(file_path):
1374
+ submodule_paths.append(rel_path)
1375
+ else:
1376
+ rel_paths.append(rel_path)
1377
+
1378
+ # Check if repository has any commits (HEAD exists)
1379
+ try:
1380
+ self.repo.head.commit
1381
+ has_head = True
1382
+ except Exception:
1383
+ has_head = False
1384
+
1385
+ # Unstage all files using appropriate method
1386
+ all_rel_paths = rel_paths + submodule_paths
1387
+
1388
+ if has_head:
1389
+ # Use git restore --staged for all files (works for both regular files and submodules)
1390
+ if all_rel_paths:
1391
+ self.repo.git.restore('--staged', *all_rel_paths)
1392
+ else:
1393
+ # For repositories with no commits, use git rm --cached
1394
+ if all_rel_paths:
1395
+ self.repo.git.rm('--cached', *all_rel_paths)
1396
+
1397
+ logger.info("Successfully unstaged %d files (%d submodules, %d regular)",
1398
+ len(file_paths), len(submodule_paths), len(rel_paths))
1399
+ return True
1400
+
1401
+ except Exception as e:
1402
+ logger.error("Error unstaging files %s: %s", file_paths, e)
1403
+ raise RuntimeError(f"Failed to unstage files: {e}")
1404
+
1405
+ def revert_files(self, file_paths: List[str]) -> bool:
1406
+ """Revert multiple files to their HEAD version in one atomic operation."""
1407
+ if not self.is_git_repo or not self.repo:
1408
+ raise RuntimeError("Not a git repository")
1409
+
1410
+ if not file_paths:
1411
+ logger.info("No files provided for reverting")
1412
+ return True
1413
+
1414
+ try:
1415
+ # Check if repository has any commits (HEAD exists)
1416
+ try:
1417
+ self.repo.head.commit
1418
+ has_head = True
1419
+ except Exception:
1420
+ has_head = False
1421
+
1422
+ if has_head:
1423
+ # Convert to relative paths and restore all files at once
1424
+ rel_paths = [os.path.relpath(file_path, self.repo.working_dir) for file_path in file_paths]
1425
+ # Filter out submodules - we don't revert submodules as they don't have working directory changes
1426
+ regular_files = []
1427
+ for i, file_path in enumerate(file_paths):
1428
+ if not self._is_submodule(file_path):
1429
+ regular_files.append(rel_paths[i])
1430
+
1431
+ if regular_files:
1432
+ self.repo.git.restore(*regular_files)
1433
+ logger.info("Successfully reverted %d files", len(regular_files))
1434
+ else:
1435
+ # For repositories with no commits, remove files to "revert" them
1436
+ removed_count = 0
1437
+ for file_path in file_paths:
1438
+ if not self._is_submodule(file_path) and os.path.exists(file_path):
1439
+ os.remove(file_path)
1440
+ removed_count += 1
1441
+ logger.info("Successfully removed %d files (no HEAD to revert to)", removed_count)
1442
+
1443
+ return True
1444
+
1445
+ except Exception as e:
1446
+ logger.error("Error reverting files %s: %s", file_paths, e)
1447
+ raise RuntimeError(f"Failed to revert files: {e}")
1448
+
1449
+ def stage_all_changes(self) -> bool:
1450
+ """Stage all changes (modified, deleted, untracked) in one atomic operation."""
1451
+ if not self.is_git_repo or not self.repo:
1452
+ raise RuntimeError("Not a git repository")
1453
+
1454
+ try:
1455
+ # Use git add . to stage everything - this is the most efficient way
1456
+ self.repo.git.add('.')
1457
+ logger.info("Successfully staged all changes using 'git add .'")
1458
+ return True
1459
+
1460
+ except Exception as e:
1461
+ logger.error("Error staging all changes: %s", e)
1462
+ raise RuntimeError(f"Failed to stage all changes: {e}")
1463
+
1464
+ def unstage_all_changes(self) -> bool:
1465
+ """Unstage all staged changes in one atomic operation."""
1466
+ if not self.is_git_repo or not self.repo:
1467
+ raise RuntimeError("Not a git repository")
1468
+
1469
+ try:
1470
+ # Check if repository has any commits (HEAD exists)
1471
+ try:
1472
+ self.repo.head.commit
1473
+ has_head = True
1474
+ except Exception:
1475
+ has_head = False
1476
+
1477
+ if has_head:
1478
+ # Use git restore --staged . to unstage everything
1479
+ self.repo.git.restore('--staged', '.')
1480
+ else:
1481
+ # For repositories with no commits, remove everything from index
1482
+ self.repo.git.rm('--cached', '-r', '.')
1483
+
1484
+ logger.info("Successfully unstaged all changes")
1485
+ return True
1486
+
1487
+ except Exception as e:
1488
+ logger.error("Error unstaging all changes: %s", e)
1489
+ raise RuntimeError(f"Failed to unstage all changes: {e}")
1490
+
1491
+ def revert_all_changes(self) -> bool:
1492
+ """Revert all working directory changes in one atomic operation."""
1493
+ if not self.is_git_repo or not self.repo:
1494
+ raise RuntimeError("Not a git repository")
1495
+
1496
+ try:
1497
+ # Check if repository has any commits (HEAD exists)
1498
+ try:
1499
+ self.repo.head.commit
1500
+ has_head = True
1501
+ except Exception:
1502
+ has_head = False
1503
+
1504
+ if has_head:
1505
+ # Use git restore . to revert all working directory changes
1506
+ self.repo.git.restore('.')
1507
+ logger.info("Successfully reverted all working directory changes")
1508
+ else:
1509
+ logger.warning("Cannot revert changes in repository with no commits")
1510
+ return False
1511
+
1512
+ return True
1513
+
1514
+ except Exception as e:
1515
+ logger.error("Error reverting all changes: %s", e)
1516
+ raise RuntimeError(f"Failed to revert all changes: {e}")
1517
+
1220
1518
  def commit_changes(self, message: str) -> bool:
1221
1519
  """Commit staged changes with the given message."""
1222
1520
  if not self.is_git_repo or not self.repo: