import os
import shutil
from pathlib import Path, PurePath
import json
from lxml import html

# from kml2geojson import build_layers, build_feature_collection, disambiguate, to_filename, STYLE_TYPES
from zipfile import ZipFile

import xml.dom.minidom as md
import xml.dom.minicompat as mc
import re
import pathlib as pl
from typing import Optional, TextIO, BinaryIO

#: Atomic KML geometry types supported.
#: MultiGeometry is handled separately.
GEOTYPES = [
    "Polygon",
    "LineString",
    "Point",
    "Track",
    "gx:Track",
]

#: Supported style types
STYLE_TYPES = [
    "svg",
    "leaflet",
]

SPACE = re.compile(r"\s+")


def get(node: md.Document, name: str) -> mc.NodeList:
    """
    Given a KML Document Object Model (DOM) node, return a list of its sub-nodes that have the given tag name.
    """
    return node.getElementsByTagName(name)


def get1(node: md.Document, name: str) -> md.Element | None:
    """
    Return the first element of ``get(node, name)``, if it exists.
    Otherwise return ``None``.
    """
    s = get(node, name)
    if s:
        return s[0]
    else:
        return None


def attr(node: md.Document, name: str) -> str:
    """
    Return as a string the value of the given DOM node's attribute named by ``name``, if it exists.
    Otherwise, return an empty string.
    """
    return node.getAttribute(name)


def val(node: md.Document) -> str:
    """
    Normalize the given DOM node and return the value of its first child (the string content of the node) stripped of leading and trailing whitespace.
    """
    try:
        node.normalize()
        return node.firstChild.wholeText.strip()  # Handles CDATASection too
    except AttributeError:
        return ""


def valf(node: md.Document) -> float:
    """
    Cast ``val(node)`` as a float.
    Return ``None`` if that does not work.
    """
    try:
        return float(val(node))
    except ValueError:
        return None


def numarray(a: list) -> list[float]:
    """
    Cast the given list into a list of floats.
    """
    return [float(aa) for aa in a]


def coords1(s: str) -> list[float]:
    """
    Convert the given KML string containing one coordinate tuple into a list of floats.

    EXAMPLE::

        >>> coords1(' -112.2,36.0,2357 ')
        [-112.2, 36.0, 2357.0]

    """
    return numarray(re.sub(SPACE, "", s).split(","))


def coords(s: str) -> list[list[float]]:
    """
    Convert the given KML string containing multiple coordinate tuples into a list of lists of floats.

    EXAMPLE::

        >>> coords('''
        ... -112.0,36.1,0
        ... -113.0,36.0,0
        ... ''')
        [[-112.0, 36.1, 0.0], [-113.0, 36.0, 0.0]]

    """
    s = s.split()  # sub(TRIM_SPACE, '', v).split()
    return [coords1(ss) for ss in s]


def gx_coords1(s: str) -> list[float]:
    """
    Convert the given KML string containing one gx coordinate tuple into a list of floats.

    EXAMPLE::

        >>> gx_coords1('-113.0 36.0 0')
        [-113.0, 36.0, 0.0]

    """
    return numarray(s.split(" "))


def gx_coords(node: md.Document) -> dict:
    """
    Given a KML DOM node, grab its <gx:coord> and <gx:timestamp><when>subnodes, and convert them into a dictionary with the keys and values

    - ``'coordinates'``: list of lists of float coordinates
    - ``'times'``: list of timestamps corresponding to the coordinates

    """
    els = get(node, "gx:coord")
    coordinates = []
    times = []
    coordinates = [gx_coords1(val(el)) for el in els]
    time_els = get(node, "when")
    times = [val(t) for t in time_els]
    return {
        "coordinates": coordinates,
        "times": times,
    }


def disambiguate(names: list[str], mark: str = "1") -> list[str]:
    """
    Given a list of strings ``names``, return a new list of names where repeated names have been disambiguated by repeatedly appending the given mark.

    EXAMPLE::

        >>> disambiguate(['sing', 'song', 'sing', 'sing'])
        ['sing', 'song', 'sing1', 'sing11']

    """
    names_seen = set()
    new_names = []
    for name in names:
        new_name = name
        while new_name in names_seen:
            new_name += mark
        new_names.append(new_name)
        names_seen.add(new_name)

    return new_names


