prerna.theme.AdminThemeUtils Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of semoss Show documentation
SEMOSS
The newest version!
package prerna.theme;

import java.io.UnsupportedEncodingException;
import java.sql.PreparedStatement;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.UUID;

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

import com.google.gson.Gson;

import prerna.auth.User;
import prerna.auth.utils.SecurityAdminUtils;
import prerna.engine.api.IRawSelectWrapper;
import prerna.query.querystruct.SelectQueryStruct;
import prerna.query.querystruct.filters.SimpleQueryFilter;
import prerna.query.querystruct.selectors.QueryColumnSelector;
import prerna.rdf.engine.wrappers.WrapperManager;
import prerna.sablecc2.om.PixelDataType;
import prerna.util.ConnectionUtils;
import prerna.util.Constants;

public class AdminThemeUtils extends AbstractThemeUtils {
	
	private static final Logger classLogger = LogManager.getLogger(AdminThemeUtils.class);

	private static AdminThemeUtils instance = new AdminThemeUtils();

	private AdminThemeUtils() {

	}

	public static AdminThemeUtils getInstance(User user) {
		if (user == null) {
			return null;
		}
		if (SecurityAdminUtils.userIsAdmin(user)) {
			return instance;
		}
		return null;
	}

	/**
	 * Get the active theme
	 * 
	 * @return
	 */
	public static Object getActiveAdminTheme() {
		if (themeDb == null) {
			return new HashMap<>();
		}

		final String ADMIN_THEME_PREFIX = "ADMIN_THEME__";
		SelectQueryStruct qs = new SelectQueryStruct();
		qs.addSelector(new QueryColumnSelector(ADMIN_THEME_PREFIX+"ID"));
		qs.addSelector(new QueryColumnSelector(ADMIN_THEME_PREFIX+"THEME_NAME"));
		qs.addSelector(new QueryColumnSelector(ADMIN_THEME_PREFIX+"THEME_MAP"));
		qs.addSelector(new QueryColumnSelector(ADMIN_THEME_PREFIX+"IS_ACTIVE"));
		qs.addExplicitFilter(SimpleQueryFilter.makeColToValFilter(ADMIN_THEME_PREFIX+"IS_ACTIVE", "==", true, PixelDataType.BOOLEAN));

		List> retVal = null;
		IRawSelectWrapper wrapper = null;
		try {
			wrapper = WrapperManager.getInstance().getRawWrapper(themeDb, qs);
			retVal = flushRsToMap(wrapper);
		} catch (Exception e) {
			classLogger.error(Constants.STACKTRACE, e);
		}

		if (retVal == null || retVal.isEmpty()) {
			return new HashMap<>();
		}

		return retVal.get(0);
	}

	/*
	 * all other methods should be on the instance
	 * so that we cannot bypass security easily
	 */

	/**
	 * Get all the admin level themes
	 * 
	 * @return
	 */
	public List> getAdminThemes(int limit, int offset) {
		final String ADMIN_THEME_PREFIX = "ADMIN_THEME__";
		SelectQueryStruct qs = new SelectQueryStruct();
		qs.addSelector(new QueryColumnSelector(ADMIN_THEME_PREFIX+"ID"));
		qs.addSelector(new QueryColumnSelector(ADMIN_THEME_PREFIX+"THEME_NAME"));
		qs.addSelector(new QueryColumnSelector(ADMIN_THEME_PREFIX+"THEME_MAP"));
		qs.addSelector(new QueryColumnSelector(ADMIN_THEME_PREFIX+"IS_ACTIVE"));
		if(limit > 0) {
			qs.setLimit(limit);
		}
		if(offset > 0) {
			qs.setOffSet(offset);
		}
		
		IRawSelectWrapper wrapper;
		try {
			wrapper = WrapperManager.getInstance().getRawWrapper(themeDb, qs);
			return flushRsToMap(wrapper);
		} catch (Exception e) {
			classLogger.error(Constants.STACKTRACE, e);
		}

		return new ArrayList<>();
	}

	/**
	 * Set the active admin level theme This means setting the is_active boolean to
	 * true for 1 theme while the previous true value to false;
	 * 
	 * @param themeId
	 * @return
	 */
	public boolean setActiveTheme(String themeId) {
		String query = "UPDATE ADMIN_THEME SET IS_ACTIVE=FALSE WHERE IS_ACTIVE=TRUE; UPDATE ADMIN_THEME SET IS_ACTIVE=TRUE WHERE ID='"
				+ themeId + "'";
		try {
			themeDb.insertData(query);
			themeDb.commit();
		} catch (SQLException e) {
			classLogger.error(Constants.STACKTRACE, e);
			return false;
		}
		return true;
	}

