#!/usr/bin/env python3
"""
Adres Analyse - Rapport generator
Gebruikt PDOK voor geocoding, ORS voor isochronen/routes en PostGIS OSM-data voor voorzieningen.
Slaat rapport en HTML-kaarten op in /root/mkdocs/docs/docs/Huizenjacht/
"""

import json
import sys
import os
import urllib.request
import urllib.parse
import re
import base64
from datetime import datetime
from io import BytesIO

try:
    from playwright.sync_api import sync_playwright
    HAS_PLAYWRIGHT = True
except ImportError:
    HAS_PLAYWRIGHT = False

try:
    from PIL import Image
    HAS_PIL = True
except ImportError:
    HAS_PIL = False

from voorzieningen import analyse_voorzieningen, format_voorzieningen_markdown

# Config
ORS_BASE = "http://192.168.1.71:9080/ors/v2"
PDOK_BASE = "https://api.pdok.nl/bzk/locatieserver/search/v3_1"
MKDOCS_DIR = "/root/mkdocs/docs/docs/Huizenjacht"

def geocode_adres(adres):
    """Geocode adres naar coördinaten via PDOK Locatieserver"""
    params = urllib.parse.urlencode({
        'q': adres,
        'rows': 5,
        'fl': 'centroide_ll,weergavenaam,postcode,woonplaatsnaam,type,huisnummer'
    })
    
    url = f"{PDOK_BASE}/free?{params}"
    with urllib.request.urlopen(url, timeout=10) as resp:
        data = json.loads(resp.read())
    
    if not data['response']['docs']:
        raise ValueError(f"Adres niet gevonden: {adres}")
    
    doc = None
    for d in data['response']['docs']:
        if d.get('type') == 'adres':
            doc = d
            break
    
    if not doc:
        doc = data['response']['docs'][0]
    
    point = doc['centroide_ll']
    coords = point.replace('POINT(', '').replace(')', '').split()
    lon, lat = float(coords[0]), float(coords[1])
    
    return {
        'adres': doc.get('weergavenaam', adres),
        'postcode': doc.get('postcode', ''),
        'woonplaats': doc.get('woonplaatsnaam', ''),
        'lat': lat,
        'lon': lon
    }

def haal_isochroon(lon, lat, profiel='driving-car', minuten=20):
    """Haal isochroon op van ORS"""
    url = f"{ORS_BASE}/isochrones/{profiel}"
    payload = {
        "locations": [[lon, lat]],
        "range": [minuten * 60],
        "attributes": ["area", "reachfactor", "total_pop"]
    }
    
    req = urllib.request.Request(
        url,
        data=json.dumps(payload).encode(),
        headers={'Content-Type': 'application/json'},
        method='POST'
    )
    
    with urllib.request.urlopen(req, timeout=30) as resp:
        return json.loads(resp.read())

def format_oppervlakte(m2):
    """Converteer m² naar leesbaar formaat"""
    if m2 >= 1_000_000:
        return f"{m2/1_000_000:.1f} km²"
    elif m2 >= 10_000:
        return f"{m2/10_000:.1f} ha"
    else:
        return f"{m2:.0f} m²"

def schrijf_naar_mkdocs(content, adres, ext=".md"):
    """Schrijf content naar MkDocs docs map als bestand"""
    os.makedirs(MKDOCS_DIR, exist_ok=True)
    safe_name = re.sub(r'[^a-zA-Z0-9]+', '-', adres)
    safe_name = safe_name.strip('-')
    filename = f"{safe_name}_{datetime.now().strftime('%Y%m%d_%H%M%S')}{ext}"
    filepath = os.path.join(MKDOCS_DIR, filename)
    
    with open(filepath, 'w', encoding='utf-8') as f:
        f.write(content)
    
    return filepath, filename

