/*******************************************************************************
 * Copyright (c) 2015 Red Hat, Inc. Distributed under license by Red Hat, Inc.
 * All rights reserved. This program is made available under the terms of the
 * Eclipse Public License v1.0 which accompanies this distribution, and is
 * available at http://www.eclipse.org/legal/epl-v10.html
 * 
 * Contributors: Red Hat, Inc.
 ******************************************************************************/

package com.openshift.internal.restclient.model;

import static com.openshift.internal.restclient.capability.CapabilityInitializer.initializeCapabilities;

import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Optional;
import java.util.Set;
import java.util.stream.Collectors;

import org.apache.commons.lang.StringUtils;
import org.jboss.dmr.ModelNode;
import org.jboss.dmr.ModelType;

import com.openshift.internal.restclient.model.volume.EmptyDirVolumeSource;
import com.openshift.internal.restclient.model.volume.VolumeMount;
import com.openshift.internal.restclient.model.volume.VolumeSource;
import com.openshift.internal.util.JBossDmrExtentions;
import com.openshift.restclient.IClient;
import com.openshift.restclient.images.DockerImageURI;
import com.openshift.restclient.model.IContainer;
import com.openshift.restclient.model.IEnvironmentVariable;
import com.openshift.restclient.model.IPort;
import com.openshift.restclient.model.IReplicationController;
import com.openshift.restclient.model.volume.IEmptyDirVolumeSource;
import com.openshift.restclient.model.volume.IVolumeMount;
import com.openshift.restclient.model.volume.IVolumeSource;

public class ReplicationController extends KubernetesResource implements IReplicationController {

    public static final String SPEC_TEMPLATE_CONTAINERS = "spec.template.spec.containers";
    protected static final String SPEC_TEMPLATE_LABELS = "spec.template.metadata.labels";
    protected static final String VOLUMES = "spec.template.spec.volumes";
    protected static final String SPEC_REPLICAS = "spec.replicas";
    protected static final String SPEC_SELECTOR = "spec.selector";
    protected static final String STATUS_REPLICA = "status.replicas";
    protected static final String SERVICEACCOUNTNAME = "spec.template.spec.serviceAccountName";

    protected static final String IMAGE = "image";
    protected static final String ENV = "env";
    private Map<String, String[]> propertyKeys;

    public ReplicationController(ModelNode node, IClient client, Map<String, String[]> propertyKeys) {
        super(node, client, propertyKeys);
        this.propertyKeys = propertyKeys;
        initializeCapabilities(getModifiableCapabilities(), this, getClient());
    }

    @Override
    public void setEnvironmentVariable(String name, String value) {
        setEnvironmentVariable(null, name, value);
    }

    @Override
    public void setEnvironmentVariable(String containerName, String name, String value) {
        String defaultedContainerName = StringUtils.defaultIfBlank(containerName, "");
        ModelNode specContainers = get(SPEC_TEMPLATE_CONTAINERS);
        if (specContainers.isDefined()) { // should ALWAYS exist
            List<ModelNode> containers = specContainers.asList();
            if (!containers.isEmpty()) {

                Optional<ModelNode> opt = containers.stream()
                        .filter(n -> defaultedContainerName.equals(asString(n, NAME))).findFirst();
                ModelNode node = opt.isPresent() ? opt.get() : containers.get(0);
                ModelNode envNode = get(node, ENV);

                List<ModelNode> varList = new ArrayList<>();
                if (ModelType.LIST.equals(envNode.getType())) {
                    varList.addAll(envNode.asList());
                }

                // Check if variable already exists
                Optional<ModelNode> targetVar = varList.stream().filter(n -> name.equals(asString(n, NAME)))
                        .findFirst();

                ModelNode var;
                if (targetVar.isPresent()) {
                    var = targetVar.get();
                    int i = varList.indexOf(var);
                    set(var, VALUE, value);
                    varList.set(i, var);
                } else {
                    var = new ModelNode();
                    set(var, NAME, name);
                    set(var, VALUE, value);
                    varList.add(var);
                }
                envNode.set(varList);
            }
        }
    }

    @Override
    public void removeEnvironmentVariable(String name) {
        removeEnvironmentVariable(null, name);
    }

