// =============================
// Karte initialisieren
// =============================
const map = L.map('map').setView([52.52, 13.405], 13);

L.tileLayer('https://{s}.tile.openstreetmap.org/{z}/{x}/{y}.png', {
  maxZoom: 19,
  attribution: '© OpenStreetMap contributors'
}).addTo(map);

// Layer für Fahrzeuge
const vehiclesLayer = L.layerGroup().addTo(map);
const vehicleMarkers = {};

// =============================
// Farben & Symbole
// =============================
const lineColors = {
  'S45': '#CD9C53', 'S46': '#CD9C53', 'S47': '#CD9C53', 'S6': '#CD9C53', 'S65': '#CD9C53', 'S66': '#CD9C53',
  'S42': '#CB6418', 'S4': '#AD5937', 'S41': '#AD5937', 'U5': '#7E5330', 'U55': '#7E5330',
  'S9': '#992746', 'S95': '#992746', 'S96': '#992746', 'U12': ['#7DAD4C', '#DA421E'], 'U2': '#DA421E',
  'S3': '#EB7405', 'S5': '#EB7405', 'S55': '#EB7405', 'S56': '#EB7405', 'U9': '#F3791D',
  'S3-wannsee': '#F0D722', 'U4': '#F0D722', 'U1': '#7DAD4C', 'U15': '#7DAD4C',
  'S8': '#66AA22', 'S85': '#66AA22', 'S86': '#66AA22', 'S2': '#007734', 'S21': '#007734',
  'S25': '#007734', 'S26': '#007734', 'U3': '#16683D', 'U35': '#16683D',
  'S3-charlottenburg': '#009BD5', 'U7': '#009BD5', 'S10': '#0082C4', 'S19': '#0082C4',
  'S3-berlin-schoenefeld': '#0066AD', 'S35': '#0066AD', 'S36': '#0066AD', 'U8': '#224F86',
  'S7': '#816DA6', 'S75': '#816DA6', 'S76': '#816DA6', 'U6': '#8C6DAB',
  'S1': '#DA6BA2', 'S15': '#DA6BA2', 'S16': '#DA6BA2', 'U0': '#DA6BA2',
  'U10': '#808080', 'U11': '#808080', 'F10': '#528DBA', 'F11': '#528DBA',
  'F12': '#528DBA', 'F21': '#528DBA', 'F23': '#528DBA', 'F24': '#528DBA'
};

const symbols = {
  bus: '🚌', tram: '🚎', subway: '🚉', suburban: '🚋',
  regional: '🚆', express: '🚄', longDistance: '🚅', ferry: '🛥️'
};

// =============================
// Aktive Filter
// =============================
let activeProducts = Array.from(document.querySelectorAll('#controls input[type="checkbox"]'))
  .filter(c => c.checked)
  .map(c => c.value);

// Checkbox Event
document.querySelectorAll('#controls input[type="checkbox"]').forEach(cb => {
  cb.addEventListener('change', () => {
    activeProducts = Array.from(document.querySelectorAll('#controls input[type="checkbox"]'))
      .filter(c => c.checked)
      .map(c => c.value);

    Object.values(vehicleMarkers).forEach(marker => {
      const prod = marker.options.vehicleProduct || '';
      if (activeProducts.includes(prod)) {
        if (!map.hasLayer(marker)) map.addLayer(marker);
      } else {
        if (map.hasLayer(marker)) map.removeLayer(marker);
      }
    });
  });
});

// =============================
// Hilfsfunktionen
// =============================
function getLineColor(lineName, product) {
  if(lineName === 'U12') return null;
  if(lineName.startsWith('S3 ')) {
    if(lineName.toLowerCase().includes('wannsee')) return lineColors['S3-wannsee'];
    if(lineName.toLowerCase().includes('charlottenburg') || lineName.toLowerCase().includes('schönefeld')) return lineColors['S3-berlin-schoenefeld'];
    return lineColors['S3'];
  }
  if(lineColors[lineName]) return lineColors[lineName];
  if(product === 'suburban') return '#CD9C53';
  if(product === 'subway') return '#7DAD4C';
  return '#666';
}

