apps/trading.py

import os
import requests
import json
import jwt
import uuid
import hashlib
import time
import math
import asyncio
import pandas as pd
from datetime import datetime
from urllib.parse import urlencode
from fastapi import APIRouter, Request, BackgroundTasks, Form, Response

router = APIRouter()

# ---------------------------------------------------------
# [설정] 환경 변수 로드
# ---------------------------------------------------------
BITHUMB_ACCESS_KEY = os.environ.get("BITHUMB_ACCESS_KEY")
BITHUMB_SECRET_KEY = os.environ.get("BITHUMB_SECRET_KEY")
WEBHOOK_PASS = os.environ.get("WEBHOOK_PASS")
SLACK_WEBHOOK_URL = os.environ.get("SLACK_WEBHOOK_URL")
BITHUMB_API_URL = 'https://api.bithumb.com'

def get_auth_headers(params=None):
    payload = {
        'access_key': BITHUMB_ACCESS_KEY,
        'nonce': str(uuid.uuid4()),
        'timestamp': round(time.time() * 1000),
    }
    if params:
        query = urlencode(params).encode()
        hash = hashlib.sha512()
        hash.update(query)
        payload['query_hash'] = hash.hexdigest()
        payload['query_hash_alg'] = 'SHA512'
    
    jwt_token = jwt.encode(payload, BITHUMB_SECRET_KEY)
    return {'Authorization': f'Bearer {jwt_token}', 'Content-Type': 'application/json'}

# ---------------------------------------------------------
# [공통] Slack 메시지 발송 함수 (블록킷 지원)
# ---------------------------------------------------------
def send_slack_message(text, blocks=None):
    if not SLACK_WEBHOOK_URL: return
    payload = {"text": text}
    if blocks: payload["blocks"] = blocks
    try:
        requests.post(SLACK_WEBHOOK_URL, json=payload, timeout=3)
    except Exception as e:
        print(f"Slack 에러: {e}")

# ---------------------------------------------------------
# [공통] 호가 단위(Tick Size) 계산 함수
# ---------------------------------------------------------
def get_tick_size(price):
    if price >= 1000000: return 1000.0
    elif price >= 500000: return 500.0
    elif price >= 100000: return 100.0
    elif price >= 10000: return 10.0
    elif price >= 5000: return 5.0
    elif price >= 100: return 1.0
    elif price >= 10: return 0.01
    else: return 0.001

# ---------------------------------------------------------
# [기능] 미체결 매수 주문 취소
# ---------------------------------------------------------
def cancel_pending_buy_orders(market):
    try:
        params = {'market': market, 'state': 'wait'}
        headers = get_auth_headers(params)
        res = requests.get(f"{BITHUMB_API_URL}/v1/orders", params=params, headers=headers)
        orders = res.json()
        if not isinstance(orders, list): return

        for order in orders:
            if order.get('side') == 'bid':
                cancel_params = {'order_id': order['uuid']}
                cancel_headers = get_auth_headers(cancel_params)
                c_res = requests.delete(f"{BITHUMB_API_URL}/v2/orders", params=cancel_params, headers=cancel_headers)
                
                # Bithumb API 확인 후 알림
                if 'uuid' in c_res.json():
                    price = float(order.get('price', 0))
                    vol = float(order.get('remaining_volume', 0))
                    send_slack_message(f"🚫 *[매수 취소 완료]* {market}\n가격: {price:,.2f} | 수량: {vol}")
    except Exception as e: print(f"🚨 취소 에러: {e}")

# ---------------------------------------------------------
# [기능] 전체 호가창 & 자산 조회
# ---------------------------------------------------------
def get_market_depth(market):
    try:
        res = requests.get(f"{BITHUMB_API_URL}/v1/orderbook?markets={market}", timeout=2)
        return res.json()[0]['orderbook_units']
    except: return None

