GraphQL API and utilities for the rpdata project
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 

96 lines
2.4 KiB

package space
import (
"context"
"errors"
"fmt"
"git.aiterp.net/rpdata/api/internal/config"
"io"
minio "github.com/minio/minio-go"
)
type Client struct {
bucket string
urlRoot string
spaceRoot string
maxSize int64
s3 *minio.Client
}
func (client *Client) MaxSize() int64 {
return client.maxSize
}
// UploadFile uploads the file to the space. This does not do any checks on it, so the endpoints should
// ensure that's all okay.
func (client *Client) UploadFile(ctx context.Context, name string, mimeType string, reader io.Reader, size int64) error {
if size > client.maxSize {
return errors.New("file is too big")
}
_, err := client.s3.PutObjectWithContext(ctx, client.bucket, client.spaceRoot+"/"+name, reader, size, minio.PutObjectOptions{
ContentType: mimeType,
UserMetadata: map[string]string{
"x-amz-acl": "public-read",
},
})
if err != nil {
return err
}
_, err = client.s3.StatObject(client.bucket, client.spaceRoot+"/"+name, minio.StatObjectOptions{})
if err != nil {
return err
}
return nil
}
// RemoveFile removes a file from the space
func (client *Client) RemoveFile(name string) error {
return client.s3.RemoveObject(client.bucket, client.spaceRoot+"/"+name)
}
// DownloadFile opens a file for download, using the same path format as the UploadFile function. Remember to Close it!
// The returned stream also has `ReaderAt` and `Seeker`, but those are impl. details.
func (client *Client) DownloadFile(ctx context.Context, path string) (io.ReadCloser, error) {
return client.s3.GetObjectWithContext(ctx, client.bucket, client.spaceRoot+"/"+path, minio.GetObjectOptions{})
}
// URLFromPath gets the URL from the path returned by UploadFile
func (client *Client) URLFromPath(path string) string {
return client.urlRoot + path
}
// Connect connects to a S3 space.
func Connect(cfg config.Space) (*Client, error) {
client, err := minio.New(cfg.Host, cfg.AccessKey, cfg.SecretKey, true)
if err != nil {
return nil, err
}
exists, err := client.BucketExists(cfg.Bucket)
if err != nil {
return nil, err
}
if !exists {
return nil, errors.New("bucket not found")
}
urlRoot := cfg.URLRoot
if urlRoot == "" {
urlRoot = fmt.Sprintf("https://%s.%s/%s/", cfg.Bucket, cfg.Host, cfg.Root)
} else {
urlRoot += "/" + cfg.Root + "/"
}
return &Client{
bucket: cfg.Bucket,
urlRoot: urlRoot,
spaceRoot: cfg.Root,
maxSize: cfg.MaxSize,
s3: client,
}, nil
}