from .layer import Layer

class LayerGroup(Layer):
    """
    Used to group several layers and handle them as one. If you add it to the map, any layers added or removed from the group will be added/removed on the map as well. 
    """

    @property
    def layers(self):
        return self._layers

    def __init__(self):
        super().__init__()
        self._layers = []
        if self._map:
            self._initJs()

    def _initJs(self):
        leafletJsObject = 'new L.layerGroup()'
        self._createJsObject(leafletJsObject, self._map.mapWidgetIndex)

    def addLayer(self, layer):
        self._layers.append(layer)
        layer.map = self._map
        layer._initJs()
        js = '{layerGroup}.addLayer({layerName})'.format(layerGroup=self._layerName,
                layerName=layer._layerName)
        self.runJavaScriptForMapIndex(js)

    def removeLayer(self, layer):
        if not layer in self._layers:
            # TODO Raise ValueError?
            return
        self._layers.remove(layer)
        js = '{layerGroup}.removeLayer({layerName})'.format(layerGroup=self._layerName,
                layerName=layer._layerName)
        self.runJavaScriptForMapIndex(js)
    
    def clearLayers(self):
        js = '{layerGroup}.clearLayers()'.format(layerGroup=self._layerName)
        self.runJavaScriptForMapIndex(js)
    
    def toGeoJSON(self, callback):
        self.getJsResponseForMapIndex(f'{self.jsName}.toGeoJSON()', callback)