artifacts/save/main.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
}