from odoo import fields, models, api, _
from odoo.exceptions import AccessError, UserError, ValidationError
from odoo.tools.float_utils import float_is_zero
from collections import defaultdict
from datetime import timedelta, datetime
from datetime import timedelta, datetime as waktu
from itertools import groupby
import pytz, requests, json, requests
from dateutil import parser
import datetime
import hmac
import hashlib
import base64
import requests
import time
import logging
import re
from hashlib import sha256

_logger = logging.getLogger(__name__)


Request_URI = '/openapi/order/v2/list-order'
ACCESS_KEY = '24bb6a1ec618ec6a'
SECRET_KEY = '32e4a78ad05ee230'

class ShipmentGroup(models.Model):
    _name = "shipment.group"
    _description = "Shipment Group"
    _inherit = ['mail.thread']
    _rec_name = 'number'

    number = fields.Char(string='Document No', index=True, copy=False, readonly=True, tracking=True)
    shipment_line = fields.One2many('shipment.group.line', 'shipment_id', string='Shipment Group Lines', auto_join=True)
    picking_lines = fields.One2many('picking.line', 'shipment_id', string='Picking Lines', auto_join=True)
    related_count = fields.Integer(compute='_compute_related_count', string='Related Count')
    receipt = fields.Char(string='Receipt', related='picking_lines.scan_receipt')
    total_line = fields.Integer(string='Total Line', compute='_compute_total_line')
    shipment_methods = fields.Text(string='Shipment Method', help='Metode Pengiriman Barang', compute='_compute_shipment_method')
    @api.depends('picking_lines.carrier')
    def _compute_shipment_method(self):
        for rec in self:
            carriers = rec.picking_lines.mapped('carrier')
            carriers = [c for c in carriers if c]
            rec.shipment_methods = ', '.join(sorted(set(carriers))) if carriers else '-'


    @api.onchange('picking_lines')
    def _onchange_limit_lines(self):
        if len(self.picking_lines) >= 100:
            raise UserError("Maksimal 100 picking line per shipment.")

    @api.depends('picking_lines')
    def _compute_total_line(self):
        for rec in self:
            rec.total_line = len(rec.picking_lines)

    def sync_product_to_picking_line(self):
        for shipment in self:
            cancelled_picking = shipment.picking_lines.filtered(
                lambda picking: picking.status == 'CANCELLED'
            )
            if cancelled_picking:
                cancelled_names = "\n - " + "\n - ".join(cancelled_picking.mapped("name"))
                raise UserError(f"Pickings berikut telah dibatalkan:{cancelled_names}")

            for picking_line in shipment.picking_lines:
                if picking_line.picking_id:
                    # set shipment group ke stock.picking
                    picking_line.picking_id.write({
                        'shipment_group_id': shipment.id
                    })
                    picking_line.picking_id.message_post(
                        body=_(
                            "📦 <b>Dispatched</b><br/>"
                            "Picking ini telah di-dispatch melalui "
                            "<b>Shipment Group %s</b>."
                        ) % shipment.number
                    )

                shipment_lines = self.env['shipment.group.line'].search([
                    ('shipment_id', '=', shipment.id),
                    ('invoice_marketplace', '=', picking_line.invoice_marketplace),
                    ('picking_id', '=', picking_line.picking_id.id)
                ])

                # reset product shipment
                picking_line.product_shipment_lines.unlink()

                for sl in shipment_lines:
                    self.env['product.shipment.line'].create({
                        'picking_line_id': picking_line.id,
                        'product_id': sl.product_id.id,
                        'carrier': picking_line.carrier or sl.carrier,
                        'invoice_marketplace': sl.invoice_marketplace,
                        'picking_id': sl.picking_id.id,
                        'order_reference': sl.order_reference,
                    })


    # def sync_product_to_picking_line(self):
    #     for shipment in self:
    #         cancelled_picking = shipment.picking_lines.filtered(
    #             lambda picking: picking.status == 'CANCELLED'
    #         )
    #         if cancelled_picking:
    #             cancelled_names = "\n - " + "\n - ".join(cancelled_picking.mapped("name"))
    #             raise UserError(f"Pickings berikut telah dibatalkan:{cancelled_names}")

    #         for picking_line in shipment.picking_lines:
    #             shipment_lines = self.env['shipment.group.line'].search([
    #                 ('shipment_id', '=', shipment.id),
    #                 ('invoice_marketplace', '=', picking_line.invoice_marketplace),
    #                 ('picking_id', '=', picking_line.picking_id.id)
    #             ])

    #             picking_line.product_shipment_lines.unlink()

    #             for sl in shipment_lines:
    #                 self.env['product.shipment.line'].create({
    #                     'picking_line_id': picking_line.id,
    #                     'product_id': sl.product_id.id,
    #                     'carrier': sl.carrier or picking_line.carrier,
    #                     'invoice_marketplace': sl.invoice_marketplace,
    #                     'picking_id': sl.picking_id.id,
    #                     'order_reference': sl.order_reference,
    #                 })

    def _compute_related_count(self):
        for record in self:
            record.related_count = len(record.picking_lines)

    def action_view_related_line(self):
        self.ensure_one()

        pickingLines = self.env['picking.line']

        base_bu = pickingLines.search([
            ('shipment_id', '=', self.id)
        ])

        all_bu = base_bu

        return {
            'name': 'Related',
            'type': 'ir.actions.act_window',
            'res_model': 'picking.line',
            'view_mode': 'tree,form',
            'target': 'current',
            'domain': [('id', 'in', list(all_bu.ids))],
        }

    @api.constrains('picking_lines')
    def _check_picking_lines(self):
        for record in self:

            for picking_line in record.picking_lines:
                if not picking_line.picking_id:
                    continue

                for move in picking_line.picking_id.move_line_ids_without_package:
                    existing_line = record.shipment_line.filtered(
                        lambda l: l.product_id == move.product_id and
                        l.picking_id.tracking_number == picking_line.scan_receipt and
                        l.carrier == picking_line.carrier
                    )

                    if not existing_line:
                        self.env['shipment.group.line'].create({
                            'shipment_id': record.id,
                            'product_id': move.product_id.id,
                            'carrier': picking_line.picking_id.carrier,
                            'invoice_marketplace': picking_line.picking_id.invoice_mp,
                            'order_reference': picking_line.order_reference,
                            'picking_id': picking_line.picking_id.id
                        })

    def get_status(self):
        """
        Batch realtime check status order Ginee
        - 1 request untuk banyak order
        - Anti 429
        - Siap dipanggil sebelum validate picking
        """
        # 1️⃣ Kumpulin invoice marketplace
        order_map = {}
        for line in self.picking_lines:
            if line.invoice_marketplace:
                order_map[line.invoice_marketplace] = line

        if not order_map:
            return

        # 2️⃣ Prepare request
        authorization = self.sign_request()
        headers = {
            'Content-Type': 'application/json',
            'X-Advai-Country': 'ID',
            'Authorization': authorization
        }

        payload = {
            "orderNumbers": list(order_map.keys())
        }

        url = "https://api.ginee.com/openapi/order/v2/list-order"

        response = requests.post(
            url,
            headers=headers,
            data=json.dumps(payload),
            timeout=30
        )

        if response.status_code != 200:
            raise UserError(_("API request failed with status code: %s") % response.status_code)

        data = response.json()
        if data.get('code') != 'SUCCESS':
            raise UserError(_("API Error: %s - %s") % (
                data.get('code', 'UNKNOWN'),
                data.get('message', 'No error message')
            ))

        # 3️⃣ Mapping response ke picking line
        contents = data.get('data', {}).get('content', [])
        if not contents:
            return

        for item in contents:
            order_no = item.get('externalOrderId')
            order_status = item.get('orderStatus')

            picking_line = order_map.get(order_no)
            if not picking_line:
                continue

            # Simpan status
            picking_line.status = order_status if order_status == 'CANCELLED' else ''


    def sign_request(self):
        signData = '$'.join(['POST', Request_URI]) + '$'
        authorization = ACCESS_KEY + ':' + base64.b64encode(
            hmac.new(SECRET_KEY.encode('utf-8'), signData.encode('utf-8'), digestmod=sha256).digest()
        ).decode('ascii')
        return authorization


    @api.model
    def create(self, vals):
        vals['number'] = self.env['ir.sequence'].next_by_code('shipment.group') or '0'
        result = super(ShipmentGroup, self).create(vals)
        return result

