import sys
from bs4 import BeautifulSoup
import re
import unicodedata

def slugify(value):
    value = unicodedata.normalize('NFKD', value).encode('ascii', 'ignore').decode('utf-8')
    value = re.sub(r'[^\w\s-]', '', value).strip().lower()
    return re.sub(r'[-\s]+', '-', value)

# The slugs used in our DB
db_slugs = [
    "chuyen-khoa-tai-mui-hong",
    "khoa-chan-doan-hinh-anh",
    "khoa-cap-cuu",
    "khoa-dinh-duong-tiet-che",
    "khoa-duoc",
    "khoa-kham-benh",
    "don-vi-kiem-soat-nhiem-khuan",
    "khoa-ngoai-tong-hop",
    "tim-mach", # This is Khoa Nội Tổng Hợp in our DB
    "phau-thuat-gay-me-hoi-suc",
    "khoa-than-kinh-dot-quy",
    "khoa-vat-ly-tri-lieu-phuc-hoi-chuc-nang",
    "khoa-xet-nghiem",
    "phong-quan-ly-van-hanh",
    "don-vi-can-thiep-mach-dsa",
    "don-vi-cap-cuu-ngoai-vien",
    "don-vi-kham-suc-khoe-ngoai-vien"
]

manual_mapping = {
    "noi tong hop": "tim-mach",
    "phau thuat gay me hoi suc phong mo": "phau-thuat-gay-me-hoi-suc",
    "dsa": "don-vi-can-thiep-mach-dsa",
    "tai mui hong": "chuyen-khoa-tai-mui-hong",
    "kiem soat nhiem khuan": "don-vi-kiem-soat-nhiem-khuan",
    "kham suc khoe ngoai vien": "don-vi-kham-suc-khoe-ngoai-vien",
    "cap cuu ngoai vien": "don-vi-cap-cuu-ngoai-vien"
}

html_path = '/home/x79/sisvietnamvn_01/BV_DHYD_HCM/Các chuyên khoa tại Bệnh viện Đại học Y Dược TP. Hồ Chí Minh.html'
with open(html_path, 'r', encoding='utf-8') as f:
    soup = BeautifulSoup(f, 'html.parser')

icons = {}
links = soup.find_all('a', href=True)

for a in links:
    href = a.get('href')
    if '/chuyen-khoa/' in href:
        img = a.find('img')
        if img:
            src = img.get('src')
            if not src:
                # Next image srcset or similar fallback
                src = img.get('data-src') or img.get('srcset')
                if src and ' ' in src:
                    src = src.split(' ')[0]
            
            title = href.split('/')[-1] # use the slug from href
            
            # Find the actual name
            name_span = a.find('span')
            if name_span:
                title = name_span.text.strip()
            else:
                title_text = a.text.strip()
                if title_text:
                    title = title_text
                    
            if src:
                # Decode src if it's encoded or make it absolute
                if src.startswith('/_next/image?url='):
                    import urllib.parse
                    src = urllib.parse.unquote(src.replace('/_next/image?url=', '').split('&')[0])
                
                # Match it
                s_text = slugify(title)
                matched_slug = None
                for key, val in manual_mapping.items():
                    if slugify(key) in s_text:
                        matched_slug = val
                        break
                        
                if not matched_slug:
                    for s in db_slugs:
                        if s_text in s or s in s_text or s_text.replace("khoa-", "") in s:
                            matched_slug = s
                            break
                            
                if matched_slug:
                    icons[matched_slug] = src

print(f"Mapped {len(icons)} image icons to specialties")

# Generate the thymeleaf switch block for images
switch_code = '<th:block th:switch="${spec.slug}">\n'
for slug, src in icons.items():
    switch_code += f'                  <!-- {slug} -->\n'
    switch_code += f'                  <th:block th:case="\'{slug}\'">\n'
    switch_code += f'                    <img src="{src}" alt="Icon" class="w-12 h-12 object-contain" />\n'
    switch_code += f'                  </th:block>\n'

# Default case
switch_code += '                  <!-- Default -->\n'
switch_code += '                  <th:block th:case="*">\n'
switch_code += '                    <div class="w-12 h-12 rounded-full bg-primary-100 flex items-center justify-center text-primary-600">\n'
switch_code += '                      <svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="1.5" stroke-linecap="round" stroke-linejoin="round" class="w-6 h-6">\n'
switch_code += '                        <path d="M8 2v4M16 2v4M3 10h18M5 4h14a2 2 0 012 2v14a2 2 0 01-2 2H5a2 2 0 01-2-2V6a2 2 0 012-2zM9 14h6M12 11v6"/>\n'
switch_code += '                      </svg>\n'
switch_code += '                    </div>\n'
switch_code += '                  </th:block>\n'
switch_code += '                </th:block>'

with open('/tmp/icons_block_img.txt', 'w', encoding='utf-8') as f:
    f.write(switch_code)
    
print("Wrote switch block to /tmp/icons_block_img.txt")