codeshift 0.3.7__py3-none-any.whl → 0.4.0__py3-none-any.whl

codeshift/__init__.py

@@ -1,8 +1,8 @@
 """
-PyResolve - AI-powered CLI tool for migrating Python code to handle breaking dependency changes.
+Codeshift - AI-powered CLI tool for migrating Python code to handle breaking dependency changes.
 
 Don't just flag the update. Fix the break.
 """
 
 __version__ = "0.2.0"
-__author__ = "PyResolve Team"
+__author__ = "Codeshift Team"

codeshift/cli/__init__.py

@@ -1,4 +1,4 @@
-"""CLI module for PyResolve."""
+"""CLI module for Codeshift."""
 
 from codeshift.cli.main import cli
 

codeshift/cli/commands/__init__.py

@@ -1,4 +1,4 @@
-"""CLI commands for PyResolve."""
+"""CLI commands for Codeshift."""
 
 from codeshift.cli.commands.apply import apply
 from codeshift.cli.commands.diff import diff

codeshift/cli/commands/auth.py

@@ -55,7 +55,7 @@ def delete_credentials() -> None:
 def get_api_key() -> str | None:
     """Get API key from environment or saved credentials."""
     # Check environment first
-    api_key = os.environ.get("PYRESOLVE_API_KEY")
+    api_key = os.environ.get("CODESHIFT_API_KEY")
     if api_key:
         return api_key
 
@@ -99,7 +99,7 @@ def login(
     api_key: str | None,
     device: bool,
 ) -> None:
-    """Login to PyResolve to enable cloud features.
+    """Login to Codeshift to enable cloud features.
 
     \b
     Authentication methods:
@@ -148,7 +148,7 @@ def register(
     password: str | None,
     name: str | None,
 ) -> None:
-    """Create a new PyResolve account.
+    """Create a new Codeshift account.
 
     \b
     Example:
@@ -228,7 +228,7 @@ def _register_account(email: str, password: str, full_name: str | None) -> None:
                         f"[green]Account created successfully![/]\n\n"
                         f"Email: [cyan]{data['user']['email']}[/]\n"
                         f"Tier: [cyan]{data['user'].get('tier', 'free')}[/]\n\n"
-                        f"[dim]You are now logged in and ready to use PyResolve.[/]",
+                        f"[dim]You are now logged in and ready to use Codeshift.[/]",
                         title="Registration Successful",
                     )
                 )
@@ -466,7 +466,7 @@ def _login_with_device_code() -> None:
 
 @click.command()
 def logout() -> None:
-    """Logout from PyResolve and remove saved credentials."""
+    """Logout from Codeshift and remove saved credentials."""
     creds = load_credentials()
 
     if not creds:

codeshift/cli/commands/scan.py

@@ -11,10 +11,7 @@ from rich.panel import Panel
 from rich.progress import BarColumn, Progress, SpinnerColumn, TaskProgressColumn, TextColumn
 from rich.table import Table
 
-from codeshift.knowledge import (
-    generate_knowledge_base_sync,
-    is_tier_1_library,
-)
+from codeshift.knowledge import generate_knowledge_base_sync, is_tier_1_library
 from codeshift.scanner import DependencyParser
 from codeshift.utils.config import ProjectConfig
 
