prerna.util.MosfetSyncHelper 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.util;

import java.io.File;
import java.io.IOException;
import java.time.ZonedDateTime;
import java.util.List;
import java.util.Map;
import java.util.Vector;

import org.apache.logging.log4j.Logger;

import prerna.auth.utils.SecurityInsightUtils;
import prerna.engine.impl.InsightAdministrator;
import prerna.engine.impl.SmssUtilities;
import prerna.om.Insight;
import prerna.om.MosfetFile;
import prerna.project.api.IProject;

public class MosfetSyncHelper {

	// get the directory separator
	private static final String DIR_SEPARATOR = java.nio.file.FileSystems.getDefault().getSeparator();

	// ADDED
	public static final String ADD = "ADD";
	// MODIFIED
	public static final String MOD = "MOD";
	// DELETE
	public static final String DEL = "DEL";
	// RENAMED
	public static final String REN = "REN";

//	public static final String ENGINE_ID_KEY = "engineId";
//	public static final String RDBMS_ID_KEY = "rdbmsId";
//	public static final String INSIGHT_NAME_KEY = "insightName";
//	public static final String LAYOUT_KEY = "layout";
//	public static final String RECIPE_KEY = "recipe";
//	public static final String HIDDEN_KEY = "hidden";

	private MosfetSyncHelper() {

	}

	////////////////////////////////////////////////////////////////////////////////////////////////
	////////////////////////////////////////////////////////////////////////////////////////////////
	////////////////////////////////////////////////////////////////////////////////////////////////

	/**
	 * 
	 * @param projectId
	 * @param projectName
	 * @param rdbmsId
	 * @return
	 */
	public static String getMosfetFileLocation(String projectId, String projectName, String rdbmsId) {
		return DIHelper.getInstance().getProperty(Constants.BASE_FOLDER)
				+ DIR_SEPARATOR + Constants.PROJECT_FOLDER
				+ DIR_SEPARATOR + SmssUtilities.getUniqueName(projectName, projectId)
				+ DIR_SEPARATOR + "app_root" 
				+ DIR_SEPARATOR + "version" 
				+ DIR_SEPARATOR + rdbmsId;
	}
	
	/*
	 * This section for synchronizing files
	 */

	public static void synchronizeInsightChanges(Map> filesChanged, Logger logger) {
		// process add
		if(filesChanged.containsKey(ADD)) {
			processAddedFiles(filesChanged.get(ADD), logger);
		}

		// process mod
		if(filesChanged.containsKey(MOD)) {
			processModifiedFiles(filesChanged.get(MOD), logger);
		}

		// process delete
		if(filesChanged.containsKey(DEL)) {
			processDelete(filesChanged.get(DEL), logger);
		}
	}

	private static void processAddedFiles(List list, Logger logger) {
		for(String fileLocation : list) {
			File file = new File(Utility.normalizePath(fileLocation));
			MosfetFile mosfetFile;
			try {
				mosfetFile = MosfetFile.generateFromFile(file);
			} catch (IOException e) {
				outputError(logger, "MOSFET file is not in valid JSON format");
				logger.error(Constants.STACKTRACE, e);
				continue;
			}
			processAddedFile(mosfetFile, logger);
		}
	}

	private static void processAddedFile(MosfetFile mosfet, Logger logger) {
		String projectId = mosfet.getProjectId();
		String id = mosfet.getRdbmsId();

		// need to add the insight in the rdbms engine
		IProject project = Utility.getProject(projectId);
		// we want to make sure the file isn't added because we made the insight
		// and is in fact a new one made by another collaborator
		Vector ins = project.getInsight(id);
		if(ins == null || ins.isEmpty() || (ins.size() == 1 && ins.get(0) == null) ) {
			logger.info("Start processing new mosfet file");
			addInsightToEngineRdbms(project, mosfet);
			logger.info("Done processing mosfet file");
		}
	}

	private static void processModifiedFiles(List list, Logger logger) {
		for(String fileLocation : list) {
			File file = new File(Utility.normalizePath(fileLocation));
			MosfetFile mosfetFile;
			try {
				mosfetFile = MosfetFile.generateFromFile(file);
			} catch (IOException e) {
				outputError(logger, "MOSFET file is not in valid JSON format");
				logger.error(Constants.STACKTRACE, e);
				continue;
			}
			processModifiedFiles(mosfetFile, logger);
		}
	}

	private static void processModifiedFiles(MosfetFile mosfet, Logger logger) {
		logger.info("Start editing existing mosfet file");
		updateInsightInEngineRdbms(mosfet);
		logger.info("Done processing mosfet file");
	}

	private static void processDelete(List list, Logger logger) {
		for(String fileLocation : list) {
			File file = new File(Utility.normalizePath(fileLocation));
			MosfetFile mosfetFile;
			try {
				mosfetFile = MosfetFile.generateFromFile(file);
			} catch (IOException e) {
				outputError(logger, "MOSFET file is not in valid JSON format");
				logger.error(Constants.STACKTRACE, e);
				continue;
			}
			logger.info("Start deleting mosfet file");
			deleteInsightFromEngineRdbms(mosfetFile);
			logger.info("Done deleting mosfet file");
		}
	}

