mirror of
https://github.com/goreleaser/goreleaser.git
synced 2025-02-03 13:11:48 +02:00
107 lines
2.4 KiB
Go
107 lines
2.4 KiB
Go
// Package checksums provides a Pipe that creates .checksums files for
|
|
// each artifact.
|
|
package checksums
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"text/template"
|
|
|
|
"github.com/apex/log"
|
|
"golang.org/x/sync/errgroup"
|
|
|
|
"github.com/goreleaser/goreleaser/checksum"
|
|
"github.com/goreleaser/goreleaser/context"
|
|
"github.com/goreleaser/goreleaser/internal/artifact"
|
|
"github.com/goreleaser/goreleaser/internal/semaphore"
|
|
)
|
|
|
|
// Pipe for checksums
|
|
type Pipe struct{}
|
|
|
|
func (Pipe) String() string {
|
|
return "calculating checksums"
|
|
}
|
|
|
|
// Default sets the pipe defaults
|
|
func (Pipe) Default(ctx *context.Context) error {
|
|
if ctx.Config.Checksum.NameTemplate == "" {
|
|
ctx.Config.Checksum.NameTemplate = "{{ .ProjectName }}_{{ .Version }}_checksums.txt"
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// Run the pipe
|
|
func (Pipe) Run(ctx *context.Context) (err error) {
|
|
filename, err := filenameFor(ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
file, err := os.OpenFile(
|
|
filepath.Join(ctx.Config.Dist, filename),
|
|
os.O_APPEND|os.O_WRONLY|os.O_CREATE|os.O_TRUNC,
|
|
0444,
|
|
)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer file.Close() // nolint: errcheck
|
|
|
|
var g errgroup.Group
|
|
var sem = semaphore.New(ctx.Parallelism)
|
|
for _, artifact := range ctx.Artifacts.Filter(
|
|
artifact.Or(
|
|
artifact.ByType(artifact.UploadableArchive),
|
|
artifact.ByType(artifact.UploadableBinary),
|
|
artifact.ByType(artifact.LinuxPackage),
|
|
),
|
|
).List() {
|
|
sem.Acquire()
|
|
artifact := artifact
|
|
g.Go(func() error {
|
|
defer sem.Release()
|
|
return checksums(ctx, file, artifact)
|
|
})
|
|
}
|
|
ctx.Artifacts.Add(artifact.Artifact{
|
|
Type: artifact.Checksum,
|
|
Path: file.Name(),
|
|
Name: filename,
|
|
})
|
|
return g.Wait()
|
|
}
|
|
|
|
func checksums(ctx *context.Context, file *os.File, artifact artifact.Artifact) error {
|
|
log.WithField("file", artifact.Name).Info("checksumming")
|
|
sha, err := checksum.SHA256(artifact.Path)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
_, err = file.WriteString(fmt.Sprintf("%v %v\n", sha, artifact.Name))
|
|
return err
|
|
}
|
|
|
|
func filenameFor(ctx *context.Context) (string, error) {
|
|
var out bytes.Buffer
|
|
t, err := template.New("checksums").
|
|
Option("missingkey=error").
|
|
Parse(ctx.Config.Checksum.NameTemplate)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
err = t.Execute(&out, struct {
|
|
ProjectName string
|
|
Tag string
|
|
Version string
|
|
Env map[string]string
|
|
}{
|
|
ProjectName: ctx.Config.ProjectName,
|
|
Tag: ctx.Git.CurrentTag,
|
|
Version: ctx.Version,
|
|
Env: ctx.Env,
|
|
})
|
|
return out.String(), err
|
|
}
|