mirror of
https://github.com/Luzifer/cloudbox.git
synced 2024-11-09 22:50:09 +00:00
Add share functionality for S3
Signed-off-by: Knut Ahlers <knut@ahlers.me>
This commit is contained in:
parent
e8d04df376
commit
122eebbd25
1 changed files with 46 additions and 6 deletions
|
@ -2,7 +2,9 @@ package s3
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"bytes"
|
"bytes"
|
||||||
|
"context"
|
||||||
"crypto/md5"
|
"crypto/md5"
|
||||||
|
"fmt"
|
||||||
"hash"
|
"hash"
|
||||||
"io"
|
"io"
|
||||||
"net/url"
|
"net/url"
|
||||||
|
@ -12,6 +14,7 @@ import (
|
||||||
"github.com/aws/aws-sdk-go/aws/credentials"
|
"github.com/aws/aws-sdk-go/aws/credentials"
|
||||||
"github.com/aws/aws-sdk-go/aws/session"
|
"github.com/aws/aws-sdk-go/aws/session"
|
||||||
"github.com/aws/aws-sdk-go/service/s3"
|
"github.com/aws/aws-sdk-go/service/s3"
|
||||||
|
"github.com/aws/aws-sdk-go/service/s3/s3manager"
|
||||||
"github.com/pkg/errors"
|
"github.com/pkg/errors"
|
||||||
|
|
||||||
"github.com/Luzifer/cloudbox/providers"
|
"github.com/Luzifer/cloudbox/providers"
|
||||||
|
@ -19,6 +22,7 @@ import (
|
||||||
|
|
||||||
type Provider struct {
|
type Provider struct {
|
||||||
bucket string
|
bucket string
|
||||||
|
bucketRegion string
|
||||||
s3 *s3.S3
|
s3 *s3.S3
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -39,10 +43,17 @@ func New(uri string) (providers.CloudProvider, error) {
|
||||||
cfg = cfg.WithCredentials(credentials.NewStaticCredentials(user, pass, ""))
|
cfg = cfg.WithCredentials(credentials.NewStaticCredentials(user, pass, ""))
|
||||||
}
|
}
|
||||||
|
|
||||||
svc := s3.New(session.Must(session.NewSession(cfg)))
|
sess := session.Must(session.NewSession(cfg))
|
||||||
|
svc := s3.New(sess)
|
||||||
|
|
||||||
|
region, err := s3manager.GetBucketRegion(context.Background(), sess, u.Host, "us-east-1")
|
||||||
|
if err != nil {
|
||||||
|
return nil, errors.Wrap(err, "Unable to find bucket region")
|
||||||
|
}
|
||||||
|
|
||||||
return &Provider{
|
return &Provider{
|
||||||
bucket: u.Host,
|
bucket: u.Host,
|
||||||
|
bucketRegion: region,
|
||||||
s3: svc,
|
s3: svc,
|
||||||
}, nil
|
}, nil
|
||||||
}
|
}
|
||||||
|
@ -119,6 +130,7 @@ func (p *Provider) PutFile(f providers.File) (providers.File, error) {
|
||||||
}
|
}
|
||||||
|
|
||||||
if _, err = p.s3.PutObject(&s3.PutObjectInput{
|
if _, err = p.s3.PutObject(&s3.PutObjectInput{
|
||||||
|
ACL: aws.String(p.getFileACL(f.Info().RelativeName)),
|
||||||
Body: bytes.NewReader(buf.Bytes()),
|
Body: bytes.NewReader(buf.Bytes()),
|
||||||
Bucket: aws.String(p.bucket),
|
Bucket: aws.String(p.bucket),
|
||||||
Key: aws.String(f.Info().RelativeName),
|
Key: aws.String(f.Info().RelativeName),
|
||||||
|
@ -130,5 +142,33 @@ func (p *Provider) PutFile(f providers.File) (providers.File, error) {
|
||||||
}
|
}
|
||||||
|
|
||||||
func (p *Provider) Share(relativeName string) (string, error) {
|
func (p *Provider) Share(relativeName string) (string, error) {
|
||||||
return "", errors.New("Not implemented")
|
_, err := p.s3.PutObjectAcl(&s3.PutObjectAclInput{
|
||||||
|
ACL: aws.String(s3.ObjectCannedACLPublicRead),
|
||||||
|
Bucket: aws.String(p.bucket),
|
||||||
|
Key: aws.String(relativeName),
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
return "", errors.Wrap(err, "Unable to publish file")
|
||||||
|
}
|
||||||
|
|
||||||
|
return fmt.Sprintf("https://s3-%s.amazonaws.com/%s/%s", p.bucketRegion, p.bucket, relativeName), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *Provider) getFileACL(relativeName string) string {
|
||||||
|
objACL, err := p.s3.GetObjectAcl(&s3.GetObjectAclInput{
|
||||||
|
Bucket: aws.String(p.bucket),
|
||||||
|
Key: aws.String(relativeName),
|
||||||
|
})
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
return s3.ObjectCannedACLPrivate
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, g := range objACL.Grants {
|
||||||
|
if *g.Grantee.URI == "http://acs.amazonaws.com/groups/global/AllUsers" && *g.Permission == "READ" {
|
||||||
|
return s3.ObjectCannedACLPublicRead
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return s3.ObjectCannedACLPrivate
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in a new issue