/
store_client.go
71 lines (59 loc) · 1.8 KB
/
store_client.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
package storeclient
import (
"context"
"io"
"net/http"
"github.com/benchkram/bob/pkg/store-client/generated"
)
type I interface {
UploadArtifact(ctx context.Context, projectName string, artifactID string, src io.Reader, size int64) (err error)
ListArtifacts(ctx context.Context, projectName string) (artifactIds []string, err error)
GetArtifact(ctx context.Context, projectName string, artifactId string) (rc io.ReadCloser, size int64, err error)
}
type c struct {
endpoint string
client *generated.Client
clientWithResponses *generated.ClientWithResponses
}
func New(endpoint, token string) I {
c := &c{
endpoint: endpoint,
client: createClientMust(endpoint, token),
clientWithResponses: createClientWithResponsesMust(endpoint, token),
}
return c
}
func createClientWithResponsesMust(endpoint, token string) *generated.ClientWithResponses {
client, err := createClientWithResponses(endpoint, token)
if err != nil {
panic(err)
}
return client
}
func createClientWithResponses(endpoint, token string) (*generated.ClientWithResponses, error) {
return generated.NewClientWithResponses(endpoint, generated.WithRequestEditorFn(
func(ctx context.Context, req *http.Request) (err error) {
if token != "" {
req.Header.Set("Authorization", "Bearer "+token)
}
return nil
},
))
}
func createClientMust(endpoint, token string) *generated.Client {
client, err := createClient(endpoint, token)
if err != nil {
panic(err)
}
return client
}
func createClient(endpoint, token string) (*generated.Client, error) {
return generated.NewClient(endpoint, generated.WithRequestEditorFn(
func(ctx context.Context, req *http.Request) (err error) {
if token != "" {
req.Header.Set("Authorization", "Bearer "+token)
}
return nil
},
))
}