# -*- coding: utf-8 -*-
# ir_attachment.py
from odoo import models, _
from odoo.exceptions import UserError
import base64


def human_file_size(size_bytes):
    """Convert bytes to human readable format"""
    if size_bytes == 0:
        return "0 B"
    size_name = ("B", "KB", "MB", "GB", "TB")
    import math
    i = int(math.floor(math.log(size_bytes, 1024)))
    p = math.pow(1024, i)
    s = round(size_bytes / p, 2)
    return "%s %s" % (s, size_name[i])


class IrHttp(models.AbstractModel):
    _inherit = "ir.http"

    def session_info(self):
        print("session_info called ✅")

        session_info = super().session_info()

        user = self.env.user
        current_user_id = user.id

        user_id = self.env['res.users'].browse(current_user_id)

        max_attachment_size_bytes = None

        if user_id and user_id.user_select_attachment_size_unit and user_id.users_general_size_limit_attachment > 0:
            unit = user_id.user_select_attachment_size_unit
            size = user_id.users_general_size_limit_attachment

            if unit == 'kb':
                max_attachment_size_bytes = size * 1024
            elif unit == 'mb':
                max_attachment_size_bytes = size * 1024 * 1024
            else:
                max_attachment_size_bytes = size * 1024 * 1024

        else:
            enable_general_limit = self.env['ir.config_parameter'].sudo().get_param(
                'pys_attachment_size_limitation.enable_general_size_limit_attachment',
                False
            )

            if enable_general_limit:
                unit = self.env["ir.config_parameter"].sudo().get_param(
                    "pys_attachment_size_limitation.select_attachment_size_unit",
                )

                size = int(
                    self.env["ir.config_parameter"].sudo().get_param(
                        "pys_attachment_size_limitation.general_size_limit_attachment",
                        0
                    )
                )

                if unit and size > 0:
                    if unit == 'kb':
                        max_attachment_size_bytes = size * 1024
                    elif unit == 'mb':
                        max_attachment_size_bytes = size * 1024 * 1024
                    else:
                        max_attachment_size_bytes = size * 1024 * 1024

        # Ensure it's an integer
        session_info["max_attachment_size"] = int(max_attachment_size_bytes) if max_attachment_size_bytes else None

        print("max_attachment_size (in bytes):", session_info["max_attachment_size"])
        return session_info


class IrAttachment(models.Model):
    _inherit = 'ir.attachment'

    def create(self, vals_list):
        # Check size before creation
        for vals in vals_list:
            if vals.get('datas'):
                try:
                    decoded_data = base64.b64decode(vals['datas'])
                    max_size = self.env['ir.http'].session_info().get('max_attachment_size')

                    if max_size and max_size > 0 and len(decoded_data) > max_size:
                        raise UserError(_(
                            "Attachment size (%s) exceeds the maximum allowed size (%s)."
                        ) % (
                                            human_file_size(len(decoded_data)),
                                            human_file_size(max_size)
                                        ))
                except Exception as e:
                    if isinstance(e, UserError):
                        raise
                    # If base64 decoding fails, continue

        return super().create(vals_list)