cli context utility functions (#481)

Signed-off-by: David Chung <[email protected]>

Author: David Chung

pkg/cli/context.go

@@ -145,8 +145,8 @@ func (c *Context) getFromFlag(name, ftype, desc string, def interface{}) (interf
 	return nil, nil
 }
 
-// returns true if the value v is missing of the type t
-func missing(t string, v interface{}) bool {
+// Missing returns true if the value v is missing of the type t
+func Missing(t string, v interface{}) bool {
 	if v == nil {
 		return true
 	}
@@ -177,7 +177,8 @@ func parseBool(text string) (bool, error) {
 	return v > 0, err
 }
 
-func (c *Context) prompt(prompt, ftype string, optional ...interface{}) (interface{}, error) {
+// Prompt handles prompting the user using the given prompt message, type string and optional values.
+func Prompt(in io.Reader, prompt, ftype string, optional ...interface{}) (interface{}, error) {
 	def, label := "", ""
 	if len(optional) > 0 {
 		def = fmt.Sprintf("%v", optional[0])
@@ -186,7 +187,7 @@ func (c *Context) prompt(prompt, ftype string, optional ...interface{}) (interfa
 		}
 	}
 
-	input := bufio.NewReader(c.input)
+	input := bufio.NewReader(in)
 	fmt.Fprintf(os.Stderr, "%s %s: ", prompt, label)
 	text, _ := input.ReadString('\n')
 	text = strings.Trim(text, " \t\n")
@@ -255,11 +256,11 @@ func (c *Context) Funcs() []template.Function {
 				}
 				// The last value in the optional var args is the value from the previous
 				// pipeline.
-				if len(optional) > 0 && !missing(ftype, optional[len(optional)-1]) {
+				if len(optional) > 0 && !Missing(ftype, optional[len(optional)-1]) {
 					return optional[len(optional)-1], nil
 				}
 
-				return c.prompt(prompt, ftype, optional...)
+				return Prompt(c.input, prompt, ftype, optional...)
 			},
 		},
 	}

pkg/cli/context_test.go

@@ -11,12 +11,12 @@ import (
 )
 
 func TestMissing(t *testing.T) {
-	require.True(t, missing("string", ""))
-	require.True(t, missing("int", 0))
-	require.True(t, missing("float", 0.))
-	require.True(t, missing("bool", none))
-	require.False(t, missing("bool", false))
-	require.False(t, missing("bool", true))
+	require.True(t, Missing("string", ""))
+	require.True(t, Missing("int", 0))
+	require.True(t, Missing("float", 0.))
+	require.True(t, Missing("bool", none))
+	require.False(t, Missing("bool", false))
+	require.False(t, Missing("bool", true))
 }
 
 func TestContext(t *testing.T) {