import ccxt
import os
from dotenv import load_dotenv

load_dotenv()

# 初始化交易所
binance = ccxt.binance({
    'apiKey': os.getenv('BINANCE_API_KEY'),
    'secret': os.getenv('BINANCE_API_SECRET'),
    'enableRateLimit': True,
    'options': {'defaultType': 'spot'}
})

bybit = ccxt.bybit({
    'apiKey': os.getenv('BYBIT_API_KEY'),
    'secret': os.getenv('BYBIT_API_SECRET'),
    'enableRateLimit': True,
})

okx = ccxt.okx({
    'apiKey': os.getenv('OKX_API_KEY'),
    'secret': os.getenv('OKX_API_SECRET'),
    'password': os.getenv('OKX_PASSPHRASE'),
    'enableRateLimit': True,
})

# 獲取 BTC/USDT 最新價格
ticker = binance.fetch_ticker('BTC/USDT')
print(f"Binance BTC/USDT: {ticker['last']}")

# 三交易所 API 環境變量配置
# 請將此文件重命名為 .env 並填入您的實際密鑰
# ⚠️ 絕對不要將此文件提交到 Git 倉庫

# ========== Binance ==========
BINANCE_API_KEY=your_binance_api_key_here
BINANCE_API_SECRET=your_binance_api_secret_here

# ========== Bybit ==========
BYBIT_API_KEY=your_bybit_api_key_here
BYBIT_API_SECRET=your_bybit_api_secret_here

# ========== OKX ==========
OKX_API_KEY=your_okx_api_key_here
OKX_API_SECRET=your_okx_api_secret_here
OKX_PASSPHRASE=your_okx_passphrase_here

import asyncio
import json
import websockets

# ========== Binance: 訂單簿深度流 ==========
async def binance_depth():
    uri = 'wss://stream.binance.com:9443/ws/btcusdt@depth@100ms'
    async with websockets.connect(uri) as ws:
        print('[Binance] 已連接訂單簿深度流')
        async for msg in ws:
            data = json.loads(msg)
            bids = data.get('b', [])
            asks = data.get('a', [])
            if bids and asks:
                print(f'[Binance] Best Bid: {bids[0][0]} | Best Ask: {asks[0][0]}')

# ========== Bybit: V5 實時行情流 ==========
async def bybit_ticker():
    uri = 'wss://stream.bybit.com/v5/public/spot'
    async with websockets.connect(uri) as ws:
        # 訂閱 BTCUSDT 實時行情
        sub = {
            'op': 'subscribe',
            'args': ['tickers.BTCUSDT']
        }
        await ws.send(json.dumps(sub))
        print('[Bybit] 已訂閱行情流')
        async for msg in ws:
            data = json.loads(msg)
            if data.get('topic', '').startswith('tickers'):
                tick = data.get('data', {})
                print(f"[Bybit] Price: {tick.get('lastPrice', 'N/A')} | 24h: {tick.get('price24hPcnt', 'N/A')}")

# ========== OKX: 實時成交流 ==========
async def okx_trades():
    uri = 'wss://ws.okx.com:8443/ws/v5/public'
    async with websockets.connect(uri) as ws:
        # 訂閱 BTC-USDT 成交數據
        sub = {
            'op': 'subscribe',
            'args': [{ 'channel': 'trades', 'instId': 'BTC-USDT' }]
        }
        await ws.send(json.dumps(sub))
        print('[OKX] 已訂閱成交流')
        async for msg in ws:
            data = json.loads(msg)
            if data.get('arg', {}).get('channel') == 'trades':
                for trade in data.get('data', []):
                    print(f"[OKX] Side: {trade.get('side')} | Price: {trade.get('px')} | Sz: {trade.get('sz')}")

# ========== 同時監聽三所 ==========
async def main():
    await asyncio.gather(
        binance_depth(),
        bybit_ticker(),
        okx_trades()
    )

if __name__ == '__main__':
    asyncio.run(main())