feat(provider): add golang + fix semver check

Author: gorillamoe

README.md

@@ -110,9 +110,9 @@ The packages are installed in the following directories:
 - [x] Install `pkg:pypi` packages
 - [x] Update `pkg:pypi` packages
 - [x] Remove `pkg:pypi` packages
-- [ ] Install `pkg:golang` packages
-- [ ] Update `pkg:golang` packages
-- [ ] Remove `pkg:golang` packages
+- [x] Install `pkg:golang` packages
+- [x] Update `pkg:golang` packages
+- [x] Remove `pkg:golang` packages
 - [ ] Install `pkg:cargo` packages
 - [ ] Update `pkg:cargo` packages
 - [ ] Remove `pkg:cargo` packages

internal/lib/semver/root.go

@@ -1,14 +1,20 @@
 package semver
 
 import (
-	"fmt"
+	"log"
 	"strconv"
 	"strings"
 )
 
+func trimVersion(version string) string {
+	return strings.TrimPrefix(version, "v")
+}
+
 // IsGreater compares two semver version strings and
 // returns true if the second version is greater than the first.
 func IsGreater(v1, v2 string) bool {
+	v1 = trimVersion(v1)
+	v2 = trimVersion(v2)
 	// Split version strings into parts (major, minor, patch)
 	v1Parts := strings.Split(v1, ".")
 	v2Parts := strings.Split(v2, ".")
@@ -27,9 +33,9 @@ func IsGreater(v1, v2 string) bool {
 		v2Num, err2 := strconv.Atoi(v2Parts[i])
 
 		if err1 != nil || err2 != nil {
-			fmt.Println("Invalid version format.")
-			fmt.Println("v1: ", v1)
-			fmt.Println("v2: ", v2)
+			log.Println("Invalid version format.", err1, err2)
+			log.Println("v1: ", v1)
+			log.Println("v2: ", v2)
 			return false
 		}
 

internal/lib/shell_out/root.go

@@ -1,13 +1,15 @@
 package shell_out
 
 import (
+	"os"
 	"os/exec"
 )
 
 func ShellOut(command string, args []string, dir string, env []string) (int, error) {
 	cmd := exec.Command(command, args...)
 	cmd.Dir = dir
 	if env != nil {
+		env = append(env, os.Environ()...)
 		cmd.Env = append(cmd.Env, env...)
 	}
 	err := cmd.Run()

internal/lib/updater/golang_provider.go

@@ -0,0 +1,152 @@
+package updater
+
+import (
+	"encoding/json"
+	"fmt"
+	"log"
+	"os"
+	"path/filepath"
+	"regexp"
+
+	"github.com/mistweaverco/zana-client/internal/lib/files"
+	"github.com/mistweaverco/zana-client/internal/lib/local_packages_parser"
+	"github.com/mistweaverco/zana-client/internal/lib/shell_out"
+)
+
+type GolangProvider struct {
+	APP_PACKAGES_DIR string
+	PREFIX           string
+}
+
+func NewProviderGolang() *GolangProvider {
+	p := &GolangProvider{}
+	p.APP_PACKAGES_DIR = filepath.Join(files.GetAppPackagesPath(), "golang")
+	p.PREFIX = "pkg:golang/"
+	return p
+}
+
+func (p *GolangProvider) getRepo(sourceID string) string {
+	re := regexp.MustCompile("^" + p.PREFIX + "(.*)")
+	matches := re.FindStringSubmatch(sourceID)
+	if len(matches) > 1 {
+		return matches[1]
+	}
+	return ""
+}
+
+func (p *GolangProvider) generatePackageJSON() bool {
+	found := false
+	packageJSON := struct {
+		Dependencies map[string]string `json:"dependencies"`
+	}{
+		Dependencies: make(map[string]string),
+	}
+
+	localPackages := local_packages_parser.GetData(true).Packages
+	for _, pkg := range localPackages {
+		if detectProvider(pkg.SourceID) != ProviderGolang {
+			continue
+		}
+		packageJSON.Dependencies[p.getRepo(pkg.SourceID)] = pkg.Version
+		found = true
+	}
+
+	filePath := filepath.Join(p.APP_PACKAGES_DIR, "package.json")
+	file, err := os.Create(filePath)
+	if err != nil {
+		fmt.Println("Error creating package.json:", err)
+		return false
+	}
+	defer file.Close()
+
+	encoder := json.NewEncoder(file)
+	encoder.SetIndent("", "  ")
+	err = encoder.Encode(packageJSON)
+	if err != nil {
+		fmt.Println("Error encoding package.json:", err)
+		return false
+	}
+
+	return found
+}
+
+func (p *GolangProvider) Clean() bool {
+	err := os.RemoveAll(p.APP_PACKAGES_DIR)
+	if err != nil {
+		log.Println("Error removing directory:", err)
+		return false
+	}
+	return p.Sync()
+}
+
+func (p *GolangProvider) Sync() bool {
+	if _, err := os.Stat(p.APP_PACKAGES_DIR); os.IsNotExist(err) {
+		err := os.Mkdir(p.APP_PACKAGES_DIR, 0755)
+		if err != nil {
+			fmt.Println("Error creating directory:", err)
+			return false
+		}
+	}
+
+	packagesFound := p.generatePackageJSON()
+
+	if !packagesFound {
+		return true
+	}
+
+	returnResult := true
+
+	filePath := filepath.Join(p.APP_PACKAGES_DIR, "package.json")
+	packageJSON, err := os.ReadFile(filePath)
+	if err != nil {
+		log.Println("Error reading golang package.json:", err)
+		return false
+	}
+
+	pkgJSON := struct {
+		Dependencies map[string]string `json:"dependencies"`
+	}{
+		Dependencies: make(map[string]string),
+	}
+
+	err = json.Unmarshal(packageJSON, &pkgJSON)
+	if err != nil {
+		log.Println("Error unmarshalling golang package.json:", err)
+		return false
+	}
+
+	gobin := filepath.Join(p.APP_PACKAGES_DIR, "bin")
+
+	for pkg, version := range pkgJSON.Dependencies {
+		installCode, err := shell_out.ShellOut("go", []string{
+			"install",
+			pkg + "@" + version,
+		}, p.APP_PACKAGES_DIR,
+			[]string{
+				"GOBIN=" + gobin,
+			})
+		if err != nil || installCode != 0 {
+			log.Println("Error running go install:", err)
+			log.Println("Tried installing package:", pkg+"@"+version, "in:", gobin)
+			returnResult = false
+		}
+	}
+
+	return returnResult
+}
+
+func (p *GolangProvider) Install(sourceID, version string) bool {
+	err := local_packages_parser.AddLocalPackage(sourceID, version)
+	if err != nil {
+		return false
+	}
+	return p.Sync()
+}
+
+func (p *GolangProvider) Remove(sourceID string) bool {
+	err := local_packages_parser.RemoveLocalPackage(sourceID)
+	if err != nil {
+		return false
+	}
+	return p.Sync()
+}

internal/lib/updater/root.go

@@ -11,11 +11,13 @@ type Provider int
 const (
 	ProviderNPM Provider = iota
 	ProviderPyPi
+	ProviderGolang
 	ProviderUnsupported
 )
 
 var npmProvider NPMProvider = *NewProviderNPM()
 var pypiProvider PyPiProvider = *NewProviderPyPi()
+var golangProvider GolangProvider = *NewProviderGolang()
 
 func detectProvider(sourceId string) Provider {
 	var provider Provider
@@ -24,6 +26,8 @@ func detectProvider(sourceId string) Provider {
 		provider = ProviderNPM
 	case strings.HasPrefix(sourceId, pypiProvider.PREFIX):
 		provider = ProviderPyPi
+	case strings.HasPrefix(sourceId, golangProvider.PREFIX):
+		provider = ProviderGolang
 	default:
 		provider = ProviderUnsupported
 	}
@@ -42,6 +46,7 @@ func CheckIfUpdateIsAvailable(localVersion string, remoteVersion string) (bool,
 func SyncAll() {
 	npmProvider.Sync()
 	pypiProvider.Sync()
+	golangProvider.Sync()
 }
 
 func Install(sourceId string, version string) bool {
@@ -51,6 +56,8 @@ func Install(sourceId string, version string) bool {
 		return npmProvider.Install(sourceId, version)
 	case ProviderPyPi:
 		return pypiProvider.Install(sourceId, version)
+	case ProviderGolang:
+		return golangProvider.Install(sourceId, version)
 	case ProviderUnsupported:
 		// Unsupported provider
 	}
@@ -64,6 +71,8 @@ func Remove(sourceId string) bool {
 		return npmProvider.Remove(sourceId)
 	case ProviderPyPi:
 		return pypiProvider.Remove(sourceId)
+	case ProviderGolang:
+		return golangProvider.Remove(sourceId)
 	case ProviderUnsupported:
 		// Unsupported provider
 	}