/*
*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.apache.airavata.client;
import java.net.MalformedURLException;
import java.net.URI;
import java.net.URL;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Observable;
import javax.jcr.RepositoryException;
import org.apache.airavata.client.api.AiravataAPI;
import org.apache.airavata.client.api.AiravataAPIInvocationException;
import org.apache.airavata.client.api.AiravataManager;
import org.apache.airavata.client.api.ApplicationManager;
import org.apache.airavata.client.api.ExecutionManager;
import org.apache.airavata.client.api.ProvenanceManager;
import org.apache.airavata.client.api.UserManager;
import org.apache.airavata.client.api.WorkflowManager;
import org.apache.airavata.client.api.builder.DescriptorBuilder;
import org.apache.airavata.client.impl.AiravataManagerImpl;
import org.apache.airavata.client.impl.ApplicationManagerImpl;
import org.apache.airavata.client.impl.ExecutionManagerImpl;
import org.apache.airavata.client.impl.PasswordCallBackImpl;
import org.apache.airavata.client.impl.ProvenanceManagerImpl;
import org.apache.airavata.client.impl.UserManagerImpl;
import org.apache.airavata.client.impl.WorkflowManagerImpl;
import org.apache.airavata.common.exception.AiravataConfigurationException;
import org.apache.airavata.common.utils.Version;
import org.apache.airavata.common.workflow.execution.context.WorkflowContextHeaderBuilder;
import org.apache.airavata.registry.api.AiravataRegistry2;
import org.apache.airavata.registry.api.AiravataRegistryFactory;
import org.apache.airavata.registry.api.AiravataUser;
import org.apache.airavata.registry.api.Gateway;
import org.apache.airavata.registry.api.PasswordCallback;
import org.apache.airavata.registry.api.exception.RegistryException;
import org.apache.airavata.ws.monitor.MonitorConfiguration;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
public class AiravataClient extends Observable implements AiravataAPI {
private static final Logger log = LoggerFactory
.getLogger(AiravataClient.class);
public static final String REGISTRY = "JCR";
public static final String GFAC = "gfac";
public static final String PROXYSERVER = "proxyserver";
public static final String MSGBOX = "msgbox";
public static final String BROKER = "broker";
public static final String DEFAULT_GFAC_URL = "gfac.url";
public static final String DEFAULT_MYPROXY_SERVER = "myproxy.url";
public static final String DEFAULT_MESSAGE_BOX_URL = "messagebox.url";
public static final String DEFAULT_BROKER_URL = "messagebroker.url";
public static final String MYPROXYUSERNAME = "myproxy.username";
public static final String MYPROXYPASS = "myproxy.password";
public static final String WITHLISTENER = "with.Listener";
public static final String WORKFLOWSERVICEURL = "xbaya.service.url";
public static final String TRUSTED_CERT_LOCATION = "trusted.cert.location";
private AiravataClientConfiguration clientConfiguration;
private static WorkflowContextHeaderBuilder builder;
private String currentUser;
private URI regitryURI;
private PasswordCallback callBack;
private AiravataRegistry2 registry;
private Map<String, String> configuration = new HashMap<String, String>();
private AiravataManagerImpl airavataManagerImpl;
private ApplicationManagerImpl applicationManagerImpl;
private WorkflowManagerImpl workflowManagerImpl;
private ProvenanceManagerImpl provenanceManagerImpl;
private UserManagerImpl userManagerImpl;
private ExecutionManagerImpl executionManagerImpl;
private String gateway;
private boolean configCreated = false;
private static final Version API_VERSION = new Version("Airavata", 0, 7,
null, null, null);
// FIXME: Need a constructor to set registry URL
protected AiravataClient() {
}
private static HashMap<String, String> createConfig(URI registryUrl,
String username, String password) throws RepositoryException,
RegistryException, AiravataConfigurationException {
HashMap<String, String> config = new HashMap<String, String>();
if (registryUrl != null) {
config.put(AiravataClient.REGISTRY, registryUrl.toString());
}
AiravataRegistry2 registryObject = getRegistry(registryUrl, "default",
username, new PasswordCallBackImpl(username, password));
if (registryObject != null) {
URI uri = registryObject.getEventingServiceURI();
config.put(
AiravataClient.BROKER,
uri == null ? "http://localhost:8080/axis2/services/EventingService"
: uri.toString());
uri = registryObject.getMessageBoxURI();
config.put(
AiravataClient.MSGBOX,
uri == null ? "http://localhost:8080/axis2/services/MsgBoxService"
: uri.toString());
List<URI> URLList = registryObject.getWorkflowInterpreterURIs();
config.put(
AiravataClient.WORKFLOWSERVICEURL,
URLList == null || URLList.size() == 0 ? "http://localhost:8080/axis2/services/WorkflowInterpretor?wsdl"
: URLList.get(0).toString());
List<URI> urlList = registryObject.getGFacURIs();
config.put(
AiravataClient.GFAC,
urlList == null || urlList.size() == 0 ? "http://localhost:8080/axis2/services/GFacService"
: urlList.get(0).toString());
config.put(AiravataClient.WITHLISTENER, "true");
}
return config;
}
@Override
public void initialize() throws AiravataAPIInvocationException {
try {
if (!configCreated) {
configuration = createConfig(getRegitryURI(), getCurrentUser(),
getPassword());
configCreated = true;
}
updateClientConfiguration(configuration);
// At this point we do not know the workflowExperimentId
// FIXME: Registry URL is set null as its not used. Set this when we
// have rest services
builder = new WorkflowContextHeaderBuilder(
configuration.get(BROKER), configuration.get(GFAC), null,
null, null, configuration.get(MSGBOX));
// TODO: At some point this should contain the current user the
// airavata
// client is
// logged in to the Airavata system
setCurrentUser(getClientConfiguration().getJcrUsername());
} catch (Exception e) {
throw new AiravataAPIInvocationException(
"Error while initializing the Airavata API", e);
}
}
private void updateClientConfiguration(Map<String, String> configuration)
throws MalformedURLException {
AiravataClientConfiguration clientConfiguration = getClientConfiguration();
if (configuration.get(GFAC) != null) {
clientConfiguration.setGfacURL(new URL(configuration.get(GFAC)));
}
if (configuration.get(MSGBOX) != null) {
clientConfiguration.setMessageboxURL(new URL(configuration
.get(MSGBOX)));
}
if (configuration.get(BROKER) != null) {
clientConfiguration.setMessagebrokerURL(new URL(configuration
.get(BROKER)));
}
if (configuration.get(WORKFLOWSERVICEURL) != null) {
clientConfiguration.setXbayaServiceURL(new URL(configuration
.get(WORKFLOWSERVICEURL)));
}
if (configuration.get(MSGBOX) != null) {
clientConfiguration.setMessageboxURL(new URL(configuration
.get(MSGBOX)));
}
if (clientConfiguration.getRegistryURL() != null
&& clientConfiguration.getGfacURL() == null) {
try {
clientConfiguration.setGfacURL(getRegistryClient()
.getGFacURIs().get(0).toURL());
configuration.put(GFAC, clientConfiguration.getGfacURL()
.toString());
} catch (Exception e) {
e.printStackTrace();
}
}
}
public AiravataRegistry2 getRegistryClient()
throws AiravataConfigurationException, RegistryException {
if (registry == null) {
registry = getRegistry(getRegitryURI(), getGateway(),
getCurrentUser(), getCallBack());
}
return registry;
}
public static AiravataRegistry2 getRegistry(URI registryURI,
String gateway, String username, PasswordCallback callback)
throws RegistryException, AiravataConfigurationException {
return AiravataRegistryFactory.getRegistry(registryURI, new Gateway(
gateway), new AiravataUser(username), callback);
}
public AiravataClientConfiguration getClientConfiguration() {
if (clientConfiguration == null) {
clientConfiguration = new AiravataClientConfiguration();
}
return clientConfiguration;
}
public AiravataManager getAiravataManager() {
if (airavataManagerImpl == null) {
airavataManagerImpl = new AiravataManagerImpl(this);
}
return airavataManagerImpl;
}
public ApplicationManager getApplicationManager() {
if (applicationManagerImpl == null) {
applicationManagerImpl = new ApplicationManagerImpl(this);
}
return applicationManagerImpl;
}
public WorkflowManager getWorkflowManager() {
if (workflowManagerImpl == null) {
workflowManagerImpl = new WorkflowManagerImpl(this);
}
return workflowManagerImpl;
}
public ProvenanceManager getProvenanceManager() {
if (provenanceManagerImpl == null) {
provenanceManagerImpl = new ProvenanceManagerImpl(this);
}
return provenanceManagerImpl;
}
public UserManager getUserManager() {
if (userManagerImpl == null) {
userManagerImpl = new UserManagerImpl(this);
}
return userManagerImpl;
}
public ExecutionManager getExecutionManager() {
if (executionManagerImpl == null) {
executionManagerImpl = new ExecutionManagerImpl(this);
}
return executionManagerImpl;
}
public String getCurrentUser() {
return currentUser;
}
public Version getVersion() {
return API_VERSION;
}
@Override
public DescriptorBuilder getDescriptorBuilder() {
return new DescriptorBuilder();
}
public void setCurrentUser(String currentUser) {
this.currentUser = currentUser;
}
public void setRegitryURI(URI regitryURI) {
this.regitryURI = regitryURI;
}
public String getPassword() {
if (getCallBack() != null) {
return getCallBack().getPassword(getCurrentUser());
}
return null;
}
public URI getRegitryURI() {
return regitryURI;
}
public PasswordCallback getCallBack() {
return callBack;
}
public void setCallBack(PasswordCallback callBack) {
this.callBack = callBack;
}
public String getGateway() {
return gateway;
}
public void setGateway(String gateway) {
this.gateway = gateway;
}
}