RootMenu.java

package org.hive2hive.client.menu;

import org.hive2hive.client.console.H2HConsoleMenu;
import org.hive2hive.client.console.H2HConsoleMenuItem;
import org.hive2hive.client.util.ConsoleFileAgent;
import org.hive2hive.client.util.MenuContainer;
import org.hive2hive.core.api.interfaces.IUserManager;
import org.hive2hive.core.exceptions.NoPeerConnectionException;
import org.hive2hive.core.security.UserCredentials;
import org.hive2hive.processframework.exceptions.InvalidProcessStateException;
import org.hive2hive.processframework.exceptions.ProcessExecutionException;
import org.hive2hive.processframework.interfaces.IProcessComponent;

public final class RootMenu extends H2HConsoleMenu {

	public RootMenu(MenuContainer menus) {
		super(menus);
	}

	@Override
	protected void addMenuItems() {
		add(new H2HConsoleMenuItem("Connect") {
			protected void execute() {
				menus.getNodeMenu().open();
			}
		});

		add(new H2HConsoleMenuItem("Login") {
			protected boolean checkPreconditions() throws NoPeerConnectionException, InvalidProcessStateException,
					InterruptedException {
				if (!menus.getNodeMenu().createNetwork()) {
					printAbortion(displayText, "Node not connected.");
					return false;
				}
				if (!menus.getUserMenu().createUserCredentials()) {
					printAbortion(displayText, "User credentials not specified.");
					return false;
				}
				if (!menus.getUserMenu().createRootDirectory()) {
					printAbortion(displayText, "Root directory not specified.");
					return false;
				}
				if (!register()) {
					printAbortion(displayText, "Registering failed.");
					return false;
				}
				return true;
			}

			protected void execute() throws NoPeerConnectionException, InterruptedException, InvalidProcessStateException {
				ConsoleFileAgent fileAgent = new ConsoleFileAgent(menus.getUserMenu().getRootDirectory());
				IProcessComponent<Void> loginProcess = menus.getNodeMenu().getNode().getUserManager()
						.createLoginProcess(menus.getUserMenu().getUserCredentials(), fileAgent);

				try {
					loginProcess.execute();
				} catch (ProcessExecutionException e) {
					menus.getUserMenu().reset();
					menus.getFileMenu().reset();
				}
			}
		});

		add(new H2HConsoleMenuItem("Logout") {
			protected boolean checkPreconditions() throws Exception {
				return checkLogin();
			}

			protected void execute() throws Exception {
				IProcessComponent<Void> logoutProcess = menus.getNodeMenu().getNode().getUserManager().createLogoutProcess();
				logoutProcess.execute();
				menus.getUserMenu().clearCredentials();
			}
		});

		add(new H2HConsoleMenuItem("File Menu") {
			@Override
			protected boolean checkPreconditions() throws Exception {
				return checkLogin();
			}

			protected void execute() throws Exception {
				menus.getFileMenu().open();
			}
		});
	}

	@Override
	public String getInstruction() {
		return "Please select an option:";
	}

	private boolean register() throws NoPeerConnectionException, InvalidProcessStateException, InterruptedException {
		IUserManager userManager = menus.getNodeMenu().getNode().getUserManager();
		UserCredentials userCredentials = menus.getUserMenu().getUserCredentials();

		if (userManager.isRegistered(userCredentials.getUserId())) {
			return true;
		} else {
			H2HConsoleMenuItem
					.printPrecondition("You are not registered to the network. This will now happen automatically.");
			IProcessComponent<Void> registerProcess = userManager.createRegisterProcess(userCredentials);
			try {
				registerProcess.execute();
				return true;
			} catch (ProcessExecutionException e) {
				return false;
			}
		}
	}

	private boolean checkLogin() throws NoPeerConnectionException {

		if (menus.getNodeMenu().getNode() == null) {
			H2HConsoleMenuItem.printPrecondition("You are not logged in. Node is not connected to a network.");
			return false;
		}
		if (menus.getUserMenu().getUserCredentials() == null) {
			H2HConsoleMenuItem.printPrecondition("You are not logged in. No user credentials specified.");
			return false;
		}
		if (!menus.getNodeMenu().getNode().getUserManager().isLoggedIn()) {
			H2HConsoleMenuItem.printPrecondition("You are not logged in.");
			return false;
		}
		return true;
	}
}