/*
 * Copyright (c) 2013 Cisco Systems, Inc. and others.  All rights reserved.
 *
 * This program and the accompanying materials are 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
 */
package org.opendaylight.yangtools.yang.data.impl.codec;

import static java.util.Objects.requireNonNull;

import org.eclipse.jdt.annotation.NonNull;
import org.eclipse.jdt.annotation.Nullable;
import org.opendaylight.yangtools.yang.model.api.TypeDefinition;
import org.opendaylight.yangtools.yang.model.api.type.BinaryTypeDefinition;
import org.opendaylight.yangtools.yang.model.api.type.BitsTypeDefinition;
import org.opendaylight.yangtools.yang.model.api.type.BooleanTypeDefinition;
import org.opendaylight.yangtools.yang.model.api.type.DecimalTypeDefinition;
import org.opendaylight.yangtools.yang.model.api.type.EmptyTypeDefinition;
import org.opendaylight.yangtools.yang.model.api.type.EnumTypeDefinition;
import org.opendaylight.yangtools.yang.model.api.type.Int16TypeDefinition;
import org.opendaylight.yangtools.yang.model.api.type.Int32TypeDefinition;
import org.opendaylight.yangtools.yang.model.api.type.Int64TypeDefinition;
import org.opendaylight.yangtools.yang.model.api.type.Int8TypeDefinition;
import org.opendaylight.yangtools.yang.model.api.type.StringTypeDefinition;
import org.opendaylight.yangtools.yang.model.api.type.Uint16TypeDefinition;
import org.opendaylight.yangtools.yang.model.api.type.Uint32TypeDefinition;
import org.opendaylight.yangtools.yang.model.api.type.Uint64TypeDefinition;
import org.opendaylight.yangtools.yang.model.api.type.Uint8TypeDefinition;
import org.opendaylight.yangtools.yang.model.api.type.UnionTypeDefinition;
import org.slf4j.LoggerFactory;

public abstract class TypeDefinitionAwareCodec<J, T extends TypeDefinition<T>> extends AbstractDataStringCodec<J> {
    private static final boolean ENABLE_UNION_CODEC =
        !Boolean.getBoolean("org.opendaylight.yangtools.yang.data.impl.codec.disable-union");

    static {
        if (!ENABLE_UNION_CODEC) {
            LoggerFactory.getLogger(TypeDefinitionAwareCodec.class).info("Support for unions is disabled");
        }
    }

    private final @NonNull Class<J> inputClass;
    private final @NonNull T typeDefinition;

    protected TypeDefinitionAwareCodec(final Class<J> outputClass, final T typeDefinition) {
        inputClass = requireNonNull(outputClass);
        this.typeDefinition = requireNonNull(typeDefinition);
    }

    @Override
    public final Class<J> getInputClass() {
        return inputClass;
    }

    protected final @NonNull T typeDefinition() {
        return typeDefinition;
    }

    @SuppressWarnings("unchecked")
    public static TypeDefinitionAwareCodec<Object, ?> from(final TypeDefinition<?> typeDefinition) {
        return (TypeDefinitionAwareCodec<Object, ?>) fromType(typeDefinition);
    }

    // FIXME: do we want an Optional or a throws instead of @Nullable here?
    public static @Nullable TypeDefinitionAwareCodec<?, ?> fromType(final TypeDefinition<?> typeDefinition) {
        if (typeDefinition instanceof BinaryTypeDefinition binaryType) {
            return BinaryStringCodec.from(binaryType);
        } else if (typeDefinition instanceof BitsTypeDefinition bitsType) {
            return BitsStringCodec.from(bitsType);
        } else if (typeDefinition instanceof BooleanTypeDefinition booleanType) {
            return BooleanStringCodec.from(booleanType);
        } else if (typeDefinition instanceof DecimalTypeDefinition decimalType) {
            return  DecimalStringCodec.from(decimalType);
        } else if (typeDefinition instanceof EmptyTypeDefinition emptyType) {
            return new EmptyStringCodec(emptyType);
        } else if (typeDefinition instanceof EnumTypeDefinition enumType) {
            return EnumStringCodec.from(enumType);
        } else if (typeDefinition instanceof Int8TypeDefinition int8Type) {
            return AbstractIntegerStringCodec.from(int8Type);
        } else if (typeDefinition instanceof Int16TypeDefinition int16Type) {
            return AbstractIntegerStringCodec.from(int16Type);
        } else if (typeDefinition instanceof Int32TypeDefinition int32Type) {
            return AbstractIntegerStringCodec.from(int32Type);
        } else if (typeDefinition instanceof Int64TypeDefinition int64Type) {
            return AbstractIntegerStringCodec.from(int64Type);
        } else if (typeDefinition instanceof StringTypeDefinition stringType) {
            return StringStringCodec.from(stringType);
        } else if (typeDefinition instanceof Uint8TypeDefinition uint8Type) {
            return AbstractIntegerStringCodec.from(uint8Type);
        } else if (typeDefinition instanceof Uint16TypeDefinition uint16Type) {
            return AbstractIntegerStringCodec.from(uint16Type);
        } else if (typeDefinition instanceof Uint32TypeDefinition uint32Type) {
            return AbstractIntegerStringCodec.from(uint32Type);
        } else if (typeDefinition instanceof Uint64TypeDefinition uint64Type) {
            return AbstractIntegerStringCodec.from(uint64Type);
        } else if (ENABLE_UNION_CODEC && typeDefinition instanceof UnionTypeDefinition unionType) {
            return UnionStringCodec.from(unionType);
        } else {
            return null;
        }
    }
}