Recursive symlink copy procedure directory (#109)

* Recursive symlink copy procedure directory

* Copy procedure directory into a temporary
directory using symlinks

* Add more asserts to the test case

Author: 8W9aG

internal/tests/procedure_url_test.go

@@ -26,7 +26,11 @@ func TestPrepareProcedureSourceURLLocal(t *testing.T) {
 	srcDir := fmt.Sprintf("file://%s", fooDir)
 	fooDst, err := util.PrepareProcedureSourceURL(srcDir)
 	assert.NoError(t, err)
-	assert.Equal(t, fooDir, fooDst)
+	assert.DirExists(t, fooDst)
+	assert.FileExists(t, filepath.Join(fooDst, "cog.yaml"))
+	fooPy := filepath.Join(fooDst, "predict.py")
+	assert.FileExists(t, fooPy)
+	assert.Contains(t, string(must.Get(os.ReadFile(fooPy))), "'predicting foo'")
 }
 
 func TestPrepareProcedureSourceURLRemote(t *testing.T) {

internal/util/procedure.go

@@ -5,13 +5,37 @@ import (
 	"encoding/hex"
 	"fmt"
 	"io"
+	"io/fs"
 	"net/http"
 	"net/url"
 	"os"
 	"os/exec"
 	"path/filepath"
 )
 
+func copyRecursiveSymlink(srcRoot, dstRoot string) error {
+	return filepath.WalkDir(srcRoot, func(path string, d fs.DirEntry, err error) error {
+		if err != nil {
+			return err
+		}
+
+		relPath, err := filepath.Rel(srcRoot, path)
+		if err != nil {
+			return err
+		}
+
+		dstPath := filepath.Join(dstRoot, relPath)
+
+		if d.IsDir() {
+			if path != srcRoot {
+				return os.MkdirAll(dstPath, 0o755)
+			}
+			return nil
+		}
+		return os.Symlink(path, dstPath)
+	})
+}
+
 func PrepareProcedureSourceURL(srcURL string) (string, error) {
 	u, err := url.Parse(srcURL)
 	if err != nil {
@@ -26,7 +50,15 @@ func PrepareProcedureSourceURL(srcURL string) (string, error) {
 		if !stat.IsDir() {
 			return "", fmt.Errorf("invalid procedure source URL: %s", srcURL)
 		}
-		return u.Path, nil
+		tmpDir, err := os.MkdirTemp("", "procedure*")
+		if err != nil {
+			return "", err
+		}
+		err = copyRecursiveSymlink(u.Path, tmpDir)
+		if err != nil {
+			return "", err
+		}
+		return tmpDir, nil
 	} else if u.Scheme == "http" || u.Scheme == "https" {
 		// http://host/path/to/tarball
 		sha := sha256.New()