forked from grailbio/base
/
b2.go
83 lines (70 loc) · 2.27 KB
/
b2.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
// Copyright 2018 GRAIL, Inc. All rights reserved.
// Use of this source code is governed by the Apache-2.0
// license that can be found in the LICENSE file.
package ticket
import (
"encoding/base64"
"encoding/json"
"fmt"
"net/http"
"github.com/Schaudge/grailbase/common/log"
"github.com/Schaudge/grailbase/security/keycrypt"
)
const (
b2AuthorizeURL = "https://api.backblazeb2.com/b2api/v1/b2_authorize_account"
)
func (b *B2AccountAuthorizationBuilder) newB2Ticket(ctx *TicketContext) (TicketB2Ticket, error) {
log.Info(ctx.ctx, "Creating BackBlaze ticket.", "B2AccountAuthorizationBuilder", b)
b2Ticket, err := b.genB2Ticket(ctx)
if err != nil {
return TicketB2Ticket{}, err
}
return TicketB2Ticket{
Value: *b2Ticket,
}, nil
}
func (b *B2AccountAuthorizationBuilder) genB2Ticket(ctx *TicketContext) (*B2Ticket, error) {
secret, err := keycrypt.Lookup(b.ApplicationKey)
if err != nil {
return nil, err
}
applicationKey, err := secret.Get()
if err != nil {
return nil, err
}
headerForAuthorizeAccount := "Basic " + base64.StdEncoding.EncodeToString([]byte(b.AccountId+":"+string(applicationKey)))
req, err := http.NewRequest("GET", b2AuthorizeURL, nil)
if err != nil {
log.Error(ctx.ctx, "Failed to create new request.", "b2AuthorizeURL", b2AuthorizeURL, "err", err.Error())
return nil, err
}
req.Header.Set("Authorization", headerForAuthorizeAccount)
client := &http.Client{}
resp, err := client.Do(req)
if err != nil {
log.Error(ctx.ctx, "Failed to authorize.", "b2AuthorizeURL", b2AuthorizeURL, "err", err.Error())
return nil, err
}
defer resp.Body.Close()
if resp.StatusCode != http.StatusOK {
type ErrorResponse struct {
Status int
Code string
Message string
}
var er ErrorResponse
if err := json.NewDecoder(resp.Body).Decode(&er); err != nil {
log.Error(ctx.ctx, "Failed to decode response.", "errResponse", er, "err", err.Error())
return nil, err
}
err := fmt.Errorf("status %d: %s", er.Status, er.Message)
log.Error(ctx.ctx, "Request failed.", "err", err.Error())
return nil, err
}
var b2Ticket B2Ticket
if err := json.NewDecoder(resp.Body).Decode(&b2Ticket); err != nil {
log.Error(ctx.ctx, "Failed to decode BackBlaze ticket.", "b2Ticket", b2Ticket, "err", err.Error())
return nil, err
}
return &b2Ticket, nil
}