L2AttachCommand.java

package org.opennaas.extensions.network.capability.basic.shell;

import java.util.Iterator;
import java.util.List;

import org.apache.felix.gogo.commands.Argument;
import org.apache.felix.gogo.commands.Command;
import org.opennaas.core.resources.IResource;
import org.opennaas.core.resources.capability.CapabilityException;
import org.opennaas.core.resources.shell.GenericKarafCommand;
import org.opennaas.extensions.network.capability.basic.INetworkBasicCapability;
import org.opennaas.extensions.network.model.NetworkModel;
import org.opennaas.extensions.network.model.NetworkModelHelper;
import org.opennaas.extensions.network.model.topology.Interface;

@Command(scope = "net", name = "l2attach", description = "Add a resource to the network")
public class L2AttachCommand extends GenericKarafCommand {

	@Argument(index = 0, name = "network:networkName", description = "The network where to add a resource", required = true, multiValued = false)
	private String	networkId;

	@Argument(index = 1, name = "interfaceName", description = "First interface to attach", required = true, multiValued = false)
	private String	interface1Name;

	@Argument(index = 2, name = "interfaceName", description = "Second interface to attach", required = true, multiValued = false)
	private String	interface2Name;

	@Override
	protected Object doExecute() throws Exception {

		IResource network;
		try {
			network = getResourceFromFriendlyName(networkId);
			if (network == null) {
				printError("Could not load network. Please check network name is correct.");
			}
		} catch (Exception e) {
			printError("Failed to get required resources: " + e.getLocalizedMessage());
			printEndCommand();
			return null;
		}

		// get interfaces
		NetworkModel netModel = (NetworkModel) network.getModel();
		List<Interface> interfaces = NetworkModelHelper.getInterfaces(netModel.getNetworkElements());

		Iterator<Interface> it = interfaces.iterator();
		boolean found = false;
		Interface interface1 = null;
		Interface interface2 = null;
		while (it.hasNext() && !found) {
			Interface current = it.next();
			if (interface1Name.equals(current.getName())) {
				interface1 = current;
			}
			if (interface2Name.equals(current.getName())) {
				interface2 = current;
			}
			found = (interface1 != null) && (interface2 != null);
		}
		if (!found) {
			if (interface1 == null)
				printError("Failed to get required interfaces: " + interface1Name + " is not prensent in network model");
			else
				printError("Failed to get required interfaces: " + interface2Name + " is not prensent in network model");
			printEndCommand();
			return null;
		}

		INetworkBasicCapability networkCapability =
				(INetworkBasicCapability) network.getCapabilityByInterface(INetworkBasicCapability.class);

		try {
			networkCapability.l2attach(interface1, interface2);
		} catch (CapabilityException e) {
			printError("Error during attach");
			printError(e);
			printEndCommand();
			return null;
		}

		printEndCommand();
		return null;
	}
}