add as command line option and usage text

1 files changed, 11 insertions, 0 deletions

diff --git a/cmd/ponzu/main.go b/cmd/ponzu/main.go
index f44f626..494158d 100644
--- a/cmd/ponzu/main.go
+++ b/cmd/ponzu/main.go
@@ -88,6 +88,10 @@ func main() {
 		case "version", "v":
 			fmt.Println(usageVersion)
 			os.Exit(0)
+
+		case "add", "a":
+			fmt.Println(usageAdd)
+			os.Exit(0)
 		}
 
 	case "new":
@@ -265,6 +269,13 @@ func main() {
 			fmt.Println("Input not recognized. No upgrade made. Answer as 'y' or 'n' only.")
 		}
 
+	case "add", "a":
+		// expecting two args, add and the go gettable package uri
+		if len(args) < 2 {
+			fmt.Println(usageAdd)
+			os.Exit(0)
+		}
+
 	case "":
 		fmt.Println(usage)
 		fmt.Println(usageHelp)