/**
 * ============LICENSE_START=======================================================
 * org.onap.aai
 * ================================================================================
 * Copyright © 2017-2018 AT&T Intellectual Property. All rights reserved.
 * Copyright © 2017-2018 European Software Marketing Ltd.
 * ================================================================================
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *       http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 * ============LICENSE_END=========================================================
 */
package org.onap.aai.babel.xml.generator.model;

import java.util.HashSet;
import java.util.Iterator;
import java.util.Map;
import java.util.Set;
import org.onap.aai.babel.xml.generator.error.IllegalAccessException;
import org.onap.aai.babel.xml.generator.types.Cardinality;
import org.onap.aai.babel.xml.generator.types.ModelType;

public abstract class Model {
    public static final String GENERATOR_AAI_ERROR_UNSUPPORTED_WIDGET_OPERATION = "Operation Not Supported for Widgets";

    protected Set<Resource> resources = new HashSet<>();
    protected Set<Widget> widgets = new HashSet<>();
    private String modelId;
    private String modelName;
    private String modelVersion;
    private String modelNameVersionId;
    private String modelDescription;

    /**
     * Gets the object (model) corresponding to the supplied TOSCA type.
     *
     * @param toscaType the tosca type
     * @return the model for the type, or null
     */
    public static Model getModelFor(String toscaType) {
        Model modelToBeReturned = null;
        String typePrefix = toscaType;
        while (modelToBeReturned == null && typePrefix != null && typePrefix.lastIndexOf('.') != -1) {
            switch (typePrefix) {
                case "org.openecomp.resource.vf.allottedResource":
                    modelToBeReturned = new AllotedResource();
                    break;
                case "org.openecomp.resource.vfc.AllottedResource":
                    modelToBeReturned = new ProvidingService();
                    break;
                case "org.openecomp.resource.vfc":
                    modelToBeReturned = new VServerWidget();
                    break;
                case "org.openecomp.resource.cp":
                case "org.openecomp.cp":
                    modelToBeReturned = new LIntfWidget();
                    break;
                case "org.openecomp.resource.vl":
                    modelToBeReturned = new L3Network();
                    break;
                case "org.openecomp.resource.vf":
                    modelToBeReturned = new VirtualFunction();
                    break;
                case "org.openecomp.groups.vfmodule":
                case "org.openecomp.groups.VfModule":
                    modelToBeReturned = new VfModule();
                    break;
                case "org.openecomp.resource.vfc.nodes.heat.cinder":
                    modelToBeReturned = new VolumeWidget();
                    break;
                default:
                    modelToBeReturned = null;
                    break;
            }
            typePrefix = typePrefix.substring(0, typePrefix.lastIndexOf('.'));
        }

        return modelToBeReturned;
    }

    public abstract boolean addResource(Resource resource);

    public abstract boolean addWidget(Widget resource);

    /**
     * Gets widget version id.
     *
     * @return the widget version id
     */
    public String getWidgetId() {
        org.onap.aai.babel.xml.generator.types.Model model =
                this.getClass().getAnnotation(org.onap.aai.babel.xml.generator.types.Model.class);
        return Widget.getWidget(model.widget()).getId();
    }

    /**
     * Gets invariant id.
     *
     * @return the invariant id
     */
    public String getWidgetInvariantId() {
        org.onap.aai.babel.xml.generator.types.Model model =
                this.getClass().getAnnotation(org.onap.aai.babel.xml.generator.types.Model.class);
        return Widget.getWidget(model.widget()).getWidgetId();
    }

    /**
     * Gets delete flag.
     *
     * @return the delete flag
     */
    public boolean getDeleteFlag() {
        org.onap.aai.babel.xml.generator.types.Model model =
                this.getClass().getAnnotation(org.onap.aai.babel.xml.generator.types.Model.class);
        return model.dataDeleteFlag();
    }

    /**
     * Gets cardinality.
     *
     * @return the cardinality
     */
    public Cardinality getCardinality() {
        org.onap.aai.babel.xml.generator.types.Model model =
                this.getClass().getAnnotation(org.onap.aai.babel.xml.generator.types.Model.class);
        return model.cardinality();
    }

    public abstract Widget.Type getWidgetType();

    public String getModelId() {
        checkSupported();
        return modelId;
    }

    /**
     * Gets model type.
     *
     * @return the model type
     */
    public ModelType getModelType() {
        if (this instanceof Service) {
            return ModelType.SERVICE;
        } else if (this instanceof Resource) {
            return ModelType.RESOURCE;
        } else if (this instanceof Widget) {
            return ModelType.WIDGET;
        } else {
            return null;
        }
    }

    public String getModelName() {
        return modelName;
    }

    public String getModelVersion() {
        return modelVersion;
    }

    public String getModelNameVersionId() {
        checkSupported();
        return modelNameVersionId;
    }

    public String getModelDescription() {
        return modelDescription;
    }

    /**
     * Populate model identification information.
     *
     * @param modelIdentInfo the model ident info
     */
    public void populateModelIdentificationInformation(Map<String, String> modelIdentInfo) {
        Iterator<String> iter = modelIdentInfo.keySet().iterator();
        String property;
        while (iter.hasNext()) {
            property = iter.next();
            switch (property) {
                case "vfModuleModelInvariantUUID":
                case "serviceInvariantUUID":
                case "resourceInvariantUUID":
                case "invariantUUID":
                case "providing_service_invariant_uuid":
                    modelId = modelIdentInfo.get(property);
                    break;
                case "vfModuleModelUUID":
                case "resourceUUID":
                case "serviceUUID":
                case "UUID":
                case "providing_service_uuid":
                    modelNameVersionId = modelIdentInfo.get(property);
                    break;
                case "vfModuleModelVersion":
                case "serviceVersion":
                case "resourceversion":
                case "version":
                    modelVersion = modelIdentInfo.get(property);
                    break;
                case "vfModuleModelName":
                case "serviceName":
                case "resourceName":
                case "name":
                    modelName = modelIdentInfo.get(property);
                    break;
                case "serviceDescription":
                case "resourceDescription":
                case "vf_module_description":
                case "description":
                    modelDescription = modelIdentInfo.get(property);
                    break;
                case "providing_service_name":
                    modelName = modelIdentInfo.get(property);
                    modelDescription = modelIdentInfo.get(property);
                    break;
                default:
                    break;
            }
        }
    }

    public void setModelVersion(String modelVersion) {
        this.modelVersion = modelVersion;
    }

    public Set<Resource> getResources() {
        return resources;
    }

    public Set<Widget> getWidgets() {
        return widgets;
    }

    private void checkSupported() {
        if (this instanceof Widget) {
            throw new IllegalAccessException(GENERATOR_AAI_ERROR_UNSUPPORTED_WIDGET_OPERATION);
        }
    }
}