1 files changed, 201 insertions, 0 deletions
diff --git a/cmd/ponzu/new.go b/cmd/ponzu/new.go
new file mode 100644
index 0000000..2f16a53
--- /dev/null
+++ b/cmd/ponzu/new.go
@@ -0,0 +1,201 @@
+package main
+
+import (
+	"fmt"
+	"os"
+	"os/exec"
+	"path/filepath"
+	"strings"
+
+	"github.com/spf13/cobra"
+)
+
+var newCmd = &cobra.Command{
+	Use:   "new [flags] <project name>",
+	Short: "creates a project directory of the name supplied as a parameter",
+	Long: `Creates aproject  directory of 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 github.com/nilslice/proj
+> New ponzu project created at $GOPATH/src/github.com/nilslice/proj`,
+	RunE: func(cmd *cobra.Command, args []string) error {
+		projectName := "ponzu"
+		if len(args) > 0 {
+			projectName = args[0]
+		} else {
+			msg := "Please provide a project name."
+			msg += "\nThis will create a directory within your $GOPATH/src."
+			return fmt.Errorf("%s", msg)
+		}
+		return newProjectInDir(projectName)
+	},
+}
+
+// name2path transforns a project name to an absolute path
+func name2path(projectName string) (string, error) {
+	gopath, err := getGOPATH()
+	if err != nil {
+		return "", err
+	}
+	gosrc := filepath.Join(gopath, "src")
+
+	path := projectName
+	// support current directory
+	if path == "." {
+		path, err = os.Getwd()
+		if err != nil {
+			return "", err
+		}
+	} else {
+		path = filepath.Join(gosrc, path)
+	}
+
+	// make sure path is inside $GOPATH/src
+	srcrel, err := filepath.Rel(gosrc, path)
+	if err != nil {
+		return "", err
+	}
+	if len(srcrel) >= 2 && srcrel[:2] == ".." {
+		return "", fmt.Errorf("path '%s' must be inside '%s'", projectName, gosrc)
+	}
+	if srcrel == "." {
+		return "", fmt.Errorf("path '%s' must not be %s", path, filepath.Join("GOPATH", "src"))
+	}
+
+	_, err = os.Stat(path)
+	if err != nil && !os.IsNotExist(err) {
+		return "", err
+	}
+	if err == nil {
+		err = os.ErrExist
+	} else if os.IsNotExist(err) {
+		err = nil
+	}
+
+	return path, err
+}
+
+func newProjectInDir(path string) error {
+	path, err := name2path(path)
+	if err != nil && !os.IsNotExist(err) {
+		return err
+	}
+
+	// path exists, ask if it should be overwritten
+	if os.IsNotExist(err) {
+		fmt.Printf("Using '%s' as project directory\n", path)
+		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)
+		}
+
+		err = execAndWait("git", "clone", local, "--branch", "ponzu-dev", "--single-branch", path)
+		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
+	err = execAndWait("git", "clone", local, path)
+	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")
+
+	RegisterCmdlineCommand(newCmd)
+}