catalog.py

# -*- coding: utf-8 -*-
#########################################################################
#
# Copyright 2019, GeoSolutions Sas.
# All rights reserved.
#
# This source code is licensed under the MIT license found in the
# LICENSE.txt file in the root directory of this source tree.
#
#########################################################################

import logging
from datetime import datetime, timedelta
from geoserver.layer import Layer
from geoserver.resource import FeatureType
from geoserver.service import service_from_index, ServiceWmsSettings
from geoserver.store import (
    coveragestore_from_index,
    datastore_from_index,
    wmsstore_from_index,
    UnsavedDataStore,
    UnsavedCoverageStore,
    UnsavedWmsStore,
)
from geoserver.style import Style
from geoserver.support import prepare_upload_bundle, build_url
from geoserver.layergroup import LayerGroup, UnsavedLayerGroup
from geoserver.workspace import workspace_from_index, Workspace
from geoserver.security import user_from_index
from geoserver.settings import GlobalSettings
import os
import re
import base64
from requests.exceptions import HTTPError
from xml.etree.ElementTree import XML
from xml.parsers.expat import ExpatError
import requests
from urllib3 import Retry
from requests.adapters import HTTPAdapter
from six import string_types

try:
    from past.builtins import basestring
except ImportError:
    pass

try:
    from urllib.parse import urlparse, urlencode, parse_qsl
except ImportError:
    from urlparse import urlparse, parse_qsl
    from urllib import urlencode

try:
    from json.decoder import JSONDecodeError
except ImportError:
    JSONDecodeError = ValueError


logger = logging.getLogger("gsconfig.catalog")


class UploadError(Exception):
    pass


class ConflictingDataError(Exception):
    pass


class AmbiguousRequestError(Exception):
    pass


class FailedRequestError(Exception):
    pass


def _name(named):
    """Get the name out of an object.  This varies based on the type of the input:
    * the "name" of a string is itself
    * the "name" of None is itself
    * the "name" of an object with a property named name is that property -
      as long as it's a string
    * otherwise, we raise a ValueError
    """
    if isinstance(named, string_types) or named is None:
        return named
    elif hasattr(named, "name") and isinstance(named.name, string_types):
        return named.name
    else:
        raise ValueError(f"Can't interpret {named} as a name or a configuration object")


