92 lines
2.3 KiB
Go
92 lines
2.3 KiB
Go
|
package runner
|
||
|
|
|
||
|
|
import (
|
||
|
|
"archive/tar"
|
||
|
|
"bytes"
|
||
|
|
"context"
|
||
|
|
"fmt"
|
||
|
|
"io"
|
||
|
|
"io/fs"
|
||
|
|
goURL "net/url"
|
||
|
|
"os"
|
||
|
|
"path/filepath"
|
||
|
|
"strings"
|
||
|
|
|
||
|
|
"github.com/nektos/act/pkg/filecollector"
|
||
|
|
)
|
||
|
|
|
||
|
|
type LocalRepositoryCache struct {
|
||
|
|
Parent ActionCache
|
||
|
|
LocalRepositories map[string]string
|
||
|
|
CacheDirCache map[string]string
|
||
|
|
}
|
||
|
|
|
||
|
|
func (l *LocalRepositoryCache) Fetch(ctx context.Context, cacheDir, url, ref, token string) (string, error) {
|
||
|
|
if dest, ok := l.LocalRepositories[fmt.Sprintf("%s@%s", url, ref)]; ok {
|
||
|
|
l.CacheDirCache[fmt.Sprintf("%s@%s", cacheDir, ref)] = dest
|
||
|
|
return ref, nil
|
||
|
|
}
|
||
|
|
if purl, err := goURL.Parse(url); err == nil {
|
||
|
|
if dest, ok := l.LocalRepositories[fmt.Sprintf("%s@%s", strings.TrimPrefix(purl.Path, "/"), ref)]; ok {
|
||
|
|
l.CacheDirCache[fmt.Sprintf("%s@%s", cacheDir, ref)] = dest
|
||
|
|
return ref, nil
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return l.Parent.Fetch(ctx, cacheDir, url, ref, token)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (l *LocalRepositoryCache) GetTarArchive(ctx context.Context, cacheDir, sha, includePrefix string) (io.ReadCloser, error) {
|
||
|
|
// sha is mapped to ref in fetch if there is a local override
|
||
|
|
if dest, ok := l.CacheDirCache[fmt.Sprintf("%s@%s", cacheDir, sha)]; ok {
|
||
|
|
srcPath := filepath.Join(dest, includePrefix)
|
||
|
|
buf := &bytes.Buffer{}
|
||
|
|
tw := tar.NewWriter(buf)
|
||
|
|
defer tw.Close()
|
||
|
|
srcPath = filepath.Clean(srcPath)
|
||
|
|
fi, err := os.Lstat(srcPath)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
tc := &filecollector.TarCollector{
|
||
|
|
TarWriter: tw,
|
||
|
|
}
|
||
|
|
if fi.IsDir() {
|
||
|
|
srcPrefix := srcPath
|
||
|
|
if !strings.HasSuffix(srcPrefix, string(filepath.Separator)) {
|
||
|
|
srcPrefix += string(filepath.Separator)
|
||
|
|
}
|
||
|
|
fc := &filecollector.FileCollector{
|
||
|
|
Fs: &filecollector.DefaultFs{},
|
||
|
|
SrcPath: srcPath,
|
||
|
|
SrcPrefix: srcPrefix,
|
||
|
|
Handler: tc,
|
||
|
|
}
|
||
|
|
err = filepath.Walk(srcPath, fc.CollectFiles(ctx, []string{}))
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
var f io.ReadCloser
|
||
|
|
var linkname string
|
||
|
|
if fi.Mode()&fs.ModeSymlink != 0 {
|
||
|
|
linkname, err = os.Readlink(srcPath)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
f, err = os.Open(srcPath)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
defer f.Close()
|
||
|
|
}
|
||
|
|
err := tc.WriteFile(fi.Name(), fi, linkname, f)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return io.NopCloser(buf), nil
|
||
|
|
}
|
||
|
|
return l.Parent.GetTarArchive(ctx, cacheDir, sha, includePrefix)
|
||
|
|
}
|