utils/yangutils/src/main/java/org/onosproject/yangutils/translator/tojava/utils/AttributesJavaDataType.java

/*
 * Copyright 2016-present Open Networking Laboratory
 *
 * 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.
 */

package org.onosproject.yangutils.translator.tojava.utils;

import java.util.Set;
import java.util.TreeSet;

import org.onosproject.yangutils.datamodel.YangDataTypes;
import org.onosproject.yangutils.datamodel.YangType;
import org.onosproject.yangutils.translator.tojava.JavaQualifiedTypeInfo;

import static org.onosproject.yangutils.datamodel.YangDataTypes.BINARY;
import static org.onosproject.yangutils.datamodel.YangDataTypes.BITS;
import static org.onosproject.yangutils.datamodel.YangDataTypes.BOOLEAN;
import static org.onosproject.yangutils.datamodel.YangDataTypes.DECIMAL64;
import static org.onosproject.yangutils.datamodel.YangDataTypes.DERIVED;
import static org.onosproject.yangutils.datamodel.YangDataTypes.EMPTY;
import static org.onosproject.yangutils.datamodel.YangDataTypes.ENUMERATION;
import static org.onosproject.yangutils.datamodel.YangDataTypes.IDENTITYREF;
import static org.onosproject.yangutils.datamodel.YangDataTypes.INSTANCE_IDENTIFIER;
import static org.onosproject.yangutils.datamodel.YangDataTypes.INT16;
import static org.onosproject.yangutils.datamodel.YangDataTypes.INT32;
import static org.onosproject.yangutils.datamodel.YangDataTypes.INT64;
import static org.onosproject.yangutils.datamodel.YangDataTypes.INT8;
import static org.onosproject.yangutils.datamodel.YangDataTypes.LEAFREF;
import static org.onosproject.yangutils.datamodel.YangDataTypes.STRING;
import static org.onosproject.yangutils.datamodel.YangDataTypes.UINT16;
import static org.onosproject.yangutils.datamodel.YangDataTypes.UINT32;
import static org.onosproject.yangutils.datamodel.YangDataTypes.UINT64;
import static org.onosproject.yangutils.datamodel.YangDataTypes.UINT8;
import static org.onosproject.yangutils.datamodel.YangDataTypes.UNION;
import static org.onosproject.yangutils.translator.tojava.utils.JavaIdentifierSyntax.getCamelCase;
import static org.onosproject.yangutils.translator.tojava.utils.JavaIdentifierSyntax.getCaptialCase;
import static org.onosproject.yangutils.utils.UtilConstants.BOOLEAN_DATA_TYPE;
import static org.onosproject.yangutils.utils.UtilConstants.BOOLEAN_WRAPPER;
import static org.onosproject.yangutils.utils.UtilConstants.BYTE;
import static org.onosproject.yangutils.utils.UtilConstants.BYTE_WRAPPER;
import static org.onosproject.yangutils.utils.UtilConstants.INT;
import static org.onosproject.yangutils.utils.UtilConstants.INTEGER_WRAPPER;
import static org.onosproject.yangutils.utils.UtilConstants.JAVA_LANG;
import static org.onosproject.yangutils.utils.UtilConstants.LONG;
import static org.onosproject.yangutils.utils.UtilConstants.LONG_WRAPPER;
import static org.onosproject.yangutils.utils.UtilConstants.SHORT;
import static org.onosproject.yangutils.utils.UtilConstants.SHORT_WRAPPER;
import static org.onosproject.yangutils.utils.UtilConstants.STRING_DATA_TYPE;

/**
 * Represents java data types info corresponding to YANG type.
 */
public final class AttributesJavaDataType {

    private static Set<JavaQualifiedTypeInfo> importInfo = new TreeSet<>();

    /**
     * Creates an instance of attribute java data type.
     */
    private AttributesJavaDataType() {
    }

    /**
     * Returns import info.
     *
     * @return import info
     */
    public static Set<JavaQualifiedTypeInfo> getImportInfo() {

        return importInfo;
    }