def get_asset_info(currency):
    try:
        headers = get_auth_headers()
        res = requests.get(f"{BITHUMB_API_URL}/v1/accounts", headers=headers, timeout=2)
        for acc in res.json():
            if acc['currency'] == currency.upper():
                return float(acc['balance']), float(acc.get('avg_buy_price', 0))
    except: pass
    return 0.0, 0.0

def place_order(market, side, price, volume, order_type='limit'):
    params = {'market': market, 'side': side, 'order_type': order_type, 'volume': str(volume)}
    if order_type == 'limit':
        params['price'] = str(price)
        
    headers = get_auth_headers(params)
    res = requests.post(f"{BITHUMB_API_URL}/v2/orders", data=json.dumps(params), headers=headers, timeout=3)
    return res.json()

# ---------------------------------------------------------
# [기능] 1분 경과 모니터링 및 Slack 1~10호가 드롭다운 UI 생성 (+안전장치 추가)
# ---------------------------------------------------------
async def monitor_pending_order(order_id, market, side, original_price, volume):
    for i in range(1, 6):
        await asyncio.sleep(60) # 1분 대기
        
        params = {'market': market, 'uuid': order_id}
        headers = get_auth_headers(params)
        res = requests.get(f"{BITHUMB_API_URL}/v1/order", params=params, headers=headers)
        
        if res.status_code == 200:
            order_info = res.json()
            if order_info.get('state') == 'wait':
                side_kr = "매수" if side == "bid" else "매도"
                
                up_options = [{"text": {"type": "plain_text", "text": f"+{t}호가"}, "value": f"up|{t}|{market}|{order_id}"} for t in range(1, 11)]
                down_options = [{"text": {"type": "plain_text", "text": f"-{t}호가"}, "value": f"down|{t}|{market}|{order_id}"} for t in range(1, 11)]
                
                # 🛡️ 안전장치: 전송 확정 팝업 UI 정의
                confirm_dialog = {
                    "title": {"type": "plain_text", "text": "주문 정정 확인"},
                    "text": {"type": "mrkdwn", "text": "선택하신 호가로 기존 주문을 취소하고 *새로 정정 주문*을 전송하시겠습니까?"},
                    "confirm": {"type": "plain_text", "text": "전송 (확정)"},
                    "deny": {"type": "plain_text", "text": "취소"}
                }
                
                blocks = [
                    {
                        "type": "section",
                        "text": {"type": "mrkdwn", "text": f"⏳ *[{i}분 경과] {market} {side_kr} 대기중*\n가격: {original_price:,.2f} | 수량: {volume}\n주문ID: `{order_id}`"}
                    },
                    {
                        "type": "actions",
                        "elements": [
                            # 취소 버튼에도 안전장치 추가 가능!
                            {
                                "type": "button", 
                                "text": {"type": "plain_text", "text": "🚫 주문 취소"}, 
                                "style": "danger", 
                                "value": f"cancel|0|{market}|{order_id}",
                                "confirm": {
                                    "title": {"type": "plain_text", "text": "주문 취소 확인"},
                                    "text": {"type": "mrkdwn", "text": "정말로 대기 중인 주문을 취소하시겠습니까?"},
                                    "confirm": {"type": "plain_text", "text": "취소 진행"},
                                    "deny": {"type": "plain_text", "text": "닫기"}
                                }
                            },
                            # + 호가 드롭다운 (+ 팝업창 연결)
                            {
                                "type": "static_select",
                                "placeholder": {"type": "plain_text", "text": "📈 +호가 올리기"},
                                "options": up_options,
                                "confirm": confirm_dialog
                            },
                            # - 호가 드롭다운 (+ 팝업창 연결)
                            {
                                "type": "static_select",
                                "placeholder": {"type": "plain_text", "text": "📉 -호가 내리기"},
                                "options": down_options,
                                "confirm": confirm_dialog
                            }
                        ]
                    }
                ]
                send_slack_message(f"⏳ {market} {side_kr} 대기중", blocks=blocks)
            elif order_info.get('state') == 'done':
                side_kr = "매수" if side == "bid" else "매도"
                send_slack_message(f"✅ *[{side_kr} 체결 완료]* {market}\n가격: {original_price:,.2f} | 수량: {volume}")
                break
            else:
                break # 취소되었거나 예외 상태

