from pathlib import Path

from isal import igzip_threaded
from xsdata.formats.dataclass.serializers import XmlSerializer
from xsdata.formats.dataclass.serializers.config import SerializerConfig
from xsdata.formats.dataclass.serializers.writers import XmlEventWriter
import zipfile
import io
from domain.netex.model import PublicationDelivery


def export_publication_delivery_xml(
    publication_delivery: PublicationDelivery, output_filename: Path
) -> None:
    serializer_config = SerializerConfig(
        ignore_default_attributes=True, xml_declaration=True, pretty_print=True
    )
    serializer_config.ignore_default_attributes = True
    serializer = XmlSerializer(config=serializer_config, writer=XmlEventWriter)

    ns_map = {
        "": "http://www.netex.org.uk/netex",
        "gml": "http://www.opengis.net/gml/3.2",
    }

    if output_filename.name.endswith(".gz"):
        with igzip_threaded.open(  # type: ignore
            output_filename.resolve(),
            "wt",
            compresslevel=3,
            threads=3,
            block_size=2 * 10**8,
            encoding="utf-8",
        ) as out:
            serializer.write(out, publication_delivery, ns_map)
    elif output_filename.name.endswith(".zip"):
        with zipfile.ZipFile(output_filename.resolve(), "w", zipfile.ZIP_DEFLATED) as zf:
            with zf.open("publication_delivery.xml", "w") as out:
                with io.TextIOWrapper(out, "utf-8") as g:
                    serializer.write(g, publication_delivery, ns_map)
    else:
        with output_filename.open("w", encoding="utf-8") as out:
            serializer.write(out, publication_delivery, ns_map)