fix: source when reading from cache

Signed-off-by: Yves Brissaud <[email protected]>
eunomie · Oct 14, 2024 · eb09bf9 · eb09bf9
1 parent 9d9c14f
commit eb09bf9
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/runkit/cache.go b/runkit/cache.go
@@ -37,7 +37,7 @@ func NewLocalCache(cli command.Cli) *LocalCache {
 	}
 }
 
-func (c *LocalCache) Get(digest string) (*RunKit, error) {
+func (c *LocalCache) Get(digest, src string) (*RunKit, error) {
 	rk := &RunKit{
 		Files: make(map[string]string),
 	}
@@ -66,6 +66,7 @@ func (c *LocalCache) Get(digest string) (*RunKit, error) {
 	}
 
 	if found {
+		rk.src = src
 		return rk, nil
 	}
 	return nil, nil

diff --git a/runkit/read.go b/runkit/read.go
@@ -25,7 +25,7 @@ type (
 	}
 
 	Cache interface {
-		Get(digest string) (*RunKit, error)
+		Get(digest, src string) (*RunKit, error)
 		Set(digest string, runxConfig, runxDoc []byte) error
 	}
 )
@@ -60,7 +60,7 @@ func Get(ctx context.Context, cache Cache, src string) (*RunKit, error) {
 
 	indexDigest = desc.Digest.String()
 
-	cached, err = cache.Get(indexDigest)
+	cached, err = cache.Get(indexDigest, src)
 	if err == nil && cached != nil {
 		return cached, nil
 	}