2022-02-11 11:02:47 +01:00
|
|
|
package initialise
|
|
|
|
|
|
|
|
import (
|
2022-02-16 13:30:49 +01:00
|
|
|
_ "embed"
|
2022-08-31 09:52:43 +02:00
|
|
|
"fmt"
|
2022-02-11 11:02:47 +01:00
|
|
|
|
2022-02-11 14:07:32 +01:00
|
|
|
"github.com/spf13/cobra"
|
|
|
|
"github.com/spf13/viper"
|
2022-08-31 09:52:43 +02:00
|
|
|
"github.com/zitadel/logging"
|
2023-10-19 12:19:10 +02:00
|
|
|
|
2022-04-27 01:01:45 +02:00
|
|
|
"github.com/zitadel/zitadel/internal/database"
|
2023-12-20 18:13:04 +02:00
|
|
|
"github.com/zitadel/zitadel/internal/database/dialect"
|
2022-02-11 11:02:47 +01:00
|
|
|
)
|
|
|
|
|
2022-02-11 14:07:32 +01:00
|
|
|
func newZitadel() *cobra.Command {
|
|
|
|
return &cobra.Command{
|
|
|
|
Use: "zitadel",
|
feat: Configurable Unique Machine Identification (#3626)
* feat: Configurable Unique Machine Identification
This change fixes Segfault on AWS App Runner with v2 #3625
The change introduces two new dependencies:
* github.com/drone/envsubst for supporting AWS ECS, which has its metadata endpoint described by an environment variable
* github.com/jarcoal/jpath so that only relevant data from a metadata response is used to identify the machine.
The change ads new configuration (see `defaults.yaml`):
* `Machine.Identification` enables configuration of how machines are uniquely identified - I'm not sure about the top level category `Machine`, as I don't have anything else to add to it. Happy to hear suggestions for better naming or structure here.
* `Machine.Identifiation.PrivateId` turns on or off the existing private IP based identification. Default is on.
* `Machine.Identification.Hostname` turns on or off using the OS hostname to identify the machine. Great for most cloud environments, where this tends to be set to something that identifies the machine uniquely. Enabled by default.
* `Machine.Identification.Webhook` configures identification based on the response to an HTTP GET request. Request headers can be configured, a JSONPath can be set for processing the response (no JSON parsing is done if this is not set), and the URL is allowed to contain environment variables in the format `"${var}"`.
The new flow for getting a unique machine id is:
1. PrivateIP (if enabled)
2. Hostname (if enabled)
3. Webhook (if enabled, to configured URL)
4. Give up and error out.
It's important that init configures machine identity first. Otherwise we could try to get an ID before configuring it. To prevent this from causing difficult to debug issues, where for example the default configuration was used, I've ensured that
the application will generate an error if the module hasn't been configured and you try to get an ID.
Misc changes:
* Spelling and gramatical corrections to `init.go::New()` long description.
* Spelling corrections to `verify_zitadel.go::newZitadel()`.
* Updated `production.md` and `development.md` based on the new build process. I think the run instructions are also out of date, but I'll leave that for someone else.
* `id.SonyFlakeGenerator` is now a function, which sets `id.sonyFlakeGenerator`, this allows us to defer initialization until configuration has been read.
* Update internal/id/config.go
Co-authored-by: Alexei-Barnes <82444470+Alexei-Barnes@users.noreply.github.com>
* Fix authored by @livio-a for tests
Co-authored-by: Livio Amstutz <livio.a@gmail.com>
2022-05-24 15:57:57 +01:00
|
|
|
Short: "initialize ZITADEL internals",
|
|
|
|
Long: `initialize ZITADEL internals.
|
2022-02-11 14:07:32 +01:00
|
|
|
|
|
|
|
Prereqesits:
|
2022-09-05 11:24:31 +02:00
|
|
|
- cockroachDB or postgreSQL with user and database
|
2022-02-11 14:07:32 +01:00
|
|
|
`,
|
2022-09-05 11:24:31 +02:00
|
|
|
Run: func(cmd *cobra.Command, args []string) {
|
2022-09-01 09:24:26 +02:00
|
|
|
config := MustNewConfig(viper.GetViper())
|
2022-09-05 11:24:31 +02:00
|
|
|
err := verifyZitadel(config.Database)
|
|
|
|
logging.OnError(err).Fatal("unable to init zitadel")
|
2022-02-11 14:07:32 +01:00
|
|
|
},
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2023-10-19 12:19:10 +02:00
|
|
|
func VerifyZitadel(db *database.DB, config database.Config) error {
|
2022-09-01 09:24:26 +02:00
|
|
|
err := ReadStmts(config.Type())
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
2023-10-19 12:19:10 +02:00
|
|
|
logging.WithFields().Info("verify system")
|
2022-08-31 09:52:43 +02:00
|
|
|
if err := exec(db, fmt.Sprintf(createSystemStmt, config.Username()), nil); err != nil {
|
2022-03-14 07:55:09 +01:00
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
2023-10-19 12:19:10 +02:00
|
|
|
logging.WithFields().Info("verify encryption keys")
|
2022-08-31 09:52:43 +02:00
|
|
|
if err := createEncryptionKeys(db); err != nil {
|
2022-03-14 07:55:09 +01:00
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
2023-10-19 12:19:10 +02:00
|
|
|
logging.WithFields().Info("verify projections")
|
2022-08-31 09:52:43 +02:00
|
|
|
if err := exec(db, fmt.Sprintf(createProjectionsStmt, config.Username()), nil); err != nil {
|
2022-02-11 11:02:47 +01:00
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
2023-10-19 12:19:10 +02:00
|
|
|
logging.WithFields().Info("verify eventstore")
|
2022-08-31 09:52:43 +02:00
|
|
|
if err := exec(db, fmt.Sprintf(createEventstoreStmt, config.Username()), nil); err != nil {
|
2022-02-11 11:02:47 +01:00
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
2023-10-19 12:19:10 +02:00
|
|
|
logging.WithFields().Info("verify events tables")
|
2022-08-31 09:52:43 +02:00
|
|
|
if err := createEvents(db); err != nil {
|
2022-02-11 11:02:47 +01:00
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
2023-10-19 12:19:10 +02:00
|
|
|
logging.WithFields().Info("verify system sequence")
|
2022-08-31 09:52:43 +02:00
|
|
|
if err := exec(db, createSystemSequenceStmt, nil); err != nil {
|
2022-03-15 07:19:02 +01:00
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
2023-10-19 12:19:10 +02:00
|
|
|
logging.WithFields().Info("verify unique constraints")
|
2022-08-31 09:52:43 +02:00
|
|
|
if err := exec(db, createUniqueConstraints, nil); err != nil {
|
2022-03-15 07:19:02 +01:00
|
|
|
return err
|
|
|
|
}
|
2023-10-19 12:19:10 +02:00
|
|
|
|
2022-03-15 07:19:02 +01:00
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func verifyZitadel(config database.Config) error {
|
2023-02-27 22:36:43 +01:00
|
|
|
logging.WithFields("database", config.DatabaseName()).Info("verify zitadel")
|
2022-09-01 09:24:26 +02:00
|
|
|
|
2023-12-20 18:13:04 +02:00
|
|
|
db, err := database.Connect(config, false, dialect.DBPurposeQuery)
|
2022-03-15 07:19:02 +01:00
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
2022-09-01 09:24:26 +02:00
|
|
|
|
2023-10-19 12:19:10 +02:00
|
|
|
if err := VerifyZitadel(db, config); err != nil {
|
2022-09-05 11:24:31 +02:00
|
|
|
return err
|
2022-03-15 07:19:02 +01:00
|
|
|
}
|
|
|
|
|
2022-02-11 11:02:47 +01:00
|
|
|
return db.Close()
|
|
|
|
}
|
|
|
|
|
2023-10-19 12:19:10 +02:00
|
|
|
func createEncryptionKeys(db *database.DB) error {
|
2022-03-14 07:55:09 +01:00
|
|
|
tx, err := db.Begin()
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
if _, err = tx.Exec(createEncryptionKeysStmt); err != nil {
|
|
|
|
tx.Rollback()
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
return tx.Commit()
|
|
|
|
}
|
|
|
|
|
2023-10-19 12:19:10 +02:00
|
|
|
func createEvents(db *database.DB) (err error) {
|
2022-02-11 11:02:47 +01:00
|
|
|
tx, err := db.Begin()
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
2023-10-19 12:19:10 +02:00
|
|
|
defer func() {
|
|
|
|
if err != nil {
|
|
|
|
rollbackErr := tx.Rollback()
|
|
|
|
logging.OnError(rollbackErr).Debug("rollback failed")
|
|
|
|
return
|
|
|
|
}
|
|
|
|
err = tx.Commit()
|
|
|
|
}()
|
|
|
|
|
|
|
|
// if events already exists events2 is created during a setup job
|
|
|
|
var count int
|
|
|
|
row := tx.QueryRow("SELECT count(*) FROM information_schema.tables WHERE table_schema = 'eventstore' AND table_name like 'events%'")
|
|
|
|
if err = row.Scan(&count); err != nil {
|
2022-02-11 11:02:47 +01:00
|
|
|
return err
|
|
|
|
}
|
2023-10-19 12:19:10 +02:00
|
|
|
if row.Err() != nil || count >= 1 {
|
|
|
|
return row.Err()
|
|
|
|
}
|
|
|
|
_, err = tx.Exec(createEventsStmt)
|
|
|
|
return err
|
2022-02-11 11:02:47 +01:00
|
|
|
}
|