import os
import time
import requests
import json
from datetime import datetime

class KisApi:
    """한국투자증권 REST API 래퍼 (모의투자/실전투자 지원)"""
    
    def __init__(self):
        self.app_key = os.getenv("KIS_APP_KEY", "")
        self.app_secret = os.getenv("KIS_APP_SECRET", "")
        self.account_no = os.getenv("KIS_ACCOUNT_NO", "")  # 계좌번호 앞 8자리
        self.account_code = os.getenv("KIS_ACCOUNT_CODE", "01") # 계좌번호 뒤 2자리 (보통 01)
        
        # 모의투자 여부 (환경변수가 "PROD"가 아니면 기본 모의투자)
        self.is_prod = os.getenv("KIS_MODE", "DEV") == "PROD"
        
        if self.is_prod:
            self.base_url = "https://openapi.koreainvestment.com:9443"
        else:
            self.base_url = "https://openapivts.koreainvestment.com:29443"

        self.access_token = None
        self.token_expired_at = None

    def _get_headers(self, tr_id=None):
        """공통 헤더 생성"""
        self._ensure_token()
        headers = {
            "Content-Type": "application/json; charset=utf-8",
            "authorization": f"Bearer {self.access_token}",
            "appkey": self.app_key,
            "appsecret": self.app_secret,
        }
        if tr_id:
            headers["tr_id"] = tr_id
        return headers

    def _ensure_token(self):
        """토큰 유효성 검사 및 재발급"""
        now = time.time()
        # 토큰이 없거나 만료 1분 전이면 재발급
        if not self.access_token or not self.token_expired_at or now >= (self.token_expired_at - 60):
            self._issue_token()

    def _issue_token(self):
        """접근 토큰 발급"""
        url = f"{self.base_url}/oauth2/tokenP"
        payload = {
            "grant_type": "client_credentials",
            "appkey": self.app_key,
            "appsecret": self.app_secret
        }
        try:
            res = requests.post(url, json=payload, timeout=10)
            res.raise_for_status()
            data = res.json()
            
            self.access_token = data["access_token"]
            # 만료 시간 (보통 24시간인데 여유 있게 계산)
            expires_in = int(data.get("expires_in", 86400))
            self.token_expired_at = time.time() + expires_in
            print(f"[KisApi] Token issued. Expires in {expires_in} sec.")
            
        except Exception as e:
            print(f"[KisApi] Token issue failed: {e}")
            raise

    def get_balance(self):
        """주식 잔고 조회"""
        url = f"{self.base_url}/uapi/domestic-stock/v1/trading/inquire-balance"
        
        # 실전: TTTC8434R, 모의: VTTC8434R
        tr_id = "TTTC8434R" if self.is_prod else "VTTC8434R"
        
        headers = self._get_headers(tr_id=tr_id)
        
        params = {
            "CANO": self.account_no,
            "ACNT_PRDT_CD": self.account_code, #보통 01
            "AFHR_FLPR_YN": "N",
            "OFL_YN": "",
            "INQR_DVSN": "02",
            "UNPR_DVSN": "01",
            "FUND_STTL_ICLD_YN": "N",
            "FNCG_AMT_AUTO_RDPT_YN": "N",
            "PRCS_DVSN": "00", # 전일매매포함
            "CTX_AREA_FK100": "",
            "CTX_AREA_NK100": ""
        }
        
        try:
            res = requests.get(url, headers=headers, params=params, timeout=10)
            res.raise_for_status()
            data = res.json()
            
            # API 응답 코드가 0이 아니면 에러
            if data["rt_cd"] != "0":
                return {"error": data["msg1"]}
                
            output1 = data.get("output1", []) # 종목별 잔고
            output2 = data.get("output2", []) # 계좌 총 평가
            
            holdings = []
            for item in output1:
                # 보유수량이 0인 것은 제외
                qty = int(item.get("hldg_qty", 0))
                if qty > 0:
                    holdings.append({
                        "code": item.get("pdno"),
                        "name": item.get("prdt_name"),
                        "qty": qty,
                        "buy_price": float(item.get("pchs_avg_pric", 0)), # 매입단가
                        "current_price": float(item.get("prpr", 0)),      # 현재가
                        "profit_rate": float(item.get("evlu_pfls_rt", 0)),# 수익률
                    })
            
            total_eval = 0
            deposit = 0
            if output2:
                total_eval = float(output2[0].get("tot_evlu_amt", 0))    # 총 평가금액
                deposit = float(output2[0].get("dnca_tot_amt", 0))       # 예수금
                
            return {
                "holdings": holdings,
                "summary": {
                    "total_eval": total_eval,
                    "deposit": deposit
                }
            }
            
        except Exception as e:
            return {"error": str(e)}

    def order(self, stock_code: str, qty: int, price: int, buy_sell: str):
        """
        주문 실행
        buy_sell: 'buy' or 'sell'
        price: 0이면 시장가(추후 구현), 양수면 지정가
        """
        url = f"{self.base_url}/uapi/domestic-stock/v1/trading/order-cash"
        
        # 매수/매도 tr_id 구분
        # 실전: 매수 TTTC0802U, 매도 TTTC0801U
        # 모의: 매수 VTTC0802U, 매도 VTTC0801U
        if self.is_prod:
            tr_id = "TTTC0802U" if buy_sell == "buy" else "TTTC0801U"
        else:
            tr_id = "VTTC0802U" if buy_sell == "buy" else "VTTC0801U"
            
        headers = self._get_headers(tr_id=tr_id)
        
        # 가격이 0이면 시장가 "01", 아니면 지정가 "00"
        ord_dvsn = "01" if price == 0 else "00"
        
        payload = {
            "CANO": self.account_no,
            "ACNT_PRDT_CD": self.account_code,
            "PDNO": stock_code,       # 종목코드 (6자리)
            "ORD_DVSN": ord_dvsn,     # 주문구분
            "ORD_QTY": str(qty),      # 주문수량 (문자열)
            "ORD_UNPR": str(price),   # 주문단가 (문자열)
        }
        
        try:
            res = requests.post(url, headers=headers, json=payload, timeout=10)
            res.raise_for_status()
            data = res.json()
            
            if data["rt_cd"] != "0":
                return {"success": False, "message": data["msg1"]}
                
            return {
                "success": True,
                "message": data["msg1"],
                "order_no": data["output"]["ODNO"] # 주문번호
            }
            
        except Exception as e:
            return {"success": False, "message": str(e)}