joao/cmd/set.go

146 lines
3.5 KiB
Go
Raw Normal View History

// Copyright © 2022 Roberto Hidalgo <joao@un.rob.mx>
2022-12-31 06:14:08 +00:00
// SPDX-License-Identifier: Apache-2.0
package cmd
import (
2022-12-14 05:41:03 +00:00
"fmt"
"io/fs"
2023-01-11 04:50:06 +00:00
"io/ioutil"
"os"
"strings"
2022-12-19 03:09:05 +00:00
"git.rob.mx/nidito/chinampa/pkg/command"
2022-12-14 05:41:03 +00:00
opclient "git.rob.mx/nidito/joao/internal/op-client"
"git.rob.mx/nidito/joao/pkg/config"
2022-12-14 05:41:03 +00:00
"github.com/sirupsen/logrus"
)
2023-01-10 07:02:10 +00:00
var Set = &command.Command{
Path: []string{"set"},
Summary: "updates configuration values",
Description: `
2022-12-17 05:40:43 +00:00
Updates the value at PATH in a local CONFIG file. Specify --secret to keep the value secret, or --delete to delete the key at PATH.
2022-12-23 05:38:37 +00:00
Will read values from stdin (or --from a file) and store it at the PATH of CONFIG, optionally --flushing to 1Password.
`,
Arguments: command.Arguments{
{
Name: "config",
Description: "The configuration file to modify",
Required: true,
Values: &command.ValueSource{
Files: &[]string{"yaml"},
},
},
{
Name: "path",
Required: true,
Description: "A dot-delimited path to set in CONFIG",
Values: &command.ValueSource{
SuggestRaw: true,
Suggestion: true,
2022-12-23 05:38:37 +00:00
Func: config.AutocompleteKeys,
},
},
},
Options: command.Options{
"input": {
ShortName: "i",
Description: "the file to read input from",
Default: "/dev/stdin",
Values: &command.ValueSource{
Files: &[]string{},
},
},
"secret": {
Description: "Store value as a secret string",
Type: "bool",
},
2022-12-17 05:40:43 +00:00
"delete": {
Description: "Delete the value at the given PATH",
Type: "bool",
},
"json": {
Description: "Treat input as JSON-encoded",
Type: "bool",
},
2022-12-14 05:41:03 +00:00
"flush": {
2022-12-17 05:40:43 +00:00
Description: "Save to 1Password after saving to PATH",
2022-12-14 05:41:03 +00:00
Type: "bool",
},
},
Action: func(cmd *command.Command) error {
path := cmd.Arguments[0].ToValue().(string)
query := cmd.Arguments[1].ToValue().(string)
var cfg *config.Config
var err error
secret := cmd.Options["secret"].ToValue().(bool)
2022-12-17 05:40:43 +00:00
delete := cmd.Options["delete"].ToValue().(bool)
input := cmd.Options["input"].ToValue().(string)
parseJSON := cmd.Options["json"].ToValue().(bool)
2022-12-14 05:41:03 +00:00
flush := cmd.Options["flush"].ToValue().(bool)
2022-12-17 05:40:43 +00:00
if secret && delete {
return fmt.Errorf("cannot --delete and set a --secret at the same time")
}
2022-12-17 05:40:43 +00:00
if secret && parseJSON {
return fmt.Errorf("cannot set a --secret that is JSON encoded, encode individual values instead")
}
2022-12-19 03:09:05 +00:00
if delete && input != "/dev/stdin" {
2023-01-11 04:50:06 +00:00
logrus.Warn("Ignoring --input while deleting")
2022-12-17 05:40:43 +00:00
}
2022-12-17 05:40:43 +00:00
cfg, err = config.Load(path, false)
if err != nil {
return err
}
2022-12-17 05:40:43 +00:00
parts := strings.Split(query, ".")
if delete {
if err := cfg.Delete(parts); err != nil {
return err
}
} else {
2023-01-11 04:50:06 +00:00
var valueBytes []byte
if input == "/dev/stdin" {
valueBytes, err = ioutil.ReadAll(cmd.Cobra.InOrStdin())
} else {
valueBytes, err = os.ReadFile(input)
}
2022-12-17 05:40:43 +00:00
if err != nil {
return err
}
if err := cfg.Set(parts, valueBytes, secret, parseJSON); err != nil {
return err
}
}
2022-12-20 05:49:37 +00:00
b, err := cfg.AsYAML()
if err != nil {
return err
}
2022-12-17 05:40:43 +00:00
var mode fs.FileMode = 0644
2022-12-14 05:41:03 +00:00
if info, err := os.Stat(path); err == nil {
mode = info.Mode().Perm()
}
if err := os.WriteFile(path, b, mode); err != nil {
return fmt.Errorf("could not save changes to %s: %w", path, err)
}
if flush {
if err := opclient.Update(cfg.Vault, cfg.Name, cfg.ToOP()); err != nil {
return fmt.Errorf("could not flush to 1password: %w", err)
}
}
logrus.Info("Done")
return err
},
2023-01-10 07:02:10 +00:00
}