WIP: feat: cancel git clone ops gracefully

See #528

pkg/git/clone.go

@@ -1,8 +1,11 @@
 package git
 
 import (
+	"fmt"
 	"os"
+	"os/signal"
 	"strings"
+	"syscall"
 
 	"coopcloud.tech/abra/pkg/log"
 	"github.com/go-git/go-git/v5"
@@ -24,6 +27,14 @@ func gitCloneIgnoreErr(err error) bool {
 
 // Clone runs a git clone which accounts for different default branches.
 func Clone(dir, url string) error {
+	sigIntCh := make(chan os.Signal)
+	signal.Notify(sigIntCh, os.Interrupt, syscall.SIGTERM)
+	defer signal.Stop(sigIntCh)
+
+	errCh := make(chan error)
+	waitCh := make(chan struct{})
+
+	go func() {
 		if _, err := os.Stat(dir); os.IsNotExist(err) {
 			log.Debugf("git clone: %s", url)
 
@@ -36,7 +47,7 @@ func Clone(dir, url string) error {
 
 			if err != nil && gitCloneIgnoreErr(err) {
 				log.Debugf("git clone: %s cloned successfully", dir)
-			return nil
+				errCh <- nil
 			}
 
 			if err != nil {
@@ -51,11 +62,11 @@ func Clone(dir, url string) error {
 
 				if err != nil && gitCloneIgnoreErr(err) {
 					log.Debugf("git clone: %s cloned successfully", dir)
-				return nil
+					errCh <- nil
 				}
 
 				if err != nil {
-				return err
+					errCh <- err
 				}
 			}
 
@@ -64,5 +75,21 @@ func Clone(dir, url string) error {
 			log.Debugf("git clone: %s already exists", dir)
 		}
 
+		close(waitCh)
+	}()
+
+	select {
+	case <-waitCh:
+		return nil
+	case <-sigIntCh:
+		fmt.Println() // NOTE(d1): newline after ^C
+		if err := os.RemoveAll(dir); err != nil {
+			return fmt.Errorf("unable to clean up git clone of %s: %s", url, err)
+		}
+		return fmt.Errorf("git clone %s: cancelled due to interrupt", url)
+	case err := <-errCh:
+		return err
+	}
+
 	return nil
 }

pkg/git/clone_test.go

@@ -0,0 +1,34 @@
+package git
+
+import (
+	"fmt"
+	"os"
+	"path"
+	"syscall"
+	"testing"
+
+	"coopcloud.tech/abra/pkg/config"
+)
+
+func TestCancelGitClone(t *testing.T) {
+	dir := path.Join(config.RECIPES_DIR, "gitea")
+	os.RemoveAll(dir)
+
+	go func() {
+		p, err := os.FindProcess(os.Getpid())
+		if err != nil {
+			t.Fatalf("unable to find current process: %s", err)
+		}
+
+		p.Signal(syscall.SIGINT)
+	}()
+
+	gitURL := fmt.Sprintf("%s/%s.git", config.REPOS_BASE_URL, "gitea")
+	if err := Clone(dir, gitURL); err == nil {
+		t.Fatal("cloning should have been interrupted")
+	}
+
+	if _, err := os.Stat(dir); err != nil && !os.IsNotExist(err) {
+		t.Fatal("recipe repo was not deleted")
+	}
+}