feat: add --command flag to connect command (#12)

Execute a CLI command when you connect to a new session by specifying the `--command` or `-c` command.

Author: joebonneau

cmds/choose.go

@@ -65,7 +65,7 @@ func Choose() *cli.Command {
 			}
 			choice := strings.TrimSpace(cmdOutput.String())
 			// TODO: get choice from Session structs array
-			connect.Connect(choice, false)
+			connect.Connect(choice, false, "")
 			return nil
 		},
 	}

cmds/connect.go

@@ -18,14 +18,20 @@ func Connect() *cli.Command {
 				Aliases: []string{"s"},
 				Usage:   "Always switch the session (and never attach). This is useful for third-party tools like Raycast.",
 			},
+			&cli.StringFlag{
+				Name:    "command",
+				Aliases: []string{"c"},
+				Usage:   "Execute a command when connecting to a new session. Will be ignored if the session exists.",
+			},
 		},
 		Action: func(cCtx *cli.Context) error {
 			session := cCtx.Args().First()
 			alwaysSwitch := cCtx.Bool("switch")
+			command := cCtx.String("command")
 			if session == "" {
 				return cli.Exit("No session provided", 0)
 			}
-			connect.Connect(session, alwaysSwitch)
+			connect.Connect(session, alwaysSwitch, command)
 			return nil
 		},
 	}

connect/connect.go

@@ -6,12 +6,12 @@ import (
 	"github.com/joshmedeski/sesh/zoxide"
 )
 
-func Connect(choice string, alwaysSwitch bool) error {
+func Connect(choice string, alwaysSwitch bool, command string) error {
 	session := session.Determine(choice)
 	zoxide.Add(session.Path)
 	tmux.Connect(tmux.TmuxSession{
 		Name: session.Name,
 		Path: session.Path,
-	}, alwaysSwitch)
+	}, alwaysSwitch, command)
 	return nil
 }

session/determine.go

@@ -1,21 +1,18 @@
 package session
 
 import (
-	"fmt"
-	"os"
+	"log"
 )
 
 func Determine(choice string) Session {
-	path, err := DeterminPath(choice)
+	path, err := DeterminePath(choice)
 	if err != nil {
-		fmt.Println("Could't determine the session path", err)
-		os.Exit(1)
+		log.Fatal("Couldn't determine the session path", err)
 	}
 
 	name := DetermineName(path)
 	if name == "" {
-		fmt.Println("Could't determine the session name", err)
-		os.Exit(1)
+		log.Fatal("Couldn't determine the session name", err)
 	}
 
 	return Session{

session/path.go

@@ -7,7 +7,7 @@ import (
 	"github.com/joshmedeski/sesh/tmux"
 )
 
-func DeterminPath(choice string) (string, error) {
+func DeterminePath(choice string) (string, error) {
 	fullPath := dir.FullPath(choice)
 	if path.IsAbs(fullPath) {
 		return fullPath, nil

tmux/tmux.go

@@ -2,7 +2,7 @@ package tmux
 
 import (
 	"bytes"
-	"fmt"
+	"log"
 	"os"
 	"os/exec"
 	"strings"
@@ -64,6 +64,14 @@ func switchSession(session string) error {
 	return nil
 }
 
+func runPersistentCommand(session string, command string) error {
+	finalCmd := []string{"send-keys", "-t", session, command, "Enter"}
+	if _, err := tmuxCmd(finalCmd); err != nil {
+		return err
+	}
+	return nil
+}
+
 func NewSession(s TmuxSession) (string, error) {
 	out, err := tmuxCmd([]string{"new-session", "-d", "-s", s.Name, "-c", s.Path})
 	if err != nil {
@@ -72,12 +80,15 @@ func NewSession(s TmuxSession) (string, error) {
 	return out, nil
 }
 
-func Connect(s TmuxSession, alwaysSwitch bool) error {
+func Connect(s TmuxSession, alwaysSwitch bool, command string) error {
 	isSession := IsSession(s.Name)
 	if !isSession {
 		_, err := NewSession(s)
 		if err != nil {
-			fmt.Println(err)
+			log.Fatal(err)
+		}
+		if command != "" {
+			runPersistentCommand(s.Name, command)
 		}
 	}
 	isAttached := isAttached()