[rust] Check lock folder at the end of SM execution and clear it if required

Author: bonigarcia

rust/src/lock.rs

@@ -17,13 +17,16 @@
 
 use crate::logger::Logger;
 use anyhow::Error;
+use std::cell::RefCell;
 use std::fs::File;
 use std::path::{Path, PathBuf};
 
 use crate::files::{create_parent_path_if_not_exists, create_path_if_not_exists};
 use fs2::FileExt;
 use std::fs;
 
+thread_local!(static LOCK_PATH: RefCell<Option<PathBuf>> = RefCell::new(None));
+
 const LOCK_FILE: &str = "sm.lock";
 
 pub struct Lock {
@@ -51,15 +54,36 @@ impl Lock {
         log.debug(format!("Acquiring lock: {}", path.display()));
         file.lock_exclusive().unwrap_or_default();
 
+        LOCK_PATH.with(|option| {
+            *option.borrow_mut() = Some(lock_folder.to_path_buf());
+        });
+
         Ok(Self { file, path })
     }
 
     pub fn release(&mut self) {
         fs::remove_file(&self.path).unwrap_or_default();
         self.file.unlock().unwrap_or_default();
+
+        LOCK_PATH.with(|option| {
+            *option.borrow_mut() = None;
+        });
     }
 
     pub fn exists(&mut self) -> bool {
         self.path.exists()
     }
 }
+
+pub fn clear_lock_if_required() {
+    let mut lock_path: Option<PathBuf> = None;
+    LOCK_PATH.with(|option| {
+        let optional_path = &*option.borrow();
+        if optional_path.is_some() {
+            lock_path = Some(optional_path.as_ref().unwrap().to_path_buf());
+        }
+    });
+    if lock_path.is_some() {
+        fs::remove_dir_all(lock_path.unwrap()).unwrap_or_default();
+    }
+}

rust/src/main.rs

@@ -22,6 +22,7 @@ use exitcode::OK;
 use exitcode::UNAVAILABLE;
 use selenium_manager::config::{BooleanKey, StringKey, CACHE_PATH_KEY};
 use selenium_manager::grid::GridManager;
+use selenium_manager::lock::clear_lock_if_required;
 use selenium_manager::logger::{Logger, BROWSER_PATH, DRIVER_PATH};
 use selenium_manager::metadata::clear_metadata;
 use selenium_manager::TTL_SEC;
@@ -318,5 +319,6 @@ fn flush_and_exit(code: i32, log: &Logger, err: Option<Error>) -> ! {
     }
     log.set_code(code);
     log.flush();
+    clear_lock_if_required();
     exit(code);
 }