Add more controls.

src/Netzmacht/Contao/Leaflet/Mapper/Control/LayersControlMapper.php

@@ -12,6 +12,9 @@
 namespace Netzmacht\Contao\Leaflet\Mapper\Control;
 
 
+use Netzmacht\Contao\Leaflet\Mapper\DefinitionMapper;
+use Netzmacht\Contao\Leaflet\Model\LayerModel;
+
 class LayersControlMapper extends AbstractControlMapper
 {
     /**
@@ -27,4 +30,45 @@ class LayersControlMapper extends AbstractControlMapper
      * @var string
      */
     protected static $type = 'layers';
+
+    protected function buildConstructArguments(\Model $model, DefinitionMapper $mapper)
+    {
+        $arguments    = parent::buildConstructArguments($model, $mapper);
+        $arguments[1] = array();
+        $arguments[2] = array();
+
+        $definition = $this->getLayersDefinition($model);
+        $collection = LayerModel::findMultipleByIds(array_keys($definition));
+
+        if ($collection) {
+            foreach ($collection as $layer) {
+                $argument = ($definition[$layer->id] === 'overlay') ? 2 : 1;
+
+                $arguments[$argument][] = $mapper->handle($layer);
+            }
+        }
+
+        return $arguments;
+    }
+
+    /**
+     * Get layers definition from the control model.
+     *
+     * @param \Model $model The control model.
+     *
+     * @return array
+     */
+    protected function getLayersDefinition(\Model $model)
+    {
+        $layers = deserialize($model->layers, true);
+        $definition = array();
+
+        foreach ($layers as $layer) {
+            if ($layer['layer']) {
+                $definition[$layer['layer']] = $layer['mode'];
+            }
+        }
+
+        return $definition;
+    }
 }