[configure] allow configuring project type

* allow user to configure the project type for given code

Author: Pavan Kumar

configure_cmd.go

@@ -4,6 +4,8 @@ import (
 	"fmt"
 	"io"
 	"os"
+	"sort"
+	"strconv"
 
 	"github.com/facebookgo/stackerr"
 	"github.com/spf13/cobra"
@@ -119,6 +121,70 @@ func (c *configureCmd) parserEmail(e *env, args []string) error {
 	return nil
 }
 
+func (c *configureCmd) projectType(e *env, args []string) error {
+	config, err := configFromDir(e.Root)
+	if err != nil {
+		return err
+	}
+	if len(args) > 1 {
+		return stackerr.Newf("Invalid args: %v, only an optional project type argument is expected.", args)
+	}
+	validTypes := map[string]int{"parse": parseFormat}
+	invertedTypes := map[int]string{parseFormat: "parse"}
+	numKeys := len(validTypes)
+	var validKeys []string
+	for key := range validTypes {
+		validKeys = append(validKeys, key)
+	}
+	sort.Strings(validKeys)
+	var selectionString string
+	for n, key := range validKeys {
+		selectionString += fmt.Sprintf("%d: %s\n", 1+n, key)
+	}
+
+	selectedProjectType := -1
+	if len(args) != 0 {
+		projectType, ok := validTypes[args[0]]
+		if !ok {
+			return stackerr.Newf("Invalid projectType: %v, valid types are: \n%s", selectionString)
+		}
+		selectedProjectType = projectType
+	}
+
+	for i := 0; i < 3; i++ {
+		fmt.Fprintf(e.Out, `Select from the listed project types:
+%s
+Enter a number between 1 and %d: `,
+			selectionString,
+			numKeys,
+		)
+		var selection string
+		fmt.Fscanf(e.In, "%s\n", &selection)
+		num, err := strconv.Atoi(selection)
+		if err != nil || num < 1 || num > numKeys {
+			fmt.Fprintf(e.Err, "Invalid selection. Please enter a number between 1 and %d\n", numKeys)
+			continue
+		}
+		projectType, ok := validTypes[validKeys[num-1]]
+		if !ok {
+			return stackerr.Newf("Invalid projectType: %v, valid types are: \n%s", selectionString)
+		}
+		selectedProjectType = projectType
+		break
+	}
+	if selectedProjectType == -1 {
+		return stackerr.Newf("Could not make a selection. Please try again.")
+	}
+
+	config.getProjectConfig().Type = selectedProjectType
+	if err := storeProjectConfig(e, config); err != nil {
+		fmt.Fprintln(e.Err, "Could not save selected project type to project config")
+		return err
+	}
+	fmt.Fprintf(e.Out, "Successfully set project type to: %v\n", invertedTypes[selectedProjectType])
+	return nil
+}
+
 func newConfigureCmd(e *env) *cobra.Command {
 	var c configureCmd
 
@@ -143,12 +209,21 @@ func newConfigureCmd(e *env) *cobra.Command {
 	cmd.AddCommand(keyCmd)
 
 	emailCmd := &cobra.Command{
-		Use:   "email",
-		Short: "Configures the parser email for this project",
-		Long:  "Configures the parser email for current project.",
-		Run:   runWithArgs(e, c.parserEmail),
+		Use:     "email",
+		Short:   "Configures the parser email for this project",
+		Long:    "Configures the parser email for current project.",
+		Run:     runWithArgs(e, c.parserEmail),
+		Aliases: []string{"user"},
 	}
 	cmd.AddCommand(emailCmd)
 
+	projectCmd := &cobra.Command{
+		Use:   "project",
+		Short: "Set the project type to one among listed options",
+		Long:  "Set the project type to one among listed options. For instance, 'parse'",
+		Run:   runWithArgs(e, c.projectType),
+	}
+	cmd.AddCommand(projectCmd)
+
 	return cmd
 }

configure_cmd_test.go

@@ -118,3 +118,35 @@ func TestParserEmail(t *testing.T) {
 	ensure.Err(t, c.parserEmail(h.env, nil), regexp.MustCompile("Invalid args:"))
 	ensure.Err(t, c.parserEmail(h.env, []string{"a", "b"}), regexp.MustCompile("Invalid args:"))
 }
+
+func TestProjectType(t *testing.T) {
+	t.Parallel()
+	h := newHarness(t)
+	defer h.Stop()
+
+	h.makeEmptyRoot()
+	ensure.Nil(t, (&newCmd{}).cloneSampleCloudCode(h.env, &app{Name: "test"}, false, false))
+
+	c := &configureCmd{}
+	err := c.projectType(h.env, []string{"1", "2"})
+	ensure.Err(t, err, regexp.MustCompile("only an optional project type argument is expected"))
+
+	h.env.In = ioutil.NopCloser(strings.NewReader("invalid\n"))
+	err = c.projectType(h.env, nil)
+	ensure.StringContains(t, h.Err.String(), "Invalid selection. Please enter a number")
+	ensure.Err(t, err, regexp.MustCompile("Could not make a selection. Please try again."))
+	h.Err.Reset()
+	h.Out.Reset()
+
+	h.env.In = ioutil.NopCloser(strings.NewReader("0\n"))
+	err = c.projectType(h.env, nil)
+	ensure.StringContains(t, h.Err.String(), "Please enter a number between 1 and")
+	ensure.Err(t, err, regexp.MustCompile("Could not make a selection. Please try again."))
+	h.Err.Reset()
+	h.Out.Reset()
+
+	h.env.In = ioutil.NopCloser(strings.NewReader("1\n"))
+	err = c.projectType(h.env, nil)
+	ensure.StringContains(t, h.Out.String(), "Successfully set project type to: parse")
+	ensure.Nil(t, err)
+}