-
Notifications
You must be signed in to change notification settings - Fork 42
/
get_attachment.go
90 lines (77 loc) · 2.04 KB
/
get_attachment.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
// Licensed under the Apache License, Version 2.0 (the "License"); you may not
// use this file except in compliance with the License. You may obtain a copy of
// the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
// WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
// License for the specific language governing permissions and limitations under
// the License.
package cmd
import (
"io"
"github.com/spf13/cobra"
"github.com/go-kivik/kivik/v4"
"github.com/go-kivik/kivik/v4/cmd/kivik/output"
)
type getAttachment struct {
*root
}
func getAttachmentCmd(r *root) *cobra.Command {
c := &getAttachment{
root: r,
}
return &cobra.Command{
Use: "attachment [dsn]/[database]/[document]/[filename]",
Aliases: []string{"att", "attach"},
Short: "Get an attachment",
Long: `Fetch an attachment with the HTTP GET verb`,
RunE: c.RunE,
}
}
func (c *getAttachment) RunE(cmd *cobra.Command, _ []string) error {
client, err := c.client()
if err != nil {
return err
}
db, docID, filename, err := c.conf.DBDocFilename()
if err != nil {
return err
}
c.log.Debugf("[get] Will fetch document: %s/%s/%s", client.DSN(), db, docID)
var att *kivik.Attachment
err = c.retry(func() error {
var err error
att, err = client.DB(db).GetAttachment(cmd.Context(), docID, filename, c.opts())
return err
})
if err != nil {
return err
}
result := &attachment{
Reader: output.JSONReader(att),
Attachment: att,
}
return c.fmt.Output(result)
}
type attachment struct {
io.Reader
*kivik.Attachment
}
var _ output.FriendlyOutput = (*attachment)(nil)
func (a *attachment) Execute(w io.Writer) error {
_, err := io.Copy(w, a.Content)
_ = a.Content.Close()
return err
}
func (a *attachment) Read(p []byte) (int, error) {
n, err := a.Reader.Read(p)
if err == io.EOF {
if err := a.Content.Close(); err != nil {
return n, err
}
}
return n, err
}