leptos-leaflet 0.10.2

use leaflet::Map;
use leptos::{html::Div, prelude::*};
use wasm_bindgen::prelude::*;
use web_sys::HtmlDivElement;

use leaflet::LocateOptions;

use crate::core::JsWriteSignal;

use super::{provide_leaflet_context, MapEvents, PopupEvents, Position, TooltipEvents};

/// A container for the Leaflet map.
///
/// This is the main container for the Leaflet map. It provides a way to add child nodes to the map.
/// It also provides a signal to access the map instance, allowing to interact with the map from other components.
#[component]
pub fn MapContainer(
    #[prop(into, optional)] class: Signal<String>,
    #[prop(into, optional)] style: Signal<String>,
    /// Centers the map on the given location
    #[prop(into, optional)]
    center: Option<Position>,
    /// Zoom level of the map. Defaults to 10.0
    #[prop(optional, default = 10.0)]
    zoom: f64,
    /// Wether zoom controls should be added to the map.
    #[prop(optional, default = true)]
    zoom_control: bool,
    /// Wether mouse wheel zoom controls is enabled or disabled.
    #[prop(optional, default = true)]
    scroll_wheel_zoom: bool,
    /// Zoom snap of the map. Defaults to 1.0
    #[prop(optional, default = 1.0)]
    zoom_snap: f64,
    /// Zoom delta of the map. Defaults to 1.0
    #[prop(optional, default = 1.0)]
    zoom_delta: f64,
    /// Allow zoom on double_click
    #[prop(optional, default = true)]
    double_click_zoom: bool,
    /// Sets the minimum zoom level
    #[prop(optional, default = 0.0)]
    min_zoom: f64,
    /// Use geolocation from the browser to track the user
    #[prop(optional)]
    locate: bool,
    /// Tracks position of the user on the map
    #[prop(optional)]
    watch: bool,
    /// Enables high-accuracy tracking
    #[prop(optional)]
    enable_high_accuracy: bool,
    /// Sets the view of the map if geolocation is available
    #[prop(optional)]
    set_view: bool,
    #[prop(optional)] map: Option<JsWriteSignal<Option<Map>>>,
    #[prop(optional)] events: MapEvents,
    #[prop(optional)] popup_events: PopupEvents,
    #[prop(optional)] tooltip_events: TooltipEvents,
    /// An optional node ref for the map `div` container element.
    #[prop(optional)]
    node_ref: Option<NodeRef<Div>>,
    /// Wether the map should prefer canvas renderer.
    #[prop(optional)]
    prefer_canvas: bool,
    /// Inner map child nodes
    #[prop(optional)]
    children: Option<Children>,
) -> impl IntoView {
    let map_ref = node_ref.unwrap_or_default();
    let map_context = provide_leaflet_context();

    let map_load = map_ref;
    Effect::new(move |_| {
        if let Some(map_div) = map_load.get() {
            let html_node = map_div.unchecked_ref::<HtmlDivElement>();
            // Randomize the id of the map
            if html_node.id().is_empty() {
                let id = format!("map-{}", (js_sys::Math::random() * 1000.0) as u32);
                map_div.set_id(&id);
            }
            let events = events.clone();
            let popup_events = popup_events.clone();
            let tooltip_events = tooltip_events.clone();

            let options = leaflet::MapOptions::new();
            options.set_prefer_canvas(prefer_canvas);
            options.set_zoom_control(zoom_control);
            options.set_scroll_wheel_zoom(scroll_wheel_zoom);
            options.set_zoom(zoom);
            options.set_zoom_snap(zoom_snap);
            options.set_zoom_delta(zoom_delta);
            options.set_double_click_zoom(JsValue::from_bool(double_click_zoom));
            options.set_min_zoom(min_zoom);
            if let Some(center) = center {
                options.set_center(center.as_lat_lng());
            }
            let Ok(leaflet_map) = Map::new(&html_node.id(), &options) else {
                tracing::error!("Failed to create map");
                return;
            };

            // Setup events
            events.setup(&leaflet_map);
            popup_events.setup(&leaflet_map);
            tooltip_events.setup(&leaflet_map);

            if locate {
                let mut locate_options = LocateOptions::new();
                locate_options.enable_high_accuracy(enable_high_accuracy);
                locate_options.set_view(set_view);
                locate_options.watch(watch);
                leaflet_map.locate_with_options(&locate_options);
            }

            map_context.set_map(&leaflet_map);
            if let Some(map) = map {
                map.set(Some(leaflet_map));
            }
        };
    });

    on_cleanup(move || {
        if let Some(map) = map_context.map_untracked().as_ref() {
            map.remove();
        };
    });

    view! { <div class=move || class.get() node_ref=map_ref style=move || style.get()>{children.map(|child|child())}</div>}
}

#[derive(Debug, Default, Clone)]
pub struct LeafletMap {
    #[cfg(not(feature = "ssr"))]
    pub map: Option<leaflet::Map>,
}

impl LeafletMap {
    pub fn new() -> Self {
        Self {
            #[cfg(not(feature = "ssr"))]
            map: None,
        }
    }
}