function createDivIcon(vehicle) {
  const product = vehicle.line?.product || '';
  const lineName = vehicle.line?.name || '?';
  const symbol = symbols[product] || '❓';

  if(lineName === 'U12') {
    const iconHtml = `
      <div style="text-align:center; font-size:24px; position: relative; width: 30px; height: 40px;">
        <svg width="30" height="40" viewBox="0 0 30 40" xmlns="http://www.w3.org/2000/svg" style="position:absolute; top:0; left:0;">
          <defs>
            <linearGradient id="u12grad" x1="0" y1="0" x2="1" y2="1">
              <stop offset="0%" stop-color="#7DAD4C" />
              <stop offset="50%" stop-color="#7DAD4C" />
              <stop offset="50%" stop-color="#DA421E" />
              <stop offset="100%" stop-color="#DA421E" />
            </linearGradient>
          </defs>
          <rect width="30" height="40" fill="url(#u12grad)" />
        </svg>
        <div style="position:absolute; top:0; left:0; width:30px; height:40px; line-height:40px; font-size:24px; font-weight:bold; color:#fff; mix-blend-mode: difference;">
          ${symbol}
        </div>
        <div class="line-label" style="position:absolute; bottom:0; left:50%; transform: translateX(-50%); font-size:12px; background:none; color:#000; font-weight:bold; mix-blend-mode:none;">
          U12
        </div>
      </div>
    `;
    const icon = L.divIcon({ html: iconHtml, className: '', iconSize: [30, 40], iconAnchor: [15, 20] });
    icon.product = product;
    return icon;
  }

  const lineColor = getLineColor(lineName, product);
  const labelStyle = `background: ${lineColor}; color: white; padding: 2px 6px; border-radius: 3px; display: inline-block; font-weight: bold; font-size: 12px; user-select: none; font-family: Arial, sans-serif;`;

  const iconHtml = `
    <div style="text-align: center; font-size: 26px; line-height: 1;">
      <div style="color: ${lineColor}; filter: drop-shadow(0 0 1px rgba(0,0,0,0.5));">${symbol}</div>
      <div class="line-label" style="${labelStyle}">${lineName}</div>
    </div>
  `;

  const icon = L.divIcon({ html: iconHtml, className: '', iconSize: [30, 40], iconAnchor: [15, 20] });
  icon.product = product;
  return icon;
}

// =============================
// Fahrzeuge laden
// =============================
async function loadVehicles() {
  const bounds = map.getBounds();
  const bbox = `north=${bounds.getNorth()}&west=${bounds.getWest()}&south=${bounds.getSouth()}&east=${bounds.getEast()}`;

  try {
    const res = await fetch(`https://v6.vbb.transport.rest/radar?${bbox}&results=100&duration=30`);
    const result = await res.json();
    const data = result.movements || [];

    const newIds = new Set();

    data.forEach(vehicle => {
      const id = vehicle.tripId;
      const loc = vehicle.location;
      if (!loc || !loc.latitude || !loc.longitude) return;

      newIds.add(id);
      const icon = createDivIcon(vehicle);

      if (vehicleMarkers[id]) {
        vehicleMarkers[id].setLatLng([loc.latitude, loc.longitude]);
      } else {
        const marker = L.marker([loc.latitude, loc.longitude], { icon });
        marker.options.vehicleProduct = vehicle.line?.product || '';
        if (activeProducts.includes(marker.options.vehicleProduct)) {
          marker.addTo(map);
        }
        vehicleMarkers[id] = marker;
      }
    });

    Object.keys(vehicleMarkers).forEach(id => {
      if (!newIds.has(id)) {
        if (map.hasLayer(vehicleMarkers[id])) map.removeLayer(vehicleMarkers[id]);
        delete vehicleMarkers[id];
      }
    });
  } catch (e) {
    console.error('Fehler beim Laden der Fahrzeuge:', e);
  }
}

// =============================
// Geolocation & Auto-Update
// =============================
function locateAndLoad() {
  if (!navigator.geolocation) {
    loadVehicles();
    return;
  }
  navigator.geolocation.getCurrentPosition(pos => {
    map.setView([pos.coords.latitude, pos.coords.longitude], 14);
    loadVehicles();
  }, () => {
    loadVehicles();
  });
}

locateAndLoad();
setInterval(loadVehicles, 30000);
map.on('moveend', loadVehicles);