50 lines
1.7 KiB
Rust
50 lines
1.7 KiB
Rust
|
//! Instance locking functions for bootstrap
|
||
|
|
//!
|
||
|
|
//! Extracted from mod.rs
|
||
|
|
|
||
|
|
use crate::security::command_guard::SafeCommand;
|
||
|
|
use log::warn;
|
||
|
|
use std::fs;
|
||
|
|
use std::path::PathBuf;
|
||
|
|
|
||
|
|
/// Check if another instance is already running
|
||
|
|
pub fn check_single_instance() -> Result<bool, Box<dyn std::error::Error>> {
|
||
|
|
let stack_path = std::env::var("BOTSERVER_STACK_PATH")
|
||
|
|
.unwrap_or_else(|_| "./botserver-stack".to_string());
|
||
|
|
let lock_file = PathBuf::from(&stack_path).join(".lock");
|
||
|
|
if lock_file.exists() {
|
||
|
|
if let Ok(pid_str) = fs::read_to_string(&lock_file) {
|
||
|
|
if let Ok(pid) = pid_str.trim().parse::<i32>() {
|
||
|
|
let pid_str = pid.to_string();
|
||
|
|
if let Some(output) = SafeCommand::new("kill")
|
||
|
|
.and_then(|c| c.args(&["-0", &pid_str]))
|
||
|
|
.ok()
|
||
|
|
.and_then(|cmd| cmd.execute().ok())
|
||
|
|
{
|
||
|
|
if output.status.success() {
|
||
|
|
warn!("Another botserver process (PID {}) is already running on this stack", pid);
|
||
|
|
return Ok(false);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
let pid = std::process::id();
|
||
|
|
if let Some(parent) = lock_file.parent() {
|
||
|
|
fs::create_dir_all(parent).ok();
|
||
|
|
}
|
||
|
|
fs::write(&lock_file, pid.to_string()).ok();
|
||
|
|
Ok(true)
|
||
|
|
}
|
||
|
|
|
||
|
|
/// Release the instance lock
|
||
|
|
pub fn release_instance_lock() {
|
||
|
|
let stack_path = std::env::var("BOTSERVER_STACK_PATH")
|
||
|
|
.unwrap_or_else(|_| "./botserver-stack".to_string());
|
||
|
|
let lock_file = PathBuf::from(&stack_path).join(".lock");
|
||
|
|
if lock_file.exists() {
|
||
|
|
fs::remove_file(&lock_file).ok();
|
||
|
|
}
|
||
|
|
}
|