pee/controller/tmux.go

230 lines
7.6 KiB
Go
Raw Permalink Normal View History

2023-10-22 01:12:58 +05:30
package controller
import (
"fmt"
"os/exec"
"strings"
2023-10-22 01:24:16 +05:30
projectconfig "github.com/xrehpicx/pee/config"
2023-10-22 01:12:58 +05:30
"github.com/charmbracelet/log"
)
2023-10-22 14:01:39 +05:30
// CreateTmuxSession creates a Tmux session based on the given Configuration.
2023-10-22 01:12:58 +05:30
func CreateTmuxSession(config *projectconfig.Configuration) error {
sessionName := config.SessionName
// Check if the session exists
checkSessionCmd := exec.Command("tmux", "has-session", "-t", sessionName)
2023-10-22 14:01:39 +05:30
log.Debug("Ran command", "command", checkSessionCmd.String())
2023-10-22 01:12:58 +05:30
if err := checkSessionCmd.Run(); err == nil {
// If it exists, switch to the session
switchSessionCmd := exec.Command("tmux", "switch-client", "-t", sessionName)
if err := switchSessionCmd.Run(); err != nil {
return err
}
2023-10-22 14:01:39 +05:30
log.Debug("Ran command", "command", switchSessionCmd.String())
2023-10-22 01:12:58 +05:30
} else {
// If it doesn't exist, create the session
2024-01-17 21:23:03 +05:30
createSessionCmd := exec.Command("tmux", "new-session", "-d", "-s", sessionName, "-c", config.WorkingDir)
// createSessionCmd := exec.Command("tmux", "new-session", "-d", "-s", sessionName)
2023-10-22 01:12:58 +05:30
if err := createSessionCmd.Run(); err != nil {
return err
}
2023-10-22 14:01:39 +05:30
log.Debug("Ran command", "command", createSessionCmd.String())
2023-10-22 01:12:58 +05:30
// Change the working directory
changeDirCmd := exec.Command("tmux", "send-keys", "-t", sessionName, "cd "+config.WorkingDir, "Enter")
if err := changeDirCmd.Run(); err != nil {
return err
}
2023-10-22 14:01:39 +05:30
log.Debug("Ran command", "command", changeDirCmd.String())
2023-10-22 01:12:58 +05:30
2023-10-22 14:01:39 +05:30
// Create the first window outside the loop
// createWindow(config, sessionName, 0)
window := config.Windows[0]
windowName := fmt.Sprintf("%s:%d", sessionName, 1)
sendCommandsCmd := exec.Command("tmux", "send-keys", "-t", windowName, strings.Join(window.Panes[0].ShellCommand, " && "), "Enter")
2023-10-22 01:12:58 +05:30
if err := sendCommandsCmd.Run(); err != nil {
return err
}
2023-10-22 14:01:39 +05:30
log.Debug("Ran command", "command", sendCommandsCmd.String())
// Rename the window to the specified name
renameWindowCmd := exec.Command("tmux", "rename-window", "-t", windowName, window.WindowName)
if err := renameWindowCmd.Run(); err != nil {
2023-10-22 01:12:58 +05:30
return err
}
2023-10-22 14:01:39 +05:30
log.Debug("Ran command", "command", renameWindowCmd.String())
2023-10-22 01:12:58 +05:30
2023-10-22 14:01:39 +05:30
// Create and run commands for additional panes in the window
for j, pane := range window.Panes[1:] {
paneName := fmt.Sprintf("%s:%d.%d", sessionName, 1, j+2)
// Split the window horizontally
splitPaneCmd := exec.Command("tmux", "split-window", "-t", windowName, "-h", "-p", "50")
if err := splitPaneCmd.Run(); err != nil {
2023-10-22 01:12:58 +05:30
return err
}
2023-10-22 14:01:39 +05:30
log.Debug("Ran command", "command", splitPaneCmd.String())
2023-10-22 01:12:58 +05:30
2023-10-22 14:01:39 +05:30
// Select the new pane
selectPaneCmd := exec.Command("tmux", "select-pane", "-t", paneName)
if err := selectPaneCmd.Run(); err != nil {
2023-10-22 01:12:58 +05:30
return err
}
2023-10-22 14:01:39 +05:30
log.Debug("Ran command", "command", selectPaneCmd.String())
2023-10-22 01:12:58 +05:30
// Change the working directory
changeDirCmd := exec.Command("tmux", "send-keys", "-t", paneName, "cd "+config.WorkingDir, "Enter")
if err := changeDirCmd.Run(); err != nil {
return err
}
log.Debug("Ran command", "command", changeDirCmd.String())
2023-10-22 14:01:39 +05:30
// Send commands to the pane
sendCommandsCmd := exec.Command("tmux", "send-keys", "-t", paneName, strings.Join(pane.ShellCommand, " && "), "Enter")
2023-10-22 01:12:58 +05:30
if err := sendCommandsCmd.Run(); err != nil {
return err
}
2023-10-22 14:01:39 +05:30
log.Debug("Ran command", "command", sendCommandsCmd.String())
}
if window.Layout != "" {
layoutCmd := exec.Command("tmux", "select-layout", "-t", windowName, window.Layout)
if err := layoutCmd.Run(); err != nil {
return err
}
log.Debug("Ran command", "command", layoutCmd.String())
}
// Create and run commands for each window inside the loop
for i := 1; i < len(config.Windows); i++ {
createWindow(config, sessionName, i)
2023-10-22 01:12:58 +05:30
}
// Select the initial window and switch to the session
defaultWindow := sessionName + ":1"
if config.StartupWindow != "" {
defaultWindow = config.StartupWindow
}
selectWindowCmd := exec.Command("tmux", "select-window", "-t", defaultWindow)
2023-10-22 01:12:58 +05:30
if err := selectWindowCmd.Run(); err != nil {
return err
}
2023-10-22 14:01:39 +05:30
log.Debug("Ran command", "command", selectWindowCmd.String())
2023-10-22 01:12:58 +05:30
2023-10-23 00:53:32 +05:30
// Select initial pane
if config.StartupPane > 0 {
defaultPane := fmt.Sprintf("%s:%d.%d", sessionName, config.StartupPane, 1)
selectPaneCmd := exec.Command("tmux", "select-pane", "-t", defaultPane)
if err := selectPaneCmd.Run(); err != nil {
return err
}
log.Debug("Ran command", "command", selectPaneCmd.String())
}
2023-10-22 14:01:39 +05:30
if config.Attach {
switchSessionCmd := exec.Command("tmux", "switch-client", "-t", sessionName)
if err := switchSessionCmd.Run(); err != nil {
return err
}
log.Debug("Ran command", "command", switchSessionCmd.String())
}
}
return nil
}
func createWindow(config *projectconfig.Configuration, sessionName string, index int) error {
if index >= len(config.Windows) {
return nil
}
window := config.Windows[index]
windowName := fmt.Sprintf("%s:%d", sessionName, index+1)
// Create a new window
createWindowCmd := exec.Command("tmux", "new-window", "-t", sessionName, "-n", windowName)
if err := createWindowCmd.Run(); err != nil {
return err
}
log.Debug("Ran command", "command", createWindowCmd.String())
// Change the working directory for the window
changeDirCmd := exec.Command("tmux", "send-keys", "-t", windowName, "cd "+config.WorkingDir, "Enter")
if err := changeDirCmd.Run(); err != nil {
return err
}
log.Debug("Ran command", "command", changeDirCmd.String())
// Send commands to the window
sendCommandsCmd := exec.Command("tmux", "send-keys", "-t", windowName, strings.Join(window.Panes[0].ShellCommand, " && "), "Enter")
if err := sendCommandsCmd.Run(); err != nil {
return err
}
log.Debug("Ran command", "command", sendCommandsCmd.String())
// Rename the window to the specified name
renameWindowCmd := exec.Command("tmux", "rename-window", "-t", windowName, window.WindowName)
if err := renameWindowCmd.Run(); err != nil {
return err
}
log.Debug("Ran command", "command", renameWindowCmd.String())
// Create and run commands for additional panes in the window
for j, pane := range window.Panes[1:] {
paneName := fmt.Sprintf("%s:%d.%d", sessionName, index+1, j+2)
// Split the window horizontally
splitPaneCmd := exec.Command("tmux", "split-window", "-t", windowName, "-h", "-p", "50")
if err := splitPaneCmd.Run(); err != nil {
return err
}
log.Debug("Ran command", "command", splitPaneCmd.String())
// Select the new pane
selectPaneCmd := exec.Command("tmux", "select-pane", "-t", paneName)
if err := selectPaneCmd.Run(); err != nil {
return err
}
log.Debug("Ran command", "command", selectPaneCmd.String())
// Change the working directory for the pane
changeDirCmd := exec.Command("tmux", "send-keys", "-t", paneName, "cd "+config.WorkingDir, "Enter")
if err := changeDirCmd.Run(); err != nil {
return err
}
log.Debug("Ran command", "command", changeDirCmd.String())
2023-10-22 14:01:39 +05:30
// Send commands to the pane
sendCommandsCmd := exec.Command("tmux", "send-keys", "-t", paneName, strings.Join(pane.ShellCommand, " && "), "Enter")
if err := sendCommandsCmd.Run(); err != nil {
return err
}
log.Debug("Ran command", "command", sendCommandsCmd.String())
}
if window.Layout != "" {
layoutCmd := exec.Command("tmux", "select-layout", "-t", windowName, window.Layout)
if err := layoutCmd.Run(); err != nil {
2023-10-22 01:12:58 +05:30
return err
}
2023-10-22 14:01:39 +05:30
log.Debug("Ran command", "command", layoutCmd.String())
2023-10-22 01:12:58 +05:30
}
return nil
}
2023-10-23 01:22:07 +05:30
func KillTmuxSession(sessionName string) error {
killSessionCmd := exec.Command("tmux", "kill-session", "-t", sessionName)
if err := killSessionCmd.Run(); err != nil {
return err
}
log.Debug("Ran command", "command", killSessionCmd.String())
return nil
}