mirror of
https://github.com/paralus/paralus.git
synced 2026-05-09 09:56:45 +00:00
99 lines
2.7 KiB
Go
99 lines
2.7 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"fmt"
|
|
"os"
|
|
"time"
|
|
|
|
"github.com/cloudflare/cfssl/log"
|
|
"github.com/uptrace/bun"
|
|
"github.com/uptrace/bun/dialect/pgdialect"
|
|
"github.com/uptrace/bun/driver/pgdriver"
|
|
"gopkg.in/yaml.v2"
|
|
)
|
|
|
|
type Provider struct {
|
|
Id string `bun:"name"`
|
|
Provider string `bun:"provider_name,notnull"`
|
|
MapperURL string `bun:"mapper_url" yaml:"mapper_url"`
|
|
ClientId string `bun:"client_id,notnull" yaml:"client_id"`
|
|
ClientSecret string `bun:"client_secret,notnull" yaml:"client_secret"`
|
|
Scope []string `bun:"scopes,array,notnull"`
|
|
IssuerURL string `bun:"issuer_url,notnull" yaml:"issuer_url"`
|
|
AuthURL string `bun:"auth_url" yaml:"auth_url,omitempty"`
|
|
TokenURL string `bun:"token_url" yaml:"token_url,omitempty"`
|
|
RequestedClaims map[string]interface{} `bun:"type:jsonb" yaml:"requested_claims,omitempty"`
|
|
}
|
|
|
|
type Config struct {
|
|
Selfservice struct {
|
|
Methods struct {
|
|
Oidc struct {
|
|
Config struct {
|
|
Providers []Provider
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
var ProvidersDB []Provider
|
|
|
|
func sync(ctx context.Context, db *bun.DB, path string) error {
|
|
err := db.NewSelect().Model(&ProvidersDB).ModelTableExpr("authsrv_oidc_provider AS provider").Where("trash = 'f'").Scan(ctx)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to fetch providers from DB: %s", err)
|
|
}
|
|
|
|
var c Config
|
|
c.Selfservice.Methods.Oidc.Config.Providers = ProvidersDB
|
|
d, err := yaml.Marshal(&c)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to marshal: %s", err)
|
|
}
|
|
err = os.WriteFile(path, d, 0644)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to write data: %s", err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func main() {
|
|
dsn := "postgres://"
|
|
outputPath := "/etc/kratos/providers.yaml"
|
|
ctx := context.Background()
|
|
channel := "provider:changed"
|
|
|
|
if len(os.Getenv("DSN")) != 0 {
|
|
dsn = os.Getenv("DSN")
|
|
}
|
|
|
|
if len(os.Getenv("KRATOS_PROVIDER_CFG")) != 0 {
|
|
outputPath = os.Getenv("KRATOS_PROVIDER_CFG")
|
|
}
|
|
|
|
sqldb := sql.OpenDB(pgdriver.NewConnector(pgdriver.WithDSN(dsn)))
|
|
db := bun.NewDB(sqldb, pgdialect.New())
|
|
|
|
ln := pgdriver.NewListener(db)
|
|
listen:
|
|
if err := ln.Listen(ctx, channel); err != nil {
|
|
log.Errorf("error listening for notifications on channel %q: %s", channel, err)
|
|
time.Sleep(2 * time.Second)
|
|
goto listen
|
|
}
|
|
|
|
log.Infof("Started listening for notification on channel %q", channel)
|
|
for range ln.Channel() {
|
|
log.Info("A notification received")
|
|
if err := sync(ctx, db, outputPath); err != nil {
|
|
log.Errorf("sync failed: %s", err)
|
|
} else {
|
|
log.Info("Synchronized successfully")
|
|
}
|
|
}
|
|
|
|
}
|