-
Notifications
You must be signed in to change notification settings - Fork 0
/
head.go
72 lines (60 loc) · 1.32 KB
/
head.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
package cliaws
import (
"context"
"encoding/json"
"fmt"
"strings"
"github.com/aws/aws-sdk-go/aws"
"github.com/aws/aws-sdk-go/service/s3"
"github.com/alexflint/go-arg"
"github.com/nathants/cli-aws/lib"
)
func init() {
lib.Commands["s3-head"] = s3Head
lib.Args["s3-head"] = s3HeadArgs{}
}
type s3HeadArgs struct {
Path string `arg:"positional"`
}
func (s3HeadArgs) Description() string {
return "\nhead an object\n"
}
func s3Head() {
var args s3HeadArgs
arg.MustParse(&args)
ctx := context.Background()
path := lib.Last(strings.Split(args.Path, "s3://"))
parts := strings.Split(path, "/")
bucket := parts[0]
var key string
if len(parts) > 1 {
key = strings.Join(parts[1:], "/")
}
s3Client, err := lib.S3ClientBucketRegion(bucket)
if err != nil {
lib.Logger.Fatal("error: ", err)
}
out, err := s3Client.HeadObjectWithContext(ctx, &s3.HeadObjectInput{
Bucket: aws.String(bucket),
Key: aws.String(key),
})
if err != nil {
lib.Logger.Fatal("error: ", err)
}
data, err := json.Marshal(out)
if err != nil {
lib.Logger.Fatal("error: ", err)
}
var outMap map[string]interface{}
err = json.Unmarshal(data, &outMap)
if err != nil {
lib.Logger.Fatal("error: ", err)
}
val := map[string]interface{}{}
for k, v := range outMap {
if v != nil {
val[k] = v
}
}
fmt.Println(lib.Pformat(val))
}