import { LatLng } from "leaflet";
import { MapPinIcon, XIcon } from "lucide-react";
import { useEffect, useState } from "react";
import toast from "react-hot-toast";
import LeafletMap from "@/components/LeafletMap";
import { Button } from "@/components/ui/button";
import { Input } from "@/components/ui/input";
import { Popover, PopoverContent, PopoverTrigger } from "@/components/ui/popover";
import { Tooltip, TooltipContent, TooltipProvider, TooltipTrigger } from "@/components/ui/tooltip";
import { Location } from "@/types/proto/api/v1/memo_service";
import { useTranslate } from "@/utils/i18n";

interface Props {
  location?: Location;
  onChange: (location?: Location) => void;
}

interface State {
  initialized: boolean;
  placeholder: string;
  position?: LatLng;
  latInput: string;
  lngInput: string;
}

const LocationSelector = (props: Props) => {
  const t = useTranslate();
  const [state, setState] = useState<State>({
    initialized: false,
    placeholder: props.location?.placeholder || "",
    position: props.location ? new LatLng(props.location.latitude, props.location.longitude) : undefined,
    latInput: props.location ? String(props.location.latitude) : "",
    lngInput: props.location ? String(props.location.longitude) : "",
  });
  const [popoverOpen, setPopoverOpen] = useState<boolean>(false);

  useEffect(() => {
    setState((state) => ({
      ...state,
      placeholder: props.location?.placeholder || "",
      position: new LatLng(props.location?.latitude || 0, props.location?.longitude || 0),
      latInput: String(props.location?.latitude) || "",
      lngInput: String(props.location?.longitude) || "",
    }));
    // if (props.location) {
    //   setState((state) => ({
    //     ...state,
    //     placeholder: props.location.placeholder || "",
    //     position: new LatLng(props.location.latitude, props.location.longitude),
    //     latInput: String(props.location.latitude),
    //     lngInput: String(props.location.longitude),
    //   }));
    // } else {
    //   setState((state) => ({
    //     ...state,
    //     placeholder: "",
    //     position: undefined,
    //     latInput: "",
    //     lngInput: "",
    //   }));
    // }
  }, [props.location]);

  useEffect(() => {
    if (popoverOpen && !props.location) {
      const handleError = (error: any, errorMessage: string) => {
        setState({ ...state, initialized: true });
        toast.error(errorMessage);
        console.error(error);
      };

      if (navigator.geolocation) {
        navigator.geolocation.getCurrentPosition(
          (position) => {
            const lat = position.coords.latitude;
            const lng = position.coords.longitude;
            setState({ ...state, position: new LatLng(lat, lng), initialized: true });
          },
          (error) => {
            handleError(error, "Failed to get current position");
          },
        );
      } else {
        handleError("Geolocation is not supported by this browser.", "Geolocation is not supported by this browser.");
      }
    }
  }, [popoverOpen]);

  useEffect(() => {
    if (!state.position) {
      setState((prev) => ({ ...prev, placeholder: "" }));
      return;
    }

    // Sync lat/lng input values from position
    const newLat = String(state.position.lat);
    const newLng = String(state.position.lng);
    if (state.latInput !== newLat || state.lngInput !== newLng) {
      setState((prev) => ({ ...prev, latInput: newLat, lngInput: newLng }));
    }

    // Fetch reverse geocoding data.
    fetch(`https://nominatim.openstreetmap.org/reverse?lat=${state.position.lat}&lon=${state.position.lng}&format=json`)
      .then((response) => response.json())
      .then((data) => {
        if (data && data.display_name) {
          setState((prev) => ({ ...prev, placeholder: data.display_name }));
        }
      })
      .catch((error) => {
        toast.error("Failed to fetch reverse geocoding data");
        console.error("Failed to fetch reverse geocoding data:", error);
      });
  }, [state.position]);

  // Update position when lat/lng inputs change (if valid numbers)
  useEffect(() => {
    const lat = parseFloat(state.latInput);
    const lng = parseFloat(state.lngInput);
    if (Number.isFinite(lat) && Number.isFinite(lng)) {
      if (!state.position || state.position.lat !== lat || state.position.lng !== lng) {
        setState((prev) => ({ ...prev, position: new LatLng(lat, lng) }));
      }
    }
  }, [state.latInput, state.lngInput]);

  const onPositionChanged = (position: LatLng) => {
    setState((prev) => ({ ...prev, position }));
  };

  const removeLocation = (e: React.MouseEvent) => {
    e.stopPropagation();
    e.preventDefault();
    props.onChange(undefined);
  };

  return (
    <Popover open={popoverOpen} onOpenChange={setPopoverOpen}>
      <TooltipProvider>
        <Tooltip>
          <TooltipTrigger asChild>
            <PopoverTrigger asChild>
              <Button variant="ghost" size={props.location ? undefined : "icon"}>
                <MapPinIcon className="size-5 shrink-0" />
                {props.location && (
                  <>
                    <span className="ml-0.5 text-sm text-ellipsis whitespace-nowrap overflow-hidden max-w-28">
                      {props.location.placeholder}
                    </span>
                    <span className="ml-1 cursor-pointer hover:text-primary" onClick={removeLocation}>
                      <XIcon className="size-4 shrink-0" />
                    </span>
                  </>
                )}
              </Button>
            </PopoverTrigger>
          </TooltipTrigger>
          {!props.location && (
            <TooltipContent side="bottom">
              <p>{t("tooltip.select-location")}</p>
            </TooltipContent>
          )}
        </Tooltip>
      </TooltipProvider>
      <PopoverContent align="center">
        <div className="min-w-80 sm:w-lg p-1 flex flex-col justify-start items-start">
          <LeafletMap key={JSON.stringify(state.initialized)} latlng={state.position} onChange={onPositionChanged} />
          <div className="mt-2 w-full flex flex-row justify-between items-center gap-2">
            <div className="flex flex-row items-center justify-start gap-2 w-full">
              <div id="lat-long-display" className="flex flex-row items-center gap-2">
                <Input
                  placeholder="Lat"
                  type="number"
                  step="any"
                  value={state.latInput}
                  onChange={(e) => setState((prev) => ({ ...prev, latInput: e.target.value }))}
                  className="w-28"
                />
                <Input
                  placeholder="Lng"
                  type="number"
                  step="any"
                  value={state.lngInput}
                  onChange={(e) => setState((prev) => ({ ...prev, lngInput: e.target.value }))}
                  className="w-28"
                />
                <div className="relative flex-1">
                  <Input
                    placeholder="Choose a position first."
                    value={state.placeholder}
                    disabled={!state.position}
                    onChange={(e) => setState((prev) => ({ ...prev, placeholder: e.target.value }))}
                  />
                </div>
              </div>
            </div>
            <Button
              className="shrink-0"
              color="primary"
              onClick={() => {
                props.onChange(
                  Location.fromPartial({
                    placeholder: state.placeholder,
                    latitude: state.position?.lat,
                    longitude: state.position?.lng,
                  }),
                );
                setPopoverOpen(false);
              }}
              disabled={!state.position || state.placeholder.length === 0}
            >
              {t("common.confirm")}
            </Button>
          </div>
        </div>
      </PopoverContent>
    </Popover>
  );
};

export default LocationSelector;