/*******************************************************************************
 * Copyright (c) 2012 GigaSpaces Technologies Ltd. All rights reserved
 *
 * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with
 * the License. You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
 * an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
 * specific language governing permissions and limitations under the License.
 *******************************************************************************/

package org.cloudifysource.esc.installer.remoteExec;

import java.util.concurrent.TimeUnit;
import java.util.concurrent.TimeoutException;
import java.util.logging.Level;

import org.apache.tools.ant.BuildException;
import org.apache.tools.ant.ExitStatusException;
import org.apache.tools.ant.taskdefs.optional.testing.BuildTimeoutException;
import org.cloudifysource.domain.cloud.ScriptLanguages;
import org.cloudifysource.esc.installer.AgentlessInstaller;
import org.cloudifysource.esc.installer.EnvironmentFileBuilder;
import org.cloudifysource.esc.installer.InstallationDetails;
import org.cloudifysource.esc.installer.InstallerException;
import org.cloudifysource.esc.util.Utils;

/*********
 * Executor implementation for SSH remote calls. Uses Ant ssh task.
 *
 * @author barakme
 * @since 2.5.0
 */
public class SshExecutor implements RemoteExecutor {
	
	private static final int CUSTOM_ERR_CODE = 255;

	private static final java.util.logging.Logger logger =
			java.util.logging.Logger.getLogger(SshExecutor.class.getName());

	@Override
	public void execute(final String targetHost, final InstallationDetails details, final String scriptPath,
			final long endTimeMillis)
			throws InstallerException, TimeoutException, InterruptedException {

		final String fullCommand =
				(details.getScriptLanguage() == ScriptLanguages.LINUX_SHELL ? "chmod +x " + scriptPath + ";"
						+ scriptPath : ("cmd.exe /c " + EnvironmentFileBuilder.normalizeCygwinPath(scriptPath))
						.replace("\\", "\\\\"));

		// TODO - replace Ant based ssh command implementation with sshj
		try {
            logger.fine("Executing command '" + fullCommand + "' on target host " + targetHost);
			Utils.executeSSHCommand(targetHost, fullCommand, details.getUsername(), details.getPassword(),
					details.getKeyFile(), endTimeMillis - System.currentTimeMillis(), TimeUnit.MILLISECONDS);
		} catch (final BuildException e) {
			// There really should be a better way to check that this is a
			// timeout
			logger.log(Level.FINE, "The SSH execution failed with error: " + e.getMessage()
					+ ". The command that failed to execute is : " + fullCommand, e);

			if (e instanceof BuildTimeoutException) {
				final TimeoutException ex =
						new TimeoutException("SSH execution failed: " + e.getMessage());
				ex.initCause(e);
				throw ex;
			} else if (e instanceof ExitStatusException) {
				final ExitStatusException ex = (ExitStatusException) e;
				final int ec = ex.getStatus();

				if (ec == 0 || ec == CUSTOM_ERR_CODE) {
					throw new InstallerException("SSH execution failed with exit code: " + ec, e);	
				} else {
					throw new InstallerException("SSH execution failed with exit code: " + ec + ", message: " 
							+ BootstrapScriptErrors.getMessageByErrorCode(ec), e);
				}
			} else {
				throw new InstallerException("SSH execution failed.", e);
			}
		}

	}

	@Override
	public void initialize(final AgentlessInstaller installer, final InstallationDetails details) {
	}

}