feat: add aioboto3

Author: baseplate-admin

docker-compose.yml

@@ -39,7 +39,6 @@ services:
       RUSTFS_ACCESS_KEY: rustfsadmin
       RUSTFS_SECRET_KEY: rustfsadmin
       RUSTFS_CONSOLE_ENABLE: "true"
-      RUSTFS_SERVER_DOMAINS: "example.com"
     command:
       - --address
       - :9000

src/backend/app/deps.py

@@ -1,5 +1,5 @@
 from sqlmodel.ext.asyncio.session import AsyncSession
-from typing import Annotated
+from typing import Annotated, Any
 from app.db import get_session
 from fastapi import Depends, HTTPException, status
 import jwt
@@ -10,6 +10,9 @@
 from app.schemas.token import TokenPayload
 from app.settings import settings
 from app import security
+import aioboto3
+from botocore.exceptions import ClientError
+from botocore.client import BaseClient
 
 bearer_scheme = HTTPBearer(auto_error=True)
 
@@ -31,11 +34,28 @@ async def get_current_user(session: SessionDep, token: TokenDep) -> User:
     return user
 
 
+async def get_s3_client():
+    session = aioboto3.Session()
+    async with session.client(
+        "s3",
+        endpoint_url=settings.RUSTFS_ENDPOINT_URL,  # RustFS S3 API
+        aws_access_key_id=settings.RUSTFS_ACCESS_KEY,
+        aws_secret_access_key=settings.RUSTFS_SECRET_ACCESS_KEY,
+    ) as s3_client:
+        # Ensure bucket exists
+        try:
+            await s3_client.head_bucket(Bucket=settings.RUSTFS_BUCKET_NAME)
+        except ClientError:
+            await s3_client.create_bucket(Bucket=settings.RUSTFS_BUCKET_NAME)
+        yield s3_client
+
+
 SessionDep = Annotated[AsyncSession, Depends(get_session)]
 TokenDep = Annotated[
     HTTPAuthorizationCredentials,
     Depends(bearer_scheme),
 ]
 CurrentUser = Annotated[User, Depends(get_current_user)]
+S3Dep = Annotated[BaseClient, Depends(get_s3_client)]
 
-__all__ = ["SessionDep", "CurrentUser", "TokenDep"]
+__all__ = ["SessionDep", "CurrentUser", "TokenDep", "S3Dep"]

src/backend/app/settings.py

@@ -1,5 +1,3 @@
-import secrets
-
 from pydantic import PostgresDsn, computed_field
 from pydantic_settings import BaseSettings, SettingsConfigDict
 
@@ -48,6 +46,11 @@ def SQLALCHEMY_DATABASE_URI(self) -> PostgresDsn | str:
     ACCESS_TOKEN_EXPIRE_MINUTES: int = (
         60 * 60 * 24 * 8  # 60 minutes * 24 hours * 8 days = 8 days
     )
+    # RustFS
+    RUSTFS_ENDPOINT_URL: str = "http://localhost:9000"
+    RUSTFS_ACCESS_KEY: str = "rustfsadmin"
+    RUSTFS_SECRET_ACCESS_KEY: str = "rustfsadmin"
+    RUSTFS_BUCKET_NAME: str = "chithi"
 
 
 settings = Settings()  # type: ignore

src/backend/pyproject.toml

@@ -23,10 +23,12 @@ dependencies = [
     "alembic>=1.17.2",
     # Cli
     "typer>=0.20.1",
+    # Rustfs Upload
+    "aioboto3>=15.5.0",
 ]
 
 [dependency-groups]
-dev = ["fastapi-cli>=0.0.16"]
+dev = ["fastapi-cli>=0.0.16", "types-aioboto3>=15.5.0"]
 
 [tool.setuptools]
 packages = ["app"]