at-container-registry/pkg/hold/service.go

package hold

import (
	"context"
	"fmt"
	"log"

	"github.com/aws/aws-sdk-go/service/s3"
	storagedriver "github.com/distribution/distribution/v3/registry/storage/driver"
	"github.com/distribution/distribution/v3/registry/storage/driver/factory"
)

// HoldService provides presigned URLs for blob storage in a hold
type HoldService struct {
	driver       storagedriver.StorageDriver
	config       *Config
	s3Client     *s3.S3            // S3 client for presigned URLs (nil if not S3 storage)
	bucket       string            // S3 bucket name
	s3PathPrefix string            // S3 path prefix (if any)
	MultipartMgr *MultipartManager // Exported for access in route handlers
}

// NewHoldService creates a new hold service
func NewHoldService(cfg *Config) (*HoldService, error) {
	// Create storage driver from config
	ctx := context.Background()
	driver, err := factory.Create(ctx, cfg.Storage.Type(), cfg.Storage.Parameters())
	if err != nil {
		return nil, fmt.Errorf("failed to create storage driver: %w", err)
	}

	service := &HoldService{
		driver:       driver,
		config:       cfg,
		MultipartMgr: NewMultipartManager(),
	}

	// Initialize S3 client for presigned URLs (if using S3 storage)
	if err := service.initS3Client(); err != nil {
		log.Printf("WARNING: S3 presigned URLs disabled: %v", err)
	}

	return service, nil
}

// GetPresignedURL is a public wrapper around getPresignedURL for use by PDS blob store
func (s *HoldService) GetPresignedURL(ctx context.Context, operation PresignedURLOperation, digest string, did string) (string, error) {
	return s.getPresignedURL(ctx, operation, digest, did)
}