/// <reference types="google.maps" />
import { APIProvider, InfoWindow, Map as GoogleMap, useMap } from '@vis.gl/react-google-maps';
import { useEffect, useRef, useState } from 'react';
import { JobMapProps, LiveStatus, liveStatusToMapPoint, MapPoint, pointDetailRows, UTILITY_COLORS } from '../types';

const API_KEY = process.env.NEXT_PUBLIC_GOOGLE_MAPS_API_KEY || '';
const LIVE_COLOR = '#1a73e8';

const DEFAULT_CENTER = { lat: 39.5, lng: -98.35 }; // continental US
const DEFAULT_ZOOM = 4;

function colorFor(utilityType: string): string {
  return UTILITY_COLORS[utilityType] ?? UTILITY_COLORS.UNKNOWN;
}

function orderKey(p: MapPoint): number {
  return p.sequence ?? new Date(p.recordedAt).getTime();
}

// Draws points as colored circles plus one polyline per utility run.
// Imperative overlays (google.maps.Marker/Polyline) are used because
// @vis.gl/react-google-maps has no polyline component.
function PointsLayer({
  points,
  fitBounds,
  onSelect,
}: {
  points: MapPoint[];
  fitBounds: boolean;
  onSelect: (point: MapPoint) => void;
}) {
  const map = useMap();
  const overlaysRef = useRef<{ setMap: (m: google.maps.Map | null) => void }[]>([]);
  const fittedCountRef = useRef(0);
  const onSelectRef = useRef(onSelect);
  onSelectRef.current = onSelect;

  useEffect(() => {
    if (!map) {
      return;
    }

    overlaysRef.current.forEach((o) => o.setMap(null));
    overlaysRef.current = [];
    if (points.length === 0) {
      return;
    }

    const byUtility = new Map<string, MapPoint[]>();
    for (const p of points) {
      const group = byUtility.get(p.utilityType) ?? [];
      group.push(p);
      byUtility.set(p.utilityType, group);
    }

    for (const [utility, group] of byUtility) {
      const sorted = [...group].sort((a, b) => orderKey(a) - orderKey(b));
      const color = colorFor(utility);

      const line = new google.maps.Polyline({
        path: sorted.map((p) => ({ lat: p.lat, lng: p.lng })),
        strokeColor: color,
        strokeOpacity: 0.8,
        strokeWeight: 3,
        map,
      });
      overlaysRef.current.push(line);

      for (const p of sorted) {
        const marker = new google.maps.Marker({
          position: { lat: p.lat, lng: p.lng },
          map,
          cursor: 'pointer',
          icon: {
            path: google.maps.SymbolPath.CIRCLE,
            scale: 5,
            fillColor: color,
            fillOpacity: 1,
            strokeColor: '#ffffff',
            strokeWeight: 1.5,
          },
        });
        marker.addListener('click', () => onSelectRef.current(p));
        overlaysRef.current.push(marker);
      }
    }

    if (fitBounds && points.length !== fittedCountRef.current) {
      fittedCountRef.current = points.length;
      const bounds = new google.maps.LatLngBounds();
      points.forEach((p) => bounds.extend({ lat: p.lat, lng: p.lng }));
      map.fitBounds(bounds, 48);
    }
  }, [map, points, fitBounds]);

  return null;
}

