Add link subcommand. Fixes #61

cmd/puma-dev/command.go

@@ -0,0 +1,88 @@
+package main
+
+import (
+	"flag"
+	"fmt"
+	"os"
+	"path/filepath"
+	"strings"
+
+	homedir "github.com/mitchellh/go-homedir"
+	"github.com/vektra/errors"
+)
+
+func command() error {
+	switch flag.Arg(0) {
+	case "link":
+		return link()
+	default:
+		return fmt.Errorf("Unknown command: %s\n", flag.Arg(0))
+	}
+}
+
+func link() error {
+	fs := flag.NewFlagSet("link", flag.ExitOnError)
+	name := fs.String("n", "", "name to link app as")
+
+	err := fs.Parse(flag.Args()[1:])
+	if err != nil {
+		return err
+	}
+
+	var dir string
+
+	if fs.NArg() == 0 {
+		dir, err = os.Getwd()
+		if err != nil {
+			return err
+		}
+
+	} else {
+		dir, err = filepath.Abs(fs.Arg(0))
+		if err != nil {
+			return err
+		}
+
+		stat, err := os.Stat(dir)
+		if err != nil {
+			if os.IsNotExist(err) {
+				return fmt.Errorf("Invalid directory: %s", dir)
+			}
+
+			return err
+		}
+
+		if !stat.IsDir() {
+			return fmt.Errorf("Invalid directory: %s", dir)
+		}
+	}
+
+	if *name == "" {
+		*name = strings.Replace(filepath.Base(dir), ".", "_", -1)
+	}
+
+	dest, err := homedir.Expand(filepath.Join(*fDir, *name))
+	if err != nil {
+		return err
+	}
+
+	_, err = os.Stat(dest)
+	if err == nil {
+		dest, err := os.Readlink(dest)
+		if err == nil {
+			fmt.Printf("! App '%s' already exists, pointed at '%s'\n", *name, dest)
+		} else {
+			fmt.Printf("! App '%s' already exists'\n", *name)
+		}
+		return nil
+	}
+
+	err = os.Symlink(dir, dest)
+	if err != nil {
+		return errors.Context(err, "creating symlink")
+	}
+
+	fmt.Printf("+ App '%s' created, linked to '%s'\n", *name, dest)
+
+	return nil
+}

cmd/puma-dev/main.go

@@ -0,0 +1,16 @@
+package main
+
+import (
+	"flag"
+	"fmt"
+	"os"
+)
+
+func init() {
+	flag.Usage = func() {
+		fmt.Fprintf(os.Stderr, "Usage of %s:\n", os.Args[0])
+		flag.PrintDefaults()
+
+		fmt.Fprintf(os.Stderr, "\nAvailable subcommands: link\n")
+	}
+}

cmd/puma-dev/main_darwin.go

@@ -39,6 +39,16 @@ var (
 func main() {
 	flag.Parse()
 
+	if flag.NArg() > 0 {
+		err := command()
+		if err != nil {
+			fmt.Printf("Error: %s\n", err)
+			os.Exit(1)
+		}
+
+		return
+	}
+
 	if *fPow {
 		*fDomains = "dev"
 		*fDir = "~/.pow"

cmd/puma-dev/main_linux.go

@@ -28,6 +28,16 @@ var (
 func main() {
 	flag.Parse()
 
+	if flag.NArg() > 0 {
+		err := command()
+		if err != nil {
+			fmt.Printf("Error: %s\n", err)
+			os.Exit(1)
+		}
+
+		return
+	}
+
 	domains := strings.Split(*fDomains, ":")
 
 	if *fSysBind {