from fastapi import Query
from pydantic import BaseModel


class SubmarineSwap(BaseModel):
    id: str
    wallet: str
    amount: int
    payment_hash: str
    time: int
    status: str
    refund_privkey: str
    refund_address: str
    boltz_id: str
    expected_amount: int
    timeout_block_height: int
    address: str
    bip21: str
    redeem_script: str


class CreateSubmarineSwap(BaseModel):
    wallet: str = Query(...)
    refund_address: str = Query(...)
    amount: int = Query(...)


class ReverseSubmarineSwap(BaseModel):
    id: str
    wallet: str
    amount: int
    onchain_address: str
    instant_settlement: bool
    time: int
    status: str
    boltz_id: str
    preimage: str
    claim_privkey: str
    lockup_address: str
    invoice: str
    onchain_amount: int
    timeout_block_height: int
    redeem_script: str


class CreateReverseSubmarineSwap(BaseModel):
    wallet: str = Query(...)
    amount: int = Query(...)
    instant_settlement: bool = Query(...)
    onchain_address: str = Query(...)


class AutoReverseSubmarineSwap(BaseModel):
    id: str
    wallet: str
    amount: int
    balance: int
    onchain_address: str
    instant_settlement: bool
    time: int


class CreateAutoReverseSubmarineSwap(BaseModel):
    wallet: str = Query(...)
    amount: int = Query(...)
    balance: int = Query(0)
    instant_settlement: bool = Query(...)
    onchain_address: str = Query(...)