    @Override
    public void removeEnvironmentVariable(String containerName, String name) {
        if (name == null) {
            throw new IllegalArgumentException("Name cannot be null.");
        }
        String defaultedContainerName = StringUtils.defaultIfBlank(containerName, "");
        ModelNode specContainers = get(SPEC_TEMPLATE_CONTAINERS);
        if (specContainers.isDefined()) { // should ALWAYS exist
            List<ModelNode> containers = specContainers.asList();
            if (!containers.isEmpty()) {

                Optional<ModelNode> opt = containers.stream()
                        .filter(n -> defaultedContainerName.equals(asString(n, NAME))).findFirst();
                ModelNode node = opt.isPresent() ? opt.get() : containers.get(0);
                ModelNode envNode = get(node, ENV);

                List<ModelNode> varList = new ArrayList<>();
                if (ModelType.LIST.equals(envNode.getType())) {
                    varList.addAll(envNode.asList());
                }

                // Check if variable exists
                Optional<ModelNode> targetVar = varList.stream().filter(n -> name.equals(asString(n, NAME)))
                        .findFirst();

                ModelNode var;
                if (targetVar.isPresent()) {
                    var = targetVar.get();
                    int i = varList.indexOf(var);
                    varList.remove(i);
                    envNode.set(varList);
                } else {
                    // do nothing
                }
            }
        }
    }

    @Override
    public Collection<IEnvironmentVariable> getEnvironmentVariables() {
        return getEnvironmentVariables(null);
    }

    @Override
    public Collection<IEnvironmentVariable> getEnvironmentVariables(String containerName) {
        String name = StringUtils.defaultIfBlank(containerName, "");
        ModelNode specContainers = get(SPEC_TEMPLATE_CONTAINERS);
        if (specContainers.isDefined()) {
            List<ModelNode> containers = specContainers.asList();
            if (!containers.isEmpty()) {
                Optional<ModelNode> opt = containers.stream().filter(n -> name.equals(asString(n, NAME))).findFirst();
                ModelNode node = opt.isPresent() ? opt.get() : containers.get(0);
                ModelNode envNode = get(node, ENV);
                if (envNode.isDefined()) {
                    return envNode.asList().stream().map(n -> new EnvironmentVariable(n, propertyKeys))
                            .collect(Collectors.toList());
                }
            }
        }
        return Collections.emptyList();
    }

    @Override
    public int getDesiredReplicaCount() {
        return asInt(SPEC_REPLICAS);
    }

    @Override
    public int getReplicas() {
        return getDesiredReplicaCount();
    }

    @Override
    public void setDesiredReplicaCount(int count) {
        set(SPEC_REPLICAS, count);
    }

    @Override
    public void setReplicas(int count) {
        setDesiredReplicaCount(count);
    }

    @Override
    public Map<String, String> getReplicaSelector() {
        return asMap(SPEC_SELECTOR);
    }

    @Override
    public void setReplicaSelector(String key, String value) {
        Map<String, String> selector = new HashMap<>();
        selector.put(key, value);
        setReplicaSelector(selector);
    }

    @Override
    public void setReplicaSelector(Map<String, String> selector) {
        get(SPEC_SELECTOR).clear();
        set(SPEC_SELECTOR, selector);
        selector.forEach((k, v) -> addTemplateLabel(k, v));
    }

    @Override
    public int getCurrentReplicaCount() {
        return asInt(STATUS_REPLICA);
    }

    @Override
    public Collection<String> getImages() {
        ModelNode node = get(SPEC_TEMPLATE_CONTAINERS);
        if (node.getType() != ModelType.LIST) {
            return new ArrayList<>();
        }
        Collection<String> list = new ArrayList<>();
        for (ModelNode entry : node.asList()) {
            list.add(entry.get(IMAGE).asString());
        }
        return list;
    }

    @Override
    public IContainer getContainer(String name) {
        if (StringUtils.isBlank(name)) {
            return null;
        }
        ModelNode containers = get(SPEC_TEMPLATE_CONTAINERS);
        if (containers.isDefined() && containers.getType() == ModelType.LIST) {
            Optional<ModelNode> first = containers.asList().stream()
                    .filter(n -> name.equals(JBossDmrExtentions.asString(n, this.propertyKeys, NAME))).findFirst();
            if (first.isPresent()) {
                return new Container(first.get(), this.propertyKeys);
            }
        }
        return null;
    }

    @Override
    public Collection<IContainer> getContainers() {
        ModelNode containers = get(SPEC_TEMPLATE_CONTAINERS);
        if (containers.isDefined() && containers.getType() == ModelType.LIST) {
            return containers.asList().stream().map(n -> new Container(n, this.propertyKeys))
                    .collect(Collectors.toList());
        }
        return Collections.emptyList();
    }

