AdminConfigService.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.config;

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

import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;

import com.google.gson.Gson;
import com.google.gson.GsonBuilder;

import jakarta.annotation.security.PermitAll;
import jakarta.servlet.http.HttpServletRequest;
import jakarta.servlet.http.HttpServletResponse;
import jakarta.ws.rs.POST;
import jakarta.ws.rs.Path;
import jakarta.ws.rs.core.Context;
import jakarta.ws.rs.core.Response;
import prerna.auth.utils.SecurityUpdateUtils;
import prerna.engine.api.IDatabaseEngine;
import prerna.engine.api.IRawSelectWrapper;
import prerna.query.querystruct.SelectQueryStruct;
import prerna.query.querystruct.selectors.QueryColumnSelector;
import prerna.rdf.engine.wrappers.WrapperManager;
import prerna.util.Constants;
import prerna.util.SocialPropertiesUtil;
import prerna.util.SystemEngineRegistry;
import prerna.web.services.util.WebUtility;

@Path("/")
@PermitAll
public class AdminConfigService {

	private static final Logger classLogger = LogManager.getLogger(AdminConfigService.class);

	@POST
	@Path("/setInitialAdmins")
	public Response setInitialAdmins(@Context HttpServletRequest request, @Context HttpServletResponse response)
			throws IOException {
		IDatabaseEngine engine = SystemEngineRegistry.getSecurityDb();
		SelectQueryStruct qs = new SelectQueryStruct();
		qs.addSelector(new QueryColumnSelector("SMSS_USER__ID"));
		qs.setLimit(1);
		try (IRawSelectWrapper wrapper = WrapperManager.getInstance().getRawWrapper(engine, qs)) {
			boolean hasUser = wrapper.hasNext();
			// if there are users, redirect to the main semoss page
			// we do not want to allow the person to make any admin requests
			if (hasUser) {
				Map<String, String> errorMap = new HashMap<>();
				errorMap.put(Constants.ERROR_MESSAGE, "Initial admin has already been set");
				return WebUtility.getResponse(errorMap, 400);
			}
		} catch (Exception e) {
			classLogger.error("Error determining whether the initial admin has already been set", e);
			Map<String, String> errorMap = new HashMap<>();
			errorMap.put(Constants.ERROR_MESSAGE,
					"Error occurred attempting to determine if the initial admin is set. Please check the system logs for assistance");
			return WebUtility.getResponse(errorMap, 400);
		}

		String idString = request.getParameter("ids");
		if (idString == null || idString.isEmpty()) {
			Map<String, String> errorMessage = new HashMap<>();
			errorMessage.put(Constants.ERROR_MESSAGE, "Need to send valid ids");
			return WebUtility.getResponse(errorMessage, 200);
		}

		Gson GSON = new GsonBuilder().disableHtmlEscaping().create();
		List<String> ids = GSON.fromJson(idString, List.class);
		for (String id : ids) {
			SecurityUpdateUtils.registerUser(id, null, null, null, null, null, null, null, true, true, true, null, null,
					null, null);
		}

		// take the user back to the login page
		String redirectUrl = SocialPropertiesUtil.getInstance().getLoginRedirect();
		Map<String, Object> retMap = new HashMap<>();
		retMap.put("success", true);
		retMap.put("redirectUrl", redirectUrl);
		return WebUtility.getResponse(retMap, 200);
	}

}