pierre-storage 0.11.0__py3-none-any.whl → 0.12.1__py3-none-any.whl

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/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/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.dist-info → pierre_storage-0.12.1.dist-info}/METADATA

@@ -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.dist-info → pierre_storage-0.12.1.dist-info}/RECORD

@@ -1,15 +1,15 @@
 pierre_storage/__init__.py,sha256=8t6r3VUY4MSZhSuBXaoxzuxd4FLRAbVLWFxXiC_fcQY,1905
 pierre_storage/auth.py,sha256=aT0KNaUKa9fXERtQvJb8C6zL5IxrfA1KLxMBcprAq5A,2074
-pierre_storage/client.py,sha256=LPU0iZoduQvE7CDAMEKcYnMDn6WDH8vZtJw4BKO8lAg,13213
+pierre_storage/client.py,sha256=FVUB-r1Oxr112nFAXPEo_rpkGVFdxYvu_NpBXh8aclI,14908
 pierre_storage/commit.py,sha256=ks5hKScHHricJ3sx8DyLSAASM72CPmVv-tbtUgHbUF4,16766
 pierre_storage/errors.py,sha256=-vuA2BUGwyDlErFtdh2boLdk0fDFDFYBEIohJk4AsIs,2184
 pierre_storage/py.typed,sha256=8PjyZ1aVoQpRVvt71muvuq5qE-jTFZkK-GLHkhdebmc,26
 pierre_storage/repo.py,sha256=wnTDVtHx9v8tJTHJ9cHJs8cn5dImsCA3M3xxx7RqhS8,44117
-pierre_storage/types.py,sha256=MNTp4LYIActTUaxII-Qa0xjsKszVSPJKlLOvaw6nIOY,14425
-pierre_storage/version.py,sha256=HFSPY5BelU4QBXsW9MXZlFAE6Sa70XR3e_RzOQe1RO4,315
+pierre_storage/types.py,sha256=9THfT19hM_qEkozdNH5qhiL34_USubSGIRFHUw8HY9E,14661
+pierre_storage/version.py,sha256=L7upS30suP-fz3rFGhpCsKcU2OKkpynrjqNIfdNDhFc,316
 pierre_storage/webhook.py,sha256=hyjSmTlU_x35m612erXDqNXbLUh5i5As5GRw7kxylFc,7425
-pierre_storage-0.11.0.dist-info/licenses/LICENSE,sha256=CFzxoMyurfMUB0u0RaXBFZ6IDeUd6FQhKrLR_IeXtuU,1063
-pierre_storage-0.11.0.dist-info/METADATA,sha256=0TaxHV-8GRD2iX1u6JpUrZHGe7P0wKdmp1sg0F8DyVg,22367
-pierre_storage-0.11.0.dist-info/WHEEL,sha256=qELbo2s1Yzl39ZmrAibXA2jjPLUYfnVhUNTlyF1rq0Y,92
-pierre_storage-0.11.0.dist-info/top_level.txt,sha256=RzcYFaSdETlcwX-45G9Q39xUgXWZLJEWcOiK0p6ZepY,15
-pierre_storage-0.11.0.dist-info/RECORD,,
+pierre_storage-0.12.1.dist-info/licenses/LICENSE,sha256=CFzxoMyurfMUB0u0RaXBFZ6IDeUd6FQhKrLR_IeXtuU,1063
+pierre_storage-0.12.1.dist-info/METADATA,sha256=r1wvmVRSZsARL6StuI8cLWRz5b7KYbQByBjEnmv7MXk,23306
+pierre_storage-0.12.1.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+pierre_storage-0.12.1.dist-info/top_level.txt,sha256=RzcYFaSdETlcwX-45G9Q39xUgXWZLJEWcOiK0p6ZepY,15
+pierre_storage-0.12.1.dist-info/RECORD,,

{pierre_storage-0.11.0.dist-info → pierre_storage-0.12.1.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.10.1)
+Generator: setuptools (80.10.2)
 Root-Is-Purelib: true
 Tag: py3-none-any