add cobra commands

1 files changed, 176 insertions, 0 deletions

diff --git a/cmd/ponzu/new.go b/cmd/ponzu/new.go
new file mode 100644
index 0000000..3c18a21
--- /dev/null
+++ b/cmd/ponzu/new.go
@@ -0,0 +1,176 @@
+package main
+
+import (
+	"errors"
+	"fmt"
+	"os"
+	"os/exec"
+	"path/filepath"
+	"strings"
+
+	"github.com/spf13/cobra"
+)
+
+var newCmd = &cobra.Command{
+	Use:   "new <projectName>",
+	Short: "creates a 'ponzu' directory by the name supplied as a parameter",
+	Long: `Creates a 'ponzu' directory by the name supplied as a parameter
+immediately following the 'new' option in the $GOPATH/src directory. Note:
+'new' depends on the program 'git' and possibly a network connection. If
+there is no local repository to clone from at the local machine's $GOPATH,
+'new' will attempt to clone the 'github.com/ponzu-cms/ponzu' package from
+over the network.
+
+Example:
+$ ponzu new myProject
+> New ponzu project created at $GOPATH/src/myProject
+
+Errors will be reported, but successful commands return nothing.`,
+	RunE: func(cmd *cobra.Command, args []string) error {
+		if len(args) < 1 {
+			return errors.New("project name not supplied")
+		}
+
+		return newProjectInDir(args[0])
+	},
+}
+
+func newProjectInDir(path string) error {
+	// set path to be nested inside $GOPATH/src
+	gopath, err := getGOPATH()
+	if err != nil {
+		return err
+	}
+	path = filepath.Join(gopath, "src", path)
+
+	// check if anything exists at the path, ask if it should be overwritten
+	if _, err = os.Stat(path); !os.IsNotExist(err) {
+		fmt.Println("Path exists, overwrite contents? (y/N):")
+
+		answer, err := getAnswer()
+		if err != nil {
+			return err
+		}
+
+		switch answer {
+		case "n", "no", "\r\n", "\n", "":
+			fmt.Println("")
+
+		case "y", "yes":
+			err := os.RemoveAll(path)
+			if err != nil {
+				return fmt.Errorf("Failed to overwrite %s. \n%s", path, err)
+			}
+
+			return createProjectInDir(path)
+
+		default:
+			fmt.Println("Input not recognized. No files overwritten. Answer as 'y' or 'n' only.")
+		}
+
+		return nil
+	}
+
+	return createProjectInDir(path)
+}
+
+func createProjectInDir(path string) error {
+	gopath, err := getGOPATH()
+	if err != nil {
+		return err
+	}
+	repo := ponzuRepo
+	local := filepath.Join(gopath, "src", filepath.Join(repo...))
+	network := "https://" + strings.Join(repo, "/") + ".git"
+	if !strings.HasPrefix(path, gopath) {
+		path = filepath.Join(gopath, path)
+	}
+
+	// create the directory or overwrite it
+	err = os.MkdirAll(path, os.ModeDir|os.ModePerm)
+	if err != nil {
+		return err
+	}
+
+	if dev {
+		if fork != "" {
+			local = filepath.Join(gopath, "src", fork)
+		}
+
+		devClone := exec.Command("git", "clone", local, "--branch", "ponzu-dev", "--single-branch", path)
+		devClone.Stdout = os.Stdout
+		devClone.Stderr = os.Stderr
+
+		err = devClone.Start()
+		if err != nil {
+			return err
+		}
+
+		err = devClone.Wait()
+		if err != nil {
+			return err
+		}
+
+		err = vendorCorePackages(path)
+		if err != nil {
+			return err
+		}
+
+		fmt.Println("Dev build cloned from " + local + ":ponzu-dev")
+		return nil
+	}
+
+	// try to git clone the repository from the local machine's $GOPATH
+	localClone := exec.Command("git", "clone", local, path)
+	localClone.Stdout = os.Stdout
+	localClone.Stderr = os.Stderr
+
+	err = localClone.Start()
+	if err != nil {
+		return err
+	}
+	err = localClone.Wait()
+	if err != nil {
+		fmt.Println("Couldn't clone from", local, "- trying network...")
+
+		// try to git clone the repository over the network
+		networkClone := exec.Command("git", "clone", network, path)
+		networkClone.Stdout = os.Stdout
+		networkClone.Stderr = os.Stderr
+
+		err = networkClone.Start()
+		if err != nil {
+			fmt.Println("Network clone failed to start. Try again and make sure you have a network connection.")
+			return err
+		}
+		err = networkClone.Wait()
+		if err != nil {
+			fmt.Println("Network clone failure.")
+			// failed
+			return fmt.Errorf("Failed to clone files from local machine [%s] and over the network [%s].\n%s", local, network, err)
+		}
+	}
+
+	// create an internal vendor directory in ./cmd/ponzu and move content,
+	// management and system packages into it
+	err = vendorCorePackages(path)
+	if err != nil {
+		return err
+	}
+
+	gitDir := filepath.Join(path, ".git")
+	err = os.RemoveAll(gitDir)
+	if err != nil {
+		fmt.Println("Failed to remove .git directory from your project path. Consider removing it manually.")
+	}
+
+	fmt.Println("New ponzu project created at", path)
+	return nil
+}
+
+func init() {
+	newCmd.Flags().StringVar(&fork, "fork", "", "modify repo source for Ponzu core development")
+	newCmd.Flags().BoolVar(&dev, "dev", false, "modify environment for Ponzu core development")
+
+	rootCmd.AddCommand(newCmd)
+}