workon/tmux/project.go

package tmux

import (
	"errors"
	"fmt"
	"io/ioutil"
	"os"
	"os/exec"
	"path"
	"path/filepath"
	"regexp"
	"strings"
	"syscall"

	"github.com/mitchellh/go-homedir"
	"gopkg.in/yaml.v2"
)

type Project struct {
	Name           string   `yaml:"name"`
	Root           string   `yaml:"root"`
	OnProjectStart []string `yaml:"on_project_start,omitempty"`
	Windows        []Window `yaml:"windows"`
}

func StartProject(name string) {
	p, err := LoadProject(name)
	if err != nil {
		fmt.Println("Unable to load project:", err)
		os.Exit(1)
	}

	// Run startup commands
	if len(p.OnProjectStart) > 0 {
		for _, command := range p.OnProjectStart {
			args := strings.Fields(command)
			cmd := exec.Command(args[0], args[1:]...)
			cmd.Dir = p.GetRoot()
			cmd.Stdout = os.Stdout
			cmd.Stderr = os.Stderr
			err := cmd.Run()
			if err != nil {
				fmt.Println("Unable to run start command:", err)
				os.Exit(1)
			}
		}
	}

	tmux := CreateTmux(false)

	if !sessionExists(name) {
		tmux.Run("new-session", "-d", "-s", name, "-n", p.Windows[0].Name, "-c", p.Windows[0].Root)

		for index, window := range p.Windows {
			if index > 0 {
				window.Create(tmux)
			}

			window.SendCommands(tmux)
		}

		p.Windows[0].Focus(tmux)
	}

	tmux.Attach(name)
}

func ListProjects() error {
	configDir := getConfigDir()
	files, err := ioutil.ReadDir(configDir)
	if err != nil {
		return err
	}

	for _, file := range files {
		if file.IsDir() {
			continue
		}
		name := file.Name()
		ext := filepath.Ext(name)
		fmt.Printf("%s\n", name[:len(name)-len(ext)])
	}
	return nil
}

// LoadProject loads and parses the config for the given project.
func LoadProject(name string) (*Project, error) {
	project := &Project{}

	fileName := getConfigFilePath(name)

	data, err := ioutil.ReadFile(fileName)
	if err != nil {
		return project, err
	}

	err = yaml.Unmarshal(data, project)
	if len(project.Windows) < 1 {
		return project, errors.New("No windows defined")
	}

	rootPath := project.GetRoot()
	for index, window := range project.Windows {
		project.Windows[index].ID = fmt.Sprintf("%s:%d", project.Name, index)
		project.Windows[index].Root = filepath.Join(rootPath, window.Root)
	}

	return project, err
}

func NewProject(name string) error {
	project := &Project{
		Name:           name,
		Root:           "~/",
		OnProjectStart: []string{""},
		Windows:        make([]Window, 3),
	}

	project.Windows[0] = Window{
		Name:     "shell",
		Commands: []string{""},
	}

	project.Windows[1] = Window{
		Name:     "server",
		Commands: []string{""},
	}

	project.Windows[2] = Window{
		Name:     "logs",
		Commands: []string{""},
	}

	return project.Save()
}

func EditProject(name string) error {
	fileName := getConfigFilePath(name)

	_, err := os.Stat(fileName)
	if err != nil {
		return errors.New("Config file does not exist")
	}

	editorName := os.Getenv("EDITOR")
	if editorName == "" {
		return errors.New("EDITOR variable not defined")
	}

	editor, err := exec.LookPath(editorName)
	if err != nil {
		return err
	}

	return syscall.Exec(editor, []string{editorName, fileName}, os.Environ())
}

func (p *Project) Save() error {
	fileName := getConfigFilePath(p.Name)

	_, err := os.Stat(fileName)
	if err == nil {
		return errors.New("Config file already exists")
	}

	data, err := yaml.Marshal(p)
	if err != nil {
		return err
	}
	return ioutil.WriteFile(fileName, data, 0644)
}

func (p *Project) GetRoot() string {
	rootPath, err := homedir.Expand(p.Root)
	if err != nil {
		fmt.Println("Unable to find root path")
	}
	return rootPath
}

func getConfigDir() string {
	home, _ := homedir.Dir()
	return path.Join(home, ".workon")
}

func getConfigFilePath(name string) string {
	return path.Join(getConfigDir(), name+".yml")
}

func sessionExists(name string) bool {
	t := Tmux{}
	result, err := t.Exec("ls")
	if err != nil {
		return false
	}

	re := regexp.MustCompile(fmt.Sprintf("^%s:", name))
	return re.MatchString(string(result))
}