Skip to content

Commit

Permalink
chore: Update injector.go to handle empty output filename and default…
Browse files Browse the repository at this point in the history
… decode base64 to false
  • Loading branch information
h0n9 committed Jun 5, 2024
1 parent baffbae commit c1a47aa
Showing 1 changed file with 16 additions and 11 deletions.
27 changes: 16 additions & 11 deletions cli/injector/injector.go
Original file line number Diff line number Diff line change
Expand Up @@ -17,8 +17,10 @@ import (

const (
DefaultProviderName = "aws"
DefaultSecretID = ""
DefaultTemplateBase64 = "e3sgcmFuZ2UgJGssICR2IDo9IC4gfX1be3sgJGsgfX1dCnt7ICR2IH19Cgp7eyBlbmQgfX0K"
DefaultOutputFilename = "output"
DefaultOutputFilename = ""
DefaultDecodeBase64 = false
)

var Cmd = &cobra.Command{
Expand All @@ -30,7 +32,7 @@ var (
providerName string
secretID string
templateBase64 string
output string
outputFilename string
decodeBase64EncodedSecret bool
)

Expand All @@ -46,10 +48,13 @@ var runCmd = &cobra.Command{

logger.Info().Msg("initialized context")

// get envs
// check required flags
if secretID == "" {
return fmt.Errorf("failed to read 'secret-id' flag")
}
if outputFilename == "" {
return fmt.Errorf("failed to read 'output' flag")
}

logger.Info().Msg("read environment variables")

Expand Down Expand Up @@ -93,12 +98,12 @@ var runCmd = &cobra.Command{

logger.Info().Msg("initialized secret handler")

err = secretHandler.Save(secretID, output, decodeBase64EncodedSecret)
err = secretHandler.Save(secretID, outputFilename, decodeBase64EncodedSecret)
if err != nil {
return err
}

logger.Info().Msg(fmt.Sprintf("saved secret id '%s' values to '%s'", secretID, output))
logger.Info().Msg(fmt.Sprintf("saved secret id '%s' values to '%s'", secretID, outputFilename))

return nil
},
Expand All @@ -108,7 +113,7 @@ func init() {
runCmd.Flags().StringVar(
&providerName,
"provider",
"aws",
DefaultProviderName,
"cloud provider name",
)
runCmd.Flags().StringVar(
Expand All @@ -120,20 +125,20 @@ func init() {
runCmd.Flags().StringVar(
&templateBase64,
"template",
"e3sgcmFuZ2UgJGssICR2IDo9IC4gfX1be3sgJGsgfX1dCnt7ICR2IH19Cgp7eyBlbmQgfX0K",
DefaultTemplateBase64,
"base64 encoded template string",
)
runCmd.Flags().StringVar(
&output,
&outputFilename,
"output",
"secret",
DefaultOutputFilename,
"output filename",
)
runCmd.Flags().BoolVar(
&decodeBase64EncodedSecret,
"decode-b64-secret",
false,
"decode base64 encoded secret",
DefaultDecodeBase64,
"decode base64-encoded secret",
)
Cmd.AddCommand(runCmd)
}

0 comments on commit c1a47aa

Please sign in to comment.