-
Notifications
You must be signed in to change notification settings - Fork 0
/
getobject.go
98 lines (80 loc) · 2.4 KB
/
getobject.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
package main
import (
"fmt"
"os"
"context"
"github.com/knockaway/gorets/pkg/rets"
"github.com/spf13/cobra"
)
const (
goResourceFlag = "resource"
goTypeFlag = "type"
goIDFlag = "id"
)
// NewGetObjectsCmd ...
func NewGetObjectsCmd() *cobra.Command {
cmd := &cobra.Command{
Use: "getobjects",
Short: "Get Objects from a RETS server",
Run: getObjects,
}
cmd.Flags().String(goResourceFlag, "Property", "Resource type of the request")
cmd.Flags().String(goTypeFlag, "Photo", "Metadata format")
cmd.Flags().String(goIDFlag, "", "Object ID for the request ('key1:*;key2:0')")
return cmd
}
func getObjects(cmd *cobra.Command, args []string) {
connect, output, timeout := getPersistentFlagValues(search, cmd)
var err error // annoying
params := GetObjectParams{}
params.Resource, err = cmd.Flags().GetString(goResourceFlag)
handleError(getObjects, err)
params.Type, err = cmd.Flags().GetString(goTypeFlag)
handleError(getObjects, err)
params.ID, err = cmd.Flags().GetString(goIDFlag)
handleError(getObjects, err)
// should we throw an err here too?
session, err := connect.Initialize()
handleError(getObjects, err)
ctx, cancel := context.WithTimeout(context.Background(), timeout)
defer cancel()
capability, err := rets.Login(ctx, session, rets.LoginRequest{URL: connect.URL})
handleError(getObjects, err)
// make sure we close the rets connection
defer rets.Logout(ctx, session, rets.LogoutRequest{URL: capability.Logout})
// warning, this does _all_ of the photos
gor, err := rets.GetObjects(ctx, session, rets.GetObjectRequest{
URL: capability.GetObject,
GetObjectParams: rets.GetObjectParams{
Resource: params.Resource,
Type: params.Type,
ID: params.ID,
},
})
handleError(getObjects, err)
response := &rets.GetObjectResponse{Response: gor}
defer response.Close()
err = response.ForEach(func(o *rets.Object, err error) error {
if err != nil {
return err
}
// if we arent saving, then we quit
if output == "" {
return nil
}
path := fmt.Sprintf("%s/%s", output, o.ContentID)
os.MkdirAll(path, os.ModePerm)
f, err := os.Create(fmt.Sprintf("%s/%d", path, o.ObjectID))
handleError(getObjects, err)
defer f.Close()
_, err = f.Write(o.Blob)
return err
})
handleError(getObjects, err)
}
// GetObjectParams ...
type GetObjectParams struct {
Resource string `json:"resource"`
Type string `json:"type"`
ID string `json:"id"`
}