class ShipmentGroupLine(models.Model):
    _name = 'shipment.group.line'
    _description = 'Shipment Group Line'
    _order = 'shipment_id, id'

    shipment_id = fields.Many2one('shipment.group', string='Shipment Ref', required=True, ondelete='cascade')
    product_id = fields.Many2one('product.product', string='Product')
    carrier = fields.Char(string='Shipping Method')
    invoice_marketplace = fields.Char(string='Invoice Marketplace')
    picking_id = fields.Many2one('stock.picking', string='Picking')
    order_reference = fields.Char(string='Order Reference')


class PickingLine(models.Model):
    _name = 'picking.line'
    _description = 'Picking Line'
    _order = 'shipment_id, id'

    shipment_id = fields.Many2one('shipment.group', string='Shipment Ref', required=True, ondelete='cascade')
    product_shipment_lines = fields.One2many('product.shipment.line', 'picking_line_id', string='Product Shipment Lines', auto_join=True)
    picking_id = fields.Many2one('stock.picking', string='Picking')
    scan_receipt = fields.Char(string="Scan Receipt")
    invoice_marketplace = fields.Char(string='Invoice Marketplace')
    carrier = fields.Char(string='Ekspedisi')
    order_reference = fields.Char(string='Order Reference')
    status = fields.Char(string='Status')

    @api.onchange('scan_receipt')
    def _check_duplicate_scan(self):
        for line in self:
            if not line.scan_receipt:
                continue

            # Cari duplikat selain dirinya
            dup = self.search([
                ('scan_receipt', '=', line.scan_receipt),
                # ('id', '!=', line.id)
            ], limit=1)

            if not dup:
                continue

            # 1. Dup di shipment yang sama (lagi create / edit)
            # if dup.shipment_id.id == line.shipment_id.id:
            #     raise ValidationError(
            #         "Receipt '%s' sudah ada di Shipment ini." %
            #         line.scan_receipt
            #     )

            # 2. Dup di shipment lain (data lama)
            raise ValidationError(
                "Receipt '%s' sudah digunakan di Shipment %s." %
                (line.scan_receipt, dup.shipment_id.number or '-')
            )


    @api.onchange('scan_receipt')
    def _onchange_scan_receipt(self):
        for line in self:
            if not line.scan_receipt:
                continue

            picking = self.env['stock.picking'].search([
                ('tracking_number', '=', line.scan_receipt),
                ('state', '=', 'done')
            ], limit=1)

            if not picking:
                picking = self.env['stock.picking'].search([
                    ('invoice_mp', '=', line.scan_receipt),
                    ('state', '=', 'done')
                ], limit=1)

            if not picking:
                raise UserError(
                    "Receipt '%s' Tidak ditemukan / Status Picking tidak Done." % line.scan_receipt
                )

            # isi field
            line.picking_id = picking
            line.carrier = picking.ginee_provider_name
            line.order_reference = picking.order_reference
            line.invoice_marketplace = picking.invoice_mp



class ProductShipmentLine(models.Model):
    _name = 'product.shipment.line'
    _description = 'Product Shipment Line'
    _order = 'picking_line_id, id'

    picking_line_id = fields.Many2one('picking.line', string='Picking Line', required=True, ondelete='cascade')
    product_id = fields.Many2one('product.product', string='Product')
    carrier = fields.Char(string='Shipping Method')
    invoice_marketplace = fields.Char(string='Invoice Marketplace')
    picking_id = fields.Many2one('stock.picking', string='Picking')
    order_reference = fields.Char(string='Order Reference')