    @Override
    public Map<String, String> getTemplateLabels() {
        return asMap(SPEC_TEMPLATE_LABELS);
    }

    @Override
    public void addTemplateLabel(String key, String value) {
        ModelNode labels = get(SPEC_TEMPLATE_LABELS);
        labels.get(key).set(value);
    }

    @Override
    public IContainer addContainer(DockerImageURI tag, Set<IPort> containerPorts, Map<String, String> envVars) {
        return addContainer(tag.getName(), tag, containerPorts, envVars, new ArrayList<String>());
    }

    @Override
    public IContainer addContainer(String name, DockerImageURI tag, Set<IPort> containerPorts,
            Map<String, String> envVars, List<String> emptyDirVolumes) {

        IContainer container = addContainer(name);
        container.setImage(tag);

        if (!emptyDirVolumes.isEmpty()) {
            Set<IVolumeMount> volumes = new HashSet<>();
            for (String path : emptyDirVolumes) {
                VolumeMount volume = new VolumeMount(new ModelNode());
                volume.setMountPath(path);
                volume.setName(String.format("%s-%s", name, emptyDirVolumes.indexOf(path) + 1));
                volumes.add(volume);
                addEmptyDirVolumeToPodSpec(volume);
            }
            container.setVolumeMounts(volumes);
        }
        if (!containerPorts.isEmpty()) {
            Set<IPort> ports = new HashSet<>();
            for (IPort port : containerPorts) {
                ports.add(new Port(new ModelNode(), port));
            }
            container.setPorts(ports);
        }
        container.setEnvVars(envVars);
        return container;
    }
    
    @Override
    public IContainer addContainer(String name) {
        ModelNode containers = get(SPEC_TEMPLATE_CONTAINERS);
        Container container = new Container(containers.add());
        container.setName(name);
        return container;
    }

    private boolean hasVolumeNamed(ModelNode volNode, String name) {
        if (volNode.isDefined()) {
            List<ModelNode> podVolumes = volNode.asList();
            for (ModelNode node : podVolumes) {
                if (name.equals(asString(node, NAME))) {
                    return true;
                }
            }
        }
        return false;
    }

    private void addEmptyDirVolumeToPodSpec(VolumeMount volume) {
        ModelNode volNode = get(VOLUMES);
        if (hasVolumeNamed(volNode, volume.getName())) {
            // already exists
            return;
        }
        IEmptyDirVolumeSource volumeSource = new EmptyDirVolumeSource(volume.getName());
        volumeSource.setMedium("");
        addVolume(volumeSource);
    }

    @Override
    public void setContainers(Collection<IContainer> containers) {
        ModelNode nodeContainers = get(SPEC_TEMPLATE_CONTAINERS);
        nodeContainers.clear();
        if (containers != null) {
            containers.forEach(c -> nodeContainers.add(ModelNode.fromJSONString(c.toJson())));
        }
    }

    @Override
    public Set<IVolumeSource> getVolumes() {
        ModelNode vol = get(VOLUMES);
        Set<IVolumeSource> volumes = new HashSet<>();
        if (vol.isDefined()) {
            for (ModelNode node : vol.asList()) {
                volumes.add(VolumeSource.create(node));
            }
        }
        return volumes;
    }

    @Override
    public void setVolumes(Set<IVolumeSource> volumeSources) {
        ModelNode vol = get(VOLUMES);
        vol.clear();
        if (volumeSources != null) {
            volumeSources.forEach(v -> vol.add(ModelNode.fromJSONString(v.toJSONString())));
        }
    }

    @Override
    public void addVolume(IVolumeSource volumeSource) {
        ModelNode volList = get(VOLUMES);
        if (hasVolumeNamed(volList, volumeSource.getName())) {
            // already exists
            return;
        }
        volList.add(ModelNode.fromJSONString(volumeSource.toJSONString()));
    }

    @SuppressWarnings("unchecked")
    @Override
    public <T extends IVolumeSource> T addVolume(String volumetype, String name) {
        ModelNode volList = get(VOLUMES);
        ModelNode node = volList.add();
        node.get(volumetype).set(new ModelNode());
        IVolumeSource source = VolumeSource.create(node);
        source.setName(name);
        return (T) source;
    }

    @Override
    public void setServiceAccountName(String serviceAccountName) {
        set(SERVICEACCOUNTNAME, serviceAccountName);
    }

    @Override
    public String getServiceAccountName() {
        return asString(SERVICEACCOUNTNAME);
    }
}
