import os
import json
import time
import MetaTrader5 as mt5
import redis

# Cấu hình thông số hệ thống
REDIS_HOST = 'localhost'
REDIS_PORT = 6379
REDIS_CHANNEL = 'trading_signals'

SYMBOL = "EURUSD"
RISK_PER_TRADE = 0.01  # Chấp nhận rủi ro 1% tài khoản cho mỗi lệnh
STOP_LOSS_PIPS = 150   # Điểm dừng lỗ (15 pips = 150 points)
TAKE_PROFIT_PIPS = 300 # Điểm chốt lời (30 pips = 300 points)

# 1. Kết nối với MetaTrader 5 Terminal
def init_mt5():
    if not mt5.initialize():
        print("❌ Khởi tạo kết nối MetaTrader 5 thất bại!")
        return False
    print("🟢 Kết nối MetaTrader 5 thành công!")
    return True

# 2. Phân hệ quản trị rủi ro: Tính toán Lot Size động theo số dư tài khoản
def calculate_lot_size(symbol, stop_loss_points):
    # Lấy thông tin tài khoản hiện tại
    account_info = mt5.account_info()
    if account_info is None:
        return 0.01 # Trả về số lot tối thiểu nếu lỗi

    balance = account_info.balance
    # Số tiền chấp nhận mất cho lệnh này
    risk_amount = balance * RISK_PER_TRADE

    # Lấy thông tin thuộc tính của sản phẩm giao dịch
    symbol_info = mt5.symbol_info(symbol)
    if symbol_info is None:
        return 0.01

    # Tính toán giá trị của 1 point trên mỗi Lot tiêu chuẩn
    point_value = symbol_info.trade_tick_value / symbol_info.trade_tick_size

    # Công thức tính Lot Size: Lot = Risk Amount / (Stop Loss in Points * Point Value)
    lot_size = risk_amount / (stop_loss_points * point_value)

    # Giới hạn lot size theo quy định của sàn (Volume Min/Max/Step)
    lot_size = max(symbol_info.volume_min, min(symbol_info.volume_max, lot_size))
    # Làm tròn theo bước nhảy volume (volume_step, thường là 0.01)
    step = symbol_info.volume_step
    lot_size = round(lot_size / step) * step

    print(f"💰 Số dư: {balance:.2f} USD | Số tiền rủi ro: {risk_amount:.2f} USD | Lot tính toán: {lot_size:.2f}")
    return lot_size

# 3. Phân hệ thực thi: Gửi lệnh đặt lên MetaTrader 5
def send_mt5_order(action, symbol, num_lots):
    symbol_info = mt5.symbol_info(symbol)
    if not symbol_info:
        print(f"❌ Không tìm thấy thông tin sản phẩm {symbol}")
        return False

    # Kích hoạt hiển thị symbol trên Market Watch
    if not symbol_info.visible:
        mt5.symbol_select(symbol, True)

    # Lấy giá hiện tại (Bid/Ask)
    tick = mt5.symbol_info_tick(symbol)
    if not tick:
        print(f"❌ Không lấy được giá Bid/Ask của {symbol}")
        return False

    # Xác định loại lệnh và mức giá dừng lỗ/chốt lời tương ứng
    if action == "BUY":
        order_type = mt5.ORDER_TYPE_BUY
        price = tick.ask
        sl = price - (STOP_LOSS_PIPS * symbol_info.point)
        tp = price + (TAKE_PROFIT_PIPS * symbol_info.point)
    elif action == "SELL":
        order_type = mt5.ORDER_TYPE_SELL
        price = tick.bid
        sl = price + (STOP_LOSS_PIPS * symbol_info.point)
        tp = price - (TAKE_PROFIT_PIPS * symbol_info.point)
    else:
        return False

    # Cấu trúc yêu cầu đặt lệnh
    request = {
        "action": mt5.TRADE_ACTION_DEAL,
        "symbol": symbol,
        "volume": num_lots,
        "type": order_type,
        "price": price,
        "sl": sl,
        "tp": tp,
        "deviation": 10, # Độ trượt giá tối đa cho phép (Slippage)
        "magic": 150926, # Mã nhận diện của riêng con Bot này
        "comment": "K15 AI Redis Follower",
        "type_time": mt5.ORDER_TIME_GTC,
        "type_filling": mt5.ORDER_FILLING_FOK,
    }

    # Gửi lệnh lên Terminal MT5
    result = mt5.order_send(request)
    if result.retcode != mt5.TRADE_RETCODE_DONE:
        print(f"❌ Đặt lệnh thất bại! Mã lỗi từ sàn (Retcode): {result.retcode}")
        return False

    print(f"🟢 Đặt lệnh {action} {num_lots} lot {symbol} thành công! ID vị thế: {result.order}")
    return True

# 4. Lắng nghe tín hiệu bất đồng bộ từ Redis Channel
def start_order_follower():
    # Khởi tạo kết nối Redis
    r = redis.Redis(host=REDIS_HOST, port=REDIS_PORT, db=0)
    pubsub = r.pubsub()
    pubsub.subscribe(REDIS_CHANNEL)

    print(f"📡 Đang lắng nghe tín hiệu giao dịch bất đồng bộ trên kênh '{REDIS_CHANNEL}'...")

    for message in pubsub.listen():
        # Bỏ qua tin nhắn đăng ký ban đầu
        if message['type'] == 'subscribe':
            continue

        try:
            # Giải mã dữ liệu JSON nhận được từ kênh Redis
            data = json.loads(message['data'].decode('utf-8'))
            print(f"n⚡ Nhận được tín hiệu mới: {data}")

            signal = data.get('signal') # "BUY" hoặc "SELL"
            symbol = data.get('symbol', SYMBOL)

            if signal in ["BUY", "SELL"]:
                # Tính toán Lot Size an toàn trước khi vào lệnh
                lots = calculate_lot_size(symbol, STOP_LOSS_PIPS)
                # Thực hiện đặt lệnh trên MT5
                send_mt5_order(signal, symbol, lots)

        except Exception as e:
            print(f"❌ Lỗi xử lý tin nhắn tín hiệu: {e}")

if __name__ == "__main__":
    if init_mt5():
        try:
            start_order_follower()
        except KeyboardInterrupt:
            print("n🛑 Đang dừng Robot Order Follower...")
        finally:
            mt5.shutdown()
            print("👋 Đã ngắt kết nối MetaTrader 5.")