fix: protect against one part install targets

pkg/commands/install/install.go

@@ -31,7 +31,7 @@ func Execute(c *cli.Context) error {
 	_ = os.MkdirAll(metadataDir, 0755)
 	_ = os.MkdirAll(downloadsDir, 0755)
 
-	source := source2.New(c.Args().First(), &source2.Options{
+	source, err := source2.New(c.Args().First(), &source2.Options{
 		OS:           c.String("os"),
 		Arch:         c.String("arch"),
 		HomeDir:      homeDir,
@@ -40,6 +40,9 @@ func Execute(c *cli.Context) error {
 		MetadataDir:  metadataDir,
 		DownloadsDir: downloadsDir,
 	})
+	if err != nil {
+		return err
+	}
 
 	fmt.Println(" source: ", source.GetSource())
 	fmt.Println("    app: ", source.GetApp())

pkg/osconfig/os.go

@@ -49,6 +49,8 @@ func New(os, arch string) *OS {
 	switch arch {
 	case AMD64:
 		newOS.Architectures = append(newOS.Architectures, "x86_64", "64bit", "64")
+	case ARM64:
+		newOS.Architectures = append(newOS.Architectures, "aarch64")
 	}
 
 	return newOS

pkg/source/github.go

@@ -49,6 +49,7 @@ func (s *GitHub) Run(ctx context.Context, version, githubToken string) error {
 
 	s.client = github.NewClient(httpcache.NewTransport(diskcache.New(cacheFile)).Client())
 	if githubToken != "" {
+		logrus.Debug("auth token provided")
 		s.client = s.client.WithAuthToken(githubToken)
 	}
 

pkg/source/source.go

@@ -2,6 +2,7 @@ package source
 
 import (
 	"context"
+	"fmt"
 	"strings"
 
 	"github.com/ekristen/distillery/pkg/asset"
@@ -88,7 +89,7 @@ func (s *Source) Cleanup() error {
 	return s.Binary.Cleanup()
 }
 
-func New(source string, opts *Options) ISource {
+func New(source string, opts *Options) (ISource, error) {
 	version := "latest"
 	versionParts := strings.Split(source, "@")
 	if len(versionParts) > 1 {
@@ -97,48 +98,53 @@ func New(source string, opts *Options) ISource {
 	}
 
 	parts := strings.Split(source, "/")
+
+	if len(parts) == 1 {
+		return nil, fmt.Errorf("invalid install source, expect format of owner/repo or owner/repo@version")
+	}
+
 	if len(parts) == 2 {
 		// could be github or homebrew or hashicorp
 		if parts[0] == "homebrew" {
 			return &Homebrew{
 				Source:  Source{Options: opts},
 				Formula: parts[1],
 				Version: version,
-			}
+			}, nil
 		} else if parts[0] == "hashicorp" {
 			return &Hashicorp{
 				Source:  Source{Options: opts},
 				Owner:   parts[1],
 				Repo:    parts[1],
 				Version: version,
-			}
+			}, nil
 		}
 
 		return &GitHub{
 			Source:  Source{Options: opts},
 			Owner:   parts[0],
 			Repo:    parts[1],
 			Version: version,
-		}
+		}, nil
 	} else if len(parts) >= 3 {
 		if strings.HasPrefix(parts[0], "github") {
 			return &GitHub{
 				Source:  Source{Options: opts},
 				Owner:   parts[1],
 				Repo:    parts[2],
 				Version: version,
-			}
+			}, nil
 		} else if strings.HasPrefix(parts[0], "gitlab") {
 			return &GitLab{
 				Source:  Source{Options: opts},
 				Owner:   parts[1],
 				Repo:    parts[2],
 				Version: version,
-			}
+			}, nil
 		}
 
-		return nil
+		return nil, nil
 	}
 
-	return nil
+	return nil, nil
 }