pierre-storage 0.11.0__tar.gz → 0.12.1__tar.gz

{pierre_storage-0.11.0/pierre_storage.egg-info → pierre_storage-0.12.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pierre-storage
-Version: 0.11.0
+Version: 0.12.1
 Summary: Pierre Git Storage SDK for Python
 Author-email: Pierre <support@pierre.io>
 License-Expression: MIT
@@ -83,6 +83,15 @@ github_repo = await storage.create_repo(
     }
 )
 # This repository will sync with github.com/octocat/Hello-World
+
+# Create a repository by forking an existing repo
+forked_repo = await storage.create_repo(
+    id="my-fork",
+    base_repo={
+        "id": "my-template-id",
+        "ref": "main",  # optional
+    },
+)
 ```
 
 ### Finding a Repository
@@ -472,6 +481,23 @@ commits = await repo.list_commits()
 3. You can then use all Pierre SDK features (diffs, commits, file access) on the synced content
 4. The provider is automatically set to `"github"` when using `base_repo`
 
+### Forking Repositories
+
+You can fork an existing repository within the same Pierre org:
+
+```python
+forked_repo = await storage.create_repo(
+    id="my-fork",
+    base_repo={
+        "id": "my-template-id",
+        "ref": "main",  # optional (branch/tag)
+        # "sha": "abc123..."  # optional commit SHA (overrides ref)
+    },
+)
+```
+
+When `default_branch` is omitted, the SDK returns `"main"`.
+
 ### Restoring Commits
 
 You can restore a repository to a previous commit:
@@ -500,7 +526,7 @@ class GitStorage:
         self,
         *,
         id: Optional[str] = None,
-        default_branch: str = "main",
+        default_branch: Optional[str] = None,  # defaults to "main"
         base_repo: Optional[BaseRepo] = None,
         ttl: Optional[int] = None,
     ) -> Repo: ...
@@ -639,6 +665,8 @@ Key types are provided via TypedDict for better IDE support:
 from pierre_storage.types import (
     GitStorageOptions,
     BaseRepo,
+    GitHubBaseRepo,
+    ForkBaseRepo,
     CommitSignature,
     CreateCommitOptions,
     ListFilesResult,
@@ -652,12 +680,19 @@ from pierre_storage.types import (
     # ... and more
 )
 
-# BaseRepo type for GitHub sync
-class BaseRepo(TypedDict, total=False):
-    provider: Literal["github"]  # Always "github"
+# BaseRepo type for GitHub sync or forks
+class GitHubBaseRepo(TypedDict, total=False):
+    provider: Literal["github"]   # Always "github"
     owner: str                    # GitHub organization or user
     name: str                     # Repository name
     default_branch: Optional[str] # Default branch (optional)
+
+class ForkBaseRepo(TypedDict, total=False):
+    id: str                       # Source repo ID
+    ref: Optional[str]            # Optional ref name
+    sha: Optional[str]            # Optional commit SHA
+
+BaseRepo = Union[GitHubBaseRepo, ForkBaseRepo]
 ```
 
 ## Webhook Validation

{pierre_storage-0.11.0 → pierre_storage-0.12.1}/README.md

@@ -47,6 +47,15 @@ github_repo = await storage.create_repo(
     }
 )
 # This repository will sync with github.com/octocat/Hello-World
