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

/*
 * Copyright 2016 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 org.onosproject.yangutils.utils.UtilConstants;

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

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

    /**
     * Default constructor.
     */
    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(YangDataTypes.INT8)) {
            return UtilConstants.BYTE;
        } else if (type.equals(YangDataTypes.INT16)) {
            return UtilConstants.SHORT;
        } else if (type.equals(YangDataTypes.INT32)) {
            return UtilConstants.INT;
        } else if (type.equals(YangDataTypes.INT64)) {
            return UtilConstants.LONG;
        } else if (type.equals(YangDataTypes.UINT8)) {
            return UtilConstants.SHORT;
        } else if (type.equals(YangDataTypes.UINT16)) {
            return UtilConstants.INT;
        } else if (type.equals(YangDataTypes.UINT32)) {
            return UtilConstants.LONG;
        } else if (type.equals(YangDataTypes.UINT64)) {
            //TODO: BIGINTEGER.
        } else if (type.equals(YangDataTypes.DECIMAL64)) {
            //TODO: DECIMAL64.
        } else if (type.equals(YangDataTypes.STRING)) {
            return UtilConstants.STRING;
        } else if (type.equals(YangDataTypes.BOOLEAN)) {
            return UtilConstants.BOOLEAN;
        } else if (type.equals(YangDataTypes.ENUMERATION)) {
            //TODO: ENUMERATION.
        } else if (type.equals(YangDataTypes.BITS)) {
            //TODO:BITS
        } else if (type.equals(YangDataTypes.BINARY)) {
            //TODO:BINARY
        } else if (type.equals(YangDataTypes.LEAFREF)) {
            //TODO:LEAFREF
        } else if (type.equals(YangDataTypes.IDENTITYREF)) {
            //TODO:IDENTITYREF
        } else if (type.equals(YangDataTypes.EMPTY)) {
            //TODO:EMPTY
        } else if (type.equals(YangDataTypes.UNION)) {
            //TODO:UNION
        } else if (type.equals(YangDataTypes.INSTANCE_IDENTIFIER)) {
            //TODO:INSTANCE_IDENTIFIER
        } else if (type.equals(YangDataTypes.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(YangDataTypes.INT8)) {
                return UtilConstants.BYTE_WRAPPER;
            } else if (type.equals(YangDataTypes.INT16)) {
                return UtilConstants.SHORT_WRAPPER;
            } else if (type.equals(YangDataTypes.INT32)) {
                return UtilConstants.INTEGER_WRAPPER;
            } else if (type.equals(YangDataTypes.INT64)) {
                return UtilConstants.LONG_WRAPPER;
            } else if (type.equals(YangDataTypes.UINT8)) {
                return UtilConstants.SHORT_WRAPPER;
            } else if (type.equals(YangDataTypes.UINT16)) {
                return UtilConstants.INTEGER_WRAPPER;
            } else if (type.equals(YangDataTypes.UINT32)) {
                return UtilConstants.LONG_WRAPPER;
            } else if (type.equals(YangDataTypes.UINT64)) {
                //TODO: BIGINTEGER.
            } else if (type.equals(YangDataTypes.DECIMAL64)) {
                //TODO: DECIMAL64.
            } else if (type.equals(YangDataTypes.STRING)) {
                return UtilConstants.STRING;
            } else if (type.equals(YangDataTypes.BOOLEAN)) {
                return UtilConstants.BOOLEAN_WRAPPER;
            } else if (type.equals(YangDataTypes.ENUMERATION)) {
                //TODO: ENUMERATION.
            } else if (type.equals(YangDataTypes.BITS)) {
                //TODO:BITS
            } else if (type.equals(YangDataTypes.BINARY)) {
                //TODO:BINARY
            } else if (type.equals(YangDataTypes.LEAFREF)) {
                //TODO:LEAFREF
            } else if (type.equals(YangDataTypes.IDENTITYREF)) {
                //TODO:IDENTITYREF
            } else if (type.equals(YangDataTypes.EMPTY)) {
                //TODO:EMPTY
            } else if (type.equals(YangDataTypes.UNION)) {
                //TODO:UNION
            } else if (type.equals(YangDataTypes.INSTANCE_IDENTIFIER)) {
                //TODO:INSTANCE_IDENTIFIER
            } else if (type.equals(YangDataTypes.DERIVED)) {
                return JavaIdentifierSyntax
                        .getCaptialCase(JavaIdentifierSyntax.getCamelCase(yangType.getDataTypeName()));
            }
        } else {
            if (type.equals(YangDataTypes.UINT64)) {
                //TODO: BIGINTEGER.
            } else if (type.equals(YangDataTypes.DECIMAL64)) {
                //TODO: DECIMAL64.
            } else if (type.equals(YangDataTypes.STRING)) {
                return UtilConstants.STRING;
            } else if (type.equals(YangDataTypes.ENUMERATION)) {
                //TODO: ENUMERATION.
            } else if (type.equals(YangDataTypes.BITS)) {
                //TODO:BITS
            } else if (type.equals(YangDataTypes.BINARY)) {
                //TODO:BINARY
            } else if (type.equals(YangDataTypes.LEAFREF)) {
                //TODO:LEAFREF
            } else if (type.equals(YangDataTypes.IDENTITYREF)) {
                //TODO:IDENTITYREF
            } else if (type.equals(YangDataTypes.EMPTY)) {
                //TODO:EMPTY
            } else if (type.equals(YangDataTypes.UNION)) {
                //TODO:UNION
            } else if (type.equals(YangDataTypes.INSTANCE_IDENTIFIER)) {
                //TODO:INSTANCE_IDENTIFIER
            } else if (type.equals(YangDataTypes.DERIVED)) {
                return JavaIdentifierSyntax
                        .getCaptialCase(JavaIdentifierSyntax.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(YangDataTypes.INT8)
                    || type.equals(YangDataTypes.INT16)
                    || type.equals(YangDataTypes.INT32)
                    || type.equals(YangDataTypes.INT64)
                    || type.equals(YangDataTypes.UINT8)
                    || type.equals(YangDataTypes.UINT16)
                    || type.equals(YangDataTypes.UINT32)
                    || type.equals(YangDataTypes.STRING)
                    || type.equals(YangDataTypes.BOOLEAN)) {
                return UtilConstants.JAVA_LANG;
            } else if (type.equals(YangDataTypes.UINT64)) {
                //TODO: BIGINTEGER.
            } else if (type.equals(YangDataTypes.DECIMAL64)) {
                //TODO: DECIMAL64.
            } else if (type.equals(YangDataTypes.ENUMERATION)) {
                //TODO: ENUMERATION.
            } else if (type.equals(YangDataTypes.BITS)) {
                //TODO:BITS
            } else if (type.equals(YangDataTypes.BINARY)) {
                //TODO:BINARY
            } else if (type.equals(YangDataTypes.LEAFREF)) {
                //TODO:LEAFREF
            } else if (type.equals(YangDataTypes.IDENTITYREF)) {
                //TODO:IDENTITYREF
            } else if (type.equals(YangDataTypes.EMPTY)) {
                //TODO:EMPTY
            } else if (type.equals(YangDataTypes.UNION)) {
                //TODO:UNION
            } else if (type.equals(YangDataTypes.INSTANCE_IDENTIFIER)) {
                //TODO:INSTANCE_IDENTIFIER
            } else if (type.equals(YangDataTypes.DERIVED)) {
                for (JavaQualifiedTypeInfo imports : getImportInfo()) {
                    if (imports.getClassInfo().equals(classInfo)) {
                        return imports.getPkgInfo();
                    }
                }
            }
        } else {

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