pee/controller/tmux.go

191 lines
6.2 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
createSessionCmd := exec.Command("tmux", "new-session", "-d", "-s", sessionName)
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
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
selectWindowCmd := exec.Command("tmux", "select-window", "-t", sessionName+":1")
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-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())
// 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
}