CR-5155 - gracefully handle repo-not-exist during clone

pkg/git/provider.go

@@ -49,7 +49,8 @@ var (
 )
 
 var supportedProviders = map[string]func(*ProviderOptions) (Provider, error){
-	"github": newGithub,
+	"github":     newGithub,
+	"github.com": newGithub,
 }
 
 // New creates a new git provider

pkg/git/provider_github.go

@@ -4,6 +4,7 @@ import (
 	"context"
 	"fmt"
 	"net/http"
+	"net/url"
 
 	g "github.com/argoproj-labs/argocd-autopilot/pkg/git/github"
 
@@ -18,13 +19,9 @@ type github struct {
 }
 
 func newGithub(opts *ProviderOptions) (Provider, error) {
-	var (
-		c   *gh.Client
-		err error
-	)
+	var c *gh.Client
 
 	hc := &http.Client{}
-
 	if opts.Auth != nil {
 		hc.Transport = &gh.BasicAuthTransport{
 			Username: opts.Auth.Username,
@@ -33,10 +30,19 @@ func newGithub(opts *ProviderOptions) (Provider, error) {
 	}
 
 	if opts.Host != "" {
-		c, err = gh.NewEnterpriseClient(opts.Host, opts.Host, hc)
+		u, err := url.Parse(opts.Host)
 		if err != nil {
 			return nil, err
 		}
+
+		if u.Hostname() == "github.com" {
+			c = gh.NewClient(hc)
+		} else {
+			c, err = gh.NewEnterpriseClient(opts.Host, opts.Host, hc)
+			if err != nil {
+				return nil, err
+			}
+		}
 	} else {
 		c = gh.NewClient(hc)
 	}

pkg/git/repository.go

@@ -5,6 +5,7 @@ import (
 	"errors"
 	"fmt"
 	"io"
+	"net/url"
 	"os"
 	"strings"
 
@@ -36,6 +37,7 @@ type (
 	}
 
 	CloneOptions struct {
+		Provider string
 		// URL clone url
 		Repo     string
 		Auth     Auth
@@ -136,10 +138,20 @@ func (o *CloneOptions) Clone(ctx context.Context) (Repository, fs.FS, error) {
 
 	r, err := clone(ctx, o)
 	if err != nil {
-		if err == transport.ErrEmptyRemoteRepository {
-			log.G(ctx).Debug("empty repository, initializing new one with specified remote")
-			r, err = initRepo(ctx, o)
+		if err != transport.ErrRepositoryNotFound && err != transport.ErrEmptyRemoteRepository {
+			return nil, nil, err
 		}
+
+		if err == transport.ErrRepositoryNotFound {
+			_, err = createRepo(ctx, o)
+			if err != nil {
+				return nil, nil, err
+			}
+		}
+
+		// just created repo OR err == transport.ErrEmptyRemoteRepository
+		log.G(ctx).Debug("empty repository, initializing new one with specified remote")
+		r, err = initRepo(ctx, o)
 	}
 
 	if err != nil {
@@ -229,6 +241,39 @@ var clone = func(ctx context.Context, opts *CloneOptions) (*repo, error) {
 	return repo, nil
 }
 
+var createRepo = func(ctx context.Context, opts *CloneOptions) (string, error) {
+	host, orgRepo, _, _, _, _, _ := util.ParseGitUrl(opts.Repo)
+	providerType := opts.Provider
+	if providerType == "" {
+		u, err := url.Parse(host)
+		if err != nil {
+			return "", err
+		}
+
+		providerType = u.Hostname()
+	}
+
+	p, err := NewProvider(&ProviderOptions{
+		Type: providerType,
+		Auth: &opts.Auth,
+		Host: host,
+	})
+	if err != nil {
+		return "", err
+	}
+
+	s := strings.Split(orgRepo, "/")
+	if len(s) != 2 {
+		return "", fmt.Errorf("Failed parsing organization and repo from '%s'", orgRepo)
+	}
+
+	return p.CreateRepository(ctx, &CreateRepoOptions{
+		Owner:   s[0],
+		Name:    s[1],
+		Private: true,
+	})
+}
+
 var initRepo = func(ctx context.Context, opts *CloneOptions) (*repo, error) {
 	ggr, err := ggInitRepo(memory.NewStorage(), opts.FS)
 	if err != nil {