step.plugins.adaptergrid.FunctionRepositoryServices Maven / Gradle / Ivy

Go to download
/*******************************************************************************
 * (C) Copyright 2016 Jerome Comte and Dorian Cransac
 *  
 * This file is part of STEP
 *  
 * STEP is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *  
 * STEP is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *  
 * You should have received a copy of the GNU Affero General Public License
 * along with STEP.  If not, see .
 *******************************************************************************/
package step.plugins.adaptergrid;

import java.util.HashMap;
import java.util.Map;

import javax.annotation.PostConstruct;
import javax.json.Json;
import javax.ws.rs.Consumes;
import javax.ws.rs.DELETE;
import javax.ws.rs.GET;
import javax.ws.rs.POST;
import javax.ws.rs.Path;
import javax.ws.rs.PathParam;
import javax.ws.rs.Produces;
import javax.ws.rs.core.MediaType;

import step.core.GlobalContext;
import step.core.artefacts.reports.ReportNode;
import step.core.deployment.AbstractServices;
import step.core.deployment.Secured;
import step.core.execution.ExecutionContext;
import step.core.execution.model.ExecutionMode;
import step.core.execution.model.ExecutionParameters;
import step.core.miscellaneous.ReportNodeAttachmentManager;
import step.functions.Function;
import step.functions.FunctionClient;
import step.functions.FunctionRepository;
import step.functions.Input;
import step.functions.Output;
import step.functions.editors.FunctionEditor;
import step.functions.editors.FunctionEditorRegistry;
import step.functions.type.FunctionTypeException;
import step.functions.type.SetupFunctionException;
import step.grid.TokenWrapper;
import step.grid.client.GridClient.AgentCommunicationException;
import step.grid.tokenpool.Interest;

@Path("/functions")
public class FunctionRepositoryServices extends AbstractServices {

	ReportNodeAttachmentManager reportNodeAttachmentManager;
	
	private FunctionClient getFunctionClient() {
		return (FunctionClient) getContext().get(GridPlugin.FUNCTIONCLIENT_KEY);
	}
	
	private FunctionRepository getFunctionRepository() {
		return getFunctionClient().getFunctionRepository();
	}
	
	@PostConstruct
	public void init() {
		reportNodeAttachmentManager = new ReportNodeAttachmentManager(getContext().getAttachmentManager());
	}

	@POST
	@Consumes(MediaType.APPLICATION_JSON)
	@Produces(MediaType.APPLICATION_JSON)
	@Path("/")
	@Secured(right="kw-write")
	public Function save(Function function) throws SetupFunctionException, FunctionTypeException {
		FunctionRepository repo = getFunctionRepository();
		if(function.getId()==null || repo.getFunctionById(function.getId().toString())==null) {
			getFunctionClient().setupFunction(function);
		} else {
			getFunctionClient().updateFunction(function);
		}
		repo.addFunction(function);
		return function;
	}
	
	@POST
	@Path("/{id}/copy")
	@Consumes(MediaType.APPLICATION_JSON)
	@Produces(MediaType.APPLICATION_JSON)
	@Secured(right="kw-write")
	public void copyFunction(@PathParam("id") String id) throws FunctionTypeException {		
		FunctionRepository repo = getFunctionRepository();
		Function source = repo.getFunctionById(id);
		if(source!=null) {
			Function copy = getFunctionClient().copyFunction(source);
			repo.addFunction(copy);
		}
	}
	
	public static ExecutionContext createContext(GlobalContext g) {
		ReportNode root = new ReportNode();
		ExecutionContext c = new ExecutionContext("");
		c.setGlobalContext(g);
		c.getReportNodeCache().put(root);
		c.setReport(root);
		ExecutionContext.setCurrentReportNode(root);
		c.setExecutionParameters(new ExecutionParameters("dummy", null, ExecutionMode.RUN));
		return c;
	}
	
	@DELETE
	@Path("/{id}")
	@Secured(right="kw-delete")
	public void delete(@PathParam("id") String functionId) throws FunctionTypeException {
		FunctionRepository repo = getFunctionRepository();
		Function function = repo.getFunctionById(functionId);
		getFunctionClient().deleteFunction(function);
		getFunctionRepository().deleteFunction(functionId);
	}
	
