ExecuteInsightResource.java

/*******************************************************************************
 * Copyright 2015 Defense Health Agency (DHA)
 *
 * If your use of this software does not include any GPLv2 components:
 * 	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.
 * ----------------------------------------------------------------------------
 * If your use of this software includes any GPLv2 components:
 * 	This program is free software; you can redistribute it and/or
 * 	modify it under the terms of the GNU General Public License
 * 	as published by the Free Software Foundation; either version 2
 * 	of the License, or (at your option) any later version.
 *
 * 	This program 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 General Public License for more details.
 *******************************************************************************/
package prerna.semoss.web.services.local;

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

import jakarta.servlet.http.HttpServletRequest;
import jakarta.servlet.http.HttpSession;
import jakarta.ws.rs.Path;
import jakarta.ws.rs.PathParam;
import jakarta.ws.rs.core.Context;
import prerna.auth.User;
import prerna.auth.utils.SecurityInsightUtils;
import prerna.auth.utils.SecurityQueryUtils;
import prerna.cluster.util.ClusterUtil;
import prerna.om.Insight;
import prerna.om.InsightStore;
import prerna.project.api.IProject;
import prerna.util.AssetUtility;
import prerna.util.Constants;
import prerna.util.Utility;
import prerna.web.services.util.WebUtility;

@Path("exec")
public class ExecuteInsightResource {

	@Path("/a-{appId}/i-{insightId}")
	public Object generateInsight(@Context HttpServletRequest request, @PathParam("appId") String projectId,
			@PathParam("insightId") String rdbmsId) {
		projectId = WebUtility.inputSanitizer(projectId);
		rdbmsId = WebUtility.inputSanitizer(rdbmsId);
		User user = null;
		HttpSession session = request.getSession(false);
		if (session == null) {
			Map<String, String> errorHash = new HashMap<String, String>();
			errorHash.put(Constants.ERROR_MESSAGE, "Invalid session to retrieve insight data");
			return WebUtility.getResponse(errorHash, 400);
		}

		user = ((User) session.getAttribute(Constants.SESSION_USER));
		if (user == null) {
			Map<String, String> errorMap = new HashMap<String, String>();
			errorMap.put("error", "User session is invalid");
			return WebUtility.getResponse(errorMap, 401);
		}

		projectId = SecurityQueryUtils.testUserEngineIdForAlias(user, projectId);
		if (!SecurityInsightUtils.userCanViewInsight(user, projectId, rdbmsId)) {
			Map<String, String> errorMap = new HashMap<String, String>();
			errorMap.put("error", "User does not have access to this insight");
			return WebUtility.getResponse(errorMap, 401);
		}

		IProject project = Utility.getProject(projectId);
		if (project == null) {
			throw new IllegalArgumentException("Cannot find project = " + projectId);
		}

		Insight newInsight = null;
		try {
			newInsight = SecurityInsightUtils.getInsight(projectId, rdbmsId);
		} catch (Exception e) {
			ClusterUtil.pullInsightsDB(projectId);
			// this is needed for the pipeline json
			ClusterUtil.pullProjectFolder(project,
					AssetUtility.getProjectVersionFolder(project.getProjectName(), projectId));
			try {
				List<Insight> in = project.getInsight(rdbmsId + "");
				newInsight = in.get(0);
			} catch (IllegalArgumentException e2) {
				Map<String, String> errorMap = new HashMap<String, String>();
				errorMap.put(Constants.ERROR_MESSAGE,
						"Error occured creating the insight. Detailed message = " + e2.getMessage());
				return WebUtility.getResponse(errorMap, 401);
			} catch (ArrayIndexOutOfBoundsException e2) {
				Map<String, String> errorMap = new HashMap<String, String>();
				errorMap.put(Constants.ERROR_MESSAGE, "Insight does not exist");
				return WebUtility.getResponse(errorMap, 401);
			}
		}

		InsightStore.getInstance().put(newInsight);
		newInsight.reRunPixelInsight(false);
		RunInsight runner = new RunInsight(newInsight);
		runner.dropInsight(true);
		return runner;
	}

}