Create a unique folder to extract the contents of .mol

adeas31 · adeas31 · commit 1a3d110c0f87 · 2020-05-05T00:12:58.000+02:00
Fixes ticket:5786 do not mix the contents of .mol while extracting different versions.

Added a boolean parameter to loadEncryptedPackage to skip unziping of .mol
diff --git a/OMCompiler/Compiler/FrontEnd/ModelicaBuiltin.mo b/OMCompiler/Compiler/FrontEnd/ModelicaBuiltin.mo
@@ -1169,6 +1169,7 @@ end parseEncryptedPackage;
 function loadEncryptedPackage
   input String fileName;
   input String workdir = "<default>" "The output directory for imported encrypted files. <default> will put the files to current working directory.";
+  input Boolean skipUnzip = false "Skips the unzip of .mol if true. In that case we expect the files are already extracted e.g., because of parseEncryptedPackage() call.";
   output Boolean success;
 external "builtin";
 annotation(Documentation(info="<html>
diff --git a/OMCompiler/Compiler/Script/CevalScript.mo b/OMCompiler/Compiler/Script/CevalScript.mo
@@ -1468,7 +1468,7 @@ algorithm
     case (cache,_,"parseEncryptedPackage",Values.STRING(filename)::Values.STRING(workdir)::_,_)
       equation
         vals = {};
-        (b, filename_1) = unZipEncryptedPackageAndCheckFile(workdir, filename);
+        (b, filename_1) = unZipEncryptedPackageAndCheckFile(workdir, filename, false);
         if (b) then
           // clear the errors before!
           Error.clearMessages() "Clear messages";
@@ -1479,9 +1479,9 @@ algorithm
         end if;
       then (cache,ValuesUtil.makeArray(vals));
 
-    case (_,_,"loadEncryptedPackage",Values.STRING(filename)::Values.STRING(workdir)::_,_)
+    case (_,_,"loadEncryptedPackage",Values.STRING(filename)::Values.STRING(workdir)::Values.BOOL(bval)::_,_)
       equation
-        (b, filename_1) = unZipEncryptedPackageAndCheckFile(workdir, filename);
+        (b, filename_1) = unZipEncryptedPackageAndCheckFile(workdir, filename, bval);
         if (b) then
           execStatReset();
           filename_1 = Testsuite.friendlyPath(filename_1);
@@ -3193,6 +3193,7 @@ end findModelicaPath2;
 protected function unZipEncryptedPackageAndCheckFile
   input String inWorkdir;
   input String filename;
+  input Boolean skipUnzip;
   output Boolean success;
   output String outFilename;
 protected
@@ -3203,7 +3204,7 @@ algorithm
   if (System.regularFileExists(filename)) then
 	  if (Util.endsWith(filename, ".mol")) then
 	    workdir := if System.directoryExists(inWorkdir) then inWorkdir else System.pwd();
-	    if (0 == System.systemCall("unzip -q -o -d \"" + workdir + "\" \"" +  filename + "\"")) then
+	    if (skipUnzip or 0 == System.systemCall("unzip -q -o -d \"" + workdir + "\" \"" +  filename + "\"")) then
 	      s1 := System.basename(filename);
 	      s2 := Util.removeLast4Char(s1);
 	      s3 := listGet(Util.stringSplitAtChar(s2," "),1);
diff --git a/OMEdit/OMEditLIB/MainWindow.cpp b/OMEdit/OMEditLIB/MainWindow.cpp
@@ -630,6 +630,13 @@ void MainWindow::beforeClosingMainWindow()
     }
     mFMUDirectoriesList.clear();
   }
+  // Delete the MOL directories
+  foreach (QString molDirectory, mMOLDirectoriesList) {
+    if (QDir().exists(molDirectory)) {
+      Utilities::removeDirectoryRecursivly(molDirectory);
+    }
+  }
+  mMOLDirectoriesList.clear();
   delete pSettings;
   // delete the OptionsDialog object
   OptionsDialog::destroy();
diff --git a/OMEdit/OMEditLIB/MainWindow.h b/OMEdit/OMEditLIB/MainWindow.h
@@ -253,6 +253,7 @@ class MainWindow : public QMainWindow
                                    const char* variables);
 
   QList<QString> mFMUDirectoriesList;
+  QList<QString> mMOLDirectoriesList;
 private:
   bool mDebug;
   bool mTestsuiteRunning;
diff --git a/OMEdit/OMEditLIB/Modeling/LibraryTreeWidget.cpp b/OMEdit/OMEditLIB/Modeling/LibraryTreeWidget.cpp
@@ -4246,7 +4246,17 @@ void LibraryWidget::openEncrytpedModelicaLibrary(QString fileName, QString encod
     MainWindow::instance()->getStatusBar()->showMessage(QString(Helper::loading).append(": ").append(fileName));
   }
 