def to_filename(s: str) -> str:
    """
    Based on `django/utils/text.py <https://github.com/django/django/blob/master/django/utils/text.py>`_.
    Return the given string converted to a string that can be used for a clean filename.
    Specifically, leading and trailing spaces are removed; other spaces are converted to underscores, and anything that is not a unicode alphanumeric, dash, underscore, or dot, is removed.

    EXAMPLE::

        >>> to_filename("%  A dbla'{-+)(ç? ")
        'A_dsbla-ç'

    """
    s = re.sub(r"(?u)[^-\w. ]", "", s)
    s = s.strip().replace(" ", "_")
    return s


# ---------------
# Main functions
# ---------------
def build_rgb_and_opacity(s: str) -> tuple:
    """
    Given a KML color string, return an equivalent RGB hex color string and an opacity float rounded to 2 decimal places.

    EXAMPLE::

        >>> build_rgb_and_opacity('ee001122')
        ('#221100', 0.93)

    """
    # Set defaults
    color = "000000"
    opacity = 1

    if s.startswith("#"):
        s = s[1:]
    if len(s) == 8:
        color = s[6:8] + s[4:6] + s[2:4]
        opacity = round(int(s[0:2], 16) / 256, 2)
    elif len(s) == 6:
        color = s[4:6] + s[2:4] + s[0:2]
    elif len(s) == 3:
        color = s[::-1]

    return "#" + color, opacity


def build_svg_style(node: md.Document) -> dict:
    """
    Given a DOM node, grab its top-level Style nodes, convert every one into a SVG style dictionary, put them in a master dictionary of the form

        #style ID -> SVG style dictionary,

    and return the result.

    The possible keys and values of each SVG style dictionary, the style options, are

    - ``iconUrl``: URL of icon
    - ``stroke``: stroke color; RGB hex string
    - ``stroke-opacity``: stroke opacity
    - ``stroke-width``:  stroke width in pixels
    - ``fill``: fill color; RGB hex string
    - ``fill-opacity``: fill opacity
    """
    d = {}
    for item in get(node, "Style"):
        style_id = "#" + attr(item, "id")
        # Create style properties
        props = {}
        for x in get(item, "PolyStyle"):
            color = val(get1(x, "color"))
            if color:
                rgb, opacity = build_rgb_and_opacity(color)
                props["fill"] = rgb
                props["fill-opacity"] = opacity
                # Set default border style
                props["stroke"] = rgb
                props["stroke-opacity"] = opacity
                props["stroke-width"] = 1
            fill = valf(get1(x, "fill"))
            if fill == 0:
                props["fill-opacity"] = fill
            elif fill == 1 and "fill-opacity" not in props:
                props["fill-opacity"] = fill
            outline = valf(get1(x, "outline"))
            if outline == 0:
                props["stroke-opacity"] = outline
            elif outline == 1 and "stroke-opacity" not in props:
                props["stroke-opacity"] = outline
        for x in get(item, "LineStyle"):
            color = val(get1(x, "color"))
            if color:
                rgb, opacity = build_rgb_and_opacity(color)
                props["stroke"] = rgb
                props["stroke-opacity"] = opacity
            width = valf(get1(x, "width"))
            if width is not None:
                props["stroke-width"] = width
        for x in get(item, "IconStyle"):
            icon = get1(x, "Icon")
            if not icon:
                continue
            # Clear previous style properties
            props = {}
            props["iconUrl"] = val(get1(icon, "href"))

        d[style_id] = props

    return d