	/**
	 * Set the active admin level theme to false so the UI resets to the base theme
	 * 
	 * @return
	 */
	public boolean setAllThemesInactive() {
		PreparedStatement ps = null;
		try {
			ps = themeDb.getPreparedStatement("UPDATE ADMIN_THEME SET IS_ACTIVE=? WHERE IS_ACTIVE=?");
			int parameterIndex = 1;
			ps.setBoolean(parameterIndex++, false);
			ps.setBoolean(parameterIndex++, false);
			ps.execute();
			if(!ps.getConnection().getAutoCommit()) {
				ps.getConnection().commit();
			}
		} catch (SQLException e) {
			classLogger.error(Constants.STACKTRACE, e);
			return false;
		} finally {
			ConnectionUtils.closeAllConnectionsIfPooling(themeDb, ps);
		}
		
		return true;
	}

	/**
	 * Insert a new admin level theme
	 * 
	 * @param themeId
	 * @param themeName
	 * @param themeMap
	 * @param isActive
	 * @return
	 */
	public String createAdminTheme(String themeName, String themeMap, boolean isActive) {
		String themeId = UUID.randomUUID().toString();
		
		PreparedStatement ps = null;
		try {
			ps = themeDb.getPreparedStatement("INSERT INTO ADMIN_THEME (ID, THEME_NAME, THEME_MAP, IS_ACTIVE) VALUES (?,?,?,?)");
			int parameterIndex = 1;
			ps.setString(parameterIndex++, themeId);
			ps.setString(parameterIndex++, themeName);
			themeDb.getQueryUtil().handleInsertionOfClob(ps.getConnection(), ps, themeMap, parameterIndex++, new Gson());
			ps.setBoolean(parameterIndex++, isActive);
			ps.execute();
			if(!ps.getConnection().getAutoCommit()) {
				ps.getConnection().commit();
			}
		} catch (UnsupportedEncodingException | SQLException e) {
			classLogger.error(Constants.STACKTRACE, e);
			return null;
		} finally {
			ConnectionUtils.closeAllConnectionsIfPooling(themeDb, ps);
		}

		if (isActive) {
			setActiveTheme(themeId);
		}
		return themeId;
	}

	/**
	 * Edit an existing admin level theme
	 * 
	 * @param themeId
	 * @param themeName
	 * @param themeMap
	 * @param isActive
	 * @return
	 */
	public boolean editAdminTheme(String themeId, String themeName, String themeMap, boolean isActive) {
		PreparedStatement ps = null;
		try {
			ps = themeDb.getPreparedStatement("UPDATE ADMIN_THEME SET THEME_NAME=?, THEME_MAP=?, IS_ACTIVE=? WHERE ID=?");
			int parameterIndex = 1;
			ps.setString(parameterIndex++, themeName);
			themeDb.getQueryUtil().handleInsertionOfClob(ps.getConnection(), ps, themeMap, parameterIndex++, new Gson());
			ps.setBoolean(parameterIndex++, isActive);
			ps.setString(parameterIndex++, themeId);
			ps.execute();
			if(!ps.getConnection().getAutoCommit()) {
				ps.getConnection().commit();
			}
		} catch (UnsupportedEncodingException | SQLException e) {
			classLogger.error(Constants.STACKTRACE, e);
			return false;
		} finally {
			ConnectionUtils.closeAllConnectionsIfPooling(themeDb, ps);
		}

		if (isActive) {
			setActiveTheme(themeId);
		}
		return true;
	}

	/**
	 * Delete a admin level theme
	 * 
	 * @param themeId
	 * @return
	 */
	public boolean deleteAdminTheme(String themeId) {
		
		PreparedStatement ps = null;
		try {
			ps = themeDb.getPreparedStatement("DELETE FROM ADMIN_THEME WHERE ID=?");
			ps.setString(1, themeId);
			ps.execute();
			if(!ps.getConnection().getAutoCommit()) {
				ps.getConnection().commit();
			}
		} catch (SQLException e) {
			classLogger.error(Constants.STACKTRACE, e);
			return false;
		} finally {
			ConnectionUtils.closeAllConnectionsIfPooling(themeDb, ps);
		}
		
		return true;
	}
}