/*
 * Copyright (c) 2014 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.groupbasedpolicy.renderer.opflex.mit;

import java.math.BigInteger;
import java.util.ArrayList;
import java.util.List;

import org.opendaylight.groupbasedpolicy.renderer.opflex.lib.messages.ManagedObject;
import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.Uri;
import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev100924.MacAddress;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.fasterxml.jackson.databind.DeserializationFeature;
import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.fasterxml.jackson.databind.annotation.JsonSerialize;
import com.fasterxml.jackson.databind.node.ArrayNode;
import com.fasterxml.jackson.databind.node.JsonNodeFactory;
import com.fasterxml.jackson.databind.node.ObjectNode;

public class MitLib {

    protected static final Logger LOG = LoggerFactory.getLogger(MitLib.class);
    private ObjectMapper objectMapper;
	private JsonNodeFactory jnf;

	@JsonSerialize
	public static class Reference {
		String subject;
		String reference_uri;
		public String getSubject() {
			return subject;
		}
		public void setSubject(String subject) {
			this.subject = subject;
		}
		public String getReference_uri() {
			return reference_uri;
		}
		public void setReference_uri(String reference_uri) {
			this.reference_uri = reference_uri;
		}

	}

	public MitLib() {
		 objectMapper = new ObjectMapper();
		 objectMapper.configure(
		            DeserializationFeature.FAIL_ON_UNKNOWN_PROPERTIES, false);
		 jnf = objectMapper.getNodeFactory();
	}

	public static final String REFERENCE_SUBJECT = "subject";
	public static final String REFERENCE_URI = "reference_uri";

	public BigInteger deserializeMoPropertyEnum(JsonNode node, PolicyPropertyInfo ppi) {

		EnumInfo ei = ppi.getEnumInfo();
		return  ei.getEnumValue(node.asText());
	}

	public String serializeMoPropertyEnum(PolicyPropertyInfo ppi, PolicyObjectInstance poi) {

		EnumInfo ei = ppi.getEnumInfo();
		return ei.getEnumValue(poi.getUint64(ppi.getPropId()));
	}

	/**
	 * Deserialize a REFERENCE property
	 *
	 * @param node
	 * @return
	 */
	public PolicyObjectInstance.PolicyReference deserializeMoPropertyRef(JsonNode node, OpflexMit mit) {

		JsonNode sn = node.findValue(REFERENCE_SUBJECT);
		if (sn == null) return null;
		JsonNode un = node.findValue(REFERENCE_URI);
		if (un == null) return null;

		PolicyClassInfo pci = mit.getClass(sn.asText());
		if (pci == null) return null;

		return new PolicyObjectInstance.PolicyReference(pci.getClassId(), new Uri(un.asText()));

	}

	/**
	 * Serialize a REFERENCE property
	 *
	 * @param reference
	 * @param mit
	 * @return
	 */
	public ObjectNode serializeMoPropertyRef(PolicyObjectInstance.PolicyReference reference, OpflexMit mit) {

		ObjectNode on = jnf.objectNode();
	    PolicyClassInfo pci = mit.getClass(reference.getClassId());
	    if (pci == null) return null;
	    PolicyUri puri = new PolicyUri(reference.getUri().getValue());
	    // walk our way up until we find a valid class
	    String identifier = puri.pop();
	    while (mit.getClass(identifier) == null) {
		    identifier = puri.pop();
	    }
		on.put(REFERENCE_SUBJECT, identifier);
		on.put(REFERENCE_URI, reference.getUri().getValue());
	    return on;
	}

	/**
	 * Take the {@link ManagedObject} and deserialize the properties
	 * into a concrete type to be used by the renderer. It also
	 * adds URIs for any children that are referenced in the
	 * properties to the MO's "children" array.
	 *
	 * @param mo
	 */
	public PolicyObjectInstance deserializeMoProperties(ManagedObject mo, OpflexMit mit) {

		/*
		 * The subject gives us the class to use for the schema
		 */
		PolicyClassInfo pci = mit.getClass(mo.getSubject());

		// sanity checks
		if (pci == null) return null;

		PolicyObjectInstance poi = new PolicyObjectInstance(pci.getClassId());

		if (mo.getProperties() == null) return poi;

		for (ManagedObject.Property prop : mo.getProperties()) {
			PolicyPropertyInfo ppi = pci.getProperty(prop.getName());
			if ((ppi == null) || (prop.getData() == null)) continue;

			JsonNode node = prop.getData();

			boolean vectored = false;
			if (ppi.getPropCardinality().equals(PolicyPropertyInfo.PropertyCardinality.VECTOR)) {
				vectored = true;
			}
			switch(ppi.getType()) {
			case STRING:
				if (vectored == true) {
					if (!node.isArray()) continue;

					List<String> sl = new ArrayList<String>();
					for (int i=0; i <node.size(); i++ ) {
						JsonNode jn = node.get(i);
						if (!jn.isTextual()) continue;

						sl.add(jn.asText());
					}
					poi.setString(ppi.getPropId(), sl);
				}
				else {
					if (!node.isTextual()) continue;

					poi.setString(ppi.getPropId(), node.asText());
				}
				break;

			case U64:
				if (vectored == true) {
					if (!node.isArray()) continue;

					List<BigInteger> bil = new ArrayList<BigInteger>();

					for (int i=0; i <node.size(); i++ ) {
						JsonNode jn = node.get(i);
						if (!jn.isBigInteger()) continue;

						bil.add(new BigInteger(jn.asText()));
					}
					poi.setUint64(ppi.getPropId(), bil);
				}
				else {
					if (!node.isBigInteger()) continue;

					poi.setUint64(ppi.getPropId(), new BigInteger(node.asText()));
				}
				break;

			case S64:
				if (vectored == true) {
					if (!node.isArray()) continue;

					List<Long> ll = new ArrayList<Long>();

					for (int i=0; i <node.size(); i++ ) {
						JsonNode jn = node.get(i);

						if (!jn.isBigInteger()) continue;

						ll.add(jn.asLong());
					}
					poi.setInt64(ppi.getPropId(), ll);
				}
				else {
					if (!node.isBigInteger()) continue;

					poi.setInt64(ppi.getPropId(), node.asLong());
				}
				break;

			case REFERENCE:
				if (vectored == true) {
					if (!node.isArray()) continue;

					List<PolicyObjectInstance.PolicyReference> prl =
							new ArrayList<PolicyObjectInstance.PolicyReference>();

					for (int i=0; i <node.size(); i++ ) {
						JsonNode jn = node.get(i);
						PolicyObjectInstance.PolicyReference pr = deserializeMoPropertyRef(jn, mit);
						if (pr == null) continue;

						prl.add(pr);
					}
					poi.setReference(ppi.getPropId(), prl);
				}
				else {
					PolicyObjectInstance.PolicyReference pr = deserializeMoPropertyRef(node, mit);
					if (pr == null) continue;

					poi.setReference(ppi.getPropId(), pr);
				}
				break;

			case ENUM8:
			case ENUM16:
			case ENUM32:
			case ENUM64:
				if (vectored == true) {
					if (!node.isArray()) continue;

					List<BigInteger> bil = new ArrayList<BigInteger>();

					for (int i=0; i <node.size(); i++ ) {
						JsonNode jn = node.get(i);
						if (!jn.isTextual()) continue;
						BigInteger bi = deserializeMoPropertyEnum(node,ppi);

						bil.add(bi);
					}
					poi.setUint64(ppi.getPropId(), bil);
				}
				else {
					if (!node.isTextual()) continue;

					BigInteger bi = deserializeMoPropertyEnum(node,ppi);
					poi.setUint64(ppi.getPropId(), bi);
				}
				break;

			case MAC:
				if (vectored == true) {
					if (!node.isArray()) continue;

					List<MacAddress> ml = new ArrayList<MacAddress>();

					for (int i=0; i <node.size(); i++ ) {
						JsonNode jn = node.get(i);

						if (!jn.isTextual()) continue;

						ml.add(new MacAddress(jn.asText()));
					}
					poi.setMacAddress(ppi.getPropId(), ml);
				}
				else {
					if (!node.isTextual()) continue;

					poi.setMacAddress(ppi.getPropId(), new MacAddress(node.asText()));
				}
				break;

			case COMPOSITE:

			}

		}

		return poi;
	}

	/**
	 * Serialize the properties contained in the {@link PolicyObjectInstance}
	 * into the properties field of the {@link ManagedObject}
	 *
	 * @param pci
	 * @param poi
	 * @param mo
	 * @param mit
	 */
	public void serializeMoProperties(PolicyClassInfo pci, PolicyObjectInstance poi, ManagedObject mo, OpflexMit mit) {

		List<PolicyPropertyInfo> ppil = pci.getProperties();
		if (ppil == null) return;

		List<ManagedObject.Property> pl = new ArrayList<ManagedObject.Property>();

		/*
		 * For serialization of values, we can cheat a bit,
		 * as the native "toString" method gives us exactly
		 * the formatting we need (including vectors).
		 */
		for (PolicyPropertyInfo ppi: ppil) {
			/*
			 * Skip any properties that aren't populated for this
			 * object instance
			 */
			if (ppi.getType() != PolicyPropertyInfo.PropertyType.COMPOSITE &&
				!poi.isSet(ppi.getPropId(), ppi.getType(), ppi.getPropCardinality())) {
				continue;
			}
			ManagedObject.Property p = null;
			boolean scalar = true;

			if (ppi.getPropCardinality() == PolicyPropertyInfo.PropertyCardinality.VECTOR) {
				scalar = false;
			}

			switch (ppi.getType()) {
			case STRING:
				p = new ManagedObject.Property();
				p.setName(ppi.getPropName());
				if (scalar == true) {
					JsonNode jn = jnf.textNode(poi.getString(ppi.getPropId()));
					p.setData(jn);
				}
				else {
					int len = poi.getStringSize(ppi.getPropId());
					ArrayNode an = jnf.arrayNode();
					for (int i = 0; i < len; i++ ) {
						an.add(poi.getString(ppi.getPropId(), i));
					}
					p.setData(an);
				}
				break;

			case S64:
				p = new ManagedObject.Property();
				p.setName(ppi.getPropName());
				if (scalar == true) {
					JsonNode jn = jnf.numberNode(poi.getInt64(ppi.getPropId()));
					p.setData(jn);
				}
				else {
					int len = poi.getInt64Size(ppi.getPropId());
					ArrayNode an = jnf.arrayNode();
					for (int i = 0; i < len; i++ ) {
						an.add(Long.valueOf(poi.getInt64(ppi.getPropId(), i)));
					}
					p.setData(an);
				}
				break;

			case ENUM8:
			case ENUM16:
			case ENUM32:
			case ENUM64:
				p = new ManagedObject.Property();
				p.setName(ppi.getPropName());
				if (scalar == true) {
					JsonNode jn = jnf.textNode(serializeMoPropertyEnum(ppi, poi));
					p.setData(jn);
				}
				else {
					int len = poi.getUint64Size(ppi.getPropId());
					ArrayNode an = jnf.arrayNode();
					for (int i = 0; i < len; i++ ) {
						an.add(serializeMoPropertyEnum(ppi, poi));
					}
					p.setData(an);
				}
				break;

			case U64:
				p = new ManagedObject.Property();
				p.setName(ppi.getPropName());
				if (scalar == true) {
					JsonNode jn = jnf.numberNode(poi.getUint64(ppi.getPropId()));
					p.setData(jn);
				}
				else {
					int len = poi.getUint64Size(ppi.getPropId());
					ArrayNode an = jnf.arrayNode();
					for (int i = 0; i < len; i++ ) {
						an.numberNode(poi.getUint64(ppi.getPropId()));
					}
					p.setData(an);
				}
				break;

			case MAC:
				p = new ManagedObject.Property();
				p.setName(ppi.getPropName());
				if (scalar == true) {
					MacAddress mac = poi.getMacAddress(ppi.getPropId());
					JsonNode jn = jnf.textNode(mac.getValue().toString());
					p.setData(jn);
				}
				else {
					int len = poi.getMacAddressSize(ppi.getPropId());
					ArrayNode an = jnf.arrayNode();
					for (int i = 0; i < len; i++ ) {
						MacAddress mac = poi.getMacAddress(ppi.getPropId());
						an.add(mac.getValue().toString());
					}
					p.setData(an);
				}
				break;

			case REFERENCE:
				p = new ManagedObject.Property();
				p.setName(ppi.getPropName());
				if (scalar == true) {
					ObjectNode on = serializeMoPropertyRef(poi.getReference(ppi.getPropId()), mit);
					p.setData(on);
				}
				else {
					int len = poi.getReferenceSize(ppi.getPropId());
					ArrayNode an = jnf.arrayNode();
					for (int i = 0; i < len; i++ ) {
						ObjectNode on = serializeMoPropertyRef(poi.getReference(ppi.getPropId(), i), mit);
						an.add(on);
					}
					p.setData(an);
				}
				break;

			case COMPOSITE:
				/*
				 * Get the URI to add to the list of children
				 */
				break;

				default:

			}
			if (p != null) {
				pl.add(p);
			}
		}

		mo.setProperties(pl);
	}
}