88 lines
2.5 KiB
Go
88 lines
2.5 KiB
Go
package azure
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"net/url"
|
|
|
|
"git.ksdemosapps.com/kylesoda/go-migrate/internal/app/config"
|
|
"github.com/Azure/azure-sdk-for-go/sdk/storage/azblob"
|
|
)
|
|
|
|
var (
|
|
ErrInvalidConnectionString = errors.New("invalid connection string")
|
|
ErrContainerNotFound = errors.New("container not found")
|
|
ErrBlobNotFound = errors.New("blob not found")
|
|
ErrInvalidInput = errors.New("invalid input parameters")
|
|
)
|
|
|
|
type Client struct {
|
|
client *azblob.Client
|
|
azureStorageConfig config.AzureStorageConfig
|
|
}
|
|
|
|
func NewClient(azureStorageConfig config.AzureStorageConfig) (*Client, error) {
|
|
protocol := "https"
|
|
if !azureStorageConfig.UseHTTPS {
|
|
protocol = "http"
|
|
}
|
|
|
|
blobEndpoint, _ := url.JoinPath(azureStorageConfig.ServiceURL, azureStorageConfig.AccountName)
|
|
connStr := fmt.Sprintf("DefaultEndpointsProtocol=%s;AccountName=%s;AccountKey=%s;BlobEndpoint=%s;",
|
|
protocol, azureStorageConfig.AccountName, azureStorageConfig.AccountKey, blobEndpoint)
|
|
|
|
client, err := azblob.NewClientFromConnectionString(connStr, nil)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("creating azure storage client: %w", err)
|
|
}
|
|
|
|
return &Client{
|
|
client: client,
|
|
azureStorageConfig: azureStorageConfig,
|
|
}, nil
|
|
}
|
|
|
|
func (c *Client) CreateContainer(ctx context.Context, containerName string) error {
|
|
if containerName == "" {
|
|
return ErrInvalidInput
|
|
}
|
|
|
|
_, err := c.client.CreateContainer(ctx, containerName, nil)
|
|
if err != nil {
|
|
return fmt.Errorf("creating container %s: %w", containerName, err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (c *Client) UploadBuffer(ctx context.Context, containerName, blobPath string, buffer []byte) error {
|
|
if containerName == "" || blobPath == "" || buffer == nil {
|
|
return ErrInvalidInput
|
|
}
|
|
|
|
_, err := c.client.UploadBuffer(ctx, containerName, blobPath, buffer, nil)
|
|
if err != nil {
|
|
return fmt.Errorf("uploading blob %s: %w", blobPath, err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (c *Client) UploadAndGetURL(ctx context.Context, blobPath string, buffer []byte) (string, error) {
|
|
if blobPath == "" || buffer == nil {
|
|
return "", ErrInvalidInput
|
|
}
|
|
|
|
fullPath := blobPath
|
|
if c.azureStorageConfig.Prefix != "" {
|
|
fullPath, _ = url.JoinPath(c.azureStorageConfig.Prefix, blobPath)
|
|
}
|
|
|
|
if err := c.UploadBuffer(ctx, c.azureStorageConfig.Container, fullPath, buffer); err != nil {
|
|
return "", err
|
|
}
|
|
|
|
blobEndpoint, _ := url.JoinPath(c.azureStorageConfig.ServiceURL, c.azureStorageConfig.AccountName)
|
|
blobURL, _ := url.JoinPath(blobEndpoint, c.azureStorageConfig.Container, fullPath)
|
|
return blobURL, nil
|
|
}
|