# ---------------------------------------------------------
# [API] Webhook 엔드포인트
# ---------------------------------------------------------
@router.post("/webhook")
async def bithumb_webhook(request: Request, bg_tasks: BackgroundTasks):
    try:
        data = await request.json()
        
        # [1] 보안 검증
        if data.get("id", "")[:4] != WEBHOOK_PASS:
            return {"status": "error", "msg": "Unauthorized"}

        # [2] 데이터 추출
        symbol = data.get("ticker", "").upper().replace("KRW", "").replace("-", "").replace("/", "").strip()
        market = f"KRW-{symbol}"
        action = data.get("action", "").lower()

        # [3] 호가 조회
        depth = get_market_depth(market)
        if not depth: return {"status": "error", "msg": "orderbook_fetch_failed"}

        current_bid = float(depth[0]['bid_price']) # 매수 1호가
        current_ask = float(depth[0]['ask_price']) # 매도 1호가

        # ------------------------------------------
        # 1. [매수]
        # ------------------------------------------
        if "buy" in action or "bid" in action:
            # 트레이딩뷰 트리거 알림
            send_slack_message(f"[📈buy-signal]{symbol}, 호가 {current_ask:,.2f}")
            
            price = float(depth[3]['bid_price']) # 매수 4호가
            amount_krw = float(data.get("amount", 5000))
            volume = math.floor((amount_krw / price) * 10000) / 10000
            
            resp = place_order(market, 'bid', price, volume)
            
            # API 크로스체크 및 실패 알림
            if 'error' in resp:
                err_msg = resp['error'].get('message', '알수없음')
                send_slack_message(f"❌ *[매수 실패]* {market}\n시도 가격: {price:,.2f} | 수량: {volume}\n사유: {err_msg}")
                return {"status": "error", "msg": err_msg}
            
            order_id = resp.get('uuid')
            send_slack_message(f"📥 *[매수 대기]* {market} (4호가)\n가격: {price:,.2f} | 수량: {volume}")
            
            # 백그라운드 1분 모니터링 실행
            bg_tasks.add_task(monitor_pending_order, order_id, market, 'bid', price, volume)

        # ------------------------------------------
        # 2. [매도]
        # ------------------------------------------
        else:
            # 트레이딩뷰 트리거 알림
            send_slack_message(f"[📉signal-sell]{symbol}, 호가 {current_bid:,.2f}")
            
            balance, avg_buy_price = get_asset_info(symbol)
            volume = math.floor(balance * 10000) / 10000
            
            cancel_pending_buy_orders(market)

            if volume > 0 and (volume * current_bid) >= 5000:
                profit_rate = ((current_bid - avg_buy_price) / avg_buy_price) * 100 if avg_buy_price > 0 else 0
                
                if profit_rate >= 5.0:
                    # 수익 5% 이상 -> 시장가 전량 매도
                    resp = place_order(market, 'ask', None, volume, order_type='market')
                    if 'error' in resp:
                        send_slack_message(f"❌ *[시장가 매도 실패]* {market}\n사유: {resp['error'].get('message', '')}")
                    else:
                        send_slack_message(f"🚀 *[시장가 매도 완료]* {market}\n수익률: {profit_rate:.2f}% | 수량: {volume}")
                else:
                    # 수익 5% 미만 -> 시그널 발생 가격의 -1호가로 지정가 매도
                    target_price = current_bid - get_tick_size(current_bid)
                    
                    resp = place_order(market, 'ask', target_price, volume, order_type='limit')
                    if 'error' in resp:
                        send_slack_message(f"❌ *[매도 실패 (-1호가)]* {market}\n사유: {resp['error'].get('message', '')}")
                    else:
                        order_id = resp.get('uuid')
                        send_slack_message(f"📤 *[매도 대기 (-1호가)]* {market}\n가격: {target_price:,.2f} | 수량: {volume}\n수익률: {profit_rate:.2f}%")
                        
                        # 백그라운드 1분 모니터링 실행
                        bg_tasks.add_task(monitor_pending_order, order_id, market, 'ask', target_price, volume)
            else:
                send_slack_message(f"⚠️ *[매도 패스]* {market}\n사유: 잔고 부족 또는 최소주문금액 미달")

        return {"status": "success"}
    except Exception as e:
        return {"status": "error", "msg": str(e)}

