78 lines
1.8 KiB
Rust
78 lines
1.8 KiB
Rust
use std::{
|
|
env::current_dir,
|
|
io::{ stdout, IsTerminal },
|
|
path::PathBuf,
|
|
process::exit
|
|
};
|
|
|
|
use tmux_interface::{
|
|
Session, Tmux,
|
|
|
|
commands,
|
|
variables::session::SessionsCtl
|
|
};
|
|
|
|
use crate::{
|
|
env,
|
|
error
|
|
};
|
|
|
|
/// return a Vec of all sessions or None
|
|
pub fn get_sessions() -> Option<Vec<Session>> {
|
|
let sessions = SessionsCtl::new().get_all();
|
|
if let Ok(sessions) = sessions {
|
|
return Some(sessions.0);
|
|
} else { return None; }
|
|
}
|
|
|
|
/// show the tmux nest text if env var is not unset
|
|
pub fn prevent_nest() {
|
|
if env::tmux() {
|
|
println!("Sessions should be nested with care; unset TMUX or use the '-n' flag to allow.");
|
|
exit(6);
|
|
}
|
|
}
|
|
|
|
/// enforce a command is being used in-session
|
|
pub fn session_enforce(cmd: &'static str) {
|
|
let tmux = var("TMUX").unwrap_or("".to_string());
|
|
if tmux.is_empty() {
|
|
error::not_in_session(cmd);
|
|
}
|
|
}
|
|
|
|
/// check whether a target session exists
|
|
pub fn session_exists<S: Into<String>>(target: S) -> bool {
|
|
let has_session = commands::HasSession::new()
|
|
.target_session(target.into());
|
|
Tmux::new().add_command(has_session)
|
|
.status()
|
|
.unwrap()
|
|
.success()
|
|
}
|
|
|
|
/// enforce a command is being run in a terminal
|
|
pub fn terminal_enforce() {
|
|
if !stdout().is_terminal() { error::not_terminal(); }
|
|
}
|
|
|
|
/// attempt to return the repo name or exit
|
|
pub fn repo_fallback() -> String {
|
|
let repo = repo_root(current_dir().unwrap());
|
|
if repo.is_none() { error::missing_target(); }
|
|
|
|
let target = repo.unwrap().file_name().unwrap().to_string_lossy().to_string();
|
|
target
|
|
}
|
|
|
|
/// recursively attempt to find a git root directory
|
|
pub fn repo_root(path: PathBuf) -> Option<PathBuf> {
|
|
// if .git dir is found, return
|
|
if path.join(".git").exists() { return Some(path); }
|
|
|
|
// otherwise, attempt to traverse
|
|
let parent = path.parent();
|
|
if let Some(parent) = parent { repo_root(parent.to_path_buf()) }
|
|
else { None }
|
|
}
|
|
|