forked from cinco/tmuxido
Initial release of tmuxido
Rust-based tmux project launcher with fzf selection, incremental mtime-based cache, per-project .tmuxido.toml session config, and Drone CI pipeline for automated binary releases.
This commit is contained in:
+267
@@ -0,0 +1,267 @@
|
||||
use anyhow::{Context, Result};
|
||||
use serde::{Deserialize, Serialize};
|
||||
use std::fs;
|
||||
use std::path::Path;
|
||||
use std::process::Command;
|
||||
|
||||
#[derive(Debug, Clone, Deserialize, Serialize)]
|
||||
pub struct Window {
|
||||
pub name: String,
|
||||
#[serde(default)]
|
||||
pub panes: Vec<String>,
|
||||
#[serde(default)]
|
||||
pub layout: Option<String>,
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone, Deserialize, Serialize)]
|
||||
pub struct SessionConfig {
|
||||
#[serde(default)]
|
||||
pub windows: Vec<Window>,
|
||||
}
|
||||
|
||||
impl SessionConfig {
|
||||
pub fn load_from_project(project_path: &Path) -> Result<Option<Self>> {
|
||||
let config_path = project_path.join(".tmuxido.toml");
|
||||
|
||||
if !config_path.exists() {
|
||||
return Ok(None);
|
||||
}
|
||||
|
||||
let content = fs::read_to_string(&config_path)
|
||||
.with_context(|| format!("Failed to read session config: {}", config_path.display()))?;
|
||||
|
||||
let config: SessionConfig = toml::from_str(&content)
|
||||
.with_context(|| format!("Failed to parse session config: {}", config_path.display()))?;
|
||||
|
||||
Ok(Some(config))
|
||||
}
|
||||
}
|
||||
|
||||
pub struct TmuxSession {
|
||||
session_name: String,
|
||||
project_path: String,
|
||||
base_index: usize,
|
||||
}
|
||||
|
||||
impl TmuxSession {
|
||||
pub fn new(project_path: &Path) -> Self {
|
||||
let session_name = project_path
|
||||
.file_name()
|
||||
.and_then(|n| n.to_str())
|
||||
.unwrap_or("project")
|
||||
.replace('.', "_")
|
||||
.replace(' ', "-");
|
||||
|
||||
let base_index = Self::get_base_index();
|
||||
|
||||
Self {
|
||||
session_name,
|
||||
project_path: project_path.display().to_string(),
|
||||
base_index,
|
||||
}
|
||||
}
|
||||
|
||||
fn get_base_index() -> usize {
|
||||
// Try to get base-index from tmux
|
||||
let output = Command::new("tmux")
|
||||
.args(["show-options", "-gv", "base-index"])
|
||||
.output();
|
||||
|
||||
if let Ok(output) = output {
|
||||
if output.status.success() {
|
||||
let index_str = String::from_utf8_lossy(&output.stdout);
|
||||
if let Ok(index) = index_str.trim().parse::<usize>() {
|
||||
return index;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Default to 0 if we can't determine
|
||||
0
|
||||
}
|
||||
|
||||
pub fn create(&self, config: &SessionConfig) -> Result<()> {
|
||||
// Check if we're already inside a tmux session
|
||||
let inside_tmux = std::env::var("TMUX").is_ok();
|
||||
|
||||
// Check if session already exists
|
||||
let session_exists = Command::new("tmux")
|
||||
.args(["has-session", "-t", &self.session_name])
|
||||
.output()
|
||||
.map(|o| o.status.success())
|
||||
.unwrap_or(false);
|
||||
|
||||
if session_exists {
|
||||
// Session exists, just switch to it
|
||||
if inside_tmux {
|
||||
Command::new("tmux")
|
||||
.args(["switch-client", "-t", &self.session_name])
|
||||
.status()
|
||||
.context("Failed to switch to existing session")?;
|
||||
} else {
|
||||
Command::new("tmux")
|
||||
.args(["attach-session", "-t", &self.session_name])
|
||||
.status()
|
||||
.context("Failed to attach to existing session")?;
|
||||
}
|
||||
return Ok(());
|
||||
}
|
||||
|
||||
// Create new session
|
||||
if config.windows.is_empty() {
|
||||
// Create simple session with one window
|
||||
self.create_simple_session()?;
|
||||
} else {
|
||||
// Create session with custom windows
|
||||
self.create_custom_session(config)?;
|
||||
}
|
||||
|
||||
// Attach or switch to the session
|
||||
if inside_tmux {
|
||||
Command::new("tmux")
|
||||
.args(["switch-client", "-t", &self.session_name])
|
||||
.status()
|
||||
.context("Failed to switch to new session")?;
|
||||
} else {
|
||||
Command::new("tmux")
|
||||
.args(["attach-session", "-t", &self.session_name])
|
||||
.status()
|
||||
.context("Failed to attach to new session")?;
|
||||
}
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
fn create_simple_session(&self) -> Result<()> {
|
||||
// Create a detached session with one window
|
||||
Command::new("tmux")
|
||||
.args([
|
||||
"new-session",
|
||||
"-d",
|
||||
"-s",
|
||||
&self.session_name,
|
||||
"-c",
|
||||
&self.project_path,
|
||||
])
|
||||
.status()
|
||||
.context("Failed to create tmux session")?;
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
fn create_custom_session(&self, config: &SessionConfig) -> Result<()> {
|
||||
// Create session with first window
|
||||
let first_window = &config.windows[0];
|
||||
Command::new("tmux")
|
||||
.args([
|
||||
"new-session",
|
||||
"-d",
|
||||
"-s",
|
||||
&self.session_name,
|
||||
"-n",
|
||||
&first_window.name,
|
||||
"-c",
|
||||
&self.project_path,
|
||||
])
|
||||
.status()
|
||||
.context("Failed to create tmux session")?;
|
||||
|
||||
// Create panes for first window if specified
|
||||
if !first_window.panes.is_empty() {
|
||||
self.create_panes(self.base_index, &first_window.panes)?;
|
||||
}
|
||||
|
||||
// Apply layout for first window if specified
|
||||
if let Some(layout) = &first_window.layout {
|
||||
self.apply_layout(self.base_index, layout)?;
|
||||
}
|
||||
|
||||
// Create additional windows
|
||||
for (index, window) in config.windows.iter().skip(1).enumerate() {
|
||||
let window_index = self.base_index + index + 1;
|
||||
|
||||
Command::new("tmux")
|
||||
.args([
|
||||
"new-window",
|
||||
"-t",
|
||||
&format!("{}:{}", self.session_name, window_index),
|
||||
"-n",
|
||||
&window.name,
|
||||
"-c",
|
||||
&self.project_path,
|
||||
])
|
||||
.status()
|
||||
.with_context(|| format!("Failed to create window: {}", window.name))?;
|
||||
|
||||
// Create panes if specified
|
||||
if !window.panes.is_empty() {
|
||||
self.create_panes(window_index, &window.panes)?;
|
||||
}
|
||||
|
||||
// Apply layout if specified
|
||||
if let Some(layout) = &window.layout {
|
||||
self.apply_layout(window_index, layout)?;
|
||||
}
|
||||
}
|
||||
|
||||
// Select the first window
|
||||
Command::new("tmux")
|
||||
.args(["select-window", "-t", &format!("{}:{}", self.session_name, self.base_index)])
|
||||
.status()
|
||||
.context("Failed to select first window")?;
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
fn create_panes(&self, window_index: usize, panes: &[String]) -> Result<()> {
|
||||
for (pane_index, command) in panes.iter().enumerate() {
|
||||
let target = format!("{}:{}", self.session_name, window_index);
|
||||
|
||||
// First pane already exists (created with the window), skip split
|
||||
if pane_index > 0 {
|
||||
// Create new pane by splitting
|
||||
Command::new("tmux")
|
||||
.args([
|
||||
"split-window",
|
||||
"-t",
|
||||
&target,
|
||||
"-c",
|
||||
&self.project_path,
|
||||
])
|
||||
.status()
|
||||
.context("Failed to split pane")?;
|
||||
}
|
||||
|
||||
// Send the command to the pane if it's not empty
|
||||
if !command.is_empty() {
|
||||
let pane_target = format!("{}:{}.{}", self.session_name, window_index, pane_index);
|
||||
Command::new("tmux")
|
||||
.args([
|
||||
"send-keys",
|
||||
"-t",
|
||||
&pane_target,
|
||||
command,
|
||||
"Enter",
|
||||
])
|
||||
.status()
|
||||
.context("Failed to send keys to pane")?;
|
||||
}
|
||||
}
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
fn apply_layout(&self, window_index: usize, layout: &str) -> Result<()> {
|
||||
Command::new("tmux")
|
||||
.args([
|
||||
"select-layout",
|
||||
"-t",
|
||||
&format!("{}:{}", self.session_name, window_index),
|
||||
layout,
|
||||
])
|
||||
.status()
|
||||
.with_context(|| format!("Failed to apply layout: {}", layout))?;
|
||||
|
||||
Ok(())
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user