func main() {
// If not set for some reason, use us-east-1 by default.
if discfgDBRegion == "" {
discfgDBRegion = "us-east-1"
}
apex.HandleFunc(func(event json.RawMessage, ctx *apex.Context) (interface{}, error) {
var m message
if err := json.Unmarshal(event, &m); err != nil {
return nil, err
}
options.Storage.AWS.Region = discfgDBRegion
// Each discfg API can be configured with a default table name.
options.CfgName = discfgDBTable
// Overwritten by the message passed to the Lambda.
if m.Name != "" {
options.CfgName = m.Name
}
options.Key = m.Key
resp := commands.GetKey(options)
// Format the expiration time (if applicable). This prevents output like "0001-01-01T00:00:00Z" when empty
// and allows for the time.RFC3339Nano format to be used whereas time.Time normally marshals to a different format.
if resp.Item.TTL > 0 {
resp.Item.OutputExpiration = resp.Item.Expiration.Format(time.RFC3339Nano)
}
// Just return the raw value for the given key if raw was passed as true
// if m.Raw == "true" {
// return resp.Item.Value, nil
// }
r := commands.FormatJSONValue(resp)
return r, nil
})
}
Use: "set",
Short: "set key value",
Long: `Sets a key value for a given discfg`,
Run: func(cmd *cobra.Command, args []string) {
setOptsFromArgs(args)
resp := commands.SetKey(Options)
commands.Out(Options, resp)
},
}
var getCmd = &cobra.Command{
Use: "get",
Short: "get key value",
Long: `Gets a key value for a given discfg`,
Run: func(cmd *cobra.Command, args []string) {
setOptsFromArgs(args)
resp := commands.GetKey(Options)
commands.Out(Options, resp)
},
}
var deleteCmd = &cobra.Command{
Use: "delete",
Short: "delete key",
Long: `Deletes a key for a given discfg`,
Run: func(cmd *cobra.Command, args []string) {
setOptsFromArgs(args)
resp := commands.DeleteKey(Options)
commands.Out(Options, resp)
},
}
var exportCmd = &cobra.Command{
Use: "export",