lnbits-legend/lnbits/core/models/audit.py

from __future__ import annotations

from datetime import datetime, timedelta, timezone
from typing import Optional

from pydantic import BaseModel, Field

from lnbits.db import FilterModel
from lnbits.settings import settings


class AuditEntry(BaseModel):
    component: Optional[str] = None
    ip_address: Optional[str] = None
    user_id: Optional[str] = None
    path: Optional[str] = None
    request_type: Optional[str] = None
    request_method: Optional[str] = None
    request_details: Optional[str] = None
    response_code: Optional[str] = None
    duration: float
    delete_at: Optional[datetime] = None
    created_at: datetime = Field(default_factory=lambda: datetime.now(timezone.utc))

    def __init__(self, **data):
        super().__init__(**data)
        retention_days = max(0, settings.lnbits_audit_retention_days) or 365
        self.delete_at = self.created_at + timedelta(days=retention_days)


class AuditFilters(FilterModel):
    __search_fields__ = [
        "ip_address",
        "user_id",
        "path",
        "request_method",
        "response_code",
        "component",
    ]
    __sort_fields__ = [
        "created_at",
        "duration",
    ]

    ip_address: Optional[str] = None
    user_id: Optional[str] = None
    path: Optional[str] = None
    request_method: Optional[str] = None
    response_code: Optional[str] = None
    component: Optional[str] = None


class AuditCountStat(BaseModel):
    field: str
    total: float


class AuditStats(BaseModel):
    request_method: list[AuditCountStat] = []
    response_code: list[AuditCountStat] = []
    component: list[AuditCountStat] = []
    long_duration: list[AuditCountStat] = []