def build_leaflet_style(node: md.Document) -> dict:
    """
    Given a DOM node, grab its top-level Style nodes, convert every one into a Leaflet style dictionary, put them in a master dictionary of the form

        #style ID -> Leaflet style dictionary,

    and return the result.

    The the possible keys and values of each Leaflet style dictionary, the style options, are

    - ``iconUrl``: URL of icon
    - ``color``: stroke color; RGB hex string
    - ``opacity``: stroke opacity
    - ``weight``:  stroke width in pixels
    - ``fillColor``: fill color; RGB hex string
    - ``fillOpacity``: fill opacity
    """
    d = {}
    for item in get(node, "Style"):
        style_id = "#" + attr(item, "id")
        # Create style properties
        props = {}
        for x in get(item, "PolyStyle"):
            color = val(get1(x, "color"))
            if color:
                rgb, opacity = build_rgb_and_opacity(color)
                props["fillColor"] = rgb
                props["fillOpacity"] = opacity
                # Set default border style
                props["color"] = rgb
                props["opacity"] = opacity
                props["weight"] = 1
            fill = valf(get1(x, "fill"))
            if fill == 0:
                props["fillOpacity"] = fill
            elif fill == 1 and "fillOpacity" not in props:
                props["fillOpacity"] = fill
            outline = valf(get1(x, "outline"))
            if outline == 0:
                props["opacity"] = outline
            elif outline == 1 and "opacity" not in props:
                props["opacity"] = outline
        for x in get(item, "LineStyle"):
            color = val(get1(x, "color"))
            if color:
                rgb, opacity = build_rgb_and_opacity(color)
                props["color"] = rgb
                props["opacity"] = opacity
            width = valf(get1(x, "width"))
            if width is not None:
                props["weight"] = width
        for x in get(item, "IconStyle"):
            icon = get1(x, "Icon")
            if not icon:
                continue
            # Clear previous style properties
            props = {}
            props["iconUrl"] = val(get1(icon, "href"))

        d[style_id] = props

    return d


def build_geometry(node: md.Document) -> dict:
    """
    Return a (decoded) GeoJSON geometry dictionary corresponding to the given KML node.
    """
    geoms = []
    times = []
    if get1(node, "MultiGeometry"):
        return build_geometry(get1(node, "MultiGeometry"))
    if get1(node, "MultiTrack"):
        return build_geometry(get1(node, "MultiTrack"))
    if get1(node, "gx:MultiTrack"):
        return build_geometry(get1(node, "gx:MultiTrack"))
    for geotype in GEOTYPES:
        geonodes = get(node, geotype)
        if not geonodes:
            continue
        for geonode in geonodes:
            if geotype == "Point":
                geoms.append(
                    {
                        "type": "Point",
                        "coordinates": coords1(val(get1(geonode, "coordinates"))),
                    }
                )
            elif geotype == "LineString":
                geoms.append(
                    {
                        "type": "LineString",
                        "coordinates": coords(val(get1(geonode, "coordinates"))),
                    }
                )
            elif geotype == "Polygon":
                rings = get(geonode, "LinearRing")
                coordinates = [coords(val(get1(ring, "coordinates"))) for ring in rings]
                geoms.append(
                    {
                        "type": "Polygon",
                        "coordinates": coordinates,
                    }
                )
            elif geotype in ["Track", "gx:Track"]:
                track = gx_coords(geonode)
                geoms.append(
                    {
                        "type": "LineString",
                        "coordinates": track["coordinates"],
                    }
                )
                if track["times"]:
                    times.append(track["times"])

    return {"geoms": geoms, "times": times}


