mirror of
https://github.com/pomerium/pomerium.git
synced 2025-07-30 06:51:30 +02:00
96 lines
2.8 KiB
Go
96 lines
2.8 KiB
Go
package cmd
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/pomerium/pomerium/config"
|
|
"github.com/pomerium/pomerium/internal/log"
|
|
"github.com/pomerium/pomerium/pkg/envoy/files"
|
|
"github.com/pomerium/pomerium/pkg/zero/importutil"
|
|
"github.com/rs/zerolog"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
func BuildImportCmd() *cobra.Command {
|
|
cmd := &cobra.Command{
|
|
Use: "import",
|
|
Short: "Import an existing configuration to a Zero cluster",
|
|
RunE: func(cmd *cobra.Command, _ []string) error {
|
|
configFlag := cmd.InheritedFlags().Lookup("config")
|
|
var configFile string
|
|
if configFlag != nil {
|
|
configFile = configFlag.Value.String()
|
|
}
|
|
if configFile == "" {
|
|
// try looking up what pid 1 is using, we are likely in a container anyway
|
|
info, err := os.ReadFile("/proc/1/cmdline")
|
|
if err == nil {
|
|
args := bytes.Split(info, []byte{0})
|
|
if len(args) > 0 && strings.Contains(string(args[0]), "pomerium") {
|
|
for i, arg := range args {
|
|
if strings.Contains(string(arg), "-config") {
|
|
if strings.Contains(string(arg), "-config=") {
|
|
configFile = strings.Split(string(arg), "=")[1]
|
|
cmd.PrintErrf("detected config file: %s\n", configFile)
|
|
} else if len(args) > i+1 {
|
|
configFile = string(args[i+1])
|
|
cmd.PrintErrf("detected config file: %s\n", configFile)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// try some common locations
|
|
if configFile == "" {
|
|
if _, err := os.Stat("/pomerium/config.yaml"); err == nil {
|
|
configFile = "/pomerium/config.yaml"
|
|
} else if _, err := os.Stat("/etc/pomerium/config.yaml"); err == nil {
|
|
configFile = "/etc/pomerium/config.yaml"
|
|
} else if _, err := os.Stat("config.yaml"); err == nil {
|
|
configFile = "config.yaml"
|
|
}
|
|
|
|
if configFile != "" {
|
|
cmd.PrintErrf("detected config file: %s\n", configFile)
|
|
}
|
|
}
|
|
}
|
|
if configFile == "" {
|
|
return fmt.Errorf("no config file provided")
|
|
}
|
|
log.SetLevel(zerolog.ErrorLevel)
|
|
src, err := config.NewFileOrEnvironmentSource(configFile, files.FullVersion())
|
|
if err != nil {
|
|
return err
|
|
}
|
|
cfg := src.GetConfig()
|
|
|
|
client := zeroClientFromContext(cmd.Context())
|
|
converted := cfg.Options.ToProto()
|
|
for i, name := range importutil.GenerateRouteNames(converted.Routes) {
|
|
converted.Routes[i].Name = name
|
|
}
|
|
resp, err := client.ImportConfig(cmd.Context(), converted)
|
|
if err != nil {
|
|
return fmt.Errorf("error importing config: %w", err)
|
|
}
|
|
if resp.Warnings != nil {
|
|
for _, warn := range *resp.Warnings {
|
|
cmd.Printf("warning: %s\n", warn)
|
|
}
|
|
}
|
|
if resp.Messages != nil {
|
|
for _, msg := range *resp.Messages {
|
|
cmd.Printf("✔ %s\n", msg)
|
|
}
|
|
}
|
|
cmd.Println("\nImport successful, return to your browser to continue setup.")
|
|
return nil
|
|
},
|
|
}
|
|
return cmd
|
|
}
|