Ask user to specify export folder for flight log charts and CSVs

nur557 · nur557 · commit 0016d8125368 · 2025-06-29T16:57:45.000+02:00
diff --git a/src/modules/flightlog.js b/src/modules/flightlog.js
@@ -1,24 +1,71 @@
 import fs from "fs";
+import path from "path";
+import { dialog } from "electron";
 
 export function exportFlightLogToCSVs(flightLog) {
-  const flightLogDir = "flight-log-export"
   let flightLogSections = ["imu", "baro", "flightInfo", "orientationInfo", "filteredDataInfo", "gnssInfo", "flightStates", "eventInfo", "voltageInfo"];
 
-  if (!fs.existsSync(flightLogDir)) {
-    fs.mkdirSync(flightLogDir);
+  let paths = dialog.showOpenDialogSync({ properties: ["openDirectory"] });
+
+  if (!paths) {
+    console.log("No directory selected for export.");
+    // TODO: after snackbar is merged print error message to user
+    return;
+  }
+
+  const exportFolderPath = path.join(paths[0]);
+
+  if (!fs.existsSync(exportFolderPath)) {
+    fs.mkdirSync(exportFolderPath);
   }
 
   for (let flightLogSection of flightLogSections) {
-    fs.writeFile(`${flightLogDir}/${flightLogSection}.csv`, objectArrayToCSV(flightLogSection, flightLog[flightLogSection]), "utf8", function (err) {
+    fs.writeFile(`${exportFolderPath}/${flightLogSection}.csv`, objectArrayToCSV(flightLogSection, flightLog[flightLogSection]), "utf8", function (err) {
       if (err) {
         console.log("An error occurred while writing CSV object to file.");
+        // TODO: after snackbar is merged print error message to user
         return console.log(err);
       }
-      console.log("CSV file has been saved.");
     });
   }
 }
 
+export function exportFlightLogChartsToHTML(flightLogChartsHTML) {
+  let paths = dialog.showOpenDialogSync({ properties: ["openDirectory"] });
+
+  if (!paths) {
+    console.log("No directory selected for export.");
+    return;
+    // TODO: after snackbar is merged print error message to user
+  }
+
+  const exportFolderPath = path.join(paths[0]);
+
+  if (!fs.existsSync(exportFolderPath)) {
+    fs.mkdirSync(exportFolderPath);
+  }
+
+  const flightLogHtmlDocument = `
+    <!DOCTYPE html>
+    <html>
+      <head>
+        <script src="https://cdn.plot.ly/plotly-2.18.2.min.js"></script>
+      </head>
+      <body>
+      ${flightLogChartsHTML}
+      </body>
+    </html>
+  `;
+
+  fs.writeFile(`${exportFolderPath}/plots.html`, flightLogHtmlDocument, 'utf8', function (err) {
+    if (err) {
+      console.log("An error occurred while writing HTML Object to File.");
+      return console.log(err);
+      // TODO: after snackbar is merged print error message to user
+    }
+  });
+}
+
 export function exportJSON(flightLog) {
   fs.writeFile("flightlog.json", JSON.stringify(flightLog), "utf8", function (err) {
     if (err) {
diff --git a/src/modules/ipc.js b/src/modules/ipc.js
@@ -1,7 +1,7 @@
 import { ipcMain, dialog } from "electron";
 import fs from "fs";
 import { parseFlightLog } from "./logparser.js"
-import { exportFlightLogToCSVs } from "./flightlog.js"
+import { exportFlightLogToCSVs, exportFlightLogChartsToHTML } from "./flightlog.js"
 import { connect, disconnect, command, cliCommand, getList } from "./serial.js";
 
 export function subscribeListeners() {
@@ -74,24 +74,8 @@ export function subscribeListeners() {
     event.sender.send("EXPORT_FLIGHTLOG_CSVS");
   });
 
-  ipcMain.on("EXPORT_FLIGHTLOG_HTML", (event, flightLogHtmlStr) => {
-
-    flightLogHtmlStr = `
-      <!DOCTYPE html>
-      <html>
-        <head>
-          <script src="https://cdn.plot.ly/plotly-2.18.2.min.js"></script>
-        </head>
-      <body>
-    ` + flightLogHtmlStr + "</body></html>"
-    fs.writeFile("plots.html", flightLogHtmlStr, 'utf8', function (err) {
-      if (err) {
-        console.log("An error occurred while writing CSV Object to File.");
-        return console.log(err);
-      }
-      console.log("HTML file has been saved.");
-    });
-
+  ipcMain.on("EXPORT_FLIGHTLOG_HTML", (event, flightLogChartsHTML) => {
+    exportFlightLogChartsToHTML(flightLogChartsHTML);
     event.sender.send("EXPORT_FLIGHTLOG_HTML");
   });
 
