-
Notifications
You must be signed in to change notification settings - Fork 1
/
aws_s3.go
62 lines (53 loc) · 1.51 KB
/
aws_s3.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
package selfupdater
import (
"crypto/md5"
"github.com/aws/aws-sdk-go/aws"
"github.com/aws/aws-sdk-go/aws/session"
"github.com/aws/aws-sdk-go/service/s3"
"github.com/aws/aws-sdk-go/service/s3/s3manager"
"github.com/pkg/errors"
"hash"
"os"
)
type S3UpdateProvider struct {
BucketName string
ExecutablePath string
ChecksumPath string
S3Downloader *s3manager.Downloader
}
func NewS3UpdateProvider(bucket string, region string, executable string, checksum string) *S3UpdateProvider {
sess := session.Must(session.NewSession(&aws.Config{
Region: aws.String(region)}))
downloader := s3manager.NewDownloader(sess)
return &S3UpdateProvider{BucketName: bucket,
ExecutablePath: executable,
ChecksumPath: checksum,
S3Downloader: downloader,
}
}
func (p *S3UpdateProvider) DownloadTo(file *os.File) error {
_, err := p.S3Downloader.Download(file,
&s3.GetObjectInput{
Bucket: aws.String(p.BucketName),
Key: aws.String(p.ExecutablePath),
})
if err != nil {
return errors.WithMessage(err, "failed to obtain remote executable")
}
return nil
}
func (p *S3UpdateProvider) RemoteChecksum() (string, error) {
buff := aws.NewWriteAtBuffer([]byte{})
numBytes, err := p.S3Downloader.Download(buff,
&s3.GetObjectInput{
Bucket: aws.String(p.BucketName),
Key: aws.String(p.ChecksumPath),
})
if err != nil {
return "", errors.WithMessage(err, "failed to obtain remote checksum")
}
return string(buff.Bytes()[:numBytes]), nil
}
func (p *S3UpdateProvider) Hash() hash.Hash {
return md5.New()
}