From b2c2c5ac59597de36027673dc11fd068714dc895 Mon Sep 17 00:00:00 2001 From: blotus Date: Fri, 3 Mar 2023 17:03:21 +0100 Subject: [PATCH] add papi_url in credentials file when enabling console_management, and remove it when disabling console_management (#2095) --- cmd/crowdsec-cli/console.go | 22 ++++++++++++++++++++++ 1 file changed, 22 insertions(+) diff --git a/cmd/crowdsec-cli/console.go b/cmd/crowdsec-cli/console.go index 387f932de..b18e753e6 100644 --- a/cmd/crowdsec-cli/console.go +++ b/cmd/crowdsec-cli/console.go @@ -14,6 +14,7 @@ import ( "github.com/go-openapi/strfmt" log "github.com/sirupsen/logrus" "github.com/spf13/cobra" + "gopkg.in/yaml.v3" "github.com/crowdsecurity/crowdsec/pkg/apiclient" "github.com/crowdsecurity/crowdsec/pkg/csconfig" @@ -250,6 +251,27 @@ func SetConsoleOpts(args []string, wanted bool) { log.Infof("%s set to %t", csconfig.CONSOLE_MANAGEMENT, wanted) csConfig.API.Server.ConsoleConfig.ReceiveDecisions = types.BoolPtr(wanted) } + if csConfig.API.Server.OnlineClient.Credentials != nil { + changed := false + if wanted && csConfig.API.Server.OnlineClient.Credentials.PapiURL == "" { + changed = true + csConfig.API.Server.OnlineClient.Credentials.PapiURL = types.PAPIBaseURL + } else if !wanted && csConfig.API.Server.OnlineClient.Credentials.PapiURL != "" { + changed = true + csConfig.API.Server.OnlineClient.Credentials.PapiURL = "" + } + if changed { + fileContent, err := yaml.Marshal(csConfig.API.Server.OnlineClient.Credentials) + if err != nil { + log.Fatalf("Cannot marshal credentials: %s", err) + } + log.Infof("Updating credentials file: %s", csConfig.API.Server.OnlineClient.CredentialsFilePath) + err = os.WriteFile(csConfig.API.Server.OnlineClient.CredentialsFilePath, fileContent, 0600) + if err != nil { + log.Fatalf("Cannot write credentials file: %s", err) + } + } + } case csconfig.SEND_CUSTOM_SCENARIOS: /*for each flag check if it's already set before setting it*/ if csConfig.API.Server.ConsoleConfig.ShareCustomScenarios != nil {