Release v2

nyekiapi/api/v2/__init__.py

@@ -0,0 +1,35 @@
+from fastapi import FastAPI
+from fastapi.responses import ORJSONResponse
+from fastapi.exceptions import RequestValidationError
+
+from nyekiapi.lib import settings
+from nyekiapi.api import docs
+from nyekiapi.api.v2.errors import (
+    BaseError,
+    NotFoundError,
+    ValidationError,
+    InternalServerError,
+)
+from nyekiapi.api.v2.profiles.router import router as example_router
+
+
+app = FastAPI(
+    debug=settings.DEBUG,
+    title="Nyeki's API",
+    version="2.0.0",
+    docs_url=None,
+    redoc_url="/docs",
+    default_response_class=ORJSONResponse,
+)
+
+
+app.include_router(example_router)
+
+app.add_exception_handler(BaseError, BaseError.handler)
+app.add_exception_handler(404, NotFoundError.handler)
+app.add_exception_handler(RequestValidationError, ValidationError.handler)
+app.add_exception_handler(500, InternalServerError.handler)
+app.add_exception_handler(Exception, InternalServerError.handler)
+
+
+docs.load_onto_fastapi("docs/v2", app)

nyekiapi/api/v2/errors.py

@@ -0,0 +1,100 @@
+from fastapi.responses import ORJSONResponse
+
+from nyekiapi.api.v2.schemas import ErrorSchema
+
+
+class BaseError(Exception):
+    status_code: int = 500
+    title: str = "Error"
+    message: str = "An error occurred."
+
+    def __init__(
+        self,
+        *,
+        title: str = ...,
+        message: str = ...,
+        status_code: int = ...,
+    ):
+        if title is not ...:
+            self.title = title
+
+        if message is not ...:
+            self.message = message
+
+        if status_code is not ...:
+            self.status_code = status_code
+
+    @classmethod
+    async def handler(cls, _request, exc: "BaseError"):
+        """Handles the exception and returns a standardized error response."""
+
+        return ORJSONResponse(
+            status_code=exc.status_code if hasattr(exc, "status_code") else cls.status_code,
+            content=ErrorSchema(
+                title=exc.title if hasattr(exc, "title") else cls.title,
+                message=exc.message if hasattr(exc, "message") else cls.message,
+            ).model_dump(mode="json"),
+        )
+
+    def __repr__(self):
+        return f"{self.__class__.__name__}(title={self.title!r}, message={self.message!r}, status_code={self.status_code!r})"
+
+
+class ErrorInitMixin:
+    def __init__(self, message: str = ..., *, title: str = ...):
+        if title is not ...:
+            self.title = title
+
+        if message is not ...:
+            self.message = message
+
+
+class BadGatewayError(ErrorInitMixin, BaseError):
+    title: str = "Bad Gateway"
+    message: str = "The server received an invalid response from the upstream server."
+    status_code: int = 502
+
+
+class InternalServerError(ErrorInitMixin, BaseError):
+    title: str = "Internal Server Error"
+    message: str = "An unexpected error occurred on the server."
+    status_code: int = 500
+
+
+class ValidationError(ErrorInitMixin, BaseError):
+    title: str = "Validation Error"
+    message: str = "One or more validation errors occurred."
+    status_code: int = 422
+
+
+class ConflictError(ErrorInitMixin, BaseError):
+    title: str = "Conflict"
+    message: str = "The request could not be completed due to a conflict with the current state of the resource."
+    status_code: int = 409
+
+
+class NotFoundError(BaseError):
+    title: str = "Not Found"
+    message: str = "The requested resource was not found."
+    status_code: int = 404
+
+    def __init__(self, resource: str = "resource"):
+        self.message = f"The requested {resource} was not found."
+
+
+class ForbiddenError(ErrorInitMixin, BaseError):
+    title: str = "Forbidden"
+    message: str = "You do not have permission to access this resource."
+    status_code: int = 403
+
+
+class PaymentRequiredError(ErrorInitMixin, BaseError):
+    title: str = "Payment Required"
+    message: str = "Payment is required to access this resource."
+    status_code: int = 402
+
+
+class UnauthorizedError(ErrorInitMixin, BaseError):
+    title: str = "Unauthorized"
+    message: str = "You must be authenticated to access this resource."
+    status_code: int = 401

nyekiapi/api/v2/profiles/router.py