def genereer_isochroon_html(adres_naam, lat, lon, auto_iso, fiets_iso):
    """Genereer een zelf-contained HTML kaart met Leaflet"""
    html = f'''<!DOCTYPE html>
<html lang="nl">
<head>
<meta charset="UTF-8">
<meta name="viewport" content="width=device-width, initial-scale=1.0">
<title>Isochroon - {adres_naam}</title>
<link rel="stylesheet" href="https://unpkg.com/leaflet@1.9.4/dist/leaflet.css"
  integrity="sha256-p4NxAoJBhIIN+hmNHrzRCf9tD/miZyoHS5obTRR9BMY=" crossorigin="" />
<script src="https://unpkg.com/leaflet@1.9.4/dist/leaflet.js"
  integrity="sha256-20nQCchB9co0qIjJZRGuk2/Z9VM+kNiyxNV1lvTlZBo=" crossorigin=""></script>
<style>
  body {{ margin: 0; padding: 0; font-family: system-ui, -apple-system, sans-serif; }}
  #map {{ height: 100vh; width: 100vw; }}
  .legend {{ background: white; padding: 10px; border-radius: 4px; box-shadow: 0 0 15px rgba(0,0,0,0.2); line-height: 1.5; }}
  .legend i {{ width: 18px; height: 18px; float: left; margin-right: 8px; opacity: 0.7; border: 1px solid #333; }}
</style>
</head>
<body>
<div id="map"></div>
<script>
var autoGeojson = {json.dumps(auto_iso, ensure_ascii=False)};
var fietsGeojson = {json.dumps(fiets_iso, ensure_ascii=False)};

var map = L.map('map').setView([{lat}, {lon}], 11);

L.tileLayer('https://{{s}}.tile.openstreetmap.org/{{z}}/{{x}}/{{y}}.png', {{
  attribution: '&copy; <a href="https://www.openstreetmap.org/copyright">OpenStreetMap</a> contributors'
}}).addTo(map);

var autoLayer = L.geoJSON(autoGeojson, {{
  style: {{ color: '#2563eb', weight: 2, fillColor: '#3b82f6', fillOpacity: 0.2 }},
  onEachFeature: function(feature, layer) {{
    var props = feature.properties;
    layer.bindPopup('<b>Auto 20 min</b><br>Oppervlakte: ' + (props.area/1000000).toFixed(1) + ' km\\u00b2<br>Reach factor: ' + props.reachfactor.toFixed(3));
  }}
}}).addTo(map);

var fietsLayer = L.geoJSON(fietsGeojson, {{
  style: {{ color: '#16a34a', weight: 2, fillColor: '#22c55e', fillOpacity: 0.2 }},
  onEachFeature: function(feature, layer) {{
    var props = feature.properties;
    layer.bindPopup('<b>Fiets 30 min</b><br>Oppervlakte: ' + (props.area/1000000).toFixed(1) + ' km\\u00b2<br>Reach factor: ' + props.reachfactor.toFixed(3));
  }}
}}).addTo(map);

L.marker([{lat}, {lon}]).addTo(map)
  .bindPopup('<b>{adres_naam}</b>').openPopup();

var legend = L.control({{position: 'bottomright'}});
legend.onAdd = function(map) {{
  var div = L.DomUtil.create('div', 'legend');
  div.innerHTML += '<h4 style="margin:0 0 5px 0">Isochronen</h4>';
  div.innerHTML += '<i style="background:#3b82f6"></i> Auto 20 min<br>';
  div.innerHTML += '<i style="background:#22c55e"></i> Fiets 30 min<br>';
  div.innerHTML += '<i style="background:#ef4444"></i> Adres';
  return div;
}};
legend.addTo(map);

map.fitBounds(autoLayer.getBounds().extend(fietsLayer.getBounds()));
</script>
</body>
</html>'''
    return html

def maak_isochroon_screenshot(html_content, width=1280, height=900, resize_width=800, quality=70):
    """Genereer een JPEG screenshot van de HTML kaart, retourneer base64 string"""
    if not HAS_PLAYWRIGHT:
        return None, "Playwright niet geïnstalleerd"
    if not HAS_PIL:
        return None, "Pillow niet geïnstalleerd"

    import tempfile

    # Write HTML to temp file
    with tempfile.NamedTemporaryFile(mode='w', suffix='.html', delete=False, encoding='utf-8') as f:
        f.write(html_content)
        html_path = f.name

    try:
        with sync_playwright() as p:
            browser = p.chromium.launch(headless=True)
            page = browser.new_page(viewport={'width': width, 'height': height})
            page.goto(f'file://{html_path}')
            page.wait_for_timeout(2500)  # Wait for tiles
            png_bytes = page.screenshot(full_page=False)
            browser.close()
    except Exception as e:
        os.unlink(html_path)
        return None, f"Screenshot fout: {e}"
    finally:
        os.unlink(html_path)

    # Compress to JPEG
    img = Image.open(BytesIO(png_bytes))
    ratio = resize_width / img.width
    resize_height = int(img.height * ratio)
    img = img.resize((resize_width, resize_height), Image.Resampling.LANCZOS)

    buf = BytesIO()
    img.save(buf, format='JPEG', quality=quality, optimize=True)
    jpeg_bytes = buf.getvalue()

    b64 = base64.b64encode(jpeg_bytes).decode('ascii')
    return b64, None

