feat: add AA

Author: Harshal6927

CONTRIBUTING.md

@@ -4,7 +4,7 @@ This document explains the architecture and design of `litestar-start`, a CLI to
 
 ## Overview
 
-`litestar-start` is an interactive CLI tool that helps developers quickly scaffold new Litestar projects with optional plugins like SQLAlchemy and JWT authentication.
+`litestar-start` is an interactive CLI tool that helps developers quickly scaffold new Litestar projects with optional plugins like AdvancedAlchemy ORM.
 
 ## Project Structure
 
@@ -98,8 +98,7 @@ All templates use **Jinja2** with `.jinja` extension. Template context includes:
 - `project_name` - Human-readable name
 - `project_slug` - Python-safe name
 - `database` - Selected database enum
-- `has_sqlalchemy` - Boolean flag
-- `has_jwt` - Boolean flag
+- `has_advanced_alchemy` - Boolean flag
 - `docker` - Boolean flag
 
 ## Adding New Features
@@ -118,8 +117,9 @@ All templates use **Jinja2** with `.jinja` extension. Template context includes:
 2. Add to `Plugin` enum in `models.py`:
    ```python
    class Plugin(StrEnum):
-       SQLALCHEMY = "SQLAlchemy"
-       JWT = "JWT"
+       ADVANCED_ALCHEMY = "AdvancedAlchemy"
+       LITESTAR_SAQ = "LitestarSAQ"
+       LITESTAR_VITE = "LitestarVite"
        NEW_PLUGIN = "NewPlugin"  # Add this
    ```
 
@@ -180,8 +180,7 @@ Templates are rendered using Jinja2 with these settings:
 | `project_slug` | `str` | Python-safe name |
 | `database` | `Database` | Selected database enum |
 | `db_config` | `DatabaseConfig` | Database configuration |
-| `has_sqlalchemy` | `bool` | SQLAlchemy plugin enabled |
-| `has_jwt` | `bool` | JWT plugin enabled |
+| `has_advanced_alchemy` | `bool` | AdvancedAlchemy plugin enabled |
 | `has_database` | `bool` | Any database selected |
 | `docker` | `bool` | Dockerfile requested |
 | `docker_infra` | `bool` | Infra compose requested |
@@ -197,7 +196,7 @@ my_project/
 │   ├── __main__.py
 │   ├── config.py
 │   └── main.py
-├── db/                    # If SQLAlchemy selected
+├── models/                # If AdvancedAlchemy selected
 │   ├── __init__.py
 │   ├── config.py
 │   └── models.py

src/Litestar/app/__init__.py.jinja src/Litestar/App/__init__.py.jinjasrc/Litestar/app/__init__.py.jinja renamed to src/Litestar/App/__init__.py.jinja

@@ -0,0 +1,20 @@
+from app.settings import get_settings
+
+settings = get_settings()
+
+{% if has_advanced_alchemy %}
+from advanced_alchemy.extensions.litestar import (
+    AsyncSessionConfig,
+    SQLAlchemyAsyncConfig,
+    SQLAlchemyPlugin,
+)
+
+session_config = AsyncSessionConfig(expire_on_commit=False)
+alchemy_config = SQLAlchemyAsyncConfig(
+    connection_string=settings.DATABASE_URL,
+    before_send_handler="autocommit",
+    session_config=session_config,
+    create_all=True,
+)
+alchemy = SQLAlchemyPlugin(config=alchemy_config)
+{% endif %}

src/Litestar/app/__main__.py.jinja src/Litestar/App/__main__.py.jinjasrc/Litestar/app/__main__.py.jinja renamed to src/Litestar/App/__main__.py.jinja

@@ -0,0 +1,9 @@
+"""Application library utilities."""
+
+{% if has_advanced_alchemy %}
+from app.lib.utils import exception_handler
+
+__all__ = ["exception_handler"]
+{% else %}
+__all__ = []
+{% endif %}

src/Litestar/App/config.py.jinja

@@ -0,0 +1,48 @@
+{% if has_advanced_alchemy %}
+from typing import Any
+
+from advanced_alchemy.exceptions import IntegrityError, NotFoundError, RepositoryError
+from advanced_alchemy.extensions.litestar.exception_handler import (
+    ConflictError,
+)
+from litestar import Request, Response
+from litestar.exceptions import (
+    HTTPException,
+    NotFoundException,
+)
+from litestar.exceptions.responses import create_exception_response
+from litestar.status_codes import (
+    HTTP_409_CONFLICT,
+    HTTP_500_INTERNAL_SERVER_ERROR,
+)
+
+
+
+class _HTTPConflictException(HTTPException):
+    """Request conflict with the current state of the target resource."""
+
+    status_code = HTTP_409_CONFLICT
+
+
+def exception_handler(
+    request: Request[Any, Any, Any],
+    exc: Exception,
+) -> Response:
+    """Handle exceptions and convert them to HTTP exceptions."""
+    http_exc: type[HTTPException]
+
+    if isinstance(exc, NotFoundError):
+        http_exc = NotFoundException
+    elif isinstance(exc, ConflictError | RepositoryError | IntegrityError):
+        http_exc = _HTTPConflictException
+    else:
+        return create_exception_response(
+            request,
+            HTTPException(
+                status_code=getattr(exc, "status_code", HTTP_500_INTERNAL_SERVER_ERROR),
+                detail=str(exc),
+            ),
+        )
+
+    return create_exception_response(request, http_exc(detail=str(exc.detail)))
+{% endif %}