// Renders the transmitter's current position as a "blue dot" (marker + soft
// accuracy halo), updated in place as new status messages arrive.
function LiveMarker({ status, onSelect }: { status: LiveStatus | null; onSelect: (point: MapPoint) => void }) {
  const map = useMap();
  const markerRef = useRef<google.maps.Marker | null>(null);
  const haloRef = useRef<google.maps.Circle | null>(null);
  const statusRef = useRef(status);
  statusRef.current = status;
  const onSelectRef = useRef(onSelect);
  onSelectRef.current = onSelect;

  useEffect(() => {
    if (!map) {
      return;
    }
    if (!status) {
      markerRef.current?.setMap(null);
      haloRef.current?.setMap(null);
      markerRef.current = null;
      haloRef.current = null;
      return;
    }

    const position = { lat: status.lat, lng: status.lng };

    if (!markerRef.current) {
      markerRef.current = new google.maps.Marker({
        map,
        position,
        zIndex: 999,
        cursor: 'pointer',
        icon: {
          path: google.maps.SymbolPath.CIRCLE,
          scale: 8,
          fillColor: LIVE_COLOR,
          fillOpacity: 1,
          strokeColor: '#ffffff',
          strokeWeight: 2,
        },
      });
      markerRef.current.addListener('click', () => {
        if (statusRef.current) {
          onSelectRef.current(liveStatusToMapPoint(statusRef.current));
        }
      });
      haloRef.current = new google.maps.Circle({
        map,
        center: position,
        radius: status.hAccuracy ?? 5,
        fillColor: LIVE_COLOR,
        fillOpacity: 0.15,
        strokeColor: LIVE_COLOR,
        strokeOpacity: 0.3,
        strokeWeight: 1,
        clickable: false,
      });
    } else {
      markerRef.current.setPosition(position);
      haloRef.current?.setCenter(position);
      haloRef.current?.setRadius(status.hAccuracy ?? 5);
    }
  }, [map, status]);

  useEffect(
    () => () => {
      markerRef.current?.setMap(null);
      haloRef.current?.setMap(null);
    },
    [],
  );

  return null;
}

function PointDetails({ point, onClose }: { point: MapPoint; onClose: () => void }) {
  return (
    <InfoWindow position={{ lat: point.lat, lng: point.lng }} onCloseClick={onClose}>
      <div style={{ fontFamily: 'sans-serif', minWidth: 200 }}>
        <table style={{ borderCollapse: 'collapse' }}>
          <tbody>
            {pointDetailRows(point).map(([label, value]) => (
              <tr key={label}>
                <td style={{ color: '#666', paddingRight: '0.75rem', whiteSpace: 'nowrap' }}>{label}</td>
                <td style={{ fontWeight: 600 }}>{value}</td>
              </tr>
            ))}
          </tbody>
        </table>
      </div>
    </InfoWindow>
  );
}

export default function GoogleJobMap({ points, liveStatus = null, fitBounds = true, heightPx = 480 }: JobMapProps) {
  const [selected, setSelected] = useState<MapPoint | null>(null);

  if (!API_KEY) {
    return (
      <div
        style={{
          height: heightPx,
          display: 'flex',
          alignItems: 'center',
          justifyContent: 'center',
          border: '1px dashed #999',
          borderRadius: 8,
          color: '#666',
        }}
      >
        Set NEXT_PUBLIC_GOOGLE_MAPS_API_KEY in .env to enable the map.
      </div>
    );
  }

  // A point can be re-fetched (new object identity) between renders; keep the
  // InfoWindow anchored to the latest copy of the same point by id. The live
  // marker's "point" is synthesized fresh from liveStatus each time.
  const selectedCurrent = (() => {
    if (!selected) {
      return null;
    }
    if (selected.id.startsWith('live-')) {
      return liveStatus ? liveStatusToMapPoint(liveStatus) : null;
    }
    return points.find((p) => p.id === selected.id) ?? selected;
  })();

  return (
    <div style={{ height: heightPx, borderRadius: 8, overflow: 'hidden' }}>
      <APIProvider apiKey={API_KEY}>
        <GoogleMap
          defaultCenter={points[0] ? { lat: points[0].lat, lng: points[0].lng } : DEFAULT_CENTER}
          defaultZoom={points[0] ? 18 : DEFAULT_ZOOM}
          mapTypeId="hybrid"
          gestureHandling="greedy"
          onClick={() => setSelected(null)}
        >
          <PointsLayer points={points} fitBounds={fitBounds} onSelect={setSelected} />
          <LiveMarker status={liveStatus} onSelect={setSelected} />
          {selectedCurrent && <PointDetails point={selectedCurrent} onClose={() => setSelected(null)} />}
        </GoogleMap>
      </APIProvider>
    </div>
  );
}