@@ -148,7 +145,7 @@ def scan(
         console.print(
             Panel(
                 "[bold]Scanning project for possible migrations[/]\n\n" f"Path: {project_path}",
-                title="PyResolve Scan",
+                title="Codeshift Scan",
             )
         )
 

codeshift/cli/commands/upgrade.py

@@ -10,12 +10,7 @@ from rich.panel import Panel
 from rich.progress import Progress, SpinnerColumn, TextColumn
 from rich.table import Table
 
-from codeshift.cli.quota import (
-    QuotaError,
-    check_quota,
-    record_usage,
-    show_quota_exceeded_message,
-)
+from codeshift.cli.quota import QuotaError, check_quota, record_usage, show_quota_exceeded_message
 from codeshift.knowledge import (
     Confidence,
     GeneratedKnowledgeBase,
@@ -130,7 +125,7 @@ def upgrade(
             f"[bold]Upgrading {knowledge.display_name}[/] to version [cyan]{target}[/]\n\n"
             f"{knowledge.description}\n"
             f"Migration guide: {knowledge.migration_guide_url or 'N/A'}",
-            title="PyResolve Migration",
+            title="Codeshift Migration",
         )
     )
 

codeshift/cli/commands/upgrade_all.py

@@ -234,7 +234,7 @@ def upgrade_all(
     console.print(
         Panel(
             "[bold]Scanning project for upgradeable dependencies[/]\n\n" f"Path: {project_path}",
-            title="PyResolve Upgrade All",
+            title="Codeshift Upgrade All",
         )
     )
 

codeshift/cli/main.py

@@ -1,4 +1,4 @@
-"""Main CLI entry point for PyResolve."""
+"""Main CLI entry point for Codeshift."""
 
 import click
 from rich.console import Console
@@ -26,7 +26,7 @@ console = Console()
 @click.version_option(version=__version__, prog_name="codeshift")
 @click.pass_context
 def cli(ctx: click.Context) -> None:
-    """PyResolve - AI-powered Python dependency migration tool.
+    """Codeshift - AI-powered Python dependency migration tool.
 
     Don't just flag the update. Fix the break.
 

codeshift/migrator/llm_migrator.py

@@ -1,6 +1,6 @@
 """LLM-based migration for complex cases.
 
-LLM migrations (Tier 2/3) are routed through the PyResolve API,
+LLM migrations (Tier 2/3) are routed through the Codeshift API,
 which handles authentication, quota checking, and billing.
 Users must have a Pro or Unlimited subscription to use LLM features.
 
@@ -11,12 +11,8 @@ import re
 from dataclasses import dataclass
 from pathlib import Path
 
-from codeshift.migrator.ast_transforms import (
-    TransformChange,
-    TransformResult,
-    TransformStatus,
-)
-from codeshift.utils.api_client import PyResolveAPIClient, get_api_client
+from codeshift.migrator.ast_transforms import TransformChange, TransformResult, TransformStatus
+from codeshift.utils.api_client import CodeshiftAPIClient, get_api_client
 from codeshift.utils.cache import LLMCache, get_llm_cache
 from codeshift.validator.syntax_checker import quick_syntax_check
 
@@ -35,10 +31,10 @@ class LLMMigrationResult:
 
 
 class LLMMigrator:
-    """Handles complex migrations using LLM via the PyResolve API.
+    """Handles complex migrations using LLM via the Codeshift API.
 
     LLM migrations require authentication and a Pro or Unlimited subscription.
-    All LLM calls are routed through the PyResolve API, which handles:
+    All LLM calls are routed through the Codeshift API, which handles:
     - Authentication and authorization
     - Quota checking and billing
     - Server-side Anthropic API calls
@@ -49,7 +45,7 @@ class LLMMigrator:
 
     def __init__(
         self,
-        client: PyResolveAPIClient | None = None,
+        client: CodeshiftAPIClient | None = None,
         cache: LLMCache | None = None,
         use_cache: bool = True,
         validate_output: bool = True,
@@ -80,7 +76,7 @@ class LLMMigrator:
         to_version: str,
         context: str | None = None,
     ) -> LLMMigrationResult:
-        """Migrate code using the LLM via PyResolve API.
+        """Migrate code using the LLM via Codeshift API.
 
         Args:
             code: Source code to migrate
@@ -114,7 +110,7 @@ class LLMMigrator:
                     used_cache=True,
                 )
 
-        # Call PyResolve API (which calls Anthropic server-side)
+        # Call Codeshift API (which calls Anthropic server-side)
         response = self.client.migrate_code(
             code=code,
             library=library,

codeshift/utils/__init__.py

@@ -1,4 +1,4 @@
-"""Utility functions and classes for PyResolve."""
+"""Utility functions and classes for Codeshift."""
 
 from codeshift.utils.config import Config, ProjectConfig
 

codeshift/utils/api_client.py

@@ -1,6 +1,6 @@
-"""PyResolve API client for LLM-powered migrations.
+"""Codeshift API client for LLM-powered migrations.
 
-This client calls the PyResolve API instead of Anthropic directly,
+This client calls the Codeshift API instead of Anthropic directly,
 ensuring that LLM features are gated behind the subscription model.
 """
 
@@ -13,7 +13,7 @@ from codeshift.cli.commands.auth import get_api_key, get_api_url
 
 @dataclass
 class APIResponse:
-    """Response from the PyResolve API."""
+    """Response from the Codeshift API."""
 
     success: bool
     content: str
@@ -22,10 +22,10 @@ class APIResponse:
     cached: bool = False
 
 
-class PyResolveAPIClient:
-    """Client for interacting with the PyResolve API for LLM migrations.
+class CodeshiftAPIClient:
+    """Client for interacting with the Codeshift API for LLM migrations.
 
-    This client routes all LLM calls through the PyResolve API,
+    This client routes all LLM calls through the Codeshift API,
     which handles:
     - Authentication and authorization
     - Quota checking and billing
@@ -41,7 +41,7 @@ class PyResolveAPIClient:
         """Initialize the API client.
 
         Args:
-            api_key: PyResolve API key. Defaults to stored credentials.
+            api_key: Codeshift API key. Defaults to stored credentials.
             api_url: API base URL. Defaults to stored URL.
             timeout: Request timeout in seconds.
         """
@@ -89,7 +89,7 @@ class PyResolveAPIClient:
         to_version: str,
         context: str | None = None,
     ) -> APIResponse:
-        """Migrate code using the PyResolve API.
+        """Migrate code using the Codeshift API.
 
         Args:
             code: Source code to migrate
@@ -249,14 +249,14 @@ class PyResolveAPIClient:
 
 
 # Singleton instance
-_default_client: PyResolveAPIClient | None = None
+_default_client: CodeshiftAPIClient | None = None
 
 
-def get_api_client() -> PyResolveAPIClient:
+def get_api_client() -> CodeshiftAPIClient:
     """Get the default API client instance."""
     global _default_client
     if _default_client is None:
-        _default_client = PyResolveAPIClient()
+        _default_client = CodeshiftAPIClient()
     return _default_client
 
 

codeshift/utils/cache.py

@@ -27,7 +27,7 @@ class CacheEntry:
 
 
 class Cache:
-    """Simple file-based cache for PyResolve."""
+    """Simple file-based cache for Codeshift."""
 
     def __init__(
         self,

{codeshift-0.3.7.dist-info → codeshift-0.4.0.dist-info}/METADATA

@@ -1,8 +1,8 @@
 Metadata-Version: 2.4
 Name: codeshift
-Version: 0.3.7
+Version: 0.4.0
 Summary: AI-powered CLI tool that migrates Python code to handle breaking dependency changes
-Author: PyResolve Team
+Author: Ragab Technologies
 License: MIT
 Project-URL: Homepage, https://github.com/Ragab-Technologies/Codeshift
 Project-URL: Repository, https://github.com/Ragab-Technologies/Codeshift
@@ -27,19 +27,6 @@ Requires-Dist: rich>=13.0
 Requires-Dist: toml>=0.10
 Requires-Dist: packaging>=23.0
 Requires-Dist: httpx>=0.25
-Requires-Dist: pytest>=8.4.2
-Requires-Dist: nox>=2025.11.12
-Requires-Dist: black>=24.10.0
-Requires-Dist: mypy>=1.19.1
-Requires-Dist: supabase>=2.27.2
-Requires-Dist: pre-commit>=4.5.1
-Provides-Extra: api
-Requires-Dist: fastapi>=0.109.0; extra == "api"
-Requires-Dist: uvicorn>=0.27.0; extra == "api"
-Requires-Dist: supabase>=2.3.0; extra == "api"
-Requires-Dist: stripe>=7.0.0; extra == "api"
-Requires-Dist: pydantic-settings>=2.1.0; extra == "api"
-Requires-Dist: email-validator>=2.0.0; extra == "api"
 Provides-Extra: dev
 Requires-Dist: pytest>=7.0; extra == "dev"
 Requires-Dist: pytest-cov>=4.0; extra == "dev"
@@ -50,8 +37,6 @@ Requires-Dist: nox>=2024.0; extra == "dev"
 Requires-Dist: pre-commit>=3.6.0; extra == "dev"
 Requires-Dist: types-toml>=0.10; extra == "dev"
 Requires-Dist: types-PyYAML>=6.0; extra == "dev"
-Provides-Extra: all
-Requires-Dist: codeshift[api,dev]; extra == "all"
 Dynamic: license-file
 
 # Codeshift

{codeshift-0.3.7.dist-info → codeshift-0.4.0.dist-info}/RECORD

@@ -1,33 +1,17 @@
-codeshift/__init__.py,sha256=rDp65Hj8u9ubLT3_gkhrKfdSBIlXKBQays3n_eLXl0s,202
+codeshift/__init__.py,sha256=L-ImO-zi7CvCuKCl_PYICUgTDppPXgYDaUoIQWukRzU,202
 codeshift/analyzer/__init__.py,sha256=m61k8rtOyHQNoPLDeoe0S9WSy9syvCyMgLZ07ojpoNU,188
 codeshift/analyzer/risk_assessor.py,sha256=nKplyymbsqcbsZyDLVZ2zukpQlPD5KIvO-FBJ4AZFYc,13422
-codeshift/api/__init__.py,sha256=jlBP8FeFfOkITfWFAK9Wqc0UHt6kyYAqVXzfvT_EQMk,29
-codeshift/api/auth.py,sha256=37wWEr4rvGbEB4HQIxI-m-uWa6HgjHQdCWCaaDwOpOA,5565
-codeshift/api/config.py,sha256=QYCSKT9t4D_mSQHIsEmaFXvQhy7yZW7EJNUqmRB8T5E,1989
-codeshift/api/database.py,sha256=VIsau8ccQ1ZRQAeQWF85GsdkqaDabTOnMlpgiJcLMLU,7171
-codeshift/api/main.py,sha256=Q6rVpUhEeX-4KBbUXliUgQ5RGTjHKPyiPggVfZVHmvg,2877
-codeshift/api/models/__init__.py,sha256=IXaCQUxh0hF03nh_3-fP0vpYdRab-RsI5LtKv9okELA,1143
-codeshift/api/models/auth.py,sha256=vbSspwn1UR0km_nC4dD08JCib56IDFQwAQqXiHZ6_t4,2435
-codeshift/api/models/billing.py,sha256=6Q0RMiYYqOAAvOPzJqnJ6FgV0mylPD0_GiguocO-Uo0,2107
-codeshift/api/models/migrate.py,sha256=XkFQ3bZWlXngnkbE3im3eZo5WnoPt1BCHpKSp38PIMY,1353
-codeshift/api/models/usage.py,sha256=YuP3XM9UsHE6OWHcXHY1jTWDirtln5Sx-0-VaDo6p8M,2911
-codeshift/api/routers/__init__.py,sha256=kh9jmNdH0v_A70KN2GbAb4zhHBnl992pK3m_aEDYtEw,171
-codeshift/api/routers/auth.py,sha256=1mJ3Hx7ARuZGA9iuglOUDXE8khScQfjMcL5naLE6MfM,13106
-codeshift/api/routers/billing.py,sha256=imG1W89cv204IojrGePd9-wXEIavsGZBa1_Obzuu9Qc,12322
-codeshift/api/routers/migrate.py,sha256=2FddO-XAHOVIoGGh7nGwVCd17jYl_FY-x7YnjKlYnxc,8945
-codeshift/api/routers/usage.py,sha256=sLaefM9T-3Kw-OGKOIERS_Z3btZTDx8Mxu-6zAuvF-w,9197
-codeshift/api/routers/webhooks.py,sha256=oHn-Fie4KxbPnfP7hdSe8yfp8JHNDkdyBBj2kGnbnfA,8733
-codeshift/cli/__init__.py,sha256=khf471pmz_J7-MGnarVWvmM0mCKFvdXpaPSJgx-KvD4,87
-codeshift/cli/main.py,sha256=iHNfLciYfq2F_QF-7hBFQvT7ezbIFg0zPJesWoKPT3Y,6983
+codeshift/cli/__init__.py,sha256=rf0c8oMqzYktdLH8ga2sA7rS5GsJse8dGfIbfW4DDXo,87
+codeshift/cli/main.py,sha256=iLXy2QMLXoiM88sn-XEHk9vJncB-WeKfEq0eps3ZHig,6983
 codeshift/cli/package_manager.py,sha256=K7spYHSQ6VoHlrzM3L9B2JoxBTe_9gLd57P6ME0PsSI,2807
 codeshift/cli/quota.py,sha256=zBiY3zqCGEyxbS3vnoQdgBld1emMQzLc0_5cUOWy9U8,5989
-codeshift/cli/commands/__init__.py,sha256=phw462By4_CzFoJQUdON73qF5-5Mv45GwUEOX_5SbLc,218
+codeshift/cli/commands/__init__.py,sha256=Kbs7DFUWOXkw5-9jiiR03cuUJeo5byusLr_Y3cKFE3E,218
 codeshift/cli/commands/apply.py,sha256=JqUiu6I5WD25677SXpOejKxLWNIUQUKrbOQ_JbpoEak,11213
-codeshift/cli/commands/auth.py,sha256=uXUtNEtAH17vhSlANfcFR5pCP8NQmFJ7Uw0bN30rYlM,28066
+codeshift/cli/commands/auth.py,sha256=FO60p545yxYySw6v6CcJPUh-Q5byM_AUz1pxypccrUA,28066
 codeshift/cli/commands/diff.py,sha256=4LjrVRu4lhj-aOBvClOnEnN0l2nZEU5S1_qzYoXL4dQ,6357
-codeshift/cli/commands/scan.py,sha256=JXR3MMKWOh1dlAEJ-OXh-EjCDBrBMxqzI6snZcBirqU,11386
-codeshift/cli/commands/upgrade.py,sha256=_V1BP15BJR6OZHg7jdq90kfdUpjquf_FoWNNEWNzrnI,15874
-codeshift/cli/commands/upgrade_all.py,sha256=v-fz9jWCzJWWW6d0QXTdOfd2M54okI9AA6qv7qxLPiE,18964
+codeshift/cli/commands/scan.py,sha256=Eia3xW6sVZSiKtxd7JXyjIcOUsxduLGOuFjBhhp1ut0,11373
+codeshift/cli/commands/upgrade.py,sha256=JjPKbjMNDzBDZuz39JYas48Ch4vgrrtyc6AEoPq6Q6Q,15853
+codeshift/cli/commands/upgrade_all.py,sha256=FimS7SYJeYkQvhrWACLQG4QiyyynCgi9SapUb8Ax0Ik,18964
 codeshift/knowledge/__init__.py,sha256=_YwrLgjvsJQuYajfnIhUQqFeircF0MfkI9zJBPZTupc,1221
 codeshift/knowledge/cache.py,sha256=aEx9aNDfrzCYMzlPRBzBOYiFIAGDcZJfQvcXroa5vsA,4837
 codeshift/knowledge/generator.py,sha256=WVCF0ULYmVQRHBiK6BTyS6FuArSCj-UDWzPJLujVbxU,7304
@@ -55,7 +39,7 @@ codeshift/knowledge_base/libraries/sqlalchemy.yaml,sha256=qEk1Nc2ovgCiBLPLsBrcsW
 codeshift/migrator/__init__.py,sha256=V5ATKxw4hV6Ec3g78IeHkP92-VM4l6OzH0gi-lnU09w,467
 codeshift/migrator/ast_transforms.py,sha256=fyySqSIFVHS8jTFledkbGWQ-_zcC5-S_rsv4sPu9jEA,7190
 codeshift/migrator/engine.py,sha256=QkxkAFIewmSOwFW041BtPBqEfv7x8nVH8zsS-bSB1Do,14572
-codeshift/migrator/llm_migrator.py,sha256=dyxVNu8Rf0FPM7Q_4TXt8f5nbbMqOHOIc_zbMsRm3RU,10373
+codeshift/migrator/llm_migrator.py,sha256=qWqb3Gzj4khDhlTmlHbpFplQ8zTq2c_ev52qL9Yeg90,10356
 codeshift/migrator/transforms/__init__.py,sha256=WWeJSr4dOKXwl8XmoCBhkYTxd4e0ENEwfMoLNwkGYsc,807
 codeshift/migrator/transforms/aiohttp_transformer.py,sha256=-7N9xYeoDSOy4iUujKU_0CtTfQS2Mx4P9lKEvriN6YQ,23297
 codeshift/migrator/transforms/attrs_transformer.py,sha256=_Bgg-oLUzvTENZnH1yIORH2rV9F2vJinnyRwQpBalXM,24128
@@ -75,17 +59,17 @@ codeshift/migrator/transforms/sqlalchemy_transformer.py,sha256=rgoIk4_iDCuydZhy9
 codeshift/scanner/__init__.py,sha256=GFx9yMPZVuxBC8mGOPZoINsCsJgHV4TSjiV4KSF3fPU,300
 codeshift/scanner/code_scanner.py,sha256=YGuHVI1FN0h8cGSARFlF5duFd8WBCJUSVMcqCbsjqEQ,12859
 codeshift/scanner/dependency_parser.py,sha256=Vd-wbgcG2trgLN7wntbNrGwuXvamn3u7B5SGvORdPiY,15372
-codeshift/utils/__init__.py,sha256=dNtKUzFwoMxuEl41d_V0Zggv5lprZnOowB8Qt_THwbg,148
-codeshift/utils/api_client.py,sha256=RUbKiox9dc_xc4z7IcrZ7plObYkkaKFa7Dw4qi75KGw,8010
-codeshift/utils/cache.py,sha256=WHwNToQVmzhLgmHvt9nWUlzDFmVLBp_Sax7R4rrQfOg,8793
+codeshift/utils/__init__.py,sha256=8G28m1UBDdEqF_G8GN6qRFWhpjDhiXJmFd9gSgIvkQc,148
+codeshift/utils/api_client.py,sha256=W6Us-eSqk3jl9l29lnEVa0QfuddZiFwA3D6FPmShRQw,8010
+codeshift/utils/cache.py,sha256=9vPjU54S48iKy4CDvcjgz0u9eeIx7aUAHCdk3coRF6w,8793
 codeshift/utils/config.py,sha256=8x-rEh4q99K0HvT4ZQHDQAeUT8Tc_HATkZOomBGVyIA,2454
 codeshift/utils/llm_client.py,sha256=WkT3KftJi7rsj8MXH4MVJvznugicb2XpkKqnqRET1eo,6369
 codeshift/validator/__init__.py,sha256=WRQSfJ7eLJdjR2_f_dXSaBtfawkvu1Dlu20Gh76D12c,280
 codeshift/validator/syntax_checker.py,sha256=FJeLIqhNhV7_Xj2RskHScJZks6A9fybaqv5Z1-MGDfo,5343
 codeshift/validator/test_runner.py,sha256=VX0OqkuI3AJxOUzRW2_BEjdDsMw1N4a0od-pPbSF6O8,6760
-codeshift-0.3.7.dist-info/licenses/LICENSE,sha256=SrjIFBfm7GQh90Giv7mEiPnYdgp1wsp0RnOTT6HkeVw,1066
-codeshift-0.3.7.dist-info/METADATA,sha256=CwpDXX4P62C0Cgq5cR5cVneNXQeEkJCyoNyj8P0TOY0,17312
-codeshift-0.3.7.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-codeshift-0.3.7.dist-info/entry_points.txt,sha256=AlJ8V7a2pNyu-9UiRKUWiTMIJtaYAUnlg53Y-wFHiK0,53
-codeshift-0.3.7.dist-info/top_level.txt,sha256=Ct42mtGs5foZ4MyYSksd5rXP0qFhWSZz8Y8mON0EEds,10
-codeshift-0.3.7.dist-info/RECORD,,
+codeshift-0.4.0.dist-info/licenses/LICENSE,sha256=mHKnse9JK19WRK76lYEwKB9nJWyzMzRpG4gkUtbTyac,1066
+codeshift-0.4.0.dist-info/METADATA,sha256=UDI4FyXLTM0c8C03CCOZQ4SSC3_F_kyrNpkrE8ePrac,16746
+codeshift-0.4.0.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+codeshift-0.4.0.dist-info/entry_points.txt,sha256=AlJ8V7a2pNyu-9UiRKUWiTMIJtaYAUnlg53Y-wFHiK0,53
+codeshift-0.4.0.dist-info/top_level.txt,sha256=Ct42mtGs5foZ4MyYSksd5rXP0qFhWSZz8Y8mON0EEds,10
+codeshift-0.4.0.dist-info/RECORD,,

{codeshift-0.3.7.dist-info → codeshift-0.4.0.dist-info}/licenses/LICENSE

@@ -1,6 +1,6 @@
 MIT License
 
-Copyright (c) 2024 PyResolve
+Copyright (c) 2024 Codeshift
 
 Permission is hereby granted, free of charge, to any person obtaining a copy
 of this software and associated documentation files (the "Software"), to deal

codeshift/api/__init__.py

@@ -1 +0,0 @@
-"""PyResolve Billing API."""

codeshift/api/auth.py

@@ -1,182 +0,0 @@
-"""Authentication utilities and dependencies for the PyResolve API."""
-
-import hashlib
-import secrets
-from collections.abc import Awaitable, Callable
-from typing import Annotated
-
-from fastapi import Depends, HTTPException, Security, status
-from fastapi.security import APIKeyHeader
-
-from codeshift.api.config import get_settings
-from codeshift.api.database import get_database
-
-# API Key header scheme
-api_key_header = APIKeyHeader(name="X-API-Key", auto_error=False)
-
-
-def generate_api_key() -> tuple[str, str, str]:
-    """Generate a new API key.
-
-    Returns:
-        Tuple of (full_key, key_prefix, key_hash)
-    """
-    settings = get_settings()
-
-    # Generate 32 random bytes (256 bits of entropy)
-    key_suffix = secrets.token_urlsafe(32)
-
-    # Create the full key with prefix
-    full_key = f"{settings.api_key_prefix}{key_suffix}"
-
-    # Get prefix for identification (first 12 chars including prefix)
-    key_prefix = full_key[:12]
-
-    # Hash the full key for storage
-    key_hash = hash_api_key(full_key)
-
-    return full_key, key_prefix, key_hash
-
-
-def hash_api_key(api_key: str) -> str:
-    """Hash an API key using SHA-256."""
-    return hashlib.sha256(api_key.encode()).hexdigest()
-
-
-class AuthenticatedUser:
-    """Authenticated user context."""
-
-    def __init__(
-        self,
-        user_id: str,
-        email: str,
-        tier: str,
-        api_key_id: str | None = None,
-        scopes: list[str] | None = None,
-    ):
-        self.user_id = user_id
-        self.email = email
-        self.tier = tier
-        self.api_key_id = api_key_id
-        self.scopes = scopes or []
-
-    def has_scope(self, scope: str) -> bool:
-        """Check if user has a specific scope."""
-        return scope in self.scopes or "admin" in self.scopes
-
-
-async def get_current_user(
-    api_key: Annotated[str | None, Security(api_key_header)] = None,
-) -> AuthenticatedUser:
-    """Validate API key and return the authenticated user.
-
-    Raises:
-        HTTPException: If API key is invalid or missing
-    """
-    if not api_key:
-        raise HTTPException(
-            status_code=status.HTTP_401_UNAUTHORIZED,
-            detail="Missing API key",
-            headers={"WWW-Authenticate": "ApiKey"},
-        )
-
-    # Hash the provided key
-    key_hash = hash_api_key(api_key)
-
-    # Look up the key in the database
-    db = get_database()
-    api_key_record = db.get_api_key_by_hash(key_hash)
-
-    if not api_key_record:
-        raise HTTPException(
-            status_code=status.HTTP_401_UNAUTHORIZED,
-            detail="Invalid API key",
-            headers={"WWW-Authenticate": "ApiKey"},
-        )
-
-    # Check if key is expired
-    if api_key_record.get("expires_at"):
-        from datetime import datetime, timezone
-
-        expires_at = api_key_record["expires_at"]
-        if isinstance(expires_at, str):
-            expires_at = datetime.fromisoformat(expires_at.replace("Z", "+00:00"))
-        if expires_at < datetime.now(timezone.utc):
-            raise HTTPException(
-                status_code=status.HTTP_401_UNAUTHORIZED,
-                detail="API key has expired",
-                headers={"WWW-Authenticate": "ApiKey"},
-            )
-
-    # Update last used timestamp
-    db.update_api_key_last_used(api_key_record["id"])
-
-    # Get profile data
-    profile = api_key_record.get("profiles", {})
-
-    return AuthenticatedUser(
-        user_id=api_key_record["user_id"],
-        email=profile.get("email", ""),
-        tier=profile.get("tier", "free"),
-        api_key_id=api_key_record["id"],
-        scopes=api_key_record.get("scopes", []),
-    )
-
-
-async def get_optional_user(
-    api_key: Annotated[str | None, Security(api_key_header)] = None,
-) -> AuthenticatedUser | None:
-    """Get the current user if authenticated, otherwise return None.
-
-    This is useful for endpoints that work both authenticated and unauthenticated.
-    """
-    if not api_key:
-        return None
-
-    try:
-        return await get_current_user(api_key)
-    except HTTPException:
-        return None
-
-
-def require_scope(scope: str) -> Callable[..., Awaitable[AuthenticatedUser]]:
-    """Dependency that requires a specific scope."""
-
-    async def check_scope(
-        user: Annotated[AuthenticatedUser, Depends(get_current_user)],
-    ) -> AuthenticatedUser:
-        if not user.has_scope(scope):
-            raise HTTPException(
-                status_code=status.HTTP_403_FORBIDDEN,
-                detail=f"Scope '{scope}' required",
-            )
-        return user
-
-    return check_scope
-
-
-def require_tier(minimum_tier: str) -> Callable[..., Awaitable[AuthenticatedUser]]:
-    """Dependency that requires a minimum tier."""
-    tier_levels = {"free": 0, "pro": 1, "unlimited": 2, "enterprise": 3}
-
-    async def check_tier(
-        user: Annotated[AuthenticatedUser, Depends(get_current_user)],
-    ) -> AuthenticatedUser:
-        user_level = tier_levels.get(user.tier, 0)
-        required_level = tier_levels.get(minimum_tier, 0)
-
-        if user_level < required_level:
-            raise HTTPException(
-                status_code=status.HTTP_403_FORBIDDEN,
-                detail=f"This feature requires {minimum_tier} tier or higher",
-            )
-        return user
-
-    return check_tier
-
-
-# Type aliases for dependency injection
-CurrentUser = Annotated[AuthenticatedUser, Depends(get_current_user)]
-OptionalUser = Annotated[AuthenticatedUser | None, Depends(get_optional_user)]
-ProUser = Annotated[AuthenticatedUser, Depends(require_tier("pro"))]
-UnlimitedUser = Annotated[AuthenticatedUser, Depends(require_tier("unlimited"))]