# -*- coding: utf-8 -*-

from odoo import http
from odoo.http import request, Response
import base64
import json


class QualityController(http.Controller):

    @http.route(['/api/quality/report/download'], type='http', auth='public', csrf=False, website=False)  # 移除 cors="*"
    def get_quality_report(self, retrospect_ref=None, **kwargs):
        """获取质检报告的下载接口
        
        Args:
            retrospect_ref: 追溯码
            
        Returns:
            直接返回文件下载响应
        """
        try:
            # 如果retrospect_ref为None，尝试从查询参数获取
            if not retrospect_ref:
                retrospect_ref = kwargs.get('retrospect_ref')

            # 参数验证
            if not retrospect_ref:
                return self._json_response({
                    'status': 'error',
                    'message': '追溯码不能为空'
                })

            # 查找对应的质检单
            quality_check = request.env['quality.check'].sudo().search([
                ('picking_id.retrospect_ref', '=', retrospect_ref),
                ('publish_status', '=', 'published')  # 只返回已发布的报告
            ], limit=1)

            if not quality_check:
                return self._json_response({
                    'status': 'error',
                    'message': '未找到对应的质检报告或报告未发布'
                })

            if not quality_check.report_number_id:
                return self._json_response({
                    'status': 'error',
                    'message': '质检报告文件不存在'
                })

            # 获取文件内容
            document = quality_check.report_number_id
            if not document.raw:  # 检查文件内容是否存在
                return self._json_response({
                    'status': 'error',
                    'message': '文件内容不存在'
                })

            # 构建文件名(确保有.pdf后缀)
            filename = document.name
            if not filename.lower().endswith('.pdf'):
                filename = f"{filename}.pdf"

            # 返回文件下载响应
            return Response(
                document.raw,
                headers=[
                    ('Content-Type', 'application/pdf'),
                    ('Content-Disposition', f'attachment; filename="{filename}"'),
                    ('Access-Control-Allow-Origin', '*'),
                    ('Access-Control-Allow-Methods', 'GET, OPTIONS'),
                    ('Access-Control-Allow-Headers', 'Content-Type, Authorization')
                ]
            )

        except Exception as e:
            return self._json_response({
                'status': 'error',
                'message': f'系统错误: {str(e)}'
            })

    def _json_response(self, data):
        """返回JSON格式的响应"""
        return Response(
            json.dumps(data, ensure_ascii=False),
            mimetype='application/json;charset=utf-8',
            headers=[
                ('Access-Control-Allow-Origin', '*'),
                ('Access-Control-Allow-Methods', 'GET, OPTIONS'),
                ('Access-Control-Allow-Headers', 'Content-Type, Authorization')
            ]
        )


class QualityReportController(http.Controller):
    @http.route('/quality/report/<int:document_id>', type='http', auth='public')
    def get_public_report(self, document_id, **kw):
        """提供公开访问PDF报告的控制器"""
        document = request.env['documents.document'].sudo().browse(int(document_id))

        # 安全检查：确保只有质检报告文档可以被访问
        if document.exists() and document.res_model == 'quality.check':
            # 获取PDF内容
            pdf_content = document.raw

            # 返回PDF内容
            return request.make_response(
                pdf_content,
                headers=[
                    ('Content-Type', 'application/pdf'),
                    ('Content-Disposition', f'inline; filename={document.name}.pdf')
                ]
            )
        return request.not_found()

    @http.route('/quality/report/not_published', type='http', auth='public')
    def get_not_published_report(self, **kw):
        """提供未发布报告的控制器"""
        return "报告尚未发布"