X-Git-Url: https://gerrit.simantics.org/r/gitweb?a=blobdiff_plain;f=bundles%2Forg.simantics%2Fsrc%2Forg%2Fsimantics%2FDatabaseBaselines.java;h=f781372ad374df903fe4803cdaa6f873beffaef4;hb=4304cd5a89b97964500ceab8f139a75bb1708bea;hp=ca272be6b3029b6593242df0ddf46b3e00b89ebb;hpb=77921feee3f8331ab54796ff0832921405bea874;p=simantics%2Fplatform.git

diff --git a/bundles/org.simantics/src/org/simantics/DatabaseBaselines.java b/bundles/org.simantics/src/org/simantics/DatabaseBaselines.java
index ca272be6b..f781372ad 100644
--- a/bundles/org.simantics/src/org/simantics/DatabaseBaselines.java
+++ b/bundles/org.simantics/src/org/simantics/DatabaseBaselines.java
@@ -13,6 +13,7 @@ import java.util.zip.ZipEntry;
 import java.util.zip.ZipFile;
 import java.util.zip.ZipOutputStream;
 
+import org.eclipse.core.runtime.Platform;
 import org.simantics.utils.FileUtils;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
@@ -26,14 +27,119 @@ public class DatabaseBaselines {
 	private static final Logger LOGGER = LoggerFactory.getLogger(DatabaseBaselines.class);
 
 	private static final boolean REQUIRE_INDEX_IN_BASELINE = false;
-	
+
 	private static final String DB_DIRECTORY = "db"; //$NON-NLS-1$
 	private static final String INDEX_DIRECTORY = ".metadata/.plugins/org.simantics.db.indexing"; //$NON-NLS-1$
 
 	private static final DateTimeFormatter TIMESTAMP_FORMAT = DateTimeFormatter.ofPattern("d. MMM yyyy HH:mm:ss");
 
-	public static Path packageBaseline(Path fromWorkspace, Path packageFile) throws IOException {
-		return compressZip(fromWorkspace, collectBaselinePaths(fromWorkspace), packageFile);
+	public static boolean handleBaselineDatabase(Path installLocation, boolean databaseExists) throws PlatformException {
+		Path workspaceLocation = Platform.getLocation().toFile().toPath();
+		Path baselineIndicatorFile = workspaceLocation.resolve(".baselined");
+		if (Files.isRegularFile(baselineIndicatorFile)) {
+			// This means that the workspace has already been initialized from
+			// a database baseline and further initialization is not necessary.
+			return true;
+		}
+
+		Path baseline = resolveBaselineFile(installLocation);
+		if (baseline == null) {
+			baseline = getAutomaticBaselinePath();
+			if(baseline == null)
+				return false;
+			if(databaseExists)
+				return false;
+			if(!existsAutomaticBaseline())
+				return false;
+		}
+
+		DatabaseBaselines.validateBaselineFile(baseline);
+		DatabaseBaselines.validateWorkspaceForBaselineInitialization(workspaceLocation);
+		DatabaseBaselines.initializeWorkspaceWithBaseline(baseline, workspaceLocation, baselineIndicatorFile);
+		return true;
+
+	}
+
+	private static Path resolveBaselineFile(Path installLocation) throws PlatformException {
+		String dbBaselineArchive = System.getProperty("org.simantics.db.baseline", null);
+		if (dbBaselineArchive == null)
+			return null;
+
+		Path baseline = Paths.get(dbBaselineArchive);
+		if (baseline.isAbsolute()) {
+			if (!Files.isRegularFile(baseline))
+				throw new PlatformException("Specified database baseline archive " + baseline
+						+ " does not exist. Cannot initialize workspace database from baseline.");
+			return baseline;
+		}
+
+		// Relative path resolution order:
+		// 1. from the platform "install location"
+		// 2. from working directory
+		//Path installLocation = tryGetInstallLocation();
+		if (installLocation != null) {
+			Path installedBaseline = installLocation.resolve(dbBaselineArchive);
+			if (Files.isRegularFile(installedBaseline))
+				return installedBaseline;
+		}
+		if (!Files.isRegularFile(baseline))
+			throw new PlatformException("Specified database baseline archive " + baseline
+					+ " does not exist in either the install location (" + installLocation
+					+ ") or the working directory (" + Paths.get(".").toAbsolutePath()
+					+ "). Cannot initialize workspace database.");
+		return null;
+	}
+
+	private static boolean useAutomaticBaseline() {
+		return getAutomaticBaselinePath() != null;
+	}
+
+	private static Path getAutomaticBaselinePath() {
+		if("true".equals(System.getProperty("org.simantics.db.baseline.automatic")))
+			return Paths.get("automatic_baseline", "baseline.zip");
+		else
+			return null;
+	}
+
+	private static boolean existsAutomaticBaseline() {
+		Path baselineFile = getAutomaticBaselinePath();
+		if(baselineFile == null)
+			return false;
+		return Files.exists(baselineFile);
+	}
+
+	public static boolean shouldCreateAutomaticBaseline(boolean existsDatabase) throws PlatformException {
+		if(!useAutomaticBaseline()) {
+			// Are we using this feature? 
+			return false;
+		}
+		if(existsDatabase) {
+			// Baseline can only be created after db initialization
+			return false;
+		}
+		if(existsAutomaticBaseline()) {
+			// Existing baselines should not be automatically overridden
+			return false;
+		}
+		return true;
+	}
+
+	public static void createAutomaticBaseline(Path dbLocation) throws PlatformException {
+
+		if(existsAutomaticBaseline())
+			return;
+
+		try {
+			DatabaseBaselines.packageBaseline(dbLocation.getParent(), getAutomaticBaselinePath());
+		} catch (IOException e) {
+			LOGGER.error("Error while creating automatic baseline", e);
+		}
+
+	}
+
+	public static Path packageBaseline(Path fromWorkspace, Path packageFilePath) throws IOException {
+		Files.createDirectories(packageFilePath.getParent());
+		return compressZip(fromWorkspace, collectBaselinePaths(fromWorkspace), packageFilePath);
 	}
 
 	private static List<Path> collectBaselinePaths(Path workspace) throws IOException {
@@ -95,7 +201,7 @@ public class DatabaseBaselines {
 				throw new PlatformException("Database location " + db + " already exists. Cannot re-initialize workspace from baseline.");
 			if (REQUIRE_INDEX_IN_BASELINE) {
 				Path index = workspaceLocation.resolve(INDEX_DIRECTORY);
-				if (!Files.exists(index) || !isEmptyDirectory(index))
+				if (Files.exists(index) || !isEmptyDirectory(index))
 					throw new PlatformException("Index location " + index + " already exists. Cannot re-initialize workspace from baseline.");
 			}
 		} catch (IOException e) {