    /**
     * Adds import info to the import info set.
     *
     * @param importData import info
     */
    public static void addImportInfo(JavaQualifiedTypeInfo importData) {

        getImportInfo().add(importData);
    }

    /**
     * Returns java type.
     *
     * @param yangType YANG type
     * @return java type
     */
    public static String getJavaDataType(YangType<?> yangType) {

        YangDataTypes type = yangType.getDataType();

        if (type.equals(INT8)) {
            return BYTE;
        } else if (type.equals(INT16)) {
            return SHORT;
        } else if (type.equals(INT32)) {
            return INT;
        } else if (type.equals(INT64)) {
            return LONG;
        } else if (type.equals(UINT8)) {
            return SHORT;
        } else if (type.equals(UINT16)) {
            return INT;
        } else if (type.equals(UINT32)) {
            return LONG;
        } else if (type.equals(UINT64)) {
            //TODO: BIGINTEGER.
        } else if (type.equals(DECIMAL64)) {
            //TODO: DECIMAL64.
        } else if (type.equals(STRING)) {
            return STRING_DATA_TYPE;
        } else if (type.equals(BOOLEAN)) {
            return BOOLEAN_DATA_TYPE;
        } else if (type.equals(ENUMERATION)) {
            //TODO: ENUMERATION.
        } else if (type.equals(BITS)) {
            //TODO:BITS
        } else if (type.equals(BINARY)) {
            //TODO:BINARY
        } else if (type.equals(LEAFREF)) {
            //TODO:LEAFREF
        } else if (type.equals(IDENTITYREF)) {
            //TODO:IDENTITYREF
        } else if (type.equals(EMPTY)) {
            //TODO:EMPTY
        } else if (type.equals(UNION)) {
            //TODO:UNION
        } else if (type.equals(INSTANCE_IDENTIFIER)) {
            //TODO:INSTANCE_IDENTIFIER
        } else if (type.equals(DERIVED)) {
            return yangType.getDataTypeName();
        }
        return null;
    }

    /**
     * Returns java import class.
     *
     * @param yangType YANG type
     * @param isListAttr if the attribute need to be a list
     * @return java import class
     */
    public static String getJavaImportClass(YangType<?> yangType, boolean isListAttr) {

        YangDataTypes type = yangType.getDataType();

        if (isListAttr) {
            if (type.equals(INT8)) {
                return BYTE_WRAPPER;
            } else if (type.equals(INT16)) {
                return SHORT_WRAPPER;
            } else if (type.equals(INT32)) {
                return INTEGER_WRAPPER;
            } else if (type.equals(INT64)) {
                return LONG_WRAPPER;
            } else if (type.equals(UINT8)) {
                return SHORT_WRAPPER;
            } else if (type.equals(UINT16)) {
                return INTEGER_WRAPPER;
            } else if (type.equals(UINT32)) {
                return LONG_WRAPPER;
            } else if (type.equals(UINT64)) {
                //TODO: BIGINTEGER.
            } else if (type.equals(DECIMAL64)) {
                //TODO: DECIMAL64.
            } else if (type.equals(STRING)) {
                return STRING_DATA_TYPE;
            } else if (type.equals(BOOLEAN)) {
                return BOOLEAN_WRAPPER;
            } else if (type.equals(ENUMERATION)) {
                //TODO: ENUMERATION.
            } else if (type.equals(BITS)) {
                //TODO:BITS
            } else if (type.equals(BINARY)) {
                //TODO:BINARY
            } else if (type.equals(LEAFREF)) {
                //TODO:LEAFREF
            } else if (type.equals(IDENTITYREF)) {
                //TODO:IDENTITYREF
            } else if (type.equals(EMPTY)) {
                //TODO:EMPTY
            } else if (type.equals(UNION)) {
                //TODO:UNION
            } else if (type.equals(INSTANCE_IDENTIFIER)) {
                //TODO:INSTANCE_IDENTIFIER
            } else if (type.equals(DERIVED)) {
                return getCaptialCase(getCamelCase(yangType.getDataTypeName()));
            }
        } else {
            if (type.equals(UINT64)) {
                //TODO: BIGINTEGER.
            } else if (type.equals(DECIMAL64)) {
                //TODO: DECIMAL64.
            } else if (type.equals(STRING)) {
                return STRING_DATA_TYPE;
            } else if (type.equals(ENUMERATION)) {
                //TODO: ENUMERATION.
            } else if (type.equals(BITS)) {
                //TODO:BITS
            } else if (type.equals(BINARY)) {
                //TODO:BINARY
            } else if (type.equals(LEAFREF)) {
                //TODO:LEAFREF
            } else if (type.equals(IDENTITYREF)) {
                //TODO:IDENTITYREF
            } else if (type.equals(EMPTY)) {
                //TODO:EMPTY
            } else if (type.equals(UNION)) {
                //TODO:UNION
            } else if (type.equals(INSTANCE_IDENTIFIER)) {
                //TODO:INSTANCE_IDENTIFIER
            } else if (type.equals(DERIVED)) {
                return getCaptialCase(getCamelCase(yangType.getDataTypeName()));
            }
        }
        return null;
    }