	public static void addInsightToEngineRdbms(IProject project, MosfetFile mosfet) {
		String projectId = mosfet.getProjectId();
		String id = mosfet.getRdbmsId();
		String insightName = mosfet.getInsightName();
		String layout = mosfet.getLayout();
		List recipe = mosfet.getRecipe();
		boolean global = mosfet.isGlobal();
		boolean cacheable = mosfet.isCacheable();
		int cacheMinutes = mosfet.getCacheMinutes();
		String cacheCron = mosfet.getCacheCron();
		ZonedDateTime cachedOn = mosfet.getCachedOn();
		boolean cacheEncrypt = mosfet.isCacheEncrypt();
		String schemaName = mosfet.getSchemaName();

		InsightAdministrator admin = new InsightAdministrator(project.getInsightDatabase());
		// just put the recipe into an array
		admin.addInsight(id, insightName, layout, recipe, global, cacheable, cacheMinutes, cacheCron, cachedOn, cacheEncrypt, schemaName);
		SecurityInsightUtils.addInsight(projectId, id, insightName, global, layout, cacheable, cacheMinutes, cacheCron, cachedOn, cacheEncrypt, recipe, schemaName);

		// also sync the metadata
		String description = mosfet.getDescription();
		if(description != null) {
			admin.updateInsightDescription(id, description);
			SecurityInsightUtils.updateInsightDescription(projectId, id, description);
		}
		String[] tags = mosfet.getTags();
		if(tags != null && tags.length > 0) {
			admin.updateInsightTags(id, tags);
			SecurityInsightUtils.updateInsightTags(id, id, tags);
		}
	}

	public static void updateInsightInEngineRdbms(MosfetFile mosfet) {
		String projectId = mosfet.getProjectId();
		String id = mosfet.getRdbmsId();
		String insightName = mosfet.getInsightName();
		String layout = mosfet.getLayout();
		List recipe = mosfet.getRecipe();
		boolean global = mosfet.isGlobal();
		boolean cacheable = mosfet.isCacheable();
		int cacheMinutes = mosfet.getCacheMinutes();
		String cacheCron = mosfet.getCacheCron();
		ZonedDateTime cachedOn = mosfet.getCachedOn();
		boolean cacheEncrypt = mosfet.isCacheEncrypt();
		String schemaName = mosfet.getSchemaName();
		
		IProject project = Utility.getProject(projectId);

		InsightAdministrator admin = new InsightAdministrator(project.getInsightDatabase());
		// just put the recipe into an array
		admin.updateInsight(id, insightName, layout, recipe, global, cacheable, cacheMinutes, cacheCron, cachedOn, cacheEncrypt, schemaName);
		SecurityInsightUtils.updateInsight(projectId, id, insightName, global, layout, cacheable, cacheMinutes, cacheCron, cachedOn, cacheEncrypt, recipe, schemaName);

		// also sync the metadata
		String description = mosfet.getDescription();
		if(description != null) {
			admin.updateInsightDescription(id, description);
			SecurityInsightUtils.updateInsightDescription(projectId, id, description);
		}
		String[] tags = mosfet.getTags();
		if(tags != null && tags.length > 0) {
			admin.updateInsightTags(id, tags);
			SecurityInsightUtils.updateInsightTags(id, id, tags);
		}
	}

	public static void deleteInsightFromEngineRdbms(MosfetFile mosfet) {
		String projectId = mosfet.getProjectId();
		String id = mosfet.getRdbmsId();

		IProject project = Utility.getProject(projectId);
		InsightAdministrator admin = new InsightAdministrator(project.getInsightDatabase());
		admin.dropInsight(id);

		SecurityInsightUtils.deleteInsight(projectId, id);
	}

	private static void outputError(Logger logger, String errorMessage) {
		if(logger != null) {
			logger.info("ERROR!!! " + errorMessage);
		}
	}

	////////////////////////////////////////////////////////////////////////////////////////////////
	////////////////////////////////////////////////////////////////////////////////////////////////
	////////////////////////////////////////////////////////////////////////////////////////////////
	////////////////////////////////////////////////////////////////////////////////////////////////

	/*
	 * These methods only update the mosfet file itself and not the metadata (security db or insights db)
	 */

	/**
	 * Get the insight name for the input mosfet file
	 * @param mosfetFile
	 * @return
	 * @throws IOException 
	 */
	public static String getInsightName(File mosfetFile) throws IOException {
		MosfetFile mosfet = MosfetFile.generateFromFile(mosfetFile);
		return mosfet.getInsightName();
	}