-  QStringList classesList = MainWindow::instance()->getOMCProxy()->parseEncryptedPackage(fileName, Utilities::tempDirectory());
+  // create unique directory to extract .mol contents
+  QFileInfo fileInfo(fileName);
+  QString tempDirectoryPath = Utilities::tempDirectory();
+  QString outputDirectory = QString("%1/%2%3").arg(Utilities::tempDirectory(), fileInfo.baseName(), QDateTime::currentDateTime().toString("yyyyMMddhhmmsszzz"));
+  if (!QDir().exists(outputDirectory)) {
+    QDir().mkpath(outputDirectory);
+    MainWindow::instance()->mMOLDirectoriesList.append(outputDirectory);
+    tempDirectoryPath = outputDirectory;
+  }
+
+  QStringList classesList = MainWindow::instance()->getOMCProxy()->parseEncryptedPackage(fileName, tempDirectoryPath);
   if (!classesList.isEmpty()) {
     if (multipleTopLevelClasses(classesList, fileName)) {
       if (showProgress) {
@@ -4277,7 +4287,8 @@ void LibraryWidget::openEncrytpedModelicaLibrary(QString fileName, QString encod
       pMessageBox->exec();
     } else { // if no conflicting model found then just load the file simply
       // load the encrypted package in OMC
-      if (MainWindow::instance()->getOMCProxy()->loadEncryptedPackage(fileName, Utilities::tempDirectory())) {
+      // we pass true for skipUnzip as we have alredy extracted mol with parseEncryptedPackage earlier.
+      if (MainWindow::instance()->getOMCProxy()->loadEncryptedPackage(fileName, tempDirectoryPath, true)) {
         // create library tree nodes for loaded models
         int progressvalue = 0;
         if (showProgress) {
diff --git a/OMEdit/OMEditLIB/OMC/OMCProxy.cpp b/OMEdit/OMEditLIB/OMC/OMCProxy.cpp
@@ -3175,9 +3175,9 @@ QList<QString> OMCProxy::parseEncryptedPackage(QString fileName, QString working
  * \param workingDirectory
  * \return
  */
-bool OMCProxy::loadEncryptedPackage(QString fileName, QString workingDirectory)
+bool OMCProxy::loadEncryptedPackage(QString fileName, QString workingDirectory, bool skipUnzip)
 {
-  bool result = mpOMCInterface->loadEncryptedPackage(fileName, workingDirectory);
+  bool result = mpOMCInterface->loadEncryptedPackage(fileName, workingDirectory, skipUnzip);
   printMessagesStringInternal();
   return result;
 }
diff --git a/OMEdit/OMEditLIB/OMC/OMCProxy.h b/OMEdit/OMEditLIB/OMC/OMCProxy.h
@@ -265,7 +265,7 @@ class OMCProxy : public QObject
   QList<QList<QString > > getUses(QString className);
   bool buildEncryptedPackage(QString className, bool encrypt = true);
   QList<QString> parseEncryptedPackage(QString fileName, QString workingDirectory);
-  bool loadEncryptedPackage(QString fileName, QString workingDirectory);
+  bool loadEncryptedPackage(QString fileName, QString workingDirectory, bool skipUnzip);
 signals:
   void commandFinished();
 public slots:

Original file line number	Diff line number	Diff line change
`@@ -630,6 +630,13 @@ void MainWindow::beforeClosingMainWindow()`
`630`	`630`	`}`
`631`	`631`	`mFMUDirectoriesList.clear();`
`632`	`632`	`}`
	`633`	`+ // Delete the MOL directories`
	`634`	`+ foreach (QString molDirectory, mMOLDirectoriesList) {`
	`635`	`+ if (QDir().exists(molDirectory)) {`
	`636`	`+ Utilities::removeDirectoryRecursivly(molDirectory);`
	`637`	`+ }`
	`638`	`+ }`
	`639`	`+ mMOLDirectoriesList.clear();`
`633`	`640`	`delete pSettings;`
`634`	`641`	`// delete the OptionsDialog object`
`635`	`642`	`OptionsDialog::destroy();`
Original file line number	Diff line number	Diff line change
`@@ -3175,9 +3175,9 @@ QList<QString> OMCProxy::parseEncryptedPackage(QString fileName, QString working`
`3175`	`3175`	`* \param workingDirectory`
`3176`	`3176`	`* \return`
`3177`	`3177`	`*/`
`3178`		`-bool OMCProxy::loadEncryptedPackage(QString fileName, QString workingDirectory)`
	`3178`	`+bool OMCProxy::loadEncryptedPackage(QString fileName, QString workingDirectory, bool skipUnzip)`
`3179`	`3179`	`{`
`3180`		`- bool result = mpOMCInterface->loadEncryptedPackage(fileName, workingDirectory);`
	`3180`	`+ bool result = mpOMCInterface->loadEncryptedPackage(fileName, workingDirectory, skipUnzip);`
`3181`	`3181`	`printMessagesStringInternal();`
`3182`	`3182`	`return result;`
`3183`	`3183`	`}`