Add sub-projects

packages/sshecret-backend/src/sshecret_backend/models.py

@@ -0,0 +1,91 @@
+#!/usr/bin/env python3
+
+import uuid
+from datetime import datetime
+from sqlalchemy import Engine, Column, DateTime, func
+from sqlmodel import Field, Relationship, SQLModel
+
+
+class Client(SQLModel, table=True):
+    """Client model."""
+
+    id: uuid.UUID = Field(default_factory=uuid.uuid4, primary_key=True)
+    name: str = Field(unique=True)
+    fingerprint: str
+    created_at: datetime = Field(
+        default=None,
+        sa_column=Column(
+            DateTime(timezone=True), server_default=func.now(), nullable=True
+        ),
+    )
+    updated_at: datetime | None = Field(
+        default=None,
+        sa_column=Column(DateTime(timezone=True), onupdate=func.now(), nullable=True),
+    )
+
+    secrets: list["ClientSecret"] = Relationship(
+        back_populates="client", passive_deletes="all"
+    )
+
+
+class ClientSecret(SQLModel, table=True):
+    """A client secret."""
+
+    id: uuid.UUID = Field(default_factory=uuid.uuid4, primary_key=True)
+    name: str
+    client_id: uuid.UUID | None = Field(foreign_key="client.id", ondelete="CASCADE")
+    client: Client | None = Relationship(back_populates="secrets")
+    secret: str
+    invalidated: bool = Field(default=False)
+    created_at: datetime = Field(
+        default=None,
+        sa_column=Column(
+            DateTime(timezone=True), server_default=func.now(), nullable=True
+        ),
+    )
+    updated_at: datetime | None = Field(
+        default=None,
+        sa_column=Column(DateTime(timezone=True), onupdate=func.now(), nullable=True),
+    )
+
+
+class AuditLog(SQLModel, table=True):
+    """Audit log.
+
+    This is implemented without any foreign keys to avoid losing data on
+    deletions.
+    """
+
+    id: uuid.UUID = Field(default_factory=uuid.uuid4, primary_key=True)
+    object: str | None = None
+    object_id: str | None = None
+    operation: str
+    client_id: uuid.UUID | None = None
+    client_name: str | None = None
+    message: str
+    origin: str | None = None
+    timestamp: datetime | None = Field(
+        default=None,
+        sa_column=Column(
+            DateTime(timezone=True), server_default=func.now(), nullable=True
+        ),
+    )
+
+
+class APIClient(SQLModel, table=True):
+    """Stores API Keys."""
+
+    id: uuid.UUID = Field(default_factory=uuid.uuid4, primary_key=True)
+    token: str
+    read_write: bool
+    created_at: datetime = Field(
+        default=None,
+        sa_column=Column(
+            DateTime(timezone=True), server_default=func.now(), nullable=True
+        ),
+    )
+
+
+def init_db(engine: Engine) -> None:
+    """Create database."""
+    SQLModel.metadata.create_all(engine)