import aiohttp

from backend.config import BILLING_URL
from .schemas import *


class BillingClient:
    def __init__(self, api_key: str):
        self.api_key = api_key
        self.headers = {
            'Authorization': f'Bearer {self.api_key}'
        }
        self.base_url = BILLING_URL

    async def _method(self, http_method, method, **kwargs):
        async with aiohttp.ClientSession(headers=self.headers) as session:
            async with session.request(http_method, self.base_url + method, **kwargs) as response:
                return await response.json()

    async def create(self, request: CreateBillRequestSchema) -> CreateBillingResponseSchema:
        json_data = request.model_dump()
        response = await self._method('POST', '/create', json=json_data)
        return CreateBillingResponseSchema.model_validate(response)

    async def delete(self, request: DeleteBillRequestSchema) -> DeleteBillResponseSchema:
        method = f'/delete/{request.listener_transaction_id}'
        response = await self._method('POST', method)
        return DeleteBillResponseSchema.model_validate(response)

    async def notify_received(self, request: NotifyReceivedBillRequestSchema) -> NotifyReceivedBillResponseSchema:
        json_data = request.model_dump()
        response = await self._method('POST', '/notify-received', json=json_data)
        return NotifyReceivedBillResponseSchema.model_validate(response)