	/**
	 * Only generate the mosfet file
	 * @param projectId
	 * @param projectName
	 * @param rdbmsId
	 * @param insightName
	 * @param layout
	 * @param recipe
	 * @param hidden
	 * @param cacheable
	 * @param cacheMinutes
	 * @param cacheEncrypt
	 * @param description
	 * @param tags
	 * @return
	 * @throws IOException
	 */
	public static File makeMosfitFile(String projectId, String projectName, String rdbmsId, String insightName, 
			String layout, List recipe, boolean hidden, 
			boolean cacheable, int cacheMinutes, String cacheCron, ZonedDateTime cachedOn, boolean cacheEncrypt,
			String description, List tags, String schemaName) throws IOException {
		return makeMosfitFile(projectId, projectName, rdbmsId, insightName, layout, recipe, hidden, 
				cacheable, cacheMinutes, cacheCron, cachedOn, cacheEncrypt, 
				description, tags, schemaName, false);
	}
	
	/**
	 * Only generate the mosfet file
	 * @param projectId
	 * @param projectName
	 * @param rdbmsId
	 * @param insightName
	 * @param layout
	 * @param recipe
	 * @param hidden
	 * @param description
	 * @param tags
	 * @param forceDelete
	 * @return
	 * @throws IOException 
	 */
	public static File makeMosfitFile(String projectId, String projectName, String rdbmsId, String insightName, 
			String layout, List recipe, boolean hidden, 
			boolean cacheable, int cacheMinutes, String cacheCron, ZonedDateTime cachedOn, boolean cacheEncrypt,
			String description, List tags, String schemaName, boolean forceDelete) throws IOException {
		MosfetFile mosfet = new MosfetFile();
		mosfet.setProjectId(projectId);
		mosfet.setRdbmsId(rdbmsId);
		mosfet.setInsightName(insightName);
		mosfet.setLayout(layout);
		mosfet.setRecipe(recipe);
		mosfet.setGlobal(hidden);
		mosfet.setCacheable(cacheable);
		mosfet.setCacheMinutes(cacheMinutes);
		mosfet.setCacheCron(cacheCron);
		mosfet.setCachedOn(cachedOn);
		mosfet.setCacheEncrypt(cacheEncrypt);
		if(description != null) {
			mosfet.setDescription(description);
		}
		if(tags != null && !tags.isEmpty()) {
			mosfet.setTags(tags.toArray(new String[tags.size()]));
		}
		mosfet.setSchemaName(schemaName);

		String mosfetPath = getMosfetFileLocation(projectId, projectName, rdbmsId);

		mosfet.write(mosfetPath, forceDelete);
		return new File(mosfetPath + DIR_SEPARATOR + MosfetFile.RECIPE_FILE);
	}

	/**
	 * Only update the mosfet file
	 * @param mosfetFile
	 * @param projectId
	 * @param projectName
	 * @param rdbmsId
	 * @param insightName
	 * @param layout
	 * @param imageFileName
	 * @param recipe
	 * @param hidden
	 * @param description
	 * @param tags
	 * @return
	 * @throws IOException 
	 */
	public static File updateMosfitFile(File mosfetFile, String projectId, String projectName, String rdbmsId, String insightName,
			String layout, String imageFileName, List recipe, boolean hidden, 
			boolean cacheable, int cacheMinutes, String cacheCron, ZonedDateTime cachedOn, boolean cacheEncrypt,
			String description, List tags) throws IOException {
		MosfetFile mosfet = new MosfetFile();
		mosfet.setProjectId(projectId);
		mosfet.setRdbmsId(rdbmsId);
		mosfet.setInsightName(insightName);
		mosfet.setLayout(layout);
		mosfet.setRecipe(recipe);
		mosfet.setGlobal(hidden);
		mosfet.setCacheable(cacheable);
		mosfet.setCacheMinutes(cacheMinutes);
		mosfet.setCacheCron(cacheCron);
		mosfet.setCachedOn(cachedOn);
		mosfet.setCacheEncrypt(cacheEncrypt);
		if(description != null) {
			mosfet.setDescription(description);
		}
		if(tags != null && !tags.isEmpty()) {
			mosfet.setTags(tags.toArray(new String[tags.size()]));
		}

		mosfet.write(mosfetFile.getParentFile().getAbsolutePath(), true);
		return mosfetFile;
	}

	/**
	 * Only update the mosfet file insight name
	 * @param mosfetFile
	 * @param newInsightName
	 * @return
	 * @throws IOException
	 */
	public static File updateMosfitFileInsightName(File mosfetFile, String newInsightName) throws IOException {
		MosfetFile mosfet = MosfetFile.generateFromFile(mosfetFile);
		mosfet.setInsightName(newInsightName);
		mosfet.write(mosfetFile.getParentFile().getAbsolutePath(), true);
		return mosfetFile;
	}
	
	/**
	 * Only update the mosfet file cached on
	 * @param mosfetFile
	 * @param cachedOn
	 * @return
	 * @throws IOException
	 */
	public static File updateMosfitFileCachedOn(File mosfetFile, ZonedDateTime cachedOn) throws IOException {
		MosfetFile mosfet = MosfetFile.generateFromFile(mosfetFile);
		mosfet.setCachedOn(cachedOn);
		mosfet.write(mosfetFile.getParentFile().getAbsolutePath(), true);
		return mosfetFile;
	}

}