from generalresearch.models.thl.payout import UserPayoutEvent
from generalresearch.models.thl.task_status import TaskStatusResponse
from generalresearch.models.thl.wallet.cashout_method import (
    CashoutRequestResponse,
    CashoutRequestInfo,
)

from generalresearch.models.thl.user_profile import UserProfile
from generalresearch.currency import USDCent

from jb.config import settings

from generalresearch.models.thl.definitions import PayoutStatus


from typing import Optional
import requests

from jb.models.auth import User


def get_task_status(tsid: str) -> Optional[TaskStatusResponse]:
    url = f"{settings.fsb_host}{settings.product_id}/status/{tsid}/"
    d = requests.get(url).json()
    if d.get("msg") == "invalid tsid":
        return None

    return TaskStatusResponse.model_validate(d)


def user_cashout_request(
    user: User, amount: USDCent, cashout_method_id: str
) -> CashoutRequestInfo:
    assert isinstance(amount, USDCent)
    assert USDCent(0) < amount < USDCent(10_00)
    url = f"{settings.fsb_host}{settings.product_id}/cashout/"

    body: dict[str, str | int] = {
        "bpuid": user.product_user_id,
        "amount": int(amount),
        "cashout_method_id": cashout_method_id,
    }
    res = requests.post(url, json=body)
    d = res.json()

    return CashoutRequestResponse.model_validate(d).cashout


def manage_pending_cashout(
    cashout_id: str, payout_status: PayoutStatus
) -> UserPayoutEvent:
    url = f"{settings.fsb_host}{settings.fsb_host_private_route}/thl/manage_cashout/"
    body = {
        "payout_id": cashout_id,
        "new_status": payout_status.value,
    }
    res = requests.post(url, json=body)
    d = res.json()

    return UserPayoutEvent.model_validate(d)


def get_wallet_balance(amt_worker_id: str) -> USDCent:
    # This will raise an Exception if wallet balance is negative
    url = f"{settings.fsb_host}{settings.product_id}/wallet/"
    params = {"bpuid": amt_worker_id}
    return USDCent(requests.get(url, params=params).json()["wallet"]["amount"])


def get_wallet_balance_if_non_negative(amt_worker_id: str) -> Optional[USDCent]:
    url = f"{settings.fsb_host}{settings.product_id}/wallet/"
    params = {"bpuid": amt_worker_id}
    amt = requests.get(url, params=params).json()["wallet"]["amount"]
    if amt >= 0:
        return USDCent(amt)
    return None