UpdateApplicationCommand.java

/*******************************************************************************
 * Copyright (c) 2014 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
 * 
 * Contributors:
 * IBM Corporation - initial API and implementation
 *******************************************************************************/
package org.eclipse.orion.server.cf.commands;

import java.net.URI;

import javax.servlet.http.HttpServletResponse;

import org.apache.commons.httpclient.methods.PutMethod;
import org.apache.commons.httpclient.methods.StringRequestEntity;
import org.eclipse.core.runtime.IStatus;
import org.eclipse.core.runtime.Status;
import org.eclipse.core.runtime.URIUtil;
import org.eclipse.orion.server.cf.CFProtocolConstants;
import org.eclipse.orion.server.cf.manifest.v2.InvalidAccessException;
import org.eclipse.orion.server.cf.manifest.v2.ManifestParseTree;
import org.eclipse.orion.server.cf.manifest.v2.utils.ManifestUtils;
import org.eclipse.orion.server.cf.objects.App;
import org.eclipse.orion.server.cf.objects.Target;
import org.eclipse.orion.server.cf.utils.HttpUtil;
import org.eclipse.orion.server.core.ServerStatus;
import org.eclipse.osgi.util.NLS;
import org.json.JSONObject;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public class UpdateApplicationCommand extends AbstractCFCommand {
	private final Logger logger = LoggerFactory.getLogger("org.eclipse.orion.server.cf"); //$NON-NLS-1$

	private String commandName;
	private App application;

	private String appName;
	private String appCommand;
	private int appInstances;
	private int appMemory;
	private String buildPack;
	private String stack;
	private JSONObject env;

	public UpdateApplicationCommand(Target target, App app) {
		super(target);
		this.commandName = "Update application parameters";
		this.application = app;
	}

	@Override
	protected ServerStatus _doIt() {
		try {
			// get stack object
			Object stackId = JSONObject.NULL;
			if (stack != null) {
				GetStackByNameCommand getStackCommand = new GetStackByNameCommand(target, stack);
				ServerStatus getStackStatus = (ServerStatus) getStackCommand.doIt();
				if (!getStackStatus.isOK())
					return getStackStatus;
				if (getStackCommand.getStack() == null)
					return new ServerStatus(IStatus.ERROR, HttpServletResponse.SC_BAD_REQUEST, NLS.bind("Stack {0} not found", stack), null);
				stackId = getStackCommand.getStack().getGuid();
			}

			/* get application URL */
			URI targetURI = URIUtil.toURI(target.getUrl());
			URI appURI = targetURI.resolve(application.getAppJSON().getString(CFProtocolConstants.V2_KEY_URL));

			PutMethod updateApplicationMethod = new PutMethod(appURI.toString());
			ServerStatus confStatus = HttpUtil.configureHttpMethod(updateApplicationMethod, target.getCloud());
			if (!confStatus.isOK())
				return confStatus;

			updateApplicationMethod.setQueryString("async=true&inline-relations-depth=1"); //$NON-NLS-1$

			/* set request body */
			JSONObject updateAppRequest = new JSONObject();
			updateAppRequest.put(CFProtocolConstants.V2_KEY_NAME, appName);
			updateAppRequest.put(CFProtocolConstants.V2_KEY_INSTANCES, appInstances);
			updateAppRequest.put(CFProtocolConstants.V2_KEY_COMMAND, appCommand);
			updateAppRequest.put(CFProtocolConstants.V2_KEY_MEMORY, appMemory);
			updateAppRequest.put(CFProtocolConstants.V2_KEY_ENVIRONMENT_JSON, env != null ? env : new JSONObject());
			updateAppRequest.put(CFProtocolConstants.V2_KEY_BUILDPACK, buildPack != null ? buildPack : JSONObject.NULL);
			updateAppRequest.put(CFProtocolConstants.V2_KEY_STACK_GUID, stackId);

			updateApplicationMethod.setRequestEntity(new StringRequestEntity(updateAppRequest.toString(), "application/json", "utf-8")); //$NON-NLS-1$ //$NON-NLS-2$

			ServerStatus status = HttpUtil.executeMethod(updateApplicationMethod);
			if (!status.isOK())
				return status;
			GetAppCommand.expire(target, application.getName());
			return status;

		} catch (Exception e) {
			String msg = NLS.bind("An error occured when performing operation {0}", commandName); //$NON-NLS-1$
			logger.error(msg, e);
			return new ServerStatus(IStatus.ERROR, HttpServletResponse.SC_INTERNAL_SERVER_ERROR, msg, e);
		}
	}

	@Override
	protected IStatus validateParams() {
		try {

			/* read deploy parameters */
			ManifestParseTree manifest = application.getManifest();
			ManifestParseTree app = manifest.get(CFProtocolConstants.V2_KEY_APPLICATIONS).get(0);

			if (application.getName() != null)
				appName = application.getName();
			else
				appName = app.get(CFProtocolConstants.V2_KEY_NAME).getValue();

			ManifestParseTree commandNode = app.getOpt(CFProtocolConstants.V2_KEY_COMMAND);
			appCommand = (commandNode != null) ? commandNode.getValue() : ""; //$NON-NLS-1$

			ManifestParseTree instancesNode = app.getOpt(CFProtocolConstants.V2_KEY_INSTANCES);
			appInstances = (instancesNode != null) ? Integer.parseInt(instancesNode.getValue()) : 1;

			/* look for v2 memory property first */
			ManifestParseTree memoryNode = app.getOpt(CFProtocolConstants.V2_KEY_MEMORY);
			if (memoryNode != null)
				appMemory = ManifestUtils.normalizeMemoryMeasure(memoryNode.getValue());
			else {
				memoryNode = app.getOpt(CFProtocolConstants.V6_KEY_MEMORY);
				if (memoryNode != null)
					appMemory = ManifestUtils.normalizeMemoryMeasure(memoryNode.getValue());
				else
					/* default memory value */
					appMemory = 1024;
			}

			ManifestParseTree buildpackNode = app.getOpt(CFProtocolConstants.V2_KEY_BUILDPACK);
			buildPack = (buildpackNode != null) ? buildpackNode.getValue() : null;

			ManifestParseTree stackNode = app.getOpt(CFProtocolConstants.V2_KEY_STACK);
			stack = (stackNode != null) ? stackNode.getValue() : null;

			/* look for environment variables */
			ManifestParseTree envNode = manifest.getOpt(CFProtocolConstants.V2_KEY_ENV);
			if (envNode != null) {
				env = new JSONObject();
				for (ManifestParseTree var : envNode.getChildren())
					env.put(var.getLabel(), var.getValue());
			}

			/* look for environment variables in app */
			ManifestParseTree appEnvNode = app.getOpt(CFProtocolConstants.V2_KEY_ENV);
			if (appEnvNode != null) {
				env = new JSONObject();
				for (ManifestParseTree var : appEnvNode.getChildren())
					env.put(var.getLabel(), var.getValue());
			}

			return Status.OK_STATUS;

		} catch (InvalidAccessException e) {
			return new ServerStatus(IStatus.ERROR, HttpServletResponse.SC_BAD_REQUEST, e.getMessage(), null);
		} catch (Exception e) {
			String msg = NLS.bind("An error occured when performing operation {0}", commandName); //$NON-NLS-1$
			logger.error(msg, e);
			return new ServerStatus(IStatus.ERROR, HttpServletResponse.SC_INTERNAL_SERVER_ERROR, msg, e);
		}
	}
}