96 lines
2 KiB
Go
96 lines
2 KiB
Go
package main
|
|
|
|
import (
|
|
"archive/tar"
|
|
"context"
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
"os/signal"
|
|
"runtime"
|
|
"time"
|
|
|
|
"code.geekeey.de/actions/artifacts"
|
|
"code.geekeey.de/actions/sdk"
|
|
"code.geekeey.de/actions/sdk/iox"
|
|
)
|
|
|
|
func main() {
|
|
action := &SaveArtifactAction{Action: sdk.New()}
|
|
ctx, cancel := signal.NotifyContext(context.Background(), os.Interrupt)
|
|
defer cancel()
|
|
if err := action.Run(ctx); err != nil {
|
|
action.Errorf("%s", err)
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
|
|
type SaveArtifactAction struct {
|
|
*sdk.Action
|
|
name string
|
|
}
|
|
|
|
func (action *SaveArtifactAction) setup() error {
|
|
return nil
|
|
}
|
|
|
|
func (action *SaveArtifactAction) Run(ctx context.Context) error {
|
|
if err := action.setup(); err != nil {
|
|
return err
|
|
}
|
|
|
|
run, job, err := artifacts.JobInfo()
|
|
if err != nil {
|
|
return fmt.Errorf("unable to get job info: %v", err)
|
|
}
|
|
|
|
github := artifacts.NewGitHub(artifacts.NewActionClient())
|
|
|
|
expire := time.Now().Add(20 * time.Hour)
|
|
create, err := github.CreateArtifact(ctx, artifacts.CreateArtifactRequest{
|
|
RunID: run,
|
|
JobRunID: job,
|
|
Name: action.name,
|
|
Version: 4,
|
|
ExpiresAt: &expire,
|
|
})
|
|
if err != nil {
|
|
return fmt.Errorf("cannot create artifact %w", err)
|
|
}
|
|
if !create.Ok {
|
|
return fmt.Errorf("cannot get pre-signed URL")
|
|
}
|
|
|
|
rd, err := action.CreateArtifact(ctx)
|
|
|
|
res, err := artifacts.PushBlob_(ctx, rd, create.SignedUploadUrl, 1024*1024*1, runtime.NumCPU())
|
|
|
|
finish, err := github.FinalizeArtifact(ctx, artifacts.FinalizeArtifactRequest{
|
|
RunID: run,
|
|
JobRunID: job,
|
|
Name: action.name,
|
|
Size: res.Size,
|
|
Hash: res.SHA256Sum,
|
|
})
|
|
if err != nil {
|
|
return fmt.Errorf("cannot finish artifact %w", err)
|
|
}
|
|
if !create.Ok {
|
|
return fmt.Errorf("cannot finish artifact upload")
|
|
}
|
|
|
|
action.Noticef("created artifact: %s", finish.ArtifactId)
|
|
|
|
return nil
|
|
}
|
|
|
|
func (action *SaveArtifactAction) CreateArtifact(ctx context.Context) (io.Reader, error) {
|
|
rd, wr, err := os.Pipe()
|
|
go func() {
|
|
defer wr.Close()
|
|
|
|
t := tar.NewWriter(wr)
|
|
t.AddFS(iox.NewGlobFS(os.DirFS(""), ""))
|
|
}()
|
|
return rd, err
|
|
}
|