-
Notifications
You must be signed in to change notification settings - Fork 18
/
storage_metadata_add.go
102 lines (78 loc) · 2.47 KB
/
storage_metadata_add.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
package cmd
import (
"fmt"
"strings"
"github.com/spf13/cobra"
"github.com/exoscale/cli/pkg/globalstate"
"github.com/exoscale/cli/pkg/output"
"github.com/exoscale/cli/pkg/storage/sos"
)
var storageMetadataAddCmd = &cobra.Command{
Use: "add sos://BUCKET/(OBJECT|PREFIX/) KEY=VALUE...",
Short: "Add key/value metadata to an object",
Long: fmt.Sprintf(`This command adds key/value metadata to an object.
Example:
exo storage metadata add sos://my-bucket/object-a \
k1=v1 \
k2=v2
Notes:
* Adding an already existing key will overwrite its value.
* The following characters are not allowed in keys: %s
Supported output template annotations: %s`,
strings.Join(strings.Split(sos.MetadataForbiddenCharset, ""), " "),
strings.Join(output.TemplateAnnotations(&sos.ShowObjectOutput{}), ", ")),
PreRunE: func(cmd *cobra.Command, args []string) error {
if len(args) < 2 {
cmdExitOnUsageError(cmd, "invalid arguments")
}
args[0] = strings.TrimPrefix(args[0], sos.BucketPrefix)
if !strings.Contains(args[0], "/") {
cmdExitOnUsageError(cmd, fmt.Sprintf("invalid argument: %q", args[0]))
}
for _, kv := range args[1:] {
if !strings.Contains(kv, "=") {
cmdExitOnUsageError(cmd, fmt.Sprintf("invalid argument: %q", kv))
}
}
return nil
},
RunE: func(cmd *cobra.Command, args []string) error {
var (
bucket string
prefix string
metadata = make(map[string]string)
)
recursive, err := cmd.Flags().GetBool("recursive")
if err != nil {
return err
}
parts := strings.SplitN(args[0], "/", 2)
bucket, prefix = parts[0], parts[1]
for _, kv := range args[1:] {
parts = strings.Split(kv, "=")
metadata[parts[0]] = parts[1]
}
storage, err := sos.NewStorageClient(
gContext,
sos.ClientOptZoneFromBucket(gContext, bucket),
)
if err != nil {
return fmt.Errorf("unable to initialize storage client: %w", err)
}
if err := storage.AddObjectsMetadata(gContext, bucket, prefix, metadata, recursive); err != nil {
return fmt.Errorf("unable to add metadata to object: %w", err)
}
if !globalstate.Quiet && !recursive && !strings.HasSuffix(prefix, "/") {
return printOutput(storage.ShowObject(gContext, bucket, prefix))
}
if !globalstate.Quiet {
fmt.Println("Metadata added successfully")
}
return nil
},
}
func init() {
storageMetadataAddCmd.Flags().BoolP("recursive", "r", false,
"add metadata recursively (with object prefix only)")
storageMetadataCmd.AddCommand(storageMetadataAddCmd)
}