src/Litestar/App/lib/__init__.py.jinja

@@ -4,11 +4,10 @@ from litestar import Litestar, get
 from litestar.openapi import OpenAPIConfig
 
 from app.config import settings
-{% if has_sqlalchemy %}
-from db.config import db_plugin
-{% endif %}
-{% if has_jwt %}
-from auth.guards import jwt_auth
+{% if has_advanced_alchemy %}
+from advanced_alchemy.exceptions import RepositoryError
+from app.config import alchemy
+from app.lib import exception_handler
 {% endif %}
 
 
@@ -32,10 +31,11 @@ app = Litestar(
         path="/api",
     ),
     debug=settings.DEBUG,
-{% if has_sqlalchemy %}
-    plugins=[db_plugin],
-{% endif %}
-{% if has_jwt %}
-    on_app_init=[jwt_auth.on_app_init],
+{% if has_advanced_alchemy %}
+    exception_handlers={
+        Exception: exception_handler,
+        RepositoryError: exception_handler,
+    },
+    plugins=[alchemy],
 {% endif %}
 )

src/Litestar/App/lib/utils.py.jinja

@@ -2,24 +2,10 @@
 
 import os
 from dataclasses import dataclass, field
+from functools import lru_cache
 from pathlib import Path
 
-from dotenv import load_dotenv
-
-# Load environment variables
-load_dotenv()
-
-
-def _get_bool(key: str, default: bool = False) -> bool:
-    """Get a boolean from environment."""
-    value = os.getenv(key, str(default)).lower()
-    return value in ("true", "1", "yes", "on")
-
-
-def _get_int(key: str, default: int) -> int:
-    """Get an integer from environment."""
-    return int(os.getenv(key, str(default)))
-
+TRUTHY = ("true", "1", "yes")
 
 @dataclass
 class Settings:
@@ -28,11 +14,11 @@ class Settings:
     # Application
     APP_NAME: str = field(default_factory=lambda: os.getenv("APP_NAME", "{{ project_name }}"))
     APP_ENV: str = field(default_factory=lambda: os.getenv("APP_ENV", "development"))
-    DEBUG: bool = field(default_factory=lambda: _get_bool("DEBUG", True))
+    DEBUG: bool = field(default_factory=lambda: os.getenv("DEBUG", "true").lower() in TRUTHY)
 
     # Server
     HOST: str = field(default_factory=lambda: os.getenv("HOST", "0.0.0.0"))
-    PORT: int = field(default_factory=lambda: _get_int("PORT", 8000))
+    PORT: int = field(default_factory=lambda: int(os.getenv("PORT", "8000")))
 
 {% if has_database %}
     # Database
@@ -60,19 +46,24 @@ class Settings:
 {% endif %}
 {% endif %}
 
-{% if has_jwt %}
-    # JWT
-    JWT_SECRET: str = field(
-        default_factory=lambda: os.getenv("JWT_SECRET", "your-super-secret-key-change-in-production")
-    )
-    JWT_ALGORITHM: str = field(default_factory=lambda: os.getenv("JWT_ALGORITHM", "HS256"))
-    JWT_EXPIRATION_MINUTES: int = field(default_factory=lambda: _get_int("JWT_EXPIRATION_MINUTES", 60))
-{% endif %}
+    @classmethod
+    def from_env(cls, dotenv_filename: str) -> "Settings":
+        env_file = (
+            Path(dotenv_filename) if Path(dotenv_filename).is_absolute() else Path(f"{os.curdir}/{dotenv_filename}")
+        )
+
+        if env_file.is_file():
+            from dotenv import load_dotenv
+
+            console.print(
+                f"[yellow]Loading environment configuration from {dotenv_filename}[/]",
+                markup=True,
+            )
 
-    @property
-    def is_production(self) -> bool:
-        """Check if running in production."""
-        return self.APP_ENV == "production"
+            load_dotenv(env_file, override=True)
+        return Settings()
 
 
-settings = Settings()
+@lru_cache(maxsize=1, typed=True)
+def get_settings() -> Settings:
+    return Settings.from_env(dotenv_filename=".env")

src/Litestar/app/main.py.jinja src/Litestar/App/main.py.jinjasrc/Litestar/app/main.py.jinja renamed to src/Litestar/App/main.py.jinja

@@ -17,10 +17,3 @@ DATABASE_URL=sqlite+aiosqlite:///./app.db
 DATABASE_URL=mysql+asyncmy://myuser:mypassword@localhost:3306/mydb
 {% endif %}
 {% endif %}
-
-{% if has_jwt %}
-# JWT
-JWT_SECRET=your-super-secret-key-change-in-production
-JWT_ALGORITHM=HS256
-JWT_EXPIRATION_MINUTES=60
-{% endif %}

src/Litestar/app/config.py.jinja src/Litestar/App/settings.py.jinjasrc/Litestar/app/config.py.jinja renamed to src/Litestar/App/settings.py.jinja

@@ -14,7 +14,6 @@ dist/
 downloads/
 eggs/
 .eggs/
-lib/
 lib64/
 parts/
 sdist/