/*
 * 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.protocol.pcep.ietf.stateful07;

import org.opendaylight.protocol.pcep.impl.object.PCEPOpenObjectParser;
import org.opendaylight.protocol.pcep.spi.TlvRegistry;
import org.opendaylight.protocol.util.ByteArray;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.Tlvs1;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.Tlvs1Builder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.stateful.capability.tlv.Stateful;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.Tlv;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.open.object.Open;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.open.object.open.Tlvs;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.open.object.open.TlvsBuilder;

/**
 * Parser for {@link Open}
 */
public class Stateful07OpenObjectParser extends PCEPOpenObjectParser {

	public Stateful07OpenObjectParser(final TlvRegistry tlvReg) {
		super(tlvReg);
	}

	@Override
	public void addTlv(final TlvsBuilder tbuilder, final Tlv tlv) {
		super.addTlv(tbuilder, tlv);
		final Tlvs1Builder statefulBuilder = new Tlvs1Builder();
		if (tbuilder.getAugmentation(Tlvs1.class) != null) {
			final Tlvs1 t = tbuilder.getAugmentation(Tlvs1.class);
			if (t.getStateful() != null) {
				statefulBuilder.setStateful(t.getStateful());
			}
		}
		if (tlv instanceof Stateful) {
			statefulBuilder.setStateful((Stateful) tlv);
		}
		tbuilder.addAugmentation(Tlvs1.class, statefulBuilder.build());
	}

	@Override
	public byte[] serializeTlvs(final Tlvs tlvs) {
		if (tlvs == null) {
			return new byte[0];
		}
		final byte[] prev = super.serializeTlvs(tlvs);
		int finalLength = prev.length;
		byte[] ofListBytes = null;
		byte[] statefulBytes = null;
		if (tlvs.getOfList() != null) {
			ofListBytes = serializeTlv(tlvs.getOfList());
			finalLength += ofListBytes.length;
		}
		if (tlvs.getAugmentation(Tlvs1.class) != null) {
			final Tlvs1 statefulTlvs = tlvs.getAugmentation(Tlvs1.class);
			if (statefulTlvs.getStateful() != null) {
				statefulBytes = serializeTlv(statefulTlvs.getStateful());
				finalLength += statefulBytes.length;
			}
		}
		final byte[] result = new byte[finalLength];
		ByteArray.copyWhole(prev, result, 0);
		int offset = prev.length;
		if (ofListBytes != null) {
			ByteArray.copyWhole(ofListBytes, result, offset);
			offset += ofListBytes.length;
		}
		if (statefulBytes != null) {
			ByteArray.copyWhole(statefulBytes, result, offset);
			offset += statefulBytes.length;
		}
		return result;
	}
}