80 lines
2.2 KiB
Rust
80 lines
2.2 KiB
Rust
#![allow(dead_code)]
|
|
use crate::executor::RunType;
|
|
#[cfg(target_os = "linux")]
|
|
use crate::steps::linux::Distribution;
|
|
use crate::sudo::Sudo;
|
|
use crate::utils::{get_require_sudo_string, require_option};
|
|
use crate::{config::Config, executor::Executor};
|
|
use color_eyre::eyre::Result;
|
|
use std::env::var;
|
|
use std::path::Path;
|
|
use std::sync::Mutex;
|
|
|
|
pub struct ExecutionContext<'a> {
|
|
run_type: RunType,
|
|
sudo: Option<Sudo>,
|
|
config: &'a Config,
|
|
/// Name of a tmux session to execute commands in, if any.
|
|
/// This is used in `./steps/remote/ssh.rs`, where we want to run `topgrade` in a new
|
|
/// tmux window for each remote.
|
|
tmux_session: Mutex<Option<String>>,
|
|
/// True if topgrade is running under ssh.
|
|
under_ssh: bool,
|
|
#[cfg(target_os = "linux")]
|
|
distribution: &'a Result<Distribution>,
|
|
}
|
|
|
|
impl<'a> ExecutionContext<'a> {
|
|
pub fn new(
|
|
run_type: RunType,
|
|
sudo: Option<Sudo>,
|
|
config: &'a Config,
|
|
#[cfg(target_os = "linux")] distribution: &'a Result<Distribution>,
|
|
) -> Self {
|
|
let under_ssh = var("SSH_CLIENT").is_ok() || var("SSH_TTY").is_ok();
|
|
Self {
|
|
run_type,
|
|
sudo,
|
|
config,
|
|
tmux_session: Mutex::new(None),
|
|
under_ssh,
|
|
#[cfg(target_os = "linux")]
|
|
distribution,
|
|
}
|
|
}
|
|
|
|
pub fn execute_elevated(&self, command: &Path, interactive: bool) -> Result<Executor> {
|
|
let sudo = require_option(self.sudo.as_ref(), get_require_sudo_string())?;
|
|
Ok(sudo.execute_elevated(self, command, interactive))
|
|
}
|
|
|
|
pub fn run_type(&self) -> RunType {
|
|
self.run_type
|
|
}
|
|
|
|
pub fn sudo(&self) -> &Option<Sudo> {
|
|
&self.sudo
|
|
}
|
|
|
|
pub fn config(&self) -> &Config {
|
|
self.config
|
|
}
|
|
|
|
pub fn under_ssh(&self) -> bool {
|
|
self.under_ssh
|
|
}
|
|
|
|
pub fn set_tmux_session(&self, session_name: String) {
|
|
self.tmux_session.lock().unwrap().replace(session_name);
|
|
}
|
|
|
|
pub fn get_tmux_session(&self) -> Option<String> {
|
|
self.tmux_session.lock().unwrap().clone()
|
|
}
|
|
|
|
#[cfg(target_os = "linux")]
|
|
pub fn distribution(&self) -> &Result<Distribution> {
|
|
self.distribution
|
|
}
|
|
}
|