mirror of
https://github.com/gomods/athens
synced 2026-02-03 11:00:32 +00:00
d932d50232
* feat: add golangci-lint linting * chore: fix linter issues * feat: add linting into the workflow * docs: update lint docs * fix: cr suggestions * fix: remove old formatting and vetting scripts * fix: add docker make target * fix: action go caching * fix: depreciated actions checkout version * fix: cr suggestion * fix: cr suggestions --------- Co-authored-by: Manu Gupta <manugupt1@gmail.com>
109 lines
3.2 KiB
Go
109 lines
3.2 KiB
Go
package s3
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"io"
|
|
|
|
"github.com/aws/aws-sdk-go/aws"
|
|
"github.com/aws/aws-sdk-go/service/s3"
|
|
"github.com/gomods/athens/pkg/config"
|
|
"github.com/gomods/athens/pkg/errors"
|
|
"github.com/gomods/athens/pkg/observ"
|
|
"github.com/gomods/athens/pkg/storage"
|
|
)
|
|
|
|
// Info implements the (./pkg/storage).Getter interface.
|
|
func (s *Storage) Info(ctx context.Context, module, version string) ([]byte, error) {
|
|
const op errors.Op = "s3.Info"
|
|
ctx, span := observ.StartSpan(ctx, op.String())
|
|
defer span.End()
|
|
exists, err := s.Exists(ctx, module, version)
|
|
if err != nil {
|
|
return nil, errors.E(op, err, errors.M(module), errors.V(version))
|
|
}
|
|
if !exists {
|
|
return nil, errors.E(op, errors.M(module), errors.V(version), errors.KindNotFound)
|
|
}
|
|
|
|
infoReader, err := s.open(ctx, config.PackageVersionedName(module, version, "info"))
|
|
if err != nil {
|
|
return nil, errors.E(op, err, errors.M(module), errors.V(version))
|
|
}
|
|
defer func() { _ = infoReader.Close() }()
|
|
|
|
infoBytes, err := io.ReadAll(infoReader)
|
|
if err != nil {
|
|
return nil, errors.E(op, err, errors.M(module), errors.V(version))
|
|
}
|
|
return infoBytes, nil
|
|
}
|
|
|
|
// GoMod implements the (./pkg/storage).Getter interface.
|
|
func (s *Storage) GoMod(ctx context.Context, module, version string) ([]byte, error) {
|
|
const op errors.Op = "s3.GoMod"
|
|
ctx, span := observ.StartSpan(ctx, op.String())
|
|
defer span.End()
|
|
exists, err := s.Exists(ctx, module, version)
|
|
if err != nil {
|
|
return nil, errors.E(op, err, errors.M(module), errors.V(version))
|
|
}
|
|
if !exists {
|
|
return nil, errors.E(op, errors.M(module), errors.V(version), errors.KindNotFound)
|
|
}
|
|
|
|
modReader, err := s.open(ctx, config.PackageVersionedName(module, version, "mod"))
|
|
if err != nil {
|
|
return nil, errors.E(op, err, errors.M(module), errors.V(version))
|
|
}
|
|
defer func() { _ = modReader.Close() }()
|
|
|
|
modBytes, err := io.ReadAll(modReader)
|
|
if err != nil {
|
|
return nil, errors.E(op, fmt.Errorf("could not get new reader for mod file: %w", err), errors.M(module), errors.V(version))
|
|
}
|
|
|
|
return modBytes, nil
|
|
}
|
|
|
|
// Zip implements the (./pkg/storage).Getter interface.
|
|
func (s *Storage) Zip(ctx context.Context, module, version string) (storage.SizeReadCloser, error) {
|
|
const op errors.Op = "s3.Zip"
|
|
ctx, span := observ.StartSpan(ctx, op.String())
|
|
defer span.End()
|
|
exists, err := s.Exists(ctx, module, version)
|
|
if err != nil {
|
|
return nil, errors.E(op, err, errors.M(module), errors.V(version))
|
|
}
|
|
if !exists {
|
|
return nil, errors.E(op, errors.M(module), errors.V(version), errors.KindNotFound)
|
|
}
|
|
|
|
zipReader, err := s.open(ctx, config.PackageVersionedName(module, version, "zip"))
|
|
if err != nil {
|
|
return nil, errors.E(op, err, errors.M(module), errors.V(version))
|
|
}
|
|
|
|
return zipReader, nil
|
|
}
|
|
|
|
func (s *Storage) open(ctx context.Context, path string) (storage.SizeReadCloser, error) {
|
|
const op errors.Op = "s3.open"
|
|
ctx, span := observ.StartSpan(ctx, op.String())
|
|
defer span.End()
|
|
getParams := &s3.GetObjectInput{
|
|
Bucket: aws.String(s.bucket),
|
|
Key: aws.String(path),
|
|
}
|
|
|
|
goo, err := s.s3API.GetObjectWithContext(ctx, getParams)
|
|
if err != nil {
|
|
return nil, errors.E(op, err)
|
|
}
|
|
var size int64
|
|
if goo.ContentLength != nil {
|
|
size = *goo.ContentLength
|
|
}
|
|
return storage.NewSizer(goo.Body, size), nil
|
|
}
|