class Catalog(object):
    """
    The GeoServer catalog represents all of the information in the GeoServer
    configuration.    This includes:
    - Stores of geospatial data
    - Resources, or individual coherent datasets within stores
    - Styles for resources
    - Layers, which combine styles with resources to create a visible map layer
    - LayerGroups, which alias one or more layers for convenience
    - Workspaces, which provide logical grouping of Stores
    - Maps, which provide a set of OWS services with a subset of the server's
        Layers
    - Namespaces, which provide unique identifiers for resources
    """

    def __init__(
        self,
        service_url,
        username="admin",
        password="geoserver",
        validate_ssl_certificate=True,
        access_token=None,
        retries=3,
        backoff_factor=0.9,
    ):
        self.service_url = service_url.strip("/")
        self.username = username
        self.password = password
        self.validate_ssl_certificate = validate_ssl_certificate
        self.access_token = access_token
        self.retries = retries
        self.backoff_factor = backoff_factor
        self.setup_connection(retries=self.retries, backoff_factor=self.backoff_factor)
        self._cache = {}
        self._version = None

    def __getstate__(self):
        """http connection cannot be pickled"""
        state = dict(vars(self))
        state.pop("http", None)
        state["http"] = None
        return state

    def __setstate__(self, state):
        """restore http connection upon unpickling"""
        self.__dict__.update(state)
        self.setup_connection(retries=self.retries, backoff_factor=self.backoff_factor)

    def setup_connection(self, retries=3, backoff_factor=0.9):
        self.client = requests.session()
        self.client.verify = self.validate_ssl_certificate
        parsed_url = urlparse(self.service_url)
        retry = Retry(
            total=retries or self.retries,
            status=retries or self.retries,
            read=retries or self.retries,
            connect=retries or self.retries,
            backoff_factor=backoff_factor or self.backoff_factor,
            status_forcelist=[502, 503, 504],
            allowed_methods=frozenset(
                ["HEAD", "TRACE", "GET", "PUT", "POST", "OPTIONS", "DELETE"]
            ),
        )
        self.client.mount(f"{parsed_url.scheme}://", HTTPAdapter(max_retries=retry))

    def http_request(self, url, data=None, method="get", headers={}, files=None):
        req_method = getattr(self.client, method.lower())

        if self.access_token:
            headers["Authorization"] = f"Bearer {self.access_token}"
            parsed_url = urlparse(url)
            params = parse_qsl(parsed_url.query.strip())
            params.append(("access_token", self.access_token))
            params = urlencode(params)
            url = f"{parsed_url.scheme}://{parsed_url.netloc}{parsed_url.path}?{params}"
        elif self.username and self.password:
            valid_uname_pw = base64.b64encode(
                f"{self.username}:{self.password}".encode("utf-8")
            ).decode("ascii")
            headers["Authorization"] = f"Basic {valid_uname_pw}"

        return req_method(url, headers=headers, data=data, files=files)

    def get_version(self):
        """obtain the version or just 2.2.x if < 2.3.x
        Raises:
            FailedRequestError: If the request fails.
        """
        if self._version:
            return self._version
        url = f"{self.service_url}/about/version.xml"
        resp = self.http_request(url)
        version = None
        if resp.status_code == 200:
            content = resp.content
            if isinstance(content, bytes):
                content = content.decode("UTF-8")
            dom = XML(content)
            resources = dom.findall("resource")
            for resource in resources:
                if resource.attrib["name"] == "GeoServer":
                    try:
                        version = resource.find("Version").text
                        break
                    except AttributeError:
                        pass

        # This will raise an exception if the catalog is not available
        # If the catalog is available but could not return version information,
        # it is an old version that does not support that
        if version is None:
            # just to inform that version < 2.3.x
            version = "2.2.x"
        self._version = version
        return version

    def get_short_version(self):
        """obtain the shory geoserver version"""
        gs_version = self.get_version()
        match = re.compile(r"[^\d.]+")
        return match.sub("", gs_version).strip(".")

    def delete(self, config_object, purge=None, recurse=False):
        """
        send a delete request
        XXX [more here]
        """
        href = urlparse(config_object.href)
        netloc = urlparse(self.service_url).netloc
        rest_url = href._replace(netloc=netloc).geturl()
        # rest_url = config_object.href
        params = []

        # purge deletes the SLD from disk when a style is deleted
        if purge:
            params.append(f"purge={str(purge)}")

        # recurse deletes the resource when a layer is deleted.
        if recurse:
            params.append("recurse=true")

        if params:
            rest_url = f"{rest_url}?{'&'.join(params)}"

        headers = {"Content-type": "application/xml", "Accept": "application/xml"}
        resp = self.http_request(rest_url, method="delete", headers=headers)
        if resp.status_code != 200:
            raise FailedRequestError(
                f"Failed to make DELETE request: {resp.status_code}, {resp.text}"
            )

        self._cache.clear()

        # do we really need to return anything other than None?
        return resp

    def get_xml(self, rest_url):
        cached_response = self._cache.get(rest_url)

        def is_valid(cached_response):
            return cached_response is not None and datetime.now() - cached_response[
                0
            ] < timedelta(seconds=5)

        def parse_or_raise(xml):
            try:
                if not isinstance(xml, string_types):
                    xml = xml.decode()
                return XML(xml)
            except (ExpatError, SyntaxError) as e:
                msg = "GeoServer gave non-XML response for [GET %s]: %s"
                msg = msg % (rest_url, xml)
                raise Exception(msg, e)

        if is_valid(cached_response):
            raw_text = cached_response[1]
            return parse_or_raise(raw_text)
        else:
            resp = self.http_request(rest_url, headers={"Accept": "application/xml"})
            if resp.status_code == 200:
                content = resp.content
                if isinstance(content, bytes):
                    content = content.decode("UTF-8")
                self._cache[rest_url] = (datetime.now(), content)
                return parse_or_raise(content)
            else:
                raise FailedRequestError(resp.content)

    def reload(self):
        url = f"{self.service_url}/reload"
        resp = self.http_request(url, method="post")
        self._cache.clear()
        return resp

    def reset(self):
        url = f"{self.service_url}/reset"
        resp = self.http_request(url, method="post")
        self._cache.clear()
        return resp

    def save(self, obj, content_type="application/xml"):
        """
        saves an object to the REST service
        gets the object's REST location and the data from the object,
        then POSTS the request.
        """
        href = urlparse(obj.href)
        netloc = urlparse(self.service_url).netloc
        rest_url = href._replace(netloc=netloc).geturl()
        data = obj.message()

        headers = {"Content-type": content_type, "Accept": content_type}

        logger.debug(f"{obj.save_method} {obj.href}")
        resp = self.http_request(
            rest_url, method=obj.save_method.lower(), data=data, headers=headers
        )

        if resp.status_code not in (200, 201):
            raise FailedRequestError(
                f"Failed to save to Geoserver catalog: {resp.status_code}, {resp.text}"
            )

        self._cache.clear()
        return resp

    def _return_first_item(self, _list):
        if len(_list) == 0:
            return None
        else:
            return _list[0]

    def get_stores(self, names=None, workspaces=None):
        """
        Returns a list of stores in the catalog. If workspaces is specified will only return stores in those workspaces.
        If names is specified, will only return stores that match.
        names can either be a comma delimited string or an array.
        Will return an empty list if no stores are found.
        """

        if workspaces:
            if isinstance(workspaces, Workspace):
                workspaces = [workspaces]
            elif isinstance(workspaces, list) and [
                w for w in workspaces if isinstance(w, Workspace)
            ]:
                # nothing
                pass
            else:
                workspaces = self.get_workspaces(names=workspaces)
        else:
            workspaces = self.get_workspaces()

        stores = []
        for ws in workspaces:
            ds_list = self.get_xml(ws.datastore_url)
            cs_list = self.get_xml(ws.coveragestore_url)
            wms_list = self.get_xml(ws.wmsstore_url)
            stores.extend(
                [
                    datastore_from_index(self, ws, n)
                    for n in ds_list.findall("dataStore")
                ]
            )
            stores.extend(
                [
                    coveragestore_from_index(self, ws, n)
                    for n in cs_list.findall("coverageStore")
                ]
            )
            stores.extend(
                [wmsstore_from_index(self, ws, n) for n in wms_list.findall("wmsStore")]
            )

        if names is None:
            names = []
        elif isinstance(names, string_types):
            names = [s.strip() for s in names.split(",") if s.strip()]
        elif not isinstance(names, list):
            names = [names]
            if len(names) and not isinstance(names[0], string_types):
                names = [_n.name for _n in names]

        if stores and names:
            return [_s for _s in stores if _s.name in names]

        return stores

    def get_store(self, name, workspace=None):
        """
        Returns a single store object.
        Will return None if no store is found.
        Will raise an error if more than one store with the same name is found.
        """
        stores = self.get_stores(workspaces=[workspace], names=name)
        return self._return_first_item(stores)

    def create_datastore(self, name, workspace=None):
        if isinstance(workspace, string_types):
            workspace = self.get_workspaces(names=workspace)[0]
        elif workspace is None:
            workspace = self.get_default_workspace()
        return UnsavedDataStore(self, name, workspace)

    def create_wmsstore(self, name, workspace=None, user=None, password=None):
        if workspace is None:
            workspace = self.get_default_workspace()
        return UnsavedWmsStore(self, name, workspace, user, password)

    def create_wmslayer(self, workspace, store, name, nativeName=None):
        headers = {"Content-type": "text/xml", "Accept": "application/xml"}
        # if not provided, fallback to name - this is what geoserver will do
        # anyway but nativeName needs to be provided if name is invalid xml
        # as this will cause verification errors since geoserver 2.6.1
        if nativeName is None:
            nativeName = name

        url = store.href.replace(".xml", "/wmslayers")
        data = f"<wmsLayer><name>{name}</name><nativeName>{nativeName}</nativeName></wmsLayer>"
        resp = self.http_request(url, method="post", data=data, headers=headers)

        if resp.status_code not in (200, 201):
            raise FailedRequestError(
                f"Failed to create WMS layer: {resp.status_code}, {resp.text}"
            )

        self._cache.clear()
        return self.get_layer(name)

    def add_data_to_store(
        self, store, name, data, workspace=None, overwrite=False, charset=None
    ):
        if isinstance(store, string_types):
            store = self.get_stores(names=store, workspaces=[workspace])[0]
        if workspace is not None and workspace:
            workspace = _name(workspace)
            assert (
                store.workspace.name == workspace
            ), f"Specified store ({store}) is not in specified workspace ({workspace})!"
        else:
            workspace = store.workspace.name
        store = store.name

        if isinstance(data, dict):
            bundle = prepare_upload_bundle(name, data)
        else:
            bundle = data

        params = dict()
        if overwrite:
            params["update"] = "overwrite"
        if charset is not None and charset:
            params["charset"] = charset
        params["filename"] = f"{name}.zip"
        params["target"] = "shp"
        # params["configure"] = "all"

        headers = {"Content-Type": "application/zip", "Accept": "application/xml"}
        upload_url = build_url(
            self.service_url,
            ["workspaces", workspace, "datastores", store, "file.shp"],
            params,
        )

        try:
            with open(bundle, "rb") as f:
                data = f.read()
                resp = self.http_request(
                    upload_url, method="put", data=data, headers=headers
                )
                if resp.status_code != 201:
                    raise FailedRequestError(
                        f"Failed to add data to store {store} : {resp.status_code}, {resp.text}"
                    )
                self._cache.clear()
        finally:
            pass

    def create_featurestore(
        self, name, data, workspace=None, overwrite=False, charset=None
    ):
        if workspace is None:
            workspace = self.get_default_workspace()
        workspace = _name(workspace)

        if not overwrite:
            stores = self.get_stores(names=name, workspaces=[workspace])
            if len(stores) > 0:
                msg = f"There is already a store named {name} in workspace {workspace}"
                raise ConflictingDataError(msg)

        params = dict()
        if charset is not None and charset:
            params["charset"] = charset
        url = build_url(
            self.service_url,
            ["workspaces", workspace, "datastores", name, "file.shp"],
            params,
        )

        # PUT /workspaces/<ws>/datastores/<ds>/file.shp
        headers = {"Content-type": "application/zip", "Accept": "application/xml"}
        if isinstance(data, dict):
            logger.debug("Data is NOT a zipfile")
            archive = prepare_upload_bundle(name, data)
        else:
            logger.debug("Data is a zipfile")
            archive = data
        file_obj = open(archive, "rb")
        try:
            resp = self.http_request(url, method="put", data=file_obj, headers=headers)
            if resp.status_code != 201:
                raise FailedRequestError(
                    f"Failed to create FeatureStore {name} : {resp.status_code}, {resp.text}"
                )
            self._cache.clear()
        finally:
            file_obj.close()

    def create_imagemosaic(
        self,
        name,
        data,
        configure="first",
        workspace=None,
        overwrite=False,
        charset=None,
        coverageName=None,
    ):
        if workspace is None:
            workspace = self.get_default_workspace()
        workspace = _name(workspace)

        if not overwrite:
            store = self.get_stores(names=name, workspaces=[workspace])
            if store:
                raise ConflictingDataError(f"There is already a store named {name}")

        params = dict()
        if charset is not None and charset:
            params["charset"] = charset
        if configure.lower() not in ("first", "none", "all"):
            raise ValueError("configure most be one of: first, none, all")
        params["configure"] = configure.lower()
        if coverageName:
            params["coverageName"] = coverageName
        store_type = "file.imagemosaic"
        contet_type = "application/zip"

        if hasattr(data, "read"):
            # Adding this check only to pass tests. We should drop support for passing a file object
            upload_data = data
        elif isinstance(data, string_types):
            if os.path.splitext(data)[-1] == ".zip":
                upload_data = open(data, "rb")
            else:
                store_type = "external.imagemosaic"
                contet_type = "text/plain"
                upload_data = data if data.startswith("file:") else f"file:{data}"
        else:
            raise ValueError(f"ImageMosaic Dataset or directory: {data} is incorrect")

        url = build_url(
            self.service_url,
            ["workspaces", workspace, "coveragestores", name, store_type],
            params,
        )

        # PUT /workspaces/<ws>/coveragestores/<name>/file.imagemosaic?configure=none
        headers = {"Content-type": contet_type, "Accept": "application/xml"}

        try:
            resp = self.http_request(
                url, method="put", data=upload_data, headers=headers
            )
            if resp.status_code != 201:
                raise FailedRequestError(
                    f"Failed to create ImageMosaic {url} : {resp.status_code}, {resp.text}"
                )
            self._cache.clear()
        finally:
            if hasattr(upload_data, "close"):
                upload_data.close()

        return self.get_stores(names=name, workspaces=[workspace])[0]

    def create_coveragestore(
        self,
        name,
        workspace=None,
        path=None,
        type="GeoTIFF",
        create_layer=True,
        layer_name=None,
        source_name=None,
        upload_data=False,
        contet_type="image/tiff",
        overwrite=False,
    ):
        """
        Create a coveragestore for locally hosted rasters.
        If create_layer is set to true, will create a coverage/layer.
        layer_name and source_name are only used if create_layer ia enabled. If not specified, the raster name will be used for both.
        """
        if path is None:
            raise Exception("You must provide a full path to the raster")

        if layer_name is not None and ":" in layer_name:
            ws_name, layer_name = layer_name.split(":")

        allowed_types = [
            "ImageMosaic",
            "GeoTIFF",
            "Gtopo30",
            "WorldImage",
            "AIG",
            "ArcGrid",
            "DTED",
            "EHdr",
            "ERDASImg",
            "ENVIHdr",
            "GeoPackage (mosaic)",
            "NITF",
            "RPFTOC",
            "RST",
            "VRT",
        ]

        if type is None:
            raise Exception("Type must be declared")
        elif type not in allowed_types:
            raise Exception(f"Type must be one of {', '.join(allowed_types)}")

        if workspace is None:
            workspace = self.get_default_workspace()
        workspace = _name(workspace)

        if not overwrite:
            stores = self.get_stores(names=name, workspaces=[workspace])
            if len(stores) > 0:
                msg = f"There is already a store named {name} in workspace {workspace}"
                raise ConflictingDataError(msg)

        if upload_data is False:
            cs = UnsavedCoverageStore(self, name, workspace)
            cs.type = type
            cs.url = path if path.startswith("file:") else f"file:{path}"
            self.save(cs)

            if create_layer:
                if layer_name is None:
                    layer_name = os.path.splitext(os.path.basename(path))[0]
                if source_name is None:
                    source_name = os.path.splitext(os.path.basename(path))[0]

                data = f"<coverage><name>{layer_name}</name><nativeName>{source_name}</nativeName></coverage>"
                url = f"{self.service_url}/workspaces/{workspace}/coveragestores/{name}/coverages.xml"
                headers = {"Content-type": "application/xml"}

                resp = self.http_request(url, method="post", data=data, headers=headers)
                if resp.status_code != 201:
                    raise FailedRequestError(
                        "Failed to create coverage/layer {} for : {}, {}".format(
                            layer_name, name, resp.status_code, resp.text
                        )
                    )
                self._cache.clear()
                return self.get_resources(names=layer_name, workspaces=[workspace])[0]
        else:
            data = open(path, "rb")
            params = {"configure": "first", "coverageName": name}
            url = build_url(
                self.service_url,
                [
                    "workspaces",
                    workspace,
                    "coveragestores",
                    name,
                    f"file.{type.lower()}",
                ],
                params,
            )

            headers = {"Content-type": contet_type}
            resp = self.http_request(url, method="put", data=data, headers=headers)

            if hasattr(data, "close"):
                data.close()

            if resp.status_code != 201:
                raise FailedRequestError(
                    "Failed to create coverage/layer {} for : {}, {}".format(
                        layer_name, name, resp.status_code, resp.text
                    )
                )

        return self.get_stores(names=name, workspaces=[workspace])[0]

    def add_granule(self, data, store, workspace=None):
        """Harvest/add a granule into an existing imagemosaic"""
        ext = os.path.splitext(data)[-1]
        if ext == ".zip":
            type = "file.imagemosaic"
            upload_data = open(data, "rb")
            headers = {"Content-type": "application/zip", "Accept": "application/xml"}
        else:
            type = "external.imagemosaic"
            upload_data = data if data.startswith("file:") else f"file:{data}"
            headers = {"Content-type": "text/plain", "Accept": "application/xml"}

        params = dict()
        workspace_name = workspace
        if isinstance(store, string_types):
            store_name = store
        else:
            store_name = store.name
            workspace_name = store.workspace.name

        if workspace_name is None:
            raise ValueError("Must specify workspace")

        url = build_url(
            self.service_url,
            ["workspaces", workspace_name, "coveragestores", store_name, type],
            params,
        )

        try:
            resp = self.http_request(
                url, method="post", data=upload_data, headers=headers
            )
            if resp.status_code != 202:
                raise FailedRequestError(
                    f"Failed to add granule to mosaic {store} : {resp.status_code}, {resp.text}"
                )
            self._cache.clear()
        finally:
            if hasattr(upload_data, "close"):
                upload_data.close()

        # maybe return a list of all granules?
        return None

    def delete_granule(self, coverage, store, granule_id, workspace=None):
        """Deletes a granule of an existing imagemosaic"""
        params = dict()

        workspace_name = workspace
        if isinstance(store, string_types):
            store_name = store
        else:
            store_name = store.name
            workspace_name = store.workspace.name

        if workspace_name is None:
            raise ValueError("Must specify workspace")

        url = build_url(
            self.service_url,
            [
                "workspaces",
                workspace_name,
                "coveragestores",
                store_name,
                "coverages",
                coverage,
                "index/granules",
                granule_id,
                ".json",
            ],
            params,
        )

        # DELETE /workspaces/<ws>/coveragestores/<name>/coverages/<coverage>/index/granules/<granule_id>.json
        headers = {"Content-type": "application/json", "Accept": "application/json"}

        resp = self.http_request(url, method="delete", headers=headers)
        if resp.status_code != 200:
            raise FailedRequestError(
                f"Failed to delete granule from mosaic {store} : {resp.status_code}, {resp.text}"
            )
        self._cache.clear()

        # maybe return a list of all granules?
        return None

    def list_granules(
        self, coverage, store, workspace=None, filter=None, limit=None, offset=None
    ):
        """List granules of an imagemosaic"""
        params = dict()

        if filter is not None and filter:
            params["filter"] = filter
        if limit is not None and limit:
            params["limit"] = limit
        if offset is not None and offset:
            params["offset"] = offset

        workspace_name = workspace
        if isinstance(store, string_types):
            store_name = store
        else:
            store_name = store.name
            workspace_name = store.workspace.name

        if workspace_name is None:
            raise ValueError("Must specify workspace")

        url = build_url(
            self.service_url,
            [
                "workspaces",
                workspace_name,
                "coveragestores",
                store_name,
                "coverages",
                coverage,
                "index/granules.json",
            ],
            params,
        )

        # GET /workspaces/<ws>/coveragestores/<name>/coverages/<coverage>/index/granules.json
        headers = {"Content-type": "application/json", "Accept": "application/json"}

        resp = self.http_request(url, headers=headers)
        if resp.status_code != 200:
            raise FailedRequestError(
                f"Failed to list granules in mosaic {store} : {resp.status_code}, {resp.text}"
            )

        self._cache.clear()
        return resp.json()

    def mosaic_coverages(self, store):
        """Returns all coverages in a coverage store"""
        params = dict()
        url = build_url(
            self.service_url,
            [
                "workspaces",
                store.workspace.name,
                "coveragestores",
                store.name,
                "coverages.json",
            ],
            params,
        )
        # GET /workspaces/<ws>/coveragestores/<name>/coverages.json
        headers = {"Content-type": "application/json", "Accept": "application/json"}

        resp = self.http_request(url, headers=headers)
        if resp.status_code != 200:
            raise FailedRequestError(
                f"Failed to get mosaic coverages {store} : {resp.status_code}, {resp.text}"
            )

        self._cache.clear()
        return resp.json()

    def mosaic_coverage_schema(self, coverage, store, workspace):
        """Returns the schema of a coverage in a coverage store"""
        params = dict()
        url = build_url(
            self.service_url,
            [
                "workspaces",
                workspace,
                "coveragestores",
                store,
                "coverages",
                coverage,
                "index.json",
            ],
            params,
        )
        # GET /workspaces/<ws>/coveragestores/<name>/coverages/<coverage>/index.json

        headers = {"Content-type": "application/json", "Accept": "application/json"}

        resp = self.http_request(url, headers=headers)
        if resp.status_code != 200:
            raise FailedRequestError(
                f"Failed to get mosaic schema {store} : {resp.status_code}, {resp.text}"
            )

        self._cache.clear()
        return resp.json()

    def publish_featuretype(
        self,
        name,
        store,
        native_crs,
        srs=None,
        jdbc_virtual_table=None,
        native_name=None,
    ):
        """Publish a featuretype from data in an existing store"""
        # @todo native_srs doesn't seem to get detected, even when in the DB
        # metadata (at least for postgis in geometry_columns) and then there
        # will be a misconfigured layer
        if native_crs is None:
            raise ValueError("must specify native_crs")

        srs = srs or native_crs
        feature_type = FeatureType(self, store.workspace, store, name)
        # because name is the in FeatureType base class, work around that
        # and hack in these others that don't have xml properties
        feature_type.dirty["name"] = name
        feature_type.dirty["srs"] = srs
        feature_type.dirty["nativeCRS"] = native_crs
        feature_type.enabled = True
        feature_type.advertised = True
        feature_type.title = name

        if native_name is not None and native_name:
            feature_type.native_name = native_name

        headers = {"Content-type": "application/xml", "Accept": "application/xml"}

        resource_url = store.resource_url
        if jdbc_virtual_table is not None and jdbc_virtual_table:
            feature_type.metadata = {"JDBC_VIRTUAL_TABLE": jdbc_virtual_table}
            params = dict()
            resource_url = build_url(
                self.service_url,
                [
                    "workspaces",
                    store.workspace.name,
                    "datastores",
                    store.name,
                    "featuretypes.xml",
                ],
                params,
            )

        resp = self.http_request(
            resource_url, method="post", data=feature_type.message(), headers=headers
        )
        if resp.status_code not in (200, 201, 202):
            raise FailedRequestError(
                f"Failed to publish feature type {name} : {resp.status_code}, {resp.text}"
            )

        self._cache.clear()
        feature_type.fetch()
        return feature_type

    def get_resources(self, names=None, stores=None, workspaces=None):
        """
        Resources include feature stores, coverage stores and WMS stores, however does not include layer groups.
        names, stores and workspaces can be provided as a comma delimited strings or as arrays, and are used for filtering.
        Will always return an array.
        """
        if workspaces and not isinstance(workspaces, list):
            workspaces = [workspaces]

        if not stores:
            _stores = self.get_stores(workspaces=workspaces)
        elif not isinstance(stores, list):
            _stores = [stores]
        else:
            _stores = stores

        if isinstance(names, string_types):
            names = [s.strip() for s in names.split(",")]

        resources = []
        for s in _stores:
            try:
                if isinstance(s, string_types):
                    if workspaces:
                        for w in workspaces:
                            if self.get_store(s, workspace=w):
                                s = self.get_store(s, workspace=w)
                                if s:
                                    if names:
                                        for name in names:
                                            _res = s.get_resources(name=name)
                                            if _res:
                                                resources.append(_res)
                                    else:
                                        resources.extend(s.get_resources())
                    else:
                        s = self.get_store(s)
                        if s:
                            if names:
                                for name in names:
                                    _res = s.get_resources(name=name)
                                    if _res:
                                        resources.append(_res)
                            else:
                                resources.extend(s.get_resources())
                elif names:
                    for name in names:
                        _res = s.get_resources(name=name)
                        if _res:
                            resources.append(_res)
                else:
                    resources.extend(s.get_resources())
            except FailedRequestError:
                continue

        if resources and names:
            return [resource for resource in resources if resource.name in names]

        return resources

    def get_resource(self, name=None, store=None, workspace=None):
        """
        returns a single resource object.
        Will return None if no resource is found.
        Will raise an error if more than one resource with the same name is found.
        """

        if store:
            resources = self.get_resources(
                names=name, stores=[store], workspaces=[workspace]
            )
        else:
            resources = self.get_resources(names=name, workspaces=[workspace])
        return self._return_first_item(resources)

    def get_layer(self, name):
        try:
            lyr = Layer(self, name)
            lyr.fetch()
            return lyr
        except FailedRequestError:
            return None

    def get_layers(self, resource=None):
        if isinstance(resource, string_types):
            ws_name = None
            if self.get_short_version() >= "2.13":
                if ":" in resource:
                    ws_name, resource = resource.split(":")

            if ws_name:
                resources = self.get_resources(names=resource, workspaces=[ws_name])
            else:
                resources = self.get_resources(names=resource)
            resource = self._return_first_item(resources)
        layers_url = f"{self.service_url}/layers.xml"
        data = self.get_xml(layers_url)
        lyrs = [Layer(self, l.find("name").text) for l in data.findall("layer")]
        if resource is not None:
            lyrs = [l for l in lyrs if l.resource.href == resource.href]
        # TODO: Filter by style
        return lyrs

    def get_layergroups(self, names=None, workspaces=None):
        """
        names and workspaces can be provided as a comma delimited strings or as arrays, and are used for filtering.
        If no workspaces are provided, will return all layer groups in the catalog (global and workspace specific).
        Will always return an array.
        """

        layergroups = []

        if workspaces is None or len(workspaces) == 0:
            # Add global layergroups
            url = f"{self.service_url}/layergroups.xml"
            groups = self.get_xml(url)
            layergroups.extend(
                [
                    LayerGroup(self, g.find("name").text, None)
                    for g in groups.findall("layerGroup")
                ]
            )
            workspaces = []
        elif isinstance(workspaces, string_types):
            workspaces = [s.strip() for s in workspaces.split(",") if s.strip()]
        elif isinstance(workspaces, Workspace):
            workspaces = [workspaces]

        if not workspaces:
            workspaces = self.get_workspaces()

        for ws in workspaces:
            ws_name = _name(ws)
            url = f"{self.service_url}/workspaces/{ws_name}/layergroups.xml"
            try:
                groups = self.get_xml(url)
            except FailedRequestError as e:
                if "no such workspace" in str(e).lower():
                    continue
                else:
                    raise FailedRequestError(f"Failed to get layergroups: {e}")

            layergroups.extend(
                [
                    LayerGroup(self, g.find("name").text, ws_name)
                    for g in groups.findall("layerGroup")
                ]
            )

        if names is None:
            names = []
        elif isinstance(names, string_types):
            names = [s.strip() for s in names.split(",") if s.strip()]

        if layergroups and names:
            return [lg for lg in layergroups if lg.name in names]

        return layergroups

    def get_layergroup(self, name, workspace=None):
        """
        returns a single layergroup object.
        Will return None if no layergroup is found.
        Will raise an error if more than one layergroup with the same name is found.
        """

        layergroups = self.get_layergroups(names=name, workspaces=[workspace])
        return self._return_first_item(layergroups)

    def create_layergroup(
        self,
        name,
        layers=(),
        styles=(),
        bounds=None,
        mode="SINGLE",
        abstract=None,
        title=None,
        workspace=None,
    ):
        if self.get_layergroups(names=name, workspaces=[workspace]):
            raise ConflictingDataError(f"LayerGroup named {name} already exists!")
        else:
            return UnsavedLayerGroup(
                self, name, layers, styles, bounds, mode, abstract, title, workspace
            )

    def get_styles(self, names=None, workspaces=None, recursive=False):
        """
        names and workspaces can be provided as a comma delimited strings or as arrays, and are used for filtering.
        If no workspaces are provided, will return all styles in the catalog (global and workspace specific).
        Will always return an array.
        """
        all_styles = []

        # Get Names first to speed up recursive queries
        if names is None:
            names = []
        elif isinstance(names, string_types):
            names = [s.strip() for s in names.split(",") if s.strip()]

        if not workspaces:
            # Add global styles
            url = f"{self.service_url}/styles.xml"
            styles = self.get_xml(url)
            all_styles += self.__build_style_list(
                styles, recursive=recursive, names=names
            )
            workspaces = []
        elif isinstance(workspaces, string_types):
            workspaces = [s.strip() for s in workspaces.split(",") if s.strip()]
        elif isinstance(workspaces, Workspace):
            workspaces = [workspaces]

        if not workspaces:
            workspaces = self.get_workspaces()

        for ws in workspaces:
            if ws:
                url = f"{self.service_url}/workspaces/{_name(ws)}/styles.xml"
            else:
                url = f"{self.service_url}/styles.xml"
            try:
                styles = self.get_xml(url)
            except FailedRequestError as e:
                if "no such workspace" in str(e).lower():
                    continue
                elif f"workspace {_name(ws)} not found" in str(e).lower():
                    continue
                else:
                    raise FailedRequestError(f"Failed to get styles: {e}")
            all_styles += self.__build_style_list(
                styles, workspace=ws, recursive=recursive, names=names
            )

        if all_styles and names:
            return [style for style in all_styles if style.name in names]

        return all_styles

    def __build_style_list(
        self, styles_tree, workspace=None, recursive=False, names=None
    ):
        all_styles = []
        for s in styles_tree.findall("style"):
            try:
                style_name = s.find("name").text
                if names and style_name not in names:
                    continue
                if recursive:
                    style_xml = self.get_xml(s[1].attrib.get("href"))
                    style_format = style_xml.find("format").text
                    style_version = (
                        style_xml.find("languageVersion")
                        .find("version")
                        .text.replace(".", "")[:-1]
                    )
                    all_styles.append(
                        Style(
                            self,
                            style_name,
                            _name(workspace),
                            style_format + style_version,
                        )
                    )
                else:
                    all_styles.append(Style(self, style_name, _name(workspace)))
            except Exception:
                all_styles.append(Style(self, s.find("name").text, _name(workspace)))
        return all_styles

    def get_style(self, name, workspace=None, recursive=False):
        """
        Get single style from geoserver.
        Keyword arguments:
            name(str): name of the style
        Optional keyword arguments:
            workspace(str): name of the workspce where the style belong
        Legacy:
            recursive(bool): no longer used
        Return
            a single style object.
        Will return None if no style is found.
        Will raise an error if more than one style with the same name is found.
        """

        url = f"{self.service_url}/workspaces/{workspace}/styles/{name}.json" if workspace \
              else f"{self.service_url}/styles/{name}.json"

        try:
            resp = self.http_request(url, headers={"Accept": "application/json"})
            resp.raise_for_status()
            payload = resp.json()['style']
            extracted_workspace = payload['workspace'].get("name", workspace) if payload.get("workspace") else workspace
            return Style(
                self,
                payload['name'],
                extracted_workspace,
                payload['format'] + payload['languageVersion']['version'],
            )

        except HTTPError as e:
            if resp.status_code == 404:
                return None
            logger.exception(e)
            raise e
        except Exception as e:
            logger.exception(e)
            raise e
    
    def delete_style(self, name, workspace=None, purge=True):
        if workspace:
            '''
            If workspace is passed, we call directly the wanted style
            '''
            url = f"{self.service_url}/workspaces/{workspace}/styles/{name}?purge={purge}"
        else:
            '''
            If is not passed, we try to get the style without passing any workspace
            '''
            url = f"{self.service_url}/styles/{name}?purge={purge}"
            
        try:
            resp = self.http_request(url, method="DELETE")
            if resp.status_code != 404:
                resp.raise_for_status()
            return resp.status_code == 201
        except Exception as e:
            logger.exception(e)
            raise e

    def create_style(
        self,
        name,
        data,
        overwrite=False,
        workspace=None,
        style_format="sld10",
        raw=False,
    ):
        styles = self.get_style(name=name, workspace=workspace, recursive=True)
        if styles:
            style = styles
        else:
            style = None

        if not overwrite and style is not None and style:
            raise ConflictingDataError(f"There is already a style named {name}")

        if not style:
            xml = "<style><name>{0}</name><filename>{0}.sld</filename></style>".format(
                name
            )
            style = Style(self, name, workspace, style_format)
            headers = {"Content-type": "application/xml", "Accept": "text/plain"}
            create_url = style.create_href
            resp = self.http_request(
                create_url, method="post", data=xml, headers=headers
            )
            if resp.status_code == 406:
                headers["Accept"] = "application/xml"
                resp = self.http_request(
                    create_url, method="post", data=xml, headers=headers
                )

            if resp.status_code not in (200, 201, 202):
                raise FailedRequestError(
                    f"Failed to create style {name} : {resp.status_code}, {resp.text}"
                )

        if style:
            headers = {"Content-type": style.content_type, "Accept": "application/xml"}

            body_href = style.body_href
            if raw:
                body_href += "?raw=true"

            resp = self.http_request(
                body_href, method="put", data=data, headers=headers
            )
            if resp.status_code not in (200, 201, 202):
                body_href = f"{os.path.splitext(style.body_href)[0]}.xml"
                if raw:
                    body_href += "?raw=true"

                resp = self.http_request(
                    body_href, method="put", data=data, headers=headers
                )
                if resp.status_code not in (200, 201, 202):
                    raise FailedRequestError(
                        f"Failed to update style {name} : {resp.status_code}, {resp.text}"
                    )

            self._cache.pop(style.href, None)
            self._cache.pop(style.body_href, None)
            return style
        else:
            raise FailedRequestError(f"Failed to create style {name}")

    def create_workspace(self, name, uri):
        xml = (
            "<namespace>" "<prefix>{name}</prefix>" "<uri>{uri}</uri>" "</namespace>"
        ).format(name=name, uri=uri)

        headers = {"Content-Type": "application/xml"}
        workspace_url = f"{self.service_url}/namespaces/"

        resp = self.http_request(
            workspace_url, method="post", data=xml, headers=headers
        )
        if resp.status_code not in (200, 201, 202):
            raise FailedRequestError(
                f"Failed to create workspace {name} : {resp.status_code}, {resp.text}"
            )

        self._cache.pop(f"{self.service_url}/workspaces.xml", None)
        workspaces = self.get_workspaces(names=name)
        # Can only have one workspace with this name
        return workspaces[0] if workspaces else None

    def get_workspaces(self, names=None):
        """
        Returns a list of workspaces in the catalog.
        If names is specified, will only return workspaces that match.
        names can either be a comma delimited string or an array.
        Will return an empty list if no workspaces are found.
        """
        if names is None:
            names = []
        elif isinstance(names, string_types):
            names = [s.strip() for s in names.split(",") if s.strip()]

        data = self.get_xml(f"{self.service_url}/workspaces.xml")
        workspaces = []
        workspaces.extend(
            [workspace_from_index(self, node) for node in data.findall("workspace")]
        )

        if workspaces and names:
            return [ws for ws in workspaces if ws.name in names]

        return workspaces

    def get_workspace(self, name):
        """
        returns a single workspace object.
        Will return None if no workspace is found.
        Will raise an error if more than one workspace with the same name is found.
        """

        workspaces = self.get_workspaces(names=name)
        return self._return_first_item(workspaces)

    def get_default_workspace(self):
        ws = Workspace(self, "default")
        # must fetch and resolve the 'real' workspace from the response
        ws.fetch()
        return workspace_from_index(self, ws.dom)

    def set_default_workspace(self, name):
        if hasattr(name, "name"):
            name = name.name
        workspace = self.get_workspaces(names=name)[0]
        if workspace is not None and workspace:
            headers = {"Content-Type": "application/xml"}
            default_workspace_url = f"{self.service_url}/workspaces/default.xml"
            data = f"<workspace><name>{name}</name></workspace>"

            resp = self.http_request(
                default_workspace_url, method="put", data=data, headers=headers
            )
            if resp.status_code not in (200, 201, 202):
                raise FailedRequestError(
                    f"Failed to set default workspace {name} : {resp.status_code}, {resp.text}"
                )

            self._cache.pop(default_workspace_url, None)
            self._cache.pop(f"{self.service_url}/workspaces.xml", None)
        else:
            raise FailedRequestError(f"no workspace named {name}")

    def list_feature_type_names(self, workspace, store, filter="available"):
        if workspace is None:
            raise ValueError("Must provide workspace")

        if store is None:
            raise ValueError("Must provide store")

        filter = filter.lower()
        workspace = _name(workspace)
        store = _name(store)

        url = f"{self.service_url}/workspaces/{workspace}/datastores/{store}/featuretypes.json?list={filter}"
        resp = self.http_request(url)
        if resp.status_code != 200:
            raise FailedRequestError("Failed to query feature_type_names")

        data = []
        if filter in ("available", "available_with_geom"):
            try:
                data = resp.json()["list"]["string"]
            except JSONDecodeError:
                pass
            return data
        elif filter == "configured":
            data = resp.json()["featureTypes"]["featureType"]
            return [fn["name"] for fn in data]
        elif filter == "all":
            feature_type_names = []
            url = f"{self.service_url}/workspaces/{workspace}/datastores/{store}/featuretypes.json?list=available"
            resp = self.http_request(url)
            if resp.status_code != 200:
                raise FailedRequestError("Failed to query feature_type_names")
            feature_type_names.extend(resp.json()["list"]["string"])

            url = f"{self.service_url}/workspaces/{workspace}/datastores/{store}/featuretypes.json?list=configured"
            resp = self.http_request(url)
            if resp.status_code != 200:
                raise FailedRequestError("Failed to query feature_type_names")
            data = resp.json()["featureTypes"]["featureType"]
            feature_type_names.extend([fn["name"] for fn in data])

            return feature_type_names

    def get_services(self, ogc_type="wms"):
        """
        Returns a list of wms services in the catalog.
        Will return an empty list if no services are found.
        """

        data = self.get_xml(f"{self.service_url}/services/{ogc_type}/settings")
        services = []
        services.append(service_from_index(self, data))
        workspaces = self.get_workspaces()
        for ws in workspaces:
            try:
                data = self.get_xml(
                    f"{self.service_url}/services/{ogc_type}/workspaces/{ws.name}/settings"
                )
                services.append(service_from_index(self, data))
            except FailedRequestError as e:
                logger.debug(f"Not found {ogc_type} service for workspace {ws.name}")
        return services

    def create_user(self, username, password):
        users = self.get_users(names=username)
        if len(users) > 0:
            logging.warning(f"User {username} already exists")
            tmp_cat = Catalog(
                service_url=self.service_url, username=username, password=password
            )
            try:
                tmp_cat.get_version()
            except FailedRequestError as e:
                logger.error("And we probably have incorrect password")
                raise FailedRequestError

            return users[0]

        xml = (
            "<user>"
            "<userName>{username}</userName>"
            "<password>{password}</password>"
            "<enabled>true</enabled>"
            "</user>"
        ).format(username=username, password=password)

        headers = {"Content-Type": "application/xml"}
        users_url = f"{self.service_url}/security/usergroup/users/"

        resp = self.http_request(users_url, method="post", data=xml, headers=headers)
        if resp.status_code not in (200, 201, 202):
            raise FailedRequestError(
                f"Failed to create user {username} : {resp.status_code}, {resp.text}"
            )

        self._cache.pop(f"{self.service_url}/security/usergroup/users/", None)
        users = self.get_users(names=username)
        return users[0] if users else None

    def get_users(self, names=None):
        """
        Returns a list of users in the catalog.
        If names is specified, will only return users that match.
        names can either be a comma delimited string or an array.
        Will return an empty list if no users are found (unlikely).
        """
        if names is None:
            names = []
        elif isinstance(names, string_types):
            names = [s.strip() for s in names.split(",") if s.strip()]

        data = self.get_xml(f"{self.service_url}/security/usergroup/users/")
        users = []
        users.extend([user_from_index(self, node) for node in data.findall("user")])

        if users and names:
            return [ws for ws in users if ws.user_name in names]
        return users

    def get_master_pwd(self):
        url = f"{self.service_url}/security/masterpw.xml"
        resp = self.http_request(url)
        masterpwd = None
        if resp.status_code == 200:
            content = resp.content
            if isinstance(content, bytes):
                content = content.decode("UTF-8")
            dom = XML(content)
            masterpwd = (
                dom.find("oldMasterPassword").text
                if dom.find("oldMasterPassword") is not None
                else None
            )
        else:
            raise FailedRequestError(resp.content)

        return masterpwd

    def set_master_pwd(self, new_pwd):
        old_pwd = self.get_master_pwd()
        if old_pwd == new_pwd:
            return new_pwd

        headers = {"Content-Type": "application/xml"}
        url = f"{self.service_url}/security/masterpw.xml"
        body = (
            "<masterPassword>"
            "<oldMasterPassword>{old_pwd}</oldMasterPassword>"
            "<newMasterPassword>{new_pwd}</newMasterPassword>"
            "</masterPassword>"
        ).format(old_pwd=old_pwd, new_pwd=new_pwd)
        resp = self.http_request(url, method="put", data=body, headers=headers)
        if resp.status_code == 200:
            res = new_pwd
            self.reload()
        else:
            raise FailedRequestError(resp.content)
        return res

    def set_my_pwd(self, new_pwd):
        headers = {"Content-Type": "application/xml"}
        url = f"{self.service_url}/security/self/password.xml"
        body = (
            "<userPassword>" "<newPassword>{new_pwd}</newPassword>" "</userPassword>"
        ).format(new_pwd=new_pwd)
        resp = self.http_request(url, method="put", data=body, headers=headers)

        if resp.status_code == 200:
            res = new_pwd
            self.reload()
            self.password = new_pwd
            self.reload()
        else:
            raise FailedRequestError(resp.content)
        return res

    def get_global_settings(self):
        return GlobalSettings(self)

    def get_roles(self):
        url = f"{self.service_url}/security/roles"
        resp = self.get_xml(rest_url=url)
        roles = [x.text for x in resp.findall("role")]
        return roles

    def get_roles_user(self, username):
        url = f"{self.service_url}/security/roles/user/{username}"
        resp = self.get_xml(rest_url=url)
        roles = [x.text for x in resp.findall("role")]
        return roles

    def add_role_user(self, rolename, username):
        url = f"{self.service_url}/security/roles/role/{rolename}/user/{username}"
        resp = self.http_request(url, method="post")

        if resp.status_code != 200:
            raise FailedRequestError(resp.content)

        self._cache.clear()

    def del_role_user(self, rolename, username):
        url = f"{self.service_url}/security/roles/role/{rolename}/user/{username}"
        resp = self.http_request(url, method="delete")

        if resp.status_code != 200:
            raise FailedRequestError(resp.content)

        self._cache.clear()