package org.opendaylight.controller.cluster.datastore.node;

import org.opendaylight.controller.protobuff.messages.common.NormalizedNodeMessages;
import org.opendaylight.controller.cluster.datastore.node.utils.PathUtils;
import org.opendaylight.yangtools.yang.common.QName;
import org.opendaylight.yangtools.yang.data.api.InstanceIdentifier;
import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
import org.opendaylight.yangtools.yang.model.api.SchemaContext;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public class NormalizedNodeToNodeCodec {
    private final SchemaContext ctx;
    private static final Logger logger = LoggerFactory.getLogger(NormalizedNodeToNodeCodec.class);

    public NormalizedNodeToNodeCodec(final SchemaContext ctx){
        this.ctx = ctx;

    }

    public NormalizedNodeMessages.Container encode(InstanceIdentifier id, NormalizedNode node){
        String parentPath = "";

        if(id != null){
            parentPath = PathUtils.getParentPath(id.toString());
        }


        NormalizedNodeToProtocolBufferNode encoder = new NormalizedNodeToProtocolBufferNode();
        encoder.encode(parentPath, node);

        return encoder.getContainer();


    }

    public NormalizedNode<?,?> decode(InstanceIdentifier id, NormalizedNodeMessages.Node node){
            NodeToNormalizedNodeBuilder currentOp = NodeToNormalizedNodeBuilder.from(ctx);

            for(InstanceIdentifier.PathArgument pathArgument : id.getPath()){
                currentOp = currentOp.getChild(pathArgument);
            }

            QName nodeType = null;

            if(id.getPath().size() < 1){
                nodeType = null;
            } else {
                final InstanceIdentifier.PathArgument pathArgument = id.getPath().get(id.getPath().size() - 1);
                if(pathArgument instanceof InstanceIdentifier.AugmentationIdentifier){
                    nodeType = null;
                } else {
                    nodeType = pathArgument.getNodeType();
                }
            }
            try {
                if((node != null)&& (!node.getType().isEmpty())){
                   return currentOp.normalize(nodeType, node);
                }else{
                  return null;
               }
            } catch(RuntimeException e){
                throw e;
        }
    }


}