from __future__ import annotations

import httpx

from app.core.auth import token_manager
from app.core.config import get_base_url, settings
from app.core.logger import setup_logger
from app.core.rate_limiter import RateLimiter

logger = setup_logger("trading")

_ORDER_URL = "/uapi/domestic-stock/v1/trading/order-cash"
_MODIFY_CANCEL_URL = "/uapi/domestic-stock/v1/trading/order-rvsecncl"

_TR_IDS = {
    "real": {"buy": "TTTC0802U", "sell": "TTTC0801U", "modify": "TTTC0803U", "cancel": "TTTC0804U"},
    "vps": {"buy": "VTTC0802U", "sell": "VTTC0801U", "modify": "VTTC0803U", "cancel": "VTTC0804U"},
}


class TradingService:
    def __init__(self) -> None:
        self._client = httpx.AsyncClient(timeout=10.0)
        self._rate_limiter = RateLimiter(settings.rate_limit.requests_per_second)

    def _get_tr_id(self, side: str) -> str:
        mode = settings.kis.server_mode
        return _TR_IDS.get(mode, _TR_IDS["vps"]).get(side, "VTTC0802U")

    async def place_order(
        self,
        stock_code: str,
        side: str,
        qty: int,
        price: int = 0,
        order_type: str = "00",
    ) -> dict:
        await self._rate_limiter.acquire()

        tr_id = self._get_tr_id(side)
        headers = token_manager.get_auth_headers(tr_id)

        body = {
            "CANO": settings.kis.account_no,
            "ACNT_PRDT_CD": settings.kis.account_code,
            "PDNO": stock_code,
            "ORD_DVSN": order_type,
            "ORD_QTY": str(qty),
            "ORD_UNPR": str(price) if order_type == "00" else "0",
        }

        hashkey = await token_manager.generate_hashkey(body)
        headers["hashkey"] = hashkey

        url = f"{get_base_url()}{_ORDER_URL}"
        resp = await self._client.post(url, json=body, headers=headers)
        resp.raise_for_status()
        data = resp.json()

        result = {
            "order_no": data.get("output", {}).get("odno", ""),
            "rt_cd": data.get("rt_cd"),
            "msg_cd": data.get("msg_cd"),
            "msg": data.get("msg1"),
            "stock_code": stock_code,
            "side": side,
            "qty": qty,
            "price": price,
            "order_type": order_type,
        }

        if data.get("rt_cd") == "0":
            logger.info(
                "주문 성공: %s %s %s주 %s @ %s원",
                stock_code,
                "매수" if side == "buy" else "매도",
                qty,
                order_type,
                price,
            )
        else:
            logger.warning("주문 실패: %s - %s", stock_code, data.get("msg1"))

        return result

    async def modify_order(
        self, order_no: str, stock_code: str, qty: int, price: int, order_type: str = "00"
    ) -> dict:
        await self._rate_limiter.acquire()

        tr_id = self._get_tr_id("modify")
        headers = token_manager.get_auth_headers(tr_id)

        body = {
            "CANO": settings.kis.account_no,
            "ACNT_PRDT_CD": settings.kis.account_code,
            "ODNO": order_no,
            "PDNO": stock_code,
            "ORD_DVSN": order_type,
            "ORD_QTY": str(qty),
            "ORD_UNPR": str(price),
        }

        hashkey = await token_manager.generate_hashkey(body)
        headers["hashkey"] = hashkey

        url = f"{get_base_url()}{_MODIFY_CANCEL_URL}"
        resp = await self._client.post(url, json=body, headers=headers)
        resp.raise_for_status()
        data = resp.json()

        result = {
            "order_no": data.get("output", {}).get("odno", ""),
            "rt_cd": data.get("rt_cd"),
            "msg": data.get("msg1"),
        }

        if data.get("rt_cd") == "0":
            logger.info("정정 성공: 주문번호 %s", order_no)
        else:
            logger.warning("정정 실패: %s - %s", order_no, data.get("msg1"))

        return result

    async def cancel_order(self, order_no: str, stock_code: str, qty: int) -> dict:
        await self._rate_limiter.acquire()

        tr_id = self._get_tr_id("cancel")
        headers = token_manager.get_auth_headers(tr_id)

        body = {
            "CANO": settings.kis.account_no,
            "ACNT_PRDT_CD": settings.kis.account_code,
            "ODNO": order_no,
            "PDNO": stock_code,
            "ORD_DVSN": "00",
            "ORD_QTY": str(qty),
            "ORD_UNPR": "0",
        }

        hashkey = await token_manager.generate_hashkey(body)
        headers["hashkey"] = hashkey

        url = f"{get_base_url()}{_MODIFY_CANCEL_URL}"
        resp = await self._client.post(url, json=body, headers=headers)
        resp.raise_for_status()
        data = resp.json()

        result = {
            "rt_cd": data.get("rt_cd"),
            "msg": data.get("msg1"),
        }

        if data.get("rt_cd") == "0":
            logger.info("취소 성공: 주문번호 %s", order_no)
        else:
            logger.warning("취소 실패: %s - %s", order_no, data.get("msg1"))

        return result

    async def close(self) -> None:
        await self._client.aclose()


trading_service = TradingService()