/*******************************************************************************
 * Copyright (c) 2009 IBM Corporation 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.eclipse.ptp.rdt.server.dstore.core;

import org.eclipse.dstore.core.client.ClientConnection;
import org.eclipse.dstore.core.client.ConnectionStatus;
import org.eclipse.dstore.core.model.DataStore;
import org.eclipse.ptp.rdt.server.dstore.internal.core.DebugUtil;
import org.eclipse.ptp.rdt.server.dstore.messages.Messages;
import org.eclipse.ptp.remote.core.exception.RemoteConnectionException;
import org.eclipse.ptp.remote.launch.core.AbstractRemoteServerRunner;

public class DStoreServer extends AbstractRemoteServerRunner {
	public static String SERVER_ID = "org.eclipse.ptp.rdt.server.dstore.RemoteToolsDStoreServer"; //$NON-NLS-1$

	private enum DStoreState {
		STARTING, WAITING, STARTED
	};

	private DStoreState fState = DStoreState.STARTING;
	private ClientConnection fDStoreConnection = null;
	private int fDStorePort = 0;

	private static final String SUCCESS_STRING = "Server Started Successfully"; //$NON-NLS-1$

	public DStoreServer() {
		super(Messages.DStoreServer_0);
	}

	public DataStore getDataStore() {
		if (fDStoreConnection == null) {
			fDStoreConnection = new ClientConnection(getRemoteConnection().getName());
		}
		return fDStoreConnection.getDataStore();
	}

	/*
	 * (non-Javadoc)
	 * 
	 * @see
	 * org.eclipse.ptp.remote.launch.core.AbstractRemoteServerRunner#doFinishServer
	 * ()
	 */
	@Override
	protected void doFinishServer() {
		if (fDStoreConnection != null) {
			fDStoreConnection.disconnect();
			fDStoreConnection = null;
		}
	}

	/*
	 * (non-Javadoc)
	 * 
	 * @see
	 * org.eclipse.ptp.remote.launch.core.AbstractRemoteServerRunner#doRestartServer
	 * ()
	 */
	@Override
	protected boolean doRestartServer() {
		fState = DStoreState.STARTING;
		return true;
	}

	/*
	 * (non-Javadoc)
	 * 
	 * @see
	 * org.eclipse.ptp.remote.launch.core.AbstractRemoteServerRunner#doStartServer
	 * ()
	 */
	@Override
	protected boolean doStartServer() {
		int port;
		try {
			port = getRemoteConnection().forwardLocalPort("localhost", fDStorePort, null); //$NON-NLS-1$
		} catch (RemoteConnectionException e) {
			if (DebugUtil.SERVER_TRACING) {
				System.err.println(Messages.DStoreServer_1 + e.getLocalizedMessage());
			}
			return false;
		}
		fDStoreConnection.setHost("localhost"); //$NON-NLS-1$
		fDStoreConnection.setPort(Integer.toString(port));
		if (DebugUtil.SERVER_TRACING) {
			System.out.println(Messages.DStoreServer_2);
		}
		ConnectionStatus status = fDStoreConnection.connect(null, 0);
		DataStore dataStore = fDStoreConnection.getDataStore();
		dataStore.showTicket(null);
		dataStore.registerLocalClassLoader(getClass().getClassLoader());
		if (DebugUtil.SERVER_TRACING) {
			System.out.println(Messages.DStoreServer_3);
		}
		return status.isConnected();
	}

	/*
	 * (non-Javadoc)
	 * 
	 * @see org.eclipse.ptp.remote.launch.core.AbstractRemoteServerRunner#
	 * doVerifyServerRunningFromStderr(java.lang.String)
	 */
	@Override
	protected boolean doVerifyServerRunningFromStderr(String output) {
		switch (fState) {
		case STARTING:
			if (output.startsWith(SUCCESS_STRING)) {
				fState = DStoreState.WAITING;
			}
			break;

		case WAITING:
			if (output.matches("^[0-9]+$")) { //$NON-NLS-1$
				fDStorePort = Integer.parseInt(output);
				fState = DStoreState.STARTED;
			}
			break;

		case STARTED:
			return true;
		}

		return false;
	}

	/*
	 * (non-Javadoc)
	 * 
	 * @see org.eclipse.ptp.remote.launch.core.AbstractRemoteServerRunner#
	 * doVerifyServerRunningFromStdout(java.lang.String)
	 */
	@Override
	protected boolean doVerifyServerRunningFromStdout(String output) {
		return false;
	}
}