from flask import Blueprint, request, jsonify

# 确保这两个引用路径是存在的，如果报错说明文件没建好
try:
    from app.services.stock_service import StockService
    from app.schemas.stock_schema import stock_buy_schema
except ImportError as e:
    # 如果服务还没写好，这里会打印错误，防止整个后端起不来
    print(f"❌ 导入服务出错: {e}")
    StockService = None
    stock_buy_schema = None

stocks_bp = Blueprint('stocks', __name__)


# ------------------------------------------------------------------
# 1. 获取入库列表
# URL: /api/v1/stocks/inbound (GET)
# ------------------------------------------------------------------
@stocks_bp.route('/inbound', methods=['GET'])
def get_inbound_list():
    if not StockService:
        return jsonify({'code': 500, 'msg': '后端服务未初始化'}), 500

    try:
        page = request.args.get('page', 1, type=int)
        limit = request.args.get('pageSize', 10, type=int)

        # 调用 Service 层获取数据
        result = StockService.get_list(page, limit)

        return jsonify({
            'code': 200,
            'msg': 'success',
            'data': result
        })
    except Exception as e:
        print(f"获取列表报错: {e}")
        return jsonify({'code': 500, 'msg': '服务器内部错误'}), 500


# ------------------------------------------------------------------
# 2. 新增入库单
# URL: /api/v1/stocks/inbound (POST)
# ------------------------------------------------------------------
@stocks_bp.route('/inbound', methods=['POST'])
def create_inbound():
    if not StockService:
        return jsonify({'code': 500, 'msg': '后端服务未初始化'}), 500

    json_data = request.get_json()
    if not json_data:
        return jsonify({'code': 400, 'msg': '没有接收到数据'}), 400

    try:
        # 1. 参数校验 (Marshmallow Schema)
        data = stock_buy_schema.load(json_data)

        # 2. 调用业务逻辑
        new_stock = StockService.create_inbound(data)

        # 3. 返回成功
        # 注意：确保 new_stock 对象有 to_dict() 方法，否则这里会报错
        return jsonify({
            'code': 200,
            'msg': '入库成功',
            'data': new_stock.to_dict() if hasattr(new_stock, 'to_dict') else str(new_stock)
        }), 201

    except Exception as e:
        # 捕获校验错误或数据库错误
        print(f"入库报错: {e}")
        return jsonify({'code': 400, 'msg': str(e)}), 400


# ------------------------------------------------------------------
# 3. 更新入库单
# URL: /api/v1/stocks/inbound/<id> (PUT)
# ------------------------------------------------------------------
@stocks_bp.route('/inbound/<int:id>', methods=['PUT'])
def update_inbound(id):
    if not StockService:
        return jsonify({'code': 500, 'msg': '后端服务未初始化'}), 500

    json_data = request.get_json()
    try:
        StockService.update_inbound(id, json_data)
        return jsonify({'code': 200, 'msg': '更新成功'})
    except Exception as e:
        return jsonify({'code': 400, 'msg': str(e)}), 400


# ------------------------------------------------------------------
# 4. 删除入库单
# URL: /api/v1/stocks/inbound/<id> (DELETE)
# ------------------------------------------------------------------
@stocks_bp.route('/inbound/<int:id>', methods=['DELETE'])
def delete_inbound(id):
    if not StockService:
        return jsonify({'code': 500, 'msg': '后端服务未初始化'}), 500

    try:
        StockService.delete_inbound(id)
        return jsonify({'code': 200, 'msg': '删除成功'})
    except Exception as e:
        return jsonify({'code': 400, 'msg': str(e)}), 400