add: reference operations for cmds

Author: ja7ad

cmd/downgrade.go

@@ -1,9 +1,10 @@
 package cmd
 
 import (
-	"fmt"
-
+	"context"
+	"github.com/GoFarsi/gvm/internal"
 	"github.com/spf13/cobra"
+	"log"
 )
 
 // downgradeCmd represents the downgrade command
@@ -20,7 +21,19 @@ For example:
   $ gvm downgrade --version 1.x.x --backup
 `,
 	Run: func(cmd *cobra.Command, args []string) {
-		fmt.Println("implement me!!")
+		ver, err := cmd.Flags().GetString("version")
+		if err != nil {
+			log.Fatalln(err)
+		}
+
+		backup, err := cmd.Flags().GetBool("backup")
+		if err != nil {
+			log.Fatalln(err)
+		}
+
+		if err := internal.DowngradeGo(context.Background(), ver, backup); err != nil {
+			log.Fatalln(err)
+		}
 	},
 }
 

cmd/install.go

@@ -1,7 +1,9 @@
 package cmd
 
 import (
-	"fmt"
+	"context"
+	"github.com/GoFarsi/gvm/internal"
+	"log"
 
 	"github.com/spf13/cobra"
 )
@@ -20,7 +22,19 @@ For example:
   $ gvm install --version 1.x.x --backup
 `,
 	Run: func(cmd *cobra.Command, args []string) {
-		fmt.Println("implement me!!!")
+		ver, err := cmd.Flags().GetString("version")
+		if err != nil {
+			log.Fatalln(err)
+		}
+
+		backup, err := cmd.Flags().GetBool("backup")
+		if err != nil {
+			log.Fatalln(err)
+		}
+
+		if err := internal.InstallGo(context.Background(), ver, backup); err != nil {
+			log.Fatalln(err)
+		}
 	},
 }
 

cmd/list.go

@@ -2,8 +2,8 @@ package cmd
 
 import (
 	"fmt"
-	"github.com/GoFarsi/gvm/api"
 	"github.com/GoFarsi/gvm/errors"
+	"github.com/GoFarsi/gvm/internal"
 	"log"
 
 	"github.com/spf13/cobra"
@@ -19,7 +19,7 @@ For example:
   $ gvm list
   $ gvm list --line 10`,
 	Run: func(cmd *cobra.Command, args []string) {
-		list, err := api.NewList()
+		list, err := internal.NewList()
 		if err != nil {
 			log.Fatalln(errors.ERR_NETWORK_TIMEOUT)
 		}

cmd/upgrade.go

@@ -1,7 +1,9 @@
 package cmd
 
 import (
-	"fmt"
+	"context"
+	"github.com/GoFarsi/gvm/internal"
+	"log"
 
 	"github.com/spf13/cobra"
 )
@@ -20,7 +22,19 @@ For example:
   $ gvm upgrade --version 1.x.x --backup
 `,
 	Run: func(cmd *cobra.Command, args []string) {
-		fmt.Println("upgrade called")
+		ver, err := cmd.Flags().GetString("version")
+		if err != nil {
+			log.Fatalln(err)
+		}
+
+		backup, err := cmd.Flags().GetBool("backup")
+		if err != nil {
+			log.Fatalln(err)
+		}
+
+		if err := internal.UpgradeGo(context.Background(), ver, backup); err != nil {
+			log.Fatalln(err)
+		}
 	},
 }
 

api/download.go renamed to internal/download.go

@@ -1,10 +1,12 @@
-package api
+package internal
 
 import (
 	"context"
 	"fmt"
 	"github.com/Code-Hex/pget"
 	"github.com/GoFarsi/gvm/cli"
+	"github.com/GoFarsi/gvm/errors"
+	"log"
 	"net/http"
 	"os"
 	"runtime"
@@ -13,16 +15,19 @@ import (
 type Download struct {
 	version      string
 	downloadPath string
+	backup       bool
 }
 
 type Target struct {
 	FileName      string
+	Path          string
 	ContentLength int64
 	Url           string
+	backup        bool
 }
 
-func newDownload(ver string, downloadPath string) (*Download, error) {
-	def, err := defaultCfg()
+func newDownload(ver string, backup bool) (*Download, error) {
+	def, err := defaultCfg(backup)
 	if err != nil {
 		return nil, err
 	}
@@ -31,35 +36,46 @@ func newDownload(ver string, downloadPath string) (*Download, error) {
 		def.version = ver
 	}
 
-	if len(downloadPath) != 0 {
-		def.downloadPath = downloadPath
-	}
-
 	return def, nil
 }
 
-func defaultCfg() (*Download, error) {
+func defaultCfg(backup bool) (*Download, error) {
 	list, err := NewList()
 	if err != nil {
 		return nil, err
 	}
-	return &Download{
+
+	dl := &Download{
 		version:      list.LastVersion(),
 		downloadPath: os.TempDir(),
-	}, nil
+		backup:       backup,
+	}
+
+	if backup {
+		home, err := os.UserHomeDir()
+		if err != nil {
+			return nil, err
+		}
+		dl.downloadPath = home
+	}
+
+	return dl, nil
 }
 
-func (d *Download) Download(ctx context.Context) error {
+func (d *Download) download(ctx context.Context) (*Target, error) {
 	target := d.getFileInfo(ctx, fmt.Sprintf(filePattern, d.version, runtime.GOARCH))
 	if target == nil {
-		return nil
+		return nil, errors.ERR_CANT_FIND_ACTIVE_MIRROR
 	}
 
-	if err := target.download(ctx, d.downloadPath); err != nil {
-		return err
+	target.Path = d.downloadPath
+	target.backup = d.backup
+
+	if err := target.download(ctx); err != nil {
+		return nil, err
 	}
 
-	return nil
+	return target, nil
 }
 
 func (*Download) getFileInfo(ctx context.Context, fileName string) *Target {
@@ -92,17 +108,25 @@ func (*Download) getFileInfo(ctx context.Context, fileName string) *Target {
 	return nil
 }
 
-func (t *Target) download(ctx context.Context, downloadPath string) error {
+func (t *Target) download(ctx context.Context) error {
+	log.Printf("started download %s with size %d MB...", t.FileName, t.ContentLength/1024/1024)
 	dl := pget.New()
 	args := []string{
 		"-o",
-		downloadPath,
+		t.Path,
 		"-t",
 		"30",
 		t.Url,
 	}
 	if err := dl.Run(ctx, "", args); err != nil {
 		return err
 	}
+
+	if t.backup {
+		log.Printf("download file %s completed and backed up in %s path.", t.FileName, t.Path)
+		return nil
+	}
+
+	log.Printf("download file %s completed.", t.FileName)
 	return nil
 }

api/list.go renamed to internal/list.go

@@ -1,4 +1,4 @@
-package api
+package internal
 
 import (
 	"context"