fastapi-gsap/gsap_broker/db_models.py
Tyler J King 295fe55bf2 feat: session-scoped ACs — multiple CRs per session
Adds session_mode flag to AC lifecycle. When session_mode=true:
- AC transitions to 'active' (not 'consumed') on first CR
- Stays active for subsequent CRs during the session
- 'session_end' outcome transitions AC to 'consumed'
- Non-session ACs behave as before (consumed on first CR)

Schema:
- ACStatus: add ACTIVE enum value
- Outcome: add SESSION_END enum value
- AuthorizeRequest: add session_mode bool field
- AuthorizationContextDB: add session_mode column
- Auto-migration via ALTER TABLE on startup

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
2026-04-03 02:06:15 -04:00

55 lines
2 KiB
Python

import uuid
from datetime import datetime
from typing import Optional
from sqlmodel import Field, SQLModel, Column
from sqlalchemy import JSON
class AuthorizationContextDB(SQLModel, table=True):
__tablename__ = "authorization_contexts"
context_id: uuid.UUID = Field(default_factory=uuid.uuid4, primary_key=True)
principal_did: str
driver_id: str
playbook: str
corpus_entry_cid: str
parameters_cid: str
accord_template: str
capability_mask: int = 3
idp_vendor: str = "keycloak"
token_jti: str = ""
elevation_active: list = Field(default=[], sa_column=Column(JSON))
mfa_satisfied: bool = False
status: str = "authorized"
issued_at: datetime = Field(default_factory=datetime.utcnow)
expires_at: datetime
consumed_at: Optional[datetime] = None
session_mode: bool = False
poll_token: Optional[str] = None
chronicle_event_cid: str = ""
class CompletionReceiptDB(SQLModel, table=True):
__tablename__ = "completion_receipts"
id: uuid.UUID = Field(default_factory=uuid.uuid4, primary_key=True)
context_id: uuid.UUID = Field(foreign_key="authorization_contexts.context_id")
outcome: str
completed_at: datetime
received_at: datetime = Field(default_factory=datetime.utcnow)
failure_reason: str = ""
chronicle_session_id: str = ""
chronicle_events: list = Field(default=[], sa_column=Column(JSON))
merkle_root: str = ""
behavioral_attestation_status: str = "unavailable"
ffc_did: str = ""
ffc_signature: str = ""
signature_verified: bool = False
chronicle_event_cid: str = ""
class ElevationRequestDB(SQLModel, table=True):
__tablename__ = "elevation_requests"
id: uuid.UUID = Field(default_factory=uuid.uuid4, primary_key=True)
principal_id: str = ""
role_name: str
justification: str = ""
duration_minutes: int = 60
status: str = "pending"
requested_at: datetime = Field(default_factory=datetime.utcnow)
expires_at: Optional[datetime] = None