add root flag to sql cmd

Signed-off-by: Jason McCallister <jason@craftcms.com>
craftcms · Apr 7, 2020 · de62b78 · de62b78
1 parent a7706e6
commit de62b78
Showing 1 changed file with 18 additions and 9 deletions.
diff --git a/internal/cmd/sql.go b/internal/cmd/sql.go
@@ -7,14 +7,23 @@ import (
 	"github.com/craftcms/nitro/internal/nitro"
 )
 
-var sqlCommand = &cobra.Command{
-	Use:   "sql",
-	Short: "Enter a SQL shell",
-	RunE: func(cmd *cobra.Command, args []string) error {
-		name := config.GetString("machine", flagMachineName)
-		engine := config.GetString("database.engine", flagDatabase)
-		version := config.GetString("database.version", flagDatabaseVersion)
+var (
+	flagRootDatabaseUser bool
 
-		return nitro.Run(nitro.NewMultipassRunner("multipass"), nitro.SQL(name, engine, version, false))
-	},
+	sqlCommand = &cobra.Command{
+		Use:   "sql",
+		Short: "Enter a SQL shell",
+		RunE: func(cmd *cobra.Command, args []string) error {
+			name := config.GetString("machine", flagMachineName)
+			engine := config.GetString("database.engine", flagDatabase)
+			version := config.GetString("database.version", flagDatabaseVersion)
+			root := flagRootDatabaseUser
+
+			return nitro.Run(nitro.NewMultipassRunner("multipass"), nitro.SQL(name, engine, version, root))
+		},
+	}
+)
+
+func init() {
+	sqlCommand.Flags().BoolVarP(&flagRootDatabaseUser, "root", "r", false, "Enter a root ")
 }