    /**
     * Returns java import package.
     *
     * @param yangType YANG type
     * @param isListAttr if the attribute is of list type
     * @param classInfo java import class info
     * @return java import package
     */
    public static String getJavaImportPackage(YangType<?> yangType, boolean isListAttr, String classInfo) {

        YangDataTypes type = yangType.getDataType();

        if (isListAttr) {
            if (type.equals(INT8)
                    || type.equals(INT16)
                    || type.equals(INT32)
                    || type.equals(INT64)
                    || type.equals(UINT8)
                    || type.equals(UINT16)
                    || type.equals(UINT32)
                    || type.equals(STRING)
                    || type.equals(BOOLEAN)) {
                return JAVA_LANG;
            } else if (type.equals(UINT64)) {
                //TODO: BIGINTEGER.
            } else if (type.equals(DECIMAL64)) {
                //TODO: DECIMAL64.
            } else if (type.equals(ENUMERATION)) {
                //TODO: ENUMERATION.
            } else if (type.equals(BITS)) {
                //TODO:BITS
            } else if (type.equals(BINARY)) {
                //TODO:BINARY
            } else if (type.equals(LEAFREF)) {
                //TODO:LEAFREF
            } else if (type.equals(IDENTITYREF)) {
                //TODO:IDENTITYREF
            } else if (type.equals(EMPTY)) {
                //TODO:EMPTY
            } else if (type.equals(UNION)) {
                //TODO:UNION
            } else if (type.equals(INSTANCE_IDENTIFIER)) {
                //TODO:INSTANCE_IDENTIFIER
            } else if (type.equals(DERIVED)) {
                for (JavaQualifiedTypeInfo imports : getImportInfo()) {
                    if (imports.getClassInfo().equals(classInfo)) {
                        return imports.getPkgInfo();
                    }
                }
            }
        } else {

            if (type.equals(UINT64)) {
                //TODO: BIGINTEGER.
            } else if (type.equals(DECIMAL64)) {
                //TODO: DECIMAL64.
            } else if (type.equals(STRING)) {
                return JAVA_LANG;
            } else if (type.equals(ENUMERATION)) {
                //TODO: ENUMERATION.
            } else if (type.equals(BITS)) {
                //TODO:BITS
            } else if (type.equals(BINARY)) {
                //TODO:BINARY
            } else if (type.equals(LEAFREF)) {
                //TODO:LEAFREF
            } else if (type.equals(IDENTITYREF)) {
                //TODO:IDENTITYREF
            } else if (type.equals(EMPTY)) {
                //TODO:EMPTY
            } else if (type.equals(UNION)) {
                //TODO:UNION
            } else if (type.equals(INSTANCE_IDENTIFIER)) {
                //TODO:INSTANCE_IDENTIFIER
            } else if (type.equals(DERIVED)) {
                for (JavaQualifiedTypeInfo imports : getImportInfo()) {
                    if (imports.getClassInfo().equals(classInfo)) {
                        return imports.getPkgInfo();
                    }
                }
            }
        }
        return null;
    }
}