Add 'init' command to rig-admin to help the setup of a new Rig server

cmd/common/useridentifier.go

@@ -24,7 +24,7 @@ func PromptUserIndentifierUpdate() (*user.Update, error) {
 
 func PromptUserIndentifier() (*model.UserIdentifier, error) {
 	var err error
-	identifierStr, err := PromptGetInput("User Identifier (username, email or phone number):", ValidateAll)
+	identifierStr, err := PromptGetInput("Username, email or phone number:", ValidateAll)
 	if err != nil {
 		return nil, err
 	}

cmd/rig-admin/cmd/init.go

@@ -0,0 +1,153 @@
+package cmd
+
+import (
+	"context"
+	"fmt"
+	"strings"
+
+	"github.com/fatih/color"
+	"github.com/rigdev/rig-go-api/api/v1/project"
+	"github.com/rigdev/rig-go-api/api/v1/user"
+	"github.com/rigdev/rig-go-api/model"
+	"github.com/rigdev/rig/cmd/common"
+	"github.com/rigdev/rig/pkg/auth"
+	"github.com/rigdev/rig/pkg/errors"
+
+	auth_service "github.com/rigdev/rig/internal/service/auth"
+	project_service "github.com/rigdev/rig/internal/service/project"
+	user_service "github.com/rigdev/rig/internal/service/user"
+	"github.com/spf13/cobra"
+)
+
+func init() {
+	init := &cobra.Command{
+		Use:   "init",
+		RunE:  register(initialize),
+		Short: "Sets up the Rig server",
+	}
+	rootCmd.AddCommand(init)
+}
+
+func initialize(ctx context.Context, cmd *cobra.Command, us user_service.Service, ps project_service.Service) error {
+	printBanner("Create user")
+	if err := createUser(ctx, us); err != nil {
+		return err
+	}
+	fmt.Println()
+
+	printBanner("Project")
+	if err := createProject(ctx, ps); err != nil {
+		return err
+	}
+	fmt.Println()
+
+	fmt.Println("Rig is set setup :)")
+
+	return nil
+}
+
+func createUser(ctx context.Context, us user_service.Service) error {
+	_, total, err := us.List(ctx, &model.Pagination{
+		Limit: 1,
+	}, "")
+	if err != nil {
+		return err
+	}
+
+	shouldCreateNew := false
+	if total == 0 {
+		fmt.Println("There are no admin users, create a new one")
+		shouldCreateNew = true
+	} else {
+		pluralS := "s"
+		if total == 1 {
+			pluralS = ""
+		}
+		fmt.Printf("There are already %v admin user%s.\n", total, pluralS)
+		shouldCreateNew, err = common.PromptConfirm("Do you wish to create a new one anyways?", false)
+		if err != nil {
+			return err
+		}
+	}
+
+	if !shouldCreateNew {
+		return nil
+	}
+
+	email, err := common.PromptGetInput("Email of the new user:", common.ValidateEmail)
+	if err != nil {
+		return err
+	}
+	password, err := common.GetPasswordPrompt("Password:")
+	if err != nil {
+		return err
+	}
+
+	ctx = auth.WithClaims(ctx, auth_service.ProjectClaims{
+		UseProjectID: auth.RigProjectID,
+	})
+	_, err = us.CreateUser(ctx, &model.RegisterMethod{
+		Method: &model.RegisterMethod_System_{},
+	}, []*user.Update{
+		{Field: &user.Update_Email{Email: email}},
+		{Field: &user.Update_Password{Password: password}},
+	})
+	if err != nil {
+		return err
+	}
+	fmt.Println("User created successfully!")
+
+	return nil
+}
+
+func createProject(ctx context.Context, ps project_service.Service) error {
+	_, total, err := ps.List(ctx, &model.Pagination{
+		Limit: 1,
+	})
+	if err != nil {
+		return err
+	}
+
+	if total > 0 {
+		msg := fmt.Sprintf("There already exist %v projects. Do you want to create a new one anyways?", total)
+		use, err := common.PromptConfirm(msg, false)
+		if err != nil {
+			return err
+		}
+		if !use {
+			return nil
+		}
+	} else {
+		fmt.Println("There are no projects, create a new one.")
+	}
+
+	for {
+		projectName, err := common.PromptGetInput("Project name:", common.ValidateNonEmpty)
+		if err != nil {
+			return err
+		}
+
+		_, err = ps.CreateProject(ctx, []*project.Update{{Field: &project.Update_Name{Name: projectName}}})
+		if errors.IsAlreadyExists(err) {
+			fmt.Println("A project with that name already exists")
+			continue
+		}
+		if err != nil {
+			return err
+		}
+		break
+	}
+	fmt.Println("Project created succesfully!")
+
+	return nil
+}
+
+func printBanner(s string) {
+	color.Cyan(makeBanner(s))
+}
+func makeBanner(s string) string {
+	totalLength := 60
+	sideLength := (totalLength - len(s)) / 2
+	side := strings.Repeat("-", sideLength)
+	return side + " " + s + " " + side
+}

cmd/rig/cmd/cmd_config/context.go

@@ -83,3 +83,13 @@ func CreateContext(cfg *Config) error {
 
 	return cfg.Save()
 }
+
+func ConfigInit(cfg *Config) error {
+	if ok, err := common.PromptConfirm("Do you want to configure a new context", true); err != nil {
+		return err
+	} else if !ok {
+		return nil
+	}
+
+	return CreateContext(cfg)
+}

go.mod

@@ -47,6 +47,7 @@ require (
 
 require (
 	firebase.google.com/go v3.13.0+incompatible
+	github.com/fatih/color v1.15.0
 	github.com/go-chi/chi/v5 v5.0.10
 	github.com/google/go-containerregistry v0.16.1
 	github.com/jedib0t/go-pretty/v6 v6.4.6
@@ -64,6 +65,8 @@ require (
 	github.com/docker/cli v24.0.0+incompatible // indirect
 	github.com/docker/docker-credential-helpers v0.7.0 // indirect
 	github.com/google/gnostic-models v0.6.8 // indirect
+	github.com/mattn/go-colorable v0.1.13 // indirect
+	github.com/mattn/go-isatty v0.0.18 // indirect
 	github.com/mattn/go-runewidth v0.0.13 // indirect
 	github.com/mitchellh/go-homedir v1.1.0 // indirect
 	github.com/rivo/uniseg v0.2.0 // indirect

go.sum

@@ -174,6 +174,8 @@ github.com/envoyproxy/go-control-plane v0.9.7/go.mod h1:cwu0lG7PUMfa9snN8LXBig5y
 github.com/envoyproxy/go-control-plane v0.9.9-0.20201210154907-fd9021fe5dad/go.mod h1:cXg6YxExXjJnVBQHBLXeUAgxn2UodCpnH306RInaBQk=
 github.com/envoyproxy/go-control-plane v0.9.10-0.20210907150352-cf90f659a021/go.mod h1:AFq3mo9L8Lqqiid3OhADV3RfLJnjiw63cSpi+fDTRC0=
 github.com/envoyproxy/protoc-gen-validate v0.1.0/go.mod h1:iSmxcyjqTsJpI2R4NaDN7+kN2VEUnK/pcBlmesArF7c=
+github.com/fatih/color v1.15.0 h1:kOqh6YHBtK8aywxGerMG2Eq3H6Qgoqeo13Bk2Mv/nBs=
+github.com/fatih/color v1.15.0/go.mod h1:0h5ZqXfHYED7Bhv2ZJamyIOUej9KtShiJESRwBDUSsw=
 github.com/felixge/httpsnoop v1.0.1 h1:lvB5Jl89CsZtGIWuTcDM1E/vkVs49/Ml7JJe07l8SPQ=
 github.com/felixge/httpsnoop v1.0.1/go.mod h1:m8KPJKqk1gH5J9DgRY2ASl2lWCfGKXixSwevea8zH2U=
 github.com/frankban/quicktest v1.14.4 h1:g2rn0vABPOOXmZUj+vbmUp0lPoXEMuhTpIluN0XL9UY=
@@ -362,6 +364,11 @@ github.com/mailru/easyjson v0.7.7 h1:UGYAvKxe3sBsEDzO8ZeWOSlIQfWFlxbzLZe7hwFURr0
 github.com/mailru/easyjson v0.7.7/go.mod h1:xzfreul335JAWq5oZzymOObrkdz5UnU4kGfJJLY9Nlc=
 github.com/manifoldco/promptui v0.9.0 h1:3V4HzJk1TtXW1MTZMP7mdlwbBpIinw3HztaIlYthEiA=
 github.com/manifoldco/promptui v0.9.0/go.mod h1:ka04sppxSGFAtxX0qhlYQjISsg9mR4GWtQEhdbn6Pgg=
+github.com/mattn/go-colorable v0.1.13 h1:fFA4WZxdEF4tXPZVKMLwD8oUnCTTo08duU7wxecdEvA=
+github.com/mattn/go-colorable v0.1.13/go.mod h1:7S9/ev0klgBDR4GtXTXX8a3vIGJpMovkB8vQcUbaXHg=
+github.com/mattn/go-isatty v0.0.16/go.mod h1:kYGgaQfpe5nmfYZH+SKPsOc2e4SrIfOl2e/yFXSvRLM=
+github.com/mattn/go-isatty v0.0.18 h1:DOKFKCQ7FNG2L1rbrmstDN4QVRdS89Nkh85u68Uwp98=
+github.com/mattn/go-isatty v0.0.18/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=
 github.com/mattn/go-runewidth v0.0.13 h1:lTGmDsbAYt5DmK6OnoV7EuIF1wEIFAcxld6ypU4OSgU=
 github.com/mattn/go-runewidth v0.0.13/go.mod h1:Jdepj2loyihRzMpdS35Xk/zdY8IAYHsh153qUoGf23w=
 github.com/matttproud/golang_protobuf_extensions v1.0.1/go.mod h1:D8He9yQNgCq6Z5Ld7szi9bcBfOoFv/3dc6xSMkL2PC0=
@@ -431,6 +438,8 @@ github.com/prometheus/procfs v0.0.2/go.mod h1:TjEm7ze935MbeOT/UhFTIMYKhuLP4wbCsT
 github.com/prometheus/procfs v0.0.3/go.mod h1:4A/X28fw3Fc593LaREMrKMqOKvUAntwMDaekg4FpcdQ=
 github.com/prometheus/procfs v0.9.0 h1:wzCHvIvM5SxWqYvwgVL7yJY8Lz3PKn49KQtpgMYJfhI=
 github.com/prometheus/procfs v0.9.0/go.mod h1:+pB4zwohETzFnmlpe6yd2lSc+0/46IYZRB/chUwxUZY=
+github.com/rigdev/rig-go-api v0.0.0-20230905151115-0a42d524bfae h1:suZ1T6nszSfIP1vzfZjCjUEkfH8+Hw1CK78w/xPmkXM=
+github.com/rigdev/rig-go-api v0.0.0-20230905151115-0a42d524bfae/go.mod h1:fraLUk9ekeQvvu5bOmUmpNAsBqG2QzqWU9wzGbPwB/w=
 github.com/rigdev/rig-go-sdk v0.0.0-20230902170251-18fbb97d0692 h1:9K+F3uKbXaTC5iR6sy6tBodlmQVp8g+tB3o0hcWKtck=
 github.com/rigdev/rig-go-sdk v0.0.0-20230902170251-18fbb97d0692/go.mod h1:mC8jVtKLH3RusvgTaHSn3lJTyaVdgsSizn8Sjb5xaoM=
 github.com/rivo/uniseg v0.2.0 h1:S1pD9weZBuJdFmowNwbpi7BJ8TNftyUImj/0WQi72jY=
@@ -719,10 +728,12 @@ golang.org/x/sys v0.0.0-20220520151302-bc2c85ada10a/go.mod h1:oPkhp1MJrh7nUepCBc
 golang.org/x/sys v0.0.0-20220704084225-05e143d24a9e/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
 golang.org/x/sys v0.0.0-20220715151400-c0bba94af5f8/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
 golang.org/x/sys v0.0.0-20220722155257-8c9f86f7a55f/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
+golang.org/x/sys v0.0.0-20220811171246-fbc7d0a398ab/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
 golang.org/x/sys v0.0.0-20220906165534-d0df966e6959/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
 golang.org/x/sys v0.0.0-20220908164124-27713097b956/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
 golang.org/x/sys v0.1.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
 golang.org/x/sys v0.5.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
+golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
 golang.org/x/sys v0.10.0 h1:SqMFp9UcQJZa+pmYuAKjd9xq1f0j5rLcDIk0mj4qAsA=
 golang.org/x/sys v0.10.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
 golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo=