feat: Add create command

Author: tyzerrr

cmd/sessionizer.go

@@ -46,10 +46,12 @@ func run(ctx context.Context, cmd *cli.Command) error {
 func runWithHandler(h handler.ISessionHandler, ctx context.Context, cmd *cli.Command) error {
 	args := cmd.Args().Slice()
 	if len(args) > 0 && args[0] == "list" {
-		return h.GrabExistingSession()
+		return h.GrabExistingSession(ctx)
+	} else if len(args) > 0 && args[0] == "create" {
+		return h.CreateNewProjectSession(ctx)
 	} else if len(args) > 0 {
 		return ErrNoSuchCmd
 	} else {
-		return h.NewSession()
+		return h.NewSession(ctx)
 	}
 }

cmd/sessionizer_test.go

@@ -28,11 +28,15 @@ func newMockSessionHandler() handler.ISessionHandler {
 	return &MockSessionHandler{}
 }
 
-func (mh *MockSessionHandler) NewSession() error {
+func (mh *MockSessionHandler) NewSession(ctx context.Context) error {
 	return nil
 }
 
-func (mh *MockSessionHandler) GrabExistingSession() error {
+func (mh *MockSessionHandler) GrabExistingSession(ctx context.Context) error {
+	return nil
+}
+
+func (mh *MockSessionHandler) CreateNewProjectSession(ctx context.Context) error {
 	return nil
 }
 

go.mod

@@ -3,3 +3,9 @@ module github.com/TlexCypher/my-tmux-sessionizer
 go 1.24.0
 
 require github.com/urfave/cli/v3 v3.0.0-beta1
+
+require (
+	github.com/chzyer/readline v0.0.0-20180603132655-2972be24d48e // indirect
+	github.com/manifoldco/promptui v0.9.0 // indirect
+	golang.org/x/sys v0.0.0-20181122145206-62eef0e2fa9b // indirect
+)

go.sum

@@ -1,10 +1,18 @@
+github.com/chzyer/logex v1.1.10/go.mod h1:+Ywpsq7O8HXn0nuIou7OrIPyXbp3wmkHB+jjWRnGsAI=
+github.com/chzyer/readline v0.0.0-20180603132655-2972be24d48e h1:fY5BOSpyZCqRo5OhCuC+XN+r/bBCmeuuJtjz+bCNIf8=
+github.com/chzyer/readline v0.0.0-20180603132655-2972be24d48e/go.mod h1:nSuG5e5PlCu98SY8svDHJxuZscDgtXS6KTTbou5AhLI=
+github.com/chzyer/test v0.0.0-20180213035817-a1ea475d72b1/go.mod h1:Q3SI9o4m/ZMnBNeIyt5eFwwo7qiLfzFZmjNmxjkiQlU=
 github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
 github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
+github.com/manifoldco/promptui v0.9.0 h1:3V4HzJk1TtXW1MTZMP7mdlwbBpIinw3HztaIlYthEiA=
+github.com/manifoldco/promptui v0.9.0/go.mod h1:ka04sppxSGFAtxX0qhlYQjISsg9mR4GWtQEhdbn6Pgg=
 github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
 github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
 github.com/stretchr/testify v1.9.0 h1:HtqpIVDClZ4nwg75+f6Lvsy/wHu+3BoSGCbBAcpTsTg=
 github.com/stretchr/testify v1.9.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=
 github.com/urfave/cli/v3 v3.0.0-beta1 h1:6DTaaUarcM0wX7qj5Hcvs+5Dm3dyUTBbEwIWAjcw9Zg=
 github.com/urfave/cli/v3 v3.0.0-beta1/go.mod h1:FnIeEMYu+ko8zP1F9Ypr3xkZMIDqW3DR92yUtY39q1Y=
+golang.org/x/sys v0.0.0-20181122145206-62eef0e2fa9b h1:MQE+LT/ABUuuvEZ+YQAMSXindAdUh7slEmAkup74op4=
+golang.org/x/sys v0.0.0-20181122145206-62eef0e2fa9b/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
 gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
 gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=

handler/handler.go

@@ -3,18 +3,24 @@ package handler
 import (
 	"bufio"
 	"bytes"
+	"context"
 	"errors"
 	"fmt"
 	"io"
 	"os"
 	"os/exec"
 	"path/filepath"
 	"strings"
+
+	"github.com/manifoldco/promptui"
 )
 
+var configFiles = []string{"./.tmux-sessionizer", "~/.tmux-sessionizer"}
+
 type ISessionHandler interface {
-	NewSession() error
-	GrabExistingSession() error
+	NewSession(ctx context.Context) error
+	GrabExistingSession(ctx context.Context) error
+	CreateNewProjectSession(ctx context.Context) error
 }
 
 type SessionHandler struct{}
@@ -47,7 +53,6 @@ func (sh *SessionHandler) newTmuxCmd(name string, args ...string) *exec.Cmd {
 }
 
 func (sh *SessionHandler) readConfig() *config {
-	configFiles := []string{"./.tmux-sessionizer", "~/.tmux-sessionizer"}
 	for _, cf := range configFiles {
 		config, err := sh.parseConfig(cf)
 		if err == nil {
@@ -124,7 +129,7 @@ func (sh *SessionHandler) replaceHomeDir(fullpath string) (string, error) {
 	return strings.Replace(fullpath, home, "~", 1), nil
 }
 
-func (sh *SessionHandler) NewSession() error {
+func (sh *SessionHandler) NewSession(ctx context.Context) error {
 	config := sh.readConfig()
 	if config == nil {
 		return errors.New("failed to create new project from projects")
@@ -209,7 +214,7 @@ func (sh *SessionHandler) toFzf(input bytes.Buffer) (bytes.Buffer, error) {
 	return fzfOut, nil
 }
 
-func (sh *SessionHandler) GrabExistingSession() error {
+func (sh *SessionHandler) GrabExistingSession(ctx context.Context) error {
 	tmuxCmd := exec.Command("tmux", "list-sessions", "-F", "#{session_name}")
 	var tmuxOut bytes.Buffer
 	tmuxCmd.Stdout = &tmuxOut
@@ -230,6 +235,85 @@ func (sh *SessionHandler) GrabExistingSession() error {
 	return nil
 }
 
+func (sh *SessionHandler) CreateNewProjectSession(ctx context.Context) error {
+	parentDirCandidatesSet := make(map[string]struct{}, 0)
+	for _, configFile := range configFiles {
+		path, err := sh.expandPath(configFile)
+		if err != nil {
+			return fmt.Errorf("failed to expand path: %w", err)
+		}
+		file, err := os.Open(path)
+		if err != nil {
+			return fmt.Errorf("failed to open config file: %w", err)
+		}
+		defer file.Close()
+
+		scanner := bufio.NewScanner(file)
+
+		for scanner.Scan() {
+			line := scanner.Text()
+			if strings.HasPrefix(line, "default=") {
+				raw := strings.TrimPrefix(line, "default=")
+				projects := strings.Split(raw, ",")
+				for _, pr := range projects {
+					trimmed := strings.TrimSpace(pr)
+					parentDirCandidatesSet[trimmed] = struct{}{}
+				}
+			}
+		}
+	}
+
+	parentDirCandidates := make([]string, 0)
+	for key, _ := range parentDirCandidatesSet {
+		parentDirCandidates = append(parentDirCandidates, key)
+	}
+
+	for {
+		chooseParentDirPrompt := promptui.Select{
+			Label: "Which project do you choose to create a new project (session) ?",
+			Items: parentDirCandidates,
+		}
+		_, parentDir, err := chooseParentDirPrompt.Run()
+		if err != nil {
+			return fmt.Errorf("failed to choose a parent directory: %w", err)
+		}
+
+		newProjectNamePrompt := promptui.Prompt{
+			Label: "Enter a new project name",
+		}
+		newProjectName, err := newProjectNamePrompt.Run()
+		if err != nil {
+			return fmt.Errorf("failed to get a new project name: %w", err)
+		}
+		newProjectPath, err := sh.expandPath(filepath.Join(parentDir, fmt.Sprintf("%v/", newProjectName)))
+		if err != nil {
+			return fmt.Errorf("failed to expand path: %w", err)
+		}
+		_, err = os.Stat(newProjectPath)
+		if err == nil {
+			fmt.Printf("%v is already exist\n", newProjectPath)
+			continue
+		}
+		if os.IsNotExist(err) {
+			if err := exec.Command("mkdir", "-p", newProjectPath).Run(); err != nil {
+				return fmt.Errorf("failed to create a new project: %w", err)
+			}
+			if sh.isInSession() {
+				return sh.switchToNewClient(newProjectName, newProjectPath)
+			}
+			if err := sh.newTmuxCmd(
+				"tmux", "new-session", "-s",
+				newProjectName, "-c", newProjectPath).Run(); err != nil {
+				return fmt.Errorf("failed to create new session %w: ", err)
+			}
+			break
+		} else {
+			return fmt.Errorf("failed to create a new project: %w", err)
+		}
+	}
+	return nil
+}
+
 func (sh *SessionHandler) isInSession() bool {
 	return len(os.Getenv("TMUX")) > 0
 }