	@POST
	@Path("/search")
	@Secured(right="kw-read")
	public Function get(Map attributes) {
		return getFunctionRepository().getFunctionByAttributes(attributes);
	}
	
	@GET
	@Path("/{id}")
	@Secured(right="kw-read")
	public Function get(@PathParam("id") String functionId) {
		return getFunctionRepository().getFunctionById(functionId);
	}
	
	@GET
	@Path("/{id}/editor")
	@Secured(right="kw-read")
	public String getFunctionEditor(@PathParam("id") String functionId) {
		Function function = getFunctionRepository().getFunctionById(functionId);
		FunctionEditor editor = getContext().get(FunctionEditorRegistry.class).getFunctionEditor(function);
		if(editor!=null) {
			return editor.getEditorPath(function);
		} else {
			return null;
		}
	}
	
	@GET
	@Path("/types/{id}")
	@Consumes(MediaType.APPLICATION_JSON)
	@Produces(MediaType.APPLICATION_JSON)
	@Secured(right="kw-read")
	public Function newFunctionTypeConf(@PathParam("id") String type) {
		Function newFunction = getFunctionClient().newFunction(type);
		newFunction.setAttributes(new HashMap<>());
		newFunction.getAttributes().put(Function.NAME, "");
		newFunction.setSchema(Json.createObjectBuilder().build());
		return newFunction;
	}

	public static class GetTokenHandleParameter {
		
		Map attributes;
		Map interests;
		boolean createSession;

		public Map getAttributes() {
			return attributes;
		}
		
		public void setAttributes(Map attributes) {
			this.attributes = attributes;
		}
		
		public Map getInterests() {
			return interests;
		}
		
		public void setInterests(Map interests) {
			this.interests = interests;
		}
		
		public boolean isCreateSession() {
			return createSession;
		}
		
		public void setCreateSession(boolean createSession) {
			this.createSession = createSession;
		}
	}
	
	
	@POST
	@Produces(MediaType.APPLICATION_JSON)
	@Path("/executor/tokens/select")
	@Secured(right="kw-execute")
	public TokenWrapper getTokenHandle(GetTokenHandleParameter parameter) throws AgentCommunicationException {
		return getFunctionClient().getTokenHandle(parameter.attributes, parameter.interests, parameter.createSession);
	}

	@POST
	@Produces(MediaType.APPLICATION_JSON)
	@Path("/executor/tokens/return")
	@Secured(right="kw-execute")
	public void returnTokenHandle(TokenWrapper token) throws AgentCommunicationException {
		getFunctionClient().returnTokenHandle(token);
	}
	
	public static class CallFunctionInput {
		
		String functionId;
		Map functionAttributes;
		TokenWrapper tokenHandle;
		Input input;
		
		public CallFunctionInput() {
			super();
		}

		public String getFunctionId() {
			return functionId;
		}

		public void setFunctionId(String functionId) {
			this.functionId = functionId;
		}

		public TokenWrapper getTokenHandle() {
			return tokenHandle;
		}

		public void setTokenHandle(TokenWrapper tokenHandle) {
			this.tokenHandle = tokenHandle;
		}

		public Map getFunctionAttributes() {
			return functionAttributes;
		}

		public void setFunctionAttributes(Map functionAttributes) {
			this.functionAttributes = functionAttributes;
		}

		public Input getInput() {
			return input;
		}

		public void setInput(Input input) {
			this.input = input;
		}
		
	}

	@POST
	@Produces(MediaType.APPLICATION_JSON)
	@Path("/executor/execute")
	@Secured(right="kw-execute")
	public Output callFunction(CallFunctionInput input) {
		if(input.functionId!=null) {
			return getFunctionClient().callFunction(input.tokenHandle, input.functionId, input.input);			
		} else {
			return getFunctionClient().callFunction(input.tokenHandle, input.functionAttributes, input.input);			
		}
	}
}