From 5f0a4202c91578c2af33182f10448034783e7035 Mon Sep 17 00:00:00 2001
From: Tom Chauveau <tom.chauveau@epitech.eu>
Date: Fri, 29 Oct 2021 19:31:59 +0200
Subject: [PATCH] Complete version checking with `getUniverseCurrentVersion`

Dagger now compare the remote universe version with the one written
in dagger.mod.

I've also added logs to easily debug code

Signed-off-by: Tom Chauveau <tom.chauveau@epitech.eu>
---
 cmd/dagger/cmd/version.go | 45 ++++++++++++++++++++++++++++++++++-----
 mod/file.go               |  6 +++---
 2 files changed, 43 insertions(+), 8 deletions(-)

diff --git a/cmd/dagger/cmd/version.go b/cmd/dagger/cmd/version.go
index 14cda040..1b0076a3 100644
--- a/cmd/dagger/cmd/version.go
+++ b/cmd/dagger/cmd/version.go
@@ -1,6 +1,7 @@
 package cmd
 
 import (
+	"context"
 	"encoding/json"
 	"fmt"
 	"io/ioutil"
@@ -16,6 +17,8 @@ import (
 	"github.com/mitchellh/go-homedir"
 	"github.com/spf13/cobra"
 	"github.com/spf13/viper"
+	"go.dagger.io/dagger/cmd/dagger/cmd/common"
+	"go.dagger.io/dagger/cmd/dagger/logger"
 	"go.dagger.io/dagger/mod"
 	"go.dagger.io/dagger/version"
 	"golang.org/x/term"
@@ -46,8 +49,10 @@ var versionCmd = &cobra.Command{
 			runtime.GOOS, runtime.GOARCH,
 		)
 
-		// TODO Display universe version
-		// How can I retrieve it if it's not vendor ?
+		universeVersion, err := getUniverseCurrentVersion()
+		if err == nil {
+			fmt.Printf("universe %s\n", universeVersion.Original())
+		}
 
 		if check := viper.GetBool("check"); check {
 			versionFilePath, err := homedir.Expand(versionFile)
@@ -215,10 +220,33 @@ func getUniverseLatestVersion() (*goVersion.Version, error) {
 	return versions[0], nil
 }
 
+// Retrieve the current universe version from `cue.mod/dagger.mod`
 func getUniverseCurrentVersion() (*goVersion.Version, error) {
-	// TODO Should be replaced with the current universe version
-	// How I can fetch it
-	return goVersion.NewVersion("0.1.0")
+	project := common.CurrentProject(context.Background())
+	pathMod := path.Join(project.Path, mod.ModFilePath)
+	fileMod, err := os.Open(pathMod)
+	if err != nil {
+		return nil, err
+	}
+
+	defer fileMod.Close()
+	data, err := ioutil.ReadAll(fileMod)
+	if err != nil {
+		return nil, err
+	}
+
+	currentVersion := ""
+	modules := strings.Split(string(data), "\n")
+	for _, module := range modules {
+		if !strings.HasPrefix(module, "alpha.dagger.io") {
+			continue
+		}
+
+		// Retrieve tag
+		tag := strings.Split(module, " ")
+		currentVersion = tag[1]
+	}
+	return goVersion.NewVersion(currentVersion)
 }
 
 // Compare the universe version with the latest version online
@@ -241,8 +269,11 @@ func isUniverseVersionLatest() (string, error) {
 }
 
 func checkVersion() {
+	lg := logger.New()
+
 	if version.Version == version.DevelopmentVersion {
 		// running devel version
+		lg.Debug().Msg("ignore check version on devel version")
 		return
 	}
 
@@ -264,8 +295,10 @@ func checkVersion() {
 	}
 
 	// Check version
+	lg.Debug().Msg("check for universe latest version...")
 	universeLatestVersion, err := isUniverseVersionLatest()
 	if err != nil {
+		lg.Debug().Msg(err.Error())
 		return
 	}
 
@@ -274,8 +307,10 @@ func checkVersion() {
 	}
 
 	// Check timestamp
+	lg.Debug().Msg("check for dagger latest version...")
 	daggerLatestVersion, err := isDaggerVersionLatest()
 	if err != nil {
+		lg.Debug().Msg(err.Error())
 		return
 	}
 
diff --git a/mod/file.go b/mod/file.go
index 1cd4d89b..67a4f9ed 100644
--- a/mod/file.go
+++ b/mod/file.go
@@ -17,7 +17,7 @@ import (
 )
 
 const (
-	modFilePath  = "./cue.mod/dagger.mod"
+	ModFilePath  = "./cue.mod/dagger.mod"
 	sumFilePath  = "./cue.mod/dagger.sum"
 	lockFilePath = "./cue.mod/dagger.lock"
 	destBasePath = "./cue.mod/pkg"
@@ -31,7 +31,7 @@ type file struct {
 }
 
 func readPath(workspacePath string) (*file, error) {
-	pMod := path.Join(workspacePath, modFilePath)
+	pMod := path.Join(workspacePath, ModFilePath)
 	fMod, err := os.Open(pMod)
 	if err != nil {
 		if !errors.Is(err, fs.ErrNotExist) {
@@ -276,7 +276,7 @@ func (f *file) write() error {
 		bMod.WriteString(fmt.Sprintf("%s %s\n", r.fullPath(), r.version))
 	}
 
-	err := ioutil.WriteFile(path.Join(f.workspacePath, modFilePath), bMod.Bytes(), 0600)
+	err := ioutil.WriteFile(path.Join(f.workspacePath, ModFilePath), bMod.Bytes(), 0600)
 	if err != nil {
 		return err
 	}