def build_feature(node: md.Document) -> dict | None:
    """
    Build and return a (decoded) GeoJSON Feature corresponding to this KML node (typically a KML Placemark).
    Return ``None`` if no Feature can be built.
    """
    geoms_and_times = build_geometry(node)
    if not geoms_and_times["geoms"]:
        return None

    props = {}
    for x in get(node, "name")[:1]:
        name = val(x)
        if name:
            props["name"] = val(x)
    for x in get(node, "description")[:1]:
        desc = val(x)
        if desc:
            props["description"] = desc
    for x in get(node, "styleUrl")[:1]:
        style_url = val(x)
        if style_url[0] != "#":
            style_url = "#" + style_url
        props["styleUrl"] = style_url
    for x in get(node, "PolyStyle")[:1]:
        color = val(get1(x, "color"))
        if color:
            rgb, opacity = build_rgb_and_opacity(color)
            props["fill"] = rgb
            props["fill-opacity"] = opacity
            # Set default border style
            props["stroke"] = rgb
            props["stroke-opacity"] = opacity
            props["stroke-width"] = 1
        fill = valf(get1(x, "fill"))
        if fill == 0:
            props["fill-opacity"] = fill
        elif fill == 1 and "fill-opacity" not in props:
            props["fill-opacity"] = fill
        outline = valf(get1(x, "outline"))
        if outline == 0:
            props["stroke-opacity"] = outline
        elif outline == 1 and "stroke-opacity" not in props:
            props["stroke-opacity"] = outline
    for x in get(node, "LineStyle")[:1]:
        color = val(get1(x, "color"))
        if color:
            rgb, opacity = build_rgb_and_opacity(color)
            props["stroke"] = rgb
            props["stroke-opacity"] = opacity
        width = valf(get1(x, "width"))
        if width:
            props["stroke-width"] = width
    for x in get(node, "ExtendedData")[:1]:
        datas = get(x, "Data")
        for data in datas:
            props[attr(data, "name")] = val(get1(data, "value"))
        simple_datas = get(x, "SimpleData")
        for simple_data in simple_datas:
            props[attr(simple_data, "name")] = val(simple_data)
    for x in get(node, "TimeSpan")[:1]:
        begin = val(get1(x, "begin"))
        end = val(get1(x, "end"))
        props["timeSpan"] = {"begin": begin, "end": end}
    if geoms_and_times["times"]:
        times = geoms_and_times["times"]
        if len(times) == 1:
            props["times"] = times[0]
        else:
            props["times"] = times

    feature = {
        "type": "Feature",
        "properties": props,
    }

    geoms = geoms_and_times["geoms"]
    if len(geoms) == 1:
        feature["geometry"] = geoms[0]
    else:
        feature["geometry"] = {
            "type": "GeometryCollection",
            "geometries": geoms,
        }

    if attr(node, "id"):
        feature["id"] = attr(node, "id")

    return feature


def build_feature_collection(node: md.Document, name: Optional[str] = None) -> dict:
    """
    Build and return a (decoded) GeoJSON FeatureCollection corresponding to this KML DOM node (typically a KML Folder).
    If a name is given, store it in the FeatureCollection's ``'name'`` attribute.
    """
    # Initialize
    geojson = {
        "type": "FeatureCollection",
        "features": [],
    }

    # Build features
    for placemark in get(node, "Placemark"):
        feature = build_feature(placemark)
        if feature is not None:
            geojson["features"].append(feature)

    # Give the collection a name if requested
    if name is not None:
        geojson["name"] = name

    return geojson


def build_layers(node: md.Document, *, disambiguate_names: bool = True) -> list[dict]:
    """
    Return a list of GeoJSON FeatureCollections, one for each folder in the given KML DOM node that contains geodata.
    Name each FeatureCollection (via a ``'name'`` attribute) according to its corresponding KML folder name.

    If ``disambiguate_names == True``, then disambiguate repeated layer names via :func:`disambiguate`.

    Warning: this can produce layers with the same geodata in case the KML node has nested folders with geodata.
    """
    layers = []
    names = []
    for i, folder in enumerate(get(node, "Folder")):
        name = val(get1(folder, "name"))
        geojson = build_feature_collection(folder, name)
        if geojson["features"]:
            layers.append(geojson)
            names.append(name)

    if not layers:
        # No folders, so use the root node
        name = val(get1(node, "name"))
        geojson = build_feature_collection(node, name)
        if geojson["features"]:
            layers.append(geojson)
            names.append(name)

    if disambiguate_names:
        new_names = disambiguate(names)
        new_layers = []
        for i, layer in enumerate(layers):
            layer["name"] = new_names[i]
            new_layers.append(layer)
        layers = new_layers

    return layers