# ---------------------------------------------------------
# [API] Slack 인터랙티브 드롭다운 & 버튼 처리
# ---------------------------------------------------------
@router.post("/slack/interaction")
async def slack_interaction(payload: str = Form(...)):
    try:
        data = json.loads(payload)
        
        # 버튼을 누르거나 드롭다운을 선택했을 때 넘어오는 값 추출
        action = data['actions'][0]
        action_type = action['type']
        
        # 버튼이면 'value', 드롭다운이면 'selected_option'에서 값 추출
        if action_type == 'button':
            action_val = action['value']
        else:
            action_val = action['selected_option']['value']
            
        parts = action_val.split('|')
        cmd = parts[0]           # cancel, up, down
        ticks = int(parts[1])    # 변경할 호가 칸 수 (1~10)
        market = parts[2]
        order_id = parts[3]

        # 1. Bithumb API에서 기존 주문 상세 조회 (최신 잔여 수량 및 가격 확인용)
        params = {'market': market, 'uuid': order_id}
        headers = get_auth_headers(params)
        res = requests.get(f"{BITHUMB_API_URL}/v1/order", params=params, headers=headers)
        order_info = res.json()

        if order_info.get('state') != 'wait':
            send_slack_message(f"⚠️ 이미 체결되었거나 취소된 주문입니다.")
            return Response(status_code=200)

        old_side = order_info['side']
        old_price = float(order_info['price'])
        remain_volume = float(order_info['remaining_volume']) # 그사이 일부가 체결되었을 수 있으므로 남은 수량만 가져옴

        # 2. 기존 주문 먼저 취소
        cancel_params = {'order_id': order_id}
        cancel_headers = get_auth_headers(cancel_params)
        requests.delete(f"{BITHUMB_API_URL}/v2/orders", params=cancel_params, headers=cancel_headers)

        if cmd == "cancel":
            send_slack_message(f"✅ *[수동 취소 완료]* {market}")
            return Response(status_code=200)

        # 3. 호가 재계산 (ex: 9995원에서 3호가 올릴 때 10000원을 넘어가면 호가 단위가 바뀌는 것까지 안전하게 계산)
        new_price = old_price
        for _ in range(ticks):
            tick_size = get_tick_size(new_price)
            if cmd == "up":
                new_price += tick_size
            elif cmd == "down":
                new_price -= tick_size

        # 4. 새로운 가격으로 재주문 실행
        order_res = place_order(market, old_side, new_price, remain_volume, 'limit')
        
        if 'error' in order_res:
            send_slack_message(f"❌ *[호가 정정 실패]* {market}\n사유: {order_res['error'].get('message')}")
        else:
            side_kr = "매수" if old_side == "bid" else "매도"
            send_slack_message(f"🔄 *[{side_kr} {ticks}호가 정정 완료]* {market}\n기존: {old_price:,.2f} ➔ 변경: {new_price:,.2f} | 수량: {remain_volume}")

        return Response(status_code=200)

    except Exception as e:
        print(f"Slack Interaction Error: {e}")
        return Response(status_code=200)