mirror of
https://github.com/andwati/zitadel.git
synced 2026-10-19 06:02:52 +00:00
94 lines
2.2 KiB
Go
94 lines
2.2 KiB
Go
package initialise
|
|
|
|
import (
|
|
"database/sql"
|
|
_ "embed"
|
|
|
|
"github.com/caos/logging"
|
|
"github.com/caos/zitadel/internal/database"
|
|
"github.com/spf13/cobra"
|
|
"github.com/spf13/viper"
|
|
|
|
//sql import
|
|
_ "github.com/lib/pq"
|
|
)
|
|
|
|
var (
|
|
user string
|
|
password string
|
|
sslCert string
|
|
sslKey string
|
|
)
|
|
|
|
const (
|
|
userFlag = "user"
|
|
passwordFlag = "password"
|
|
sslCertFlag = "ssl-cert"
|
|
sslKeyFlag = "ssl-key"
|
|
)
|
|
|
|
func New() *cobra.Command {
|
|
cmd := &cobra.Command{
|
|
Use: "init",
|
|
Short: "initialize ZITADEL instance",
|
|
Long: `Sets up the minimum requirements to start ZITADEL.
|
|
|
|
Prereqesits:
|
|
- cockroachdb
|
|
|
|
The user provided by flags needs priviledge to
|
|
- create the database if it does not exist
|
|
- see other users and create a new one if the user does not exist
|
|
- grant all rights of the ZITADEL database to the user created if not yet set
|
|
`,
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
config := new(Config)
|
|
if err := viper.Unmarshal(config); err != nil {
|
|
return err
|
|
}
|
|
return initialise(config, verifyUser, verifyDB, verifyGrant)
|
|
},
|
|
}
|
|
|
|
cmd.PersistentFlags().StringVar(&password, passwordFlag, "", "password of the the provided user")
|
|
cmd.PersistentFlags().StringVar(&sslCert, sslCertFlag, "", "ssl cert from the provided user")
|
|
cmd.PersistentFlags().StringVar(&sslKey, sslKeyFlag, "", "ssl key from the provided user")
|
|
cmd.PersistentFlags().StringVar(&user, userFlag, "", "(required) the user to check if the database, user and grants exists and create if not")
|
|
cmd.MarkPersistentFlagRequired(userFlag)
|
|
|
|
cmd.AddCommand(newZitadel(), newDatabase(), newUser(), newGrant())
|
|
|
|
return cmd
|
|
}
|
|
|
|
func adminConfig(config database.Config) database.Config {
|
|
adminConfig := config
|
|
adminConfig.User = user
|
|
adminConfig.Password = password
|
|
adminConfig.SSL.Cert = sslCert
|
|
adminConfig.SSL.Key = sslKey
|
|
|
|
return adminConfig
|
|
}
|
|
|
|
func initialise(config *Config, steps ...func(*sql.DB, database.Config) error) error {
|
|
logging.Info("initialization started")
|
|
|
|
db, err := database.Connect(adminConfig(config.Database))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
for _, step := range steps {
|
|
if err = step(db, config.Database); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
if err = db.Close(); err != nil {
|
|
return err
|
|
}
|
|
|
|
return verifyZitadel(config.Database)
|
|
}
|