@@ -0,0 +1,158 @@
+from enum import StrEnum
+
+from typing import Annotated, Self
+
+from httpx import HTTPStatusError
+
+from fastapi import APIRouter, Query, Response
+
+from pydantic import BaseModel, Field, model_validator
+
+from nyekiapi.lib import settings
+from nyekiapi.api.v2.errors import NotFoundError, BadGatewayError, InternalServerError
+from nyekiapi.api.v2.schemas import ErrorSchema
+from nyekiapi.api.v2.profiles.schemas import ProfileSchema
+from nyekiapi.lib.integrations import discord, github, Profile
+
+
+router = APIRouter(tags=["Profiles"])
+
+
+sources = []
+
+if settings.DISCORD_TOKEN:
+    sources.append(("DISCORD", "discord"))
+
+if settings.GITHUB_TOKEN:
+    sources.append(("GITHUB", "github"))
+
+
+ProfileSource = StrEnum("ProfileSource", sources)
+
+
+class ProfileQuerySchema(BaseModel):
+    source: ProfileSource = Field(..., description="The source of the profile.")
+    source_id: str | None = Field(
+        None,
+        description='The ID of the profile in the source.\nSupported by: `"discord"` `"github"`',
+    )
+    source_name: str | None = Field(
+        None,
+        description='The name of the profile in the source.\nSupported by: `"github"`',
+    )
+
+    @model_validator(mode="after")
+    def validate_source(self) -> Self:
+        if self.source_id and self.source_name:
+            raise ValueError(
+                "Both source_id and source_name cannot be provided at the same time."
+            )
+
+        if self.source.value == "discord" and not self.source_id:
+            raise ValueError("source_id is required for Discord profiles.")
+
+        if self.source.value == "github" and not (self.source_name or self.source_id):
+            raise ValueError(
+                "Either source_name or source_id is required for GitHub profiles."
+            )
+
+        return self
+
+
+ProfileQuery = Annotated[ProfileQuerySchema, Query()]
+
+
+async def _get_profile(profile: ProfileQuerySchema) -> Profile:
+    try:
+        if profile.source.value == "discord":
+            return await discord.get_profile_by_id(profile.source_id)
+
+        elif profile.source.value == "github":
+            if profile.source_id:
+                return await github.get_profile_by_id(profile.source_id)
+            else:
+                return await github.get_profile_by_name(profile.source_name)
+
+    except HTTPStatusError as e:
+        if e.response.status_code == 404:
+            raise NotFoundError("profile") from e
+
+        elif e.response.status_code >= 500:
+            raise BadGatewayError() from e
+
+        else:
+            raise InternalServerError() from e
+
+
+@router.get(
+    "/profile",
+    responses={
+        404: {"model": ErrorSchema},
+        422: {"model": ErrorSchema},
+        500: {"model": ErrorSchema},
+        502: {"model": ErrorSchema},
+    },
+)
+async def get_profile(profile: ProfileQuery) -> ProfileSchema:
+    """Returns basic information about a profile from a given source."""
+
+    profile = await _get_profile(profile)
+
+    return ProfileSchema.from_integration(profile)
+
+
+@router.get(
+    "/profile/avatar",
+    responses={
+        404: {"model": ErrorSchema},
+        422: {"model": ErrorSchema},
+        500: {"model": ErrorSchema},
+        502: {"model": ErrorSchema},
+    },
+    status_code=302,
+)
+async def get_profile_avatar(response: Response, profile: ProfileQuery) -> None:
+    """Redirects to the avatar URL of a profile from a given source."""
+
+    profile = await _get_profile(profile)
+
+    response.headers["Location"] = profile.avatar_url
+
+
+@router.get(
+    "/profile/banner",
+    responses={
+        404: {"model": ErrorSchema},
+        422: {"model": ErrorSchema},
+        500: {"model": ErrorSchema},
+        502: {"model": ErrorSchema},
+    },
+    status_code=302,
+)
+async def get_profile_banner(response: Response, profile: ProfileQuery) -> None:
+    """Redirects to the banner URL of a profile from a given source."""
+
+    profile = await _get_profile(profile)
+
+    if not profile.banner_url:
+        raise NotFoundError("profile banner")
+
+    response.headers["Location"] = profile.banner_url
+
+
+@router.get(
+    "/profile/url",
+    responses={
+        404: {"model": ErrorSchema},
+        422: {"model": ErrorSchema},
+        500: {"model": ErrorSchema},
+        502: {"model": ErrorSchema},
+    },
+    status_code=302,
+)
+async def get_profile_url(response: Response, profile: ProfileQuery) -> None:
+    """Redirects to the profile URL of a profile from a given source."""
+
+    profile = await _get_profile(profile)
+
+    response.headers["Location"] = profile.url

nyekiapi/api/v2/profiles/schemas.py

@@ -0,0 +1,29 @@
+from pydantic import BaseModel, Field
+
+from nyekiapi.lib.integrations import Profile
+
+
+class ProfileSchema(BaseModel):
+    id: str = Field(
+        ..., description="The unique identifier of the profile at the source."
+    )
+    url: str = Field(..., description="The URL of the profile at the source.")
+    name: str = Field(..., description="The display name of the profile at the source.")
+    username: str = Field(..., description="The username of the profile at the source.")
+    avatar_url: str = Field(
+        ..., description="The avatar URL of the profile at the source."
+    )
+    banner_url: str | None = Field(
+        None, description="The banner URL of the profile at the source."
+    )
+
+    @classmethod
+    def from_integration(cls, profile: Profile) -> "ProfileSchema":
+        return cls(
+            id=profile.id,
+            url=profile.url,
+            name=profile.name,
+            username=profile.username,
+            avatar_url=profile.avatar_url,
+            banner_url=profile.banner_url,
+        )

nyekiapi/api/v2/schemas.py

@@ -0,0 +1,29 @@
+from typing import Annotated
+
+from pydantic import BaseModel, Field
+
+
+NanoID = Annotated[
+    str, Field(min_length=21, max_length=21, pattern="^[A-Za-z0-9_-]{21}$")
+]
+"""A nanoid identifier type for pydantic models.
+
+Example:
+```
+from pydantic import BaseModel
+
+from project.api.schemas import NanoID
+
+
+class BookSchema(BaseModel):
+    id: NanoID
+    title: str
+```
+"""
+
+
+class ErrorSchema(BaseModel):
+    """A standard error response schema."""
+
+    title: str = Field(..., description="A short, human-readable title of the error.")
+    message: str = Field(..., description="A more detailed description of the error.")