def print_isochroon_sectie(titel, icoon, iso_data, rapport_lines=None):
    """Print isochroon sectie; optioneel ook toevoegen aan rapport_lines"""
    text = f"{icoon} **{titel}**"
    print(text)
    if rapport_lines is not None:
        rapport_lines.append(text)
    if 'features' in iso_data and iso_data['features']:
        props = iso_data['features'][0]['properties']
        area_m2 = props.get('area', 0)
        reach = props.get('reachfactor', 0)
        text = f"**Oppervlakte:** {format_oppervlakte(area_m2)}"
        print(text)
        if rapport_lines is not None:
            rapport_lines.append(text)
        text = f"**Reach factor:** {reach:.3f}"
        print(text)
        if rapport_lines is not None:
            rapport_lines.append(text)
        bbox = iso_data.get('bbox', [])
        if len(bbox) == 4:
            text = f"**Bereik:** {bbox[1]:.3f}°-{bbox[3]:.3f}°N, {bbox[0]:.3f}°-{bbox[2]:.3f}°E"
            print(text)
            if rapport_lines is not None:
                rapport_lines.append(text)
    else:
        text = "❌ Geen isochroon data ontvangen"
        print(text)
        if rapport_lines is not None:
            rapport_lines.append(text)
    print()
    if rapport_lines is not None:
        rapport_lines.append("")

def genereer_rapport(adres_input):
    """Genereer volledig adres analyse rapport"""
    rapport_lines = []

    def print_line(text=""):
        print(text)
        rapport_lines.append(text)

    print_line(f"📍 **Adres Analyse Rapport**")
    print_line(f"Gegenereerd: {datetime.now().strftime('%d-%m-%Y %H:%M')}")
    print_line()

    print_line("🔍 Geocoding...")
    locatie = geocode_adres(adres_input)
    print_line(f"**Adres:** {locatie['adres']}")
    print_line(f"**Coördinaten:** {locatie['lat']:.6f}, {locatie['lon']:.6f}")
    print_line()

    auto_iso = haal_isochroon(locatie['lon'], locatie['lat'], 'driving-car', 20)
    fiets_iso = haal_isochroon(locatie['lon'], locatie['lat'], 'cycling-regular', 30)
    print_isochroon_sectie("Bereikbaarheid (Auto 20 min)", "🚗", auto_iso, rapport_lines)
    print_isochroon_sectie("Bereikbaarheid (Fiets 30 min)", "🚲", fiets_iso, rapport_lines)

    print_line("🏙️ **Voorzieningen dichtbij**")
    voorzieningen = analyse_voorzieningen(locatie['lon'], locatie['lat'], limit=5)
    print_line(format_voorzieningen_markdown("🛒 **Supermarkten**", voorzieningen['supermarkten']))
    print_line()
    print_line(format_voorzieningen_markdown("🏫 **Scholen**", voorzieningen['scholen']))
    print_line()

    # Genereer interactieve HTML kaart en sla op in MkDocs.map
    html_content = genereer_isochroon_html(
        locatie['adres'], locatie['lat'], locatie['lon'], auto_iso, fiets_iso
    )
    html_filepath, html_filename = schrijf_naar_mkdocs(html_content, locatie['adres'], ext=".html")
    print_line(f"🗺️ **Interactieve kaart opgeslagen:** `{html_filepath}`")

    # Genereer screenshot en embed in markdown
    b64_img, err = maak_isochroon_screenshot(html_content)
    if b64_img:
        print_line("")
        print_line("**🗺️ Isochroon kaart (auto + fiets):**")
        img_md = f"![Isochroon kaart](data:image/jpeg;base64,{b64_img})"
        print_line(img_md)
    else:
        print_line(f"⚠️ Kon geen screenshot maken: {err}")

    # Sla ook ruwe geojson op
    output_file = f"/tmp/isochronen_{datetime.now().strftime('%Y%m%d_%H%M%S')}.geojson"
    with open(output_file, 'w') as f:
        json.dump({'auto_20_min': auto_iso, 'fiets_30_min': fiets_iso}, f)
    print_line(f"💾 Ruwe isochroondata: `{output_file}`")
    print_line()
    print_line("---")
    print_line("_Onderdelen: geocoding, isochronen, voorzieningen_")

    # Schrijf rapport naar MkDocs map
    print_line()
    md_filepath, md_filename = schrijf_naar_mkdocs('\n'.join(rapport_lines), locatie['adres'], ext=".md")
    print_line(f"📝 **Rapport opgeslagen:** `{md_filepath}`")

if __name__ == '__main__':
    adres = sys.argv[1] if len(sys.argv) > 1 else "Griendvelden 14 Best"
    genereer_rapport(adres)