from odoo import fields, models, api, tools, _
import logging
import re
import pysolr

_logger = logging.getLogger(__name__)
# _cat_brand_solr = pysolr.Solr('http://10.148.0.5:8983/solr/url_category_brand/', always_commit=True, timeout=30)
_cat_brand_solr_dev = pysolr.Solr('http://localhost:8983/solr/url_category_brand', always_commit=True, timeout=30)


class BrandProductCategory(models.Model):
    _name = 'v.brand.product.category'
    _auto = False
    _rec_name = 'brand_id'
    brand_id = fields.Many2one('x_manufactures', string='Brand')
    category_id = fields.Many2one('product.public.category', string='Category')

    def init(self):
        tools.drop_view_if_exists(self.env.cr, self._table)
        self.env.cr.execute("""
            CREATE OR REPLACE VIEW %s
            AS select row_number() over(order by pt.x_manufacture) as id,
            pt.x_manufacture as brand_id,
            ppcptr.product_public_category_id as category_id
            from product_template pt
            join product_public_category_product_template_rel ppcptr on ppcptr.product_template_id = pt.id
            join x_manufactures xm on xm.id = pt.x_manufacture
            group by x_manufacture, ppcptr.product_public_category_id
        """ % self._table)


class FindPage(models.Model):
    _name = 'web.find.page'
    _inherit = ['mail.thread']
    _rec_name = 'url'

    brand_id = fields.Many2one('x_manufactures', string='Brand')
    category_id = fields.Many2one('product.public.category', string='Category', help='Bisa semua level Category')
    url = fields.Char(string='Url', readonly=True)
    keywords = fields.Char(string='Keywords')


    @api.model
    def create(self, vals):
        record = super(FindPage, self).create(vals)
        record._sync_to_solr()
        return record

    def write(self, vals):
        res = super(FindPage, self).write(vals)
        self._sync_to_solr()
        return res

    def _sync_to_solr(self, limit=10000):
        urls = self.env['web.find.page'].search([])
        documents = []
        catch = {}
        for url in urls:
            try:
                keyword = self.keywords
                document = {
                    'id': url.id,
                    'category_id_i': url.category_id.id,
                    'brand_id_i': url.brand_id.id,
                    'url_s': url.url,
                }
                if keyword:
                    document['keyword_s'] = keyword
                documents.append(document)
                catch = document
            except Exception as e:
                _logger.error('Failed to add document to Solr URL Category Brand: %s', e)
                _logger.error('Document Data: %s', catch)

            try:
                _cat_brand_solr_dev.add(documents)
            except Exception as e:
                _logger.error('Gagal sinkron ke Solr: %s', e)
                return True

        return True

    def _get_category_hierarchy(self, category):
        categories = []
        current_category = category
        while current_category:
            categories.insert(0, current_category)
            current_category = current_category.parent_id
        return categories


    def _generate_url(self):
        list_url = []
        keyword = self.keywords

        if keyword:
            brands = self.env['x_manufactures'].search([('x_name', 'ilike', keyword)])
            categories = self.env['product.public.category'].search([('name', 'ilike', keyword)])

            for brand in brands:
                for category in categories:
                    list_url.append(self._generate_mod_url(category, brand, keyword))
        else:
            # Default: ambil semua kombinasi brand-category dari view
            categories = self.env['v.brand.product.category'].search([])
            for category in categories:
                category_hierarchy = self._get_category_hierarchy(category.category_id)
                for level, cat in enumerate(reversed(category_hierarchy), start=1):
                    list_url.append(self._generate_mod_url(cat, category.brand_id))

        # Hapus duplikat URL
        unique_list = []
        for item in list_url:
            if item not in unique_list:
                unique_list.append(item)

        count = 0
        for item in unique_list:
            self._create_find_page(item['url'], item['category_id'], item['brand_id'])
            count += 1

        _logger.info(f"Total URL generated: {count}")

    def _create_find_page(self, url, category_id, brand_id):
        param = {
            'url': url,
            'category_id': category_id,
            'brand_id': brand_id,
        }
        find_page = self.env['web.find.page'].create(param)
        _logger.info('Created Web Find Page %s' % find_page.id)

    def _generate_mod_url(self, category, brand, keyword=None):
        cleaned_category = re.sub(r'[^\w\s]', '', category.name)
        cleaned_brand = re.sub(r'[^\w\s]', '', brand.x_name)
        cleaned_combined = f"{cleaned_category} {cleaned_brand}"
        cleaned_combined = re.sub(r'\s+', '-', cleaned_combined.strip())

        if keyword:
            slugified_keyword = re.sub(r'\s+', '-', keyword.strip().lower())
            url = f"https://indoteknik.com/shop/find/key={slugified_keyword}"
        else:
            url = f"https://indoteknik.com/shop/find/{cleaned_combined.lower()}"

        result = {
            'url': url,
            'category_id': category.id,
            'brand_id': brand.id
        }
        if keyword:
            result['keyword_s'] = keyword

        return result