def convert(
        kml_path_or_buffer: str | pl.Path | TextIO | BinaryIO,
        feature_collection_name: Optional[str] = None,
        style_type: Optional[str] = None,
        *,
        separate_folders: bool = False,
):
    """
    Given a path to a KML file or given a KML file object,
    convert it to a single GeoJSON FeatureCollection dictionary named
    ``feature_collection_name``.
    Close the KML file afterwards.

    If ``separate_folders``, then return several FeatureCollections,
    one for each folder in the KML file that contains geodata or that has a descendant
    node that contains geodata.
    Warning: this can produce FeatureCollections with the same geodata in case the KML
    file has nested folders with geodata.

    If a style type from :const:`STYLE_TYPES` is given, then also create a JSON
    dictionary that encodes into the style type the style information contained in the
    KML file.

    Return a tuple (style dict, FeatureCollection 1, ..., FeatureCollection n),
    where the style dict is present if and only if ``style_type`` is given and
    where n > 1 if and only if ``separate_folders`` and the KML file contains more than
    one folder of geodata.
    """
    # Read KML
    if isinstance(kml_path_or_buffer, (str, pl.Path)):
        kml_path_or_buffer = pl.Path(kml_path_or_buffer).resolve()
        with kml_path_or_buffer.open(encoding="utf-8", errors="ignore") as src:
            kml_str = src.read()
    else:
        kml_str = kml_path_or_buffer.read()
        kml_path_or_buffer.close()

    # Parse KML
    root = md.parseString(kml_str)

    # Build GeoJSON layers
    if separate_folders:
        result = build_layers(root)
    else:
        result = [build_feature_collection(root, name=feature_collection_name)]

    if style_type is not None:
        # Build style dictionary
        if style_type not in STYLE_TYPES:
            raise ValueError(f"style type must be one of {STYLE_TYPES}")
        else:
            builder_name = f"build_{style_type}_style"
            style_dict = globals()[builder_name](root)
            result = style_dict, *result

    return result


def kmz_convert(kmz_path, output_dir, separate_folders=False,
                style_type=None, style_filename='style.json'):
    """
    Given a path to a KML file, convert it to one or several GeoJSON FeatureCollection files and save the result(s) to the given output directory.
    If not ``separate_folders`` (the default), then create one GeoJSON file.
    Otherwise, create several GeoJSON files, one for each folder in the KML file that contains geodata or that has a descendant node that contains geodata.
    Warning: this can produce GeoJSON files with the same geodata in case the KML file has nested folders with geodata.
    If a ``style_type`` is given, then also build a JSON style file of the given style type and save it to the output directory under the name given by ``style_filename``.
    """

    # Create absolute paths
    kmz_path = Path(kmz_path).resolve()
    output_dir = Path(output_dir)
    if not output_dir.exists():
        output_dir.mkdir()
    output_dir = output_dir.resolve()

    # opening the zip file in READ mode
    with ZipFile(kmz_path, 'r') as zip:
        names = zip.namelist()
        # Find the KML file in the archive
        # There should be only one KML per KNZ
        for name in names:
            if '.kml' in name:
                kml_file = name

        kml_str = zip.read(kml_file)
    # Parse KML
    root = md.parseString(kml_str)

    # Build GeoJSON layers
    if separate_folders:
        layers = build_layers(root)
    else:
        layers = [build_feature_collection(root, name=kmz_path.stem)]

    # Handle HTML Description Tables
    for layer in layers:
        for feature in layer['features']:
            if feature['properties'].get('description'):
                if "<table>" in feature['properties']['description']:
                    tree = html.fromstring(feature['properties']['description'])

                    feature['properties']['date'] = tree.xpath('//table/tr[3]/td/text()')[0].strip()
                    feature['properties']['location'] = tree.xpath('//table/tr[5]/td/b/text()')[0].strip()
                    feature['properties']['pressure'] = float(
                        tree.xpath('//table/tr[7]/td/text()')[0].strip().split(" ")[0])
                    feature['properties']['speed'] = float(
                        tree.xpath('//table/tr[9]/td/text()')[0].strip().split(";")[2].strip().replace(" kph", ""))

                    del feature['properties']['name']
                    del feature['properties']['styleUrl']
                    del feature['properties']['description']

    return layers

    # Create filenames for layers
    '''filenames = disambiguate([to_filename(layer['name']) for layer in layers])
    filenames = [name + '.geojson' for name in filenames]

    # Write layers to files
    for i in range(len(layers)):
        path = output_dir/filenames[i]
        with path.open('w') as tgt:
            json.dump(layers[i], tgt, indent = 2)

    # Build and export style file if desired
    if style_type is not None:
        if style_type not in STYLE_TYPES:
            raise ValueError('style type must be one of {!s}'.format(
              STYLE_TYPES))
        builder_name = 'build_{!s}_style'.format(style_type)
        style_dict = globals()[builder_name](root)
        path = output_dir/style_filename
        with path.open('w') as tgt:
            json.dump(style_dict, tgt, indent=2)'''