#!/usr/bin/env python
# -*- coding: utf-8 -*-

"""
支付成功异步回调处理模块
当用户支付成功后，支付平台会把订单支付信息异步请求到回调接口
"""

import json
import asyncio
from .api import YsmPayApi

class PaymentNotify:
    """支付通知处理类"""
    
    def __init__(self, appid, appsecret):
        """
        初始化
        
        Args:
            appid: 支付通道ID
            appsecret: 密钥
        """
        self.appid = appid
        self.appsecret = appsecret
    
    async def process(self, request_data):
        """
        处理支付通知（异步）
        
        Args:
            request_data: 请求数据（字符串或字典）
        
        Returns:
            tuple: (是否成功, 消息)
        """
        # 解析请求数据
        if isinstance(request_data, str):
            try:
                data = json.loads(request_data)
            except:
                return False, "数据格式错误"
        else:
            data = request_data
        
        # 验证签名
        calc_hash = YsmPayApi.hash_sign(data, self.appsecret)
        if data.get('hash') != calc_hash:
            return False, "验签失败，签名错误"
        
        # 获取商户订单号
        mch_orderid = data.get('mch_orderid')
        
        # 处理支付结果
        if data.get('state') == 'SUCCESS':
            # 在这里处理订单业务逻辑
            # 注意：平台可能会多次调用本接口，请确保订单不会被重复处理
            
            # 示例：订单处理逻辑，这里可以使用异步数据库操作
            """
            order = await find_order_by_id(mch_orderid)
            if order and order.status != 'paid':
                await update_order_status(mch_orderid, 'paid')
                # 其他业务逻辑...
            """
            
            return True, "success"
        else:
            # 处理未支付的情况
            return False, "订单未支付"

# FastAPI Web应用示例
"""
from fastapi import FastAPI, Request, Response
from fastapi.responses import JSONResponse

app = FastAPI()

@app.post("/notify")
async def payment_notify(request: Request):
    # 配置信息
    appid = '20********'  # 支付通道ID
    appsecret = 'e605ac7******************4af5164'  # AppSecret
    
    # 创建通知处理器
    notify_handler = PaymentNotify(appid, appsecret)
    
    # 获取请求数据
    request_data = await request.body()
    
    # 处理通知
    success, message = await notify_handler.process(request_data)
    
    if success:
        return Response(content=message)
    else:
        return JSONResponse(content={"error": message}, status_code=400)

# 启动命令: uvicorn notify:app --reload
"""