Crappy code fix 13/?

1 files changed, 27 insertions, 0 deletions

diff --git a/launcher/src/lib.rs b/launcher/src/lib.rs
index 9e2ef8b..3e5d0fe 100644
--- a/launcher/src/lib.rs
+++ b/launcher/src/lib.rs
@@ -1,3 +1,5 @@
+mod constants;
+
 use std::ffi::OsStr;
 use std::path::PathBuf;
 
@@ -13,4 +15,29 @@ impl AppData {
             AppData::Asar(path) => path.as_ref()
         }
     }
+
+    pub fn locate_app(work_dir: &PathBuf) -> Option<AppData> {
+        let mut app_data: Option<AppData> = None;
+
+        let mut app_path_folder = PathBuf::from(work_dir);
+        app_path_folder.push(constants::APP_FOLDER_PATH);
+
+        let mut app_path_asar = PathBuf::from(work_dir);
+        app_path_asar.push(constants::APP_ASAR_PATH);
+
+        if app_path_folder.exists() {
+            app_data = Some(AppData::Folder(app_path_folder))
+        } else if app_path_asar.exists() {
+            app_data = Some(AppData::Asar(app_path_asar))
+        } else {
+            println!(
+                "atomic-launcher: Warning: Could not find Electron-compatible app.\
+            Looked in {} and {}. The default atomic-runtime application will be opened instead.",
+                app_path_folder.to_str().unwrap(),
+                app_path_asar.to_str().unwrap()
+            );
+        }
+
+        app_data
+    }
 }