"""
支付视图
"""
from flask import Blueprint, render_template, request, jsonify, session, redirect, url_for, flash
from app.models.payment import Payment
from app.models.order import Order
from app.utils.decorators import login_required
from config.database import db
from datetime import datetime

payment_bp = Blueprint('payment', __name__, url_prefix='/payment')


@payment_bp.route('/process', methods=['POST'])
@login_required
def process():
    """处理支付请求"""
    try:
        user_id = session['user_id']
        payment_sn = request.json.get('payment_sn')
        payment_method = request.json.get('payment_method')

        if not payment_sn:
            return jsonify({'success': False, 'message': '支付流水号不能为空'})

        # 获取支付记录
        payment = Payment.query.filter_by(payment_sn=payment_sn).first()
        if not payment:
            return jsonify({'success': False, 'message': '支付记录不存在'})

        order = payment.order
        if order.user_id != user_id:
            return jsonify({'success': False, 'message': '订单不存在'})

        if not order.can_pay():
            return jsonify({'success': False, 'message': '订单不可支付'})

        # 根据支付方式处理
        if payment_method == 'wechat':
            # 微信支付
            result = process_wechat_pay(payment)
        elif payment_method == 'alipay':
            # 支付宝支付
            result = process_alipay(payment)
        elif payment_method == 'bank':
            # 银行卡支付
            result = process_bank_pay(payment)
        else:
            return jsonify({'success': False, 'message': '不支持的支付方式'})

        return jsonify(result)

    except Exception as e:
        return jsonify({'success': False, 'message': f'支付处理失败: {str(e)}'})


def process_wechat_pay(payment):
    """处理微信支付"""
    # TODO: 接入真实的微信支付API
    # 目前返回模拟的支付二维码

    # 模拟生成支付二维码数据
    qr_code_url = f"weixin://wxpay/bizpayurl?pr={payment.payment_sn}"

    return {
        'success': True,
        'payment_type': 'qrcode',
        'qr_code_url': qr_code_url,
        'payment_sn': payment.payment_sn,
        'amount': float(payment.amount),
        'message': '请使用微信扫码支付'
    }


def process_alipay(payment):
    """处理支付宝支付"""
    # TODO: 接入真实的支付宝API
    # 目前返回模拟的跳转链接

    pay_url = f"https://mapi.alipay.com/gateway.do?service=create_direct_pay_by_user&payment_sn={payment.payment_sn}"

    return {
        'success': True,
        'payment_type': 'redirect',
        'pay_url': pay_url,
        'payment_sn': payment.payment_sn,
        'amount': float(payment.amount),
        'message': '正在跳转到支付宝...'
    }


def process_bank_pay(payment):
    """处理银行卡支付"""
    # TODO: 接入银行支付网关
    # 目前返回模拟的网银链接

    bank_url = f"https://pay.bank.com/pay?order={payment.payment_sn}"

    return {
        'success': True,
        'payment_type': 'redirect',
        'pay_url': bank_url,
        'payment_sn': payment.payment_sn,
        'amount': float(payment.amount),
        'message': '正在跳转到网银...'
    }


@payment_bp.route('/callback/wechat', methods=['POST'])
def wechat_callback():
    """微信支付回调"""
    try:
        # TODO: 验证微信支付回调签名
        # 目前模拟处理

        callback_data = request.get_data()
        # 解析回调数据，获取支付结果

        # 模拟成功的回调处理
        return handle_payment_success(request.form.get('payment_sn'), 'wechat_success_' + str(datetime.now().timestamp()))

    except Exception as e:
        return f"FAIL: {str(e)}"


@payment_bp.route('/callback/alipay', methods=['POST'])
def alipay_callback():
    """支付宝支付回调"""
    try:
        # TODO: 验证支付宝回调签名
        # 目前模拟处理

        return handle_payment_success(request.form.get('payment_sn'), 'alipay_success_' + str(datetime.now().timestamp()))

    except Exception as e:
        return f"FAIL: {str(e)}"


def handle_payment_success(payment_sn, third_party_sn):
    """处理支付成功"""
    try:
        payment = Payment.query.filter_by(payment_sn=payment_sn).first()
        if not payment:
            return "FAIL: Payment not found"

        if payment.status == Payment.STATUS_SUCCESS:
            return "SUCCESS"  # 已经处理过的支付

        # 更新支付状态
        payment.status = Payment.STATUS_SUCCESS
        payment.third_party_sn = third_party_sn
        payment.paid_at = datetime.utcnow()

        # 更新订单状态
        order = payment.order
        order.status = Order.STATUS_PENDING_SHIPMENT

        db.session.commit()

        return "SUCCESS"

    except Exception as e:
        db.session.rollback()
        return f"FAIL: {str(e)}"


@payment_bp.route('/check_status/<payment_sn>')
@login_required
def check_status(payment_sn):
    """检查支付状态"""
    try:
        user_id = session['user_id']
        payment = Payment.query.filter_by(payment_sn=payment_sn).first()

        if not payment or payment.order.user_id != user_id:
            return jsonify({'success': False, 'message': '支付记录不存在'})

        return jsonify({
            'success': True,
            'status': payment.status,
            'status_text': payment.get_status_text(),
            'paid_at': payment.paid_at.isoformat() if payment.paid_at else None
        })

    except Exception as e:
        return jsonify({'success': False, 'message': f'查询失败: {str(e)}'})


@payment_bp.route('/simulate_success/<payment_sn>', methods=['POST'])
@login_required
def simulate_success(payment_sn):
    """模拟支付成功（开发测试用）"""
    try:
        user_id = session['user_id']
        payment = Payment.query.filter_by(payment_sn=payment_sn).first()

        if not payment or payment.order.user_id != user_id:
            return jsonify({'success': False, 'message': '支付记录不存在'})

        if payment.status == Payment.STATUS_SUCCESS:
            return jsonify({'success': False, 'message': '订单已支付'})

        # 模拟支付成功
        result = handle_payment_success(payment_sn, f'SIMULATE_{datetime.now().timestamp()}')

        if result == "SUCCESS":
            return jsonify({'success': True, 'message': '支付成功'})
        else:
            return jsonify({'success': False, 'message': result})

    except Exception as e:
        return jsonify({'success': False, 'message': f'模拟支付失败: {str(e)}'})