models.py

# Data models for your extension

import time
from sqlite3 import Row
from typing import Any, Dict, List

from pydantic import BaseModel


class NWCKey(BaseModel):
    pubkey: str
    wallet: str
    description: str
    expires_at: int
    permissions: str
    created_at: int
    last_used: int

    def get_permissions(self) -> List[str]:
        try:
            return self.permissions.split(" ")
        except Exception:
            return []

    @classmethod
    def from_row(cls, row: Dict[str, Any]) -> "NWCKey":
        return cls(**row)


class NWCBudget(BaseModel):
    id: int
    pubkey: str
    budget_msats: int
    refresh_window: int
    created_at: int
    used_budget_msats: int = 0

    def get_timestamp_range(self) -> tuple[int, int]:
        c = int(time.time())
        if self.refresh_window <= 0:  # never refresh
            # return a timestamp in the future
            return c, c + 21000000
        # calculate the next refresh timestamp
        elapsed = c - self.created_at
        passed_cycles = elapsed // self.refresh_window
        last_cycle = self.created_at + (passed_cycles * self.refresh_window)
        next_cycle = last_cycle + self.refresh_window
        return last_cycle, next_cycle

    @classmethod
    def from_row(cls, row: Row) -> "NWCBudget":
        return cls(**dict(row))


class NWCLog(BaseModel):
    id: int
    pubkey: str
    payload: str
    created_at: int

    @classmethod
    def from_row(cls, row: Row) -> "NWCLog":
        return cls(**dict(row))


class NWCNewBudget(BaseModel):
    budget_msats: int
    refresh_window: int
    created_at: int


class NWCRegistrationRequest(BaseModel):
    permissions: List[str]
    description: str
    expires_at: int
    budgets: List[NWCNewBudget]


class NWCGetResponse(BaseModel):
    data: NWCKey
    budgets: List[NWCBudget]