+
+# Create a repository by forking an existing repo
+forked_repo = await storage.create_repo(
+    id="my-fork",
+    base_repo={
+        "id": "my-template-id",
+        "ref": "main",  # optional
+    },
+)
 ```
 
 ### Finding a Repository
@@ -436,6 +445,23 @@ commits = await repo.list_commits()
 3. You can then use all Pierre SDK features (diffs, commits, file access) on the synced content
 4. The provider is automatically set to `"github"` when using `base_repo`
 
+### Forking Repositories
+
+You can fork an existing repository within the same Pierre org:
+
+```python
+forked_repo = await storage.create_repo(
+    id="my-fork",
+    base_repo={
+        "id": "my-template-id",
+        "ref": "main",  # optional (branch/tag)
+        # "sha": "abc123..."  # optional commit SHA (overrides ref)
+    },
+)
+```
+
+When `default_branch` is omitted, the SDK returns `"main"`.
+
 ### Restoring Commits
 
 You can restore a repository to a previous commit:
@@ -464,7 +490,7 @@ class GitStorage:
         self,
         *,
         id: Optional[str] = None,
-        default_branch: str = "main",
+        default_branch: Optional[str] = None,  # defaults to "main"
         base_repo: Optional[BaseRepo] = None,
         ttl: Optional[int] = None,
     ) -> Repo: ...
@@ -603,6 +629,8 @@ Key types are provided via TypedDict for better IDE support:
 from pierre_storage.types import (
     GitStorageOptions,
     BaseRepo,
+    GitHubBaseRepo,
+    ForkBaseRepo,
     CommitSignature,
     CreateCommitOptions,
     ListFilesResult,
@@ -616,12 +644,19 @@ from pierre_storage.types import (
     # ... and more
 )
 
-# BaseRepo type for GitHub sync
-class BaseRepo(TypedDict, total=False):
-    provider: Literal["github"]  # Always "github"
+# BaseRepo type for GitHub sync or forks
+class GitHubBaseRepo(TypedDict, total=False):
+    provider: Literal["github"]   # Always "github"
     owner: str                    # GitHub organization or user
     name: str                     # Repository name
     default_branch: Optional[str] # Default branch (optional)
+
+class ForkBaseRepo(TypedDict, total=False):
+    id: str                       # Source repo ID
+    ref: Optional[str]            # Optional ref name
+    sha: Optional[str]            # Optional commit SHA
+
+BaseRepo = Union[GitHubBaseRepo, ForkBaseRepo]
 ```
 
 ## Webhook Validation

{pierre_storage-0.11.0 → pierre_storage-0.12.1}/pierre_storage/client.py

@@ -1,7 +1,7 @@
 """Main client for Pierre Git Storage SDK."""
 
 import uuid
-from typing import Any, Dict, Optional
+from typing import Any, Dict, Optional, cast
 from urllib.parse import urlencode
 
 import httpx
@@ -10,7 +10,10 @@ from pierre_storage.auth import generate_jwt
 from pierre_storage.errors import ApiError
 from pierre_storage.repo import DEFAULT_TOKEN_TTL_SECONDS, RepoImpl
 from pierre_storage.types import (
+    BaseRepo,
     DeleteRepoResult,
+    ForkBaseRepo,
+    GitHubBaseRepo,
     GitStorageOptions,
     ListReposResult,
     Repo,
@@ -102,17 +105,18 @@ class GitStorage:
         self,
         *,
         id: Optional[str] = None,
-        default_branch: str = "main",
-        base_repo: Optional[Dict[str, Any]] = None,
+        default_branch: Optional[str] = None,
+        base_repo: Optional[BaseRepo] = None,
         ttl: Optional[int] = None,
     ) -> Repo:
         """Create a new repository.
 
         Args:
             id: Repository ID (auto-generated if not provided)
-            default_branch: Default branch name (default: "main")
-            base_repo: Optional base repository for GitHub sync
-                       (provider, owner, name, default_branch)
+            default_branch: Default branch name (default: "main" for non-forks)
+            base_repo: Optional base repository for GitHub sync or fork
+                       GitHub: owner, name, default_branch
+                       Fork: id, ref, sha
             ttl: Token TTL in seconds
 
         Returns:
@@ -129,19 +133,52 @@ class GitStorage:
         )
 
         url = f"{self.options['api_base_url']}/api/v{self.options['api_version']}/repos"
-        body: Dict[str, Any] = {"default_branch": default_branch}
+        body: Dict[str, Any] = {}
 
         # Match backend priority: base_repo.default_branch > default_branch > 'main'
-        resolved_default_branch = default_branch
+        explicit_default_branch = default_branch is not None
+        resolved_default_branch: Optional[str] = None
+
         if base_repo:
-            # Ensure provider is set to 'github' if not provided
-            base_repo_with_provider = {
-                "provider": "github",
-                **base_repo,
-            }
-            body["base_repo"] = base_repo_with_provider
-            if base_repo.get("default_branch"):
-                resolved_default_branch = base_repo["default_branch"]
+            if "id" in base_repo:
+                fork_repo = cast(ForkBaseRepo, base_repo)
+                base_repo_token = self._generate_jwt(
+                    fork_repo["id"],
+                    {"permissions": ["git:read"], "ttl": ttl},
+                )
+                base_repo_payload: Dict[str, Any] = {
+                    "provider": "code",
+                    "owner": self.options["name"],
+                    "name": fork_repo["id"],
+                    "operation": "fork",
+                    "auth": {"token": base_repo_token},
+                }
+                if fork_repo.get("ref"):
+                    base_repo_payload["ref"] = fork_repo["ref"]
+                if fork_repo.get("sha"):
+                    base_repo_payload["sha"] = fork_repo["sha"]
+                body["base_repo"] = base_repo_payload
+                if explicit_default_branch:
+                    resolved_default_branch = default_branch
+                    body["default_branch"] = default_branch
+            else:
+                github_repo = cast(GitHubBaseRepo, base_repo)
+                # Ensure provider is set to 'github' if not provided
+                base_repo_with_provider = {
+                    "provider": "github",
+                    **github_repo,
+                }
+                body["base_repo"] = base_repo_with_provider
+                if github_repo.get("default_branch"):
+                    resolved_default_branch = github_repo["default_branch"]
+                elif explicit_default_branch:
+                    resolved_default_branch = default_branch
+                else:
+                    resolved_default_branch = "main"
+                body["default_branch"] = resolved_default_branch
+        else:
+            resolved_default_branch = default_branch if explicit_default_branch else "main"
+            body["default_branch"] = resolved_default_branch
 
         async with httpx.AsyncClient() as client:
             response = await client.post(
@@ -173,7 +210,7 @@ class GitStorage:
 
         return RepoImpl(
             repo_id,
-            resolved_default_branch,
+            resolved_default_branch or "main",
             api_base_url,
             storage_base_url,
             name,

{pierre_storage-0.11.0 → pierre_storage-0.12.1}/pierre_storage/types.py

@@ -41,7 +41,7 @@ class GitStorageOptions(TypedDict, total=False):
     default_ttl: Optional[int]
 
 
-class BaseRepo(TypedDict, total=False):
+class GitHubBaseRepo(TypedDict, total=False):
     """Base repository configuration for GitHub sync."""
 
     provider: Literal["github"]  # required
@@ -50,6 +50,17 @@ class BaseRepo(TypedDict, total=False):
     default_branch: Optional[str]
 
 
+class ForkBaseRepo(TypedDict, total=False):
+    """Base repository configuration for code storage forks."""
+
+    id: str  # required
+    ref: Optional[str]
+    sha: Optional[str]
+
+
+BaseRepo = Union[GitHubBaseRepo, ForkBaseRepo]
+
+
 class DeleteRepoResult(TypedDict):
     """Result from deleting a repository."""
 

{pierre_storage-0.11.0 → pierre_storage-0.12.1}/pierre_storage/version.py

@@ -1,7 +1,7 @@
 """Version information for Pierre Storage SDK."""
 
 PACKAGE_NAME = "code-storage-py-sdk"
-PACKAGE_VERSION = "0.9.0"
+PACKAGE_VERSION = "0.12.1"
 
 
 def get_user_agent() -> str:

{pierre_storage-0.11.0 → pierre_storage-0.12.1/pierre_storage.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pierre-storage
-Version: 0.11.0
+Version: 0.12.1
 Summary: Pierre Git Storage SDK for Python
 Author-email: Pierre <support@pierre.io>
 License-Expression: MIT
@@ -83,6 +83,15 @@ github_repo = await storage.create_repo(
     }
 )
 # This repository will sync with github.com/octocat/Hello-World
+
+# Create a repository by forking an existing repo
+forked_repo = await storage.create_repo(
+    id="my-fork",
+    base_repo={
+        "id": "my-template-id",
+        "ref": "main",  # optional
+    },
+)
 ```
 
 ### Finding a Repository
@@ -472,6 +481,23 @@ commits = await repo.list_commits()
 3. You can then use all Pierre SDK features (diffs, commits, file access) on the synced content
 4. The provider is automatically set to `"github"` when using `base_repo`
 
+### Forking Repositories
+
+You can fork an existing repository within the same Pierre org:
+
+```python
+forked_repo = await storage.create_repo(
+    id="my-fork",
+    base_repo={
+        "id": "my-template-id",
+        "ref": "main",  # optional (branch/tag)
+        # "sha": "abc123..."  # optional commit SHA (overrides ref)
+    },
+)
+```
+
+When `default_branch` is omitted, the SDK returns `"main"`.
+
 ### Restoring Commits
 
 You can restore a repository to a previous commit:
@@ -500,7 +526,7 @@ class GitStorage:
         self,
         *,
         id: Optional[str] = None,
-        default_branch: str = "main",
+        default_branch: Optional[str] = None,  # defaults to "main"
         base_repo: Optional[BaseRepo] = None,
         ttl: Optional[int] = None,
     ) -> Repo: ...
@@ -639,6 +665,8 @@ Key types are provided via TypedDict for better IDE support:
 from pierre_storage.types import (
     GitStorageOptions,
     BaseRepo,
+    GitHubBaseRepo,
+    ForkBaseRepo,
     CommitSignature,
     CreateCommitOptions,
     ListFilesResult,
@@ -652,12 +680,19 @@ from pierre_storage.types import (
     # ... and more
 )
 
-# BaseRepo type for GitHub sync
-class BaseRepo(TypedDict, total=False):
-    provider: Literal["github"]  # Always "github"
+# BaseRepo type for GitHub sync or forks
+class GitHubBaseRepo(TypedDict, total=False):
+    provider: Literal["github"]   # Always "github"
     owner: str                    # GitHub organization or user
     name: str                     # Repository name
     default_branch: Optional[str] # Default branch (optional)
+
+class ForkBaseRepo(TypedDict, total=False):
+    id: str                       # Source repo ID
+    ref: Optional[str]            # Optional ref name
+    sha: Optional[str]            # Optional commit SHA
+
+BaseRepo = Union[GitHubBaseRepo, ForkBaseRepo]
 ```
 
 ## Webhook Validation

{pierre_storage-0.11.0 → pierre_storage-0.12.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "pierre-storage"
-version = "0.11.0"
+version = "0.12.1"
 description = "Pierre Git Storage SDK for Python"
 readme = "README.md"
 license = "MIT"

{pierre_storage-0.11.0 → pierre_storage-0.12.1}/tests/test_client.py

@@ -147,6 +147,43 @@ class TestGitStorage:
             body = call_kwargs["json"]
             assert body["base_repo"]["provider"] == "github"
 
+    @pytest.mark.asyncio
+    async def test_create_repo_with_fork_base_repo(self, git_storage_options: dict) -> None:
+        """Test creating a forked repository."""
+        storage = GitStorage(git_storage_options)
+
+        mock_post_response = MagicMock()
+        mock_post_response.status_code = 200
+        mock_post_response.is_success = True
+        mock_post_response.json.return_value = {"repo_id": "test-repo"}
+
+        with patch("httpx.AsyncClient") as mock_client:
+            client_instance = mock_client.return_value.__aenter__.return_value
+            client_instance.post = AsyncMock(return_value=mock_post_response)
+
+            repo = await storage.create_repo(
+                id="test-repo",
+                base_repo={
+                    "id": "template-repo",
+                    "ref": "develop",
+                },
+            )
+            assert repo.default_branch == "main"
+
+            call_kwargs = client_instance.post.call_args[1]
+            body = call_kwargs["json"]
+            assert "default_branch" not in body
+            assert body["base_repo"]["provider"] == "code"
+            assert body["base_repo"]["owner"] == "test-customer"
+            assert body["base_repo"]["name"] == "template-repo"
+            assert body["base_repo"]["operation"] == "fork"
+            assert body["base_repo"]["ref"] == "develop"
+
+            token = body["base_repo"]["auth"]["token"]
+            payload = jwt.decode(token, options={"verify_signature": False})
+            assert payload["repo"] == "template-repo"
+            assert payload["scopes"] == ["git:read"]
+
     @pytest.mark.asyncio
     async def test_create_repo_conflict(self, git_storage_options: dict) -> None:
         """Test creating a repository that already exists."""