Merge pull request #241 from HyperloopUPV-H8/backend/version-system

[Backend] Version system and auto-update

Author: JESUSJH18

backend/cmd/VERSION.md

@@ -0,0 +1 @@
+2.2.8

backend/cmd/main.go

@@ -7,18 +7,24 @@ import (
 	"flag"
 	"fmt"
 
+	"encoding/json"
 	"log"
 	"net"
 	"net/http"
 	_ "net/http/pprof"
 	"os"
+	"os/exec"
 	"os/signal"
 	"path"
+	"path/filepath"
 	"runtime"
 	"runtime/pprof"
+
 	"strings"
 	"time"
 
+	"github.com/hashicorp/go-version"
+
 	adj_module "github.com/HyperloopUPV-H8/h9-backend/internal/adj"
 	"github.com/HyperloopUPV-H8/h9-backend/internal/common"
 	"github.com/HyperloopUPV-H8/h9-backend/internal/pod_data"
@@ -76,9 +82,25 @@ var enableSNTP = flag.Bool("sntp", false, "enables a simple SNTP server on port
 var networkDevice = flag.Int("dev", -1, "index of the network device to use, overrides device prompt")
 var blockprofile = flag.Int("blockprofile", 0, "number of block profiles to include")
 var playbackFile = flag.String("playback", "", "")
+var currentVersion string
 
 func main() {
+
+	versionFile := "VERSION.md"
+	versionData, err := os.ReadFile(versionFile)
+	if err != nil {
+		fmt.Fprintf(os.Stderr, "Error reading version file (%s): %v\n", versionFile, err)
+		os.Exit(1)
+	}
+	currentVersion = strings.TrimSpace(string(versionData))
+
+	versionFlag := flag.Bool("version", false, "Show the backend version")
 	flag.Parse()
+	if *versionFlag {
+		fmt.Println("Hyperloop UPV Backend Version:", currentVersion)
+		os.Exit(0)
+	}
+
 	traceFile := initTrace(*traceLevel, *traceFile)
 	defer traceFile.Close()
 
@@ -99,6 +121,89 @@ func main() {
 	runtime.SetBlockProfileRate(*blockprofile)
 
 	config := getConfig("./config.toml")
+	latestVersionStr, err := getLatestVersionFromGitHub()
+	if err != nil {
+		fmt.Println("Warning:", err)
+		fmt.Println("Skipping version check. Proceeding with the current version:", currentVersion)
+	} else {
+		current, err := version.NewVersion(currentVersion)
+		if err != nil {
+			fmt.Println("Invalid current version:", err)
+			return
+		}
+
+		latest, err := version.NewVersion(latestVersionStr)
+		if err != nil {
+			fmt.Println("Invalid latest version:", err)
+			return
+		}
+
+		if latest.GreaterThan(current) {
+			fmt.Printf("There is a new version available: %s (current version: %s)\n", latest, current)
+			fmt.Print("Do you want to update? (y/n): ")
+
+			var response string
+			fmt.Scanln(&response)
+
+			if strings.ToLower(response) == "y" {
+				fmt.Println("Launching updater to update the backend...")
+
+				// Get the directory of the current executable
+				execPath, err := os.Executable()
+				if err != nil {
+					fmt.Fprintf(os.Stderr, "Error getting executable path: %v\n", err)
+					os.Exit(1)
+				}
+				execDir := filepath.Dir(execPath)
+
+				backendPath := filepath.Join(execDir, "..", "..", "backend")
+
+				if _, err := os.Stat(backendPath); err == nil {
+
+					fmt.Println("Backend folder detected. Building and launching updater...")
+
+					updaterPath := filepath.Join(execDir, "..", "..", "updater")
+
+					cmd := exec.Command("go", "build", "-o", filepath.Join(updaterPath, "updater.exe"), updaterPath)
+					cmd.Stdout = os.Stdout
+					cmd.Stderr = os.Stderr
+					if err := cmd.Run(); err != nil {
+						fmt.Fprintf(os.Stderr, "Error building updater: %v\n", err)
+						os.Exit(1)
+					}
+
+					updaterExe := filepath.Join(updaterPath, "updater.exe")
+					cmd = exec.Command(updaterExe)
+					cmd.Dir = updaterPath
+					cmd.Stdout = os.Stdout
+					cmd.Stderr = os.Stderr
+					if err := cmd.Run(); err != nil {
+						fmt.Fprintf(os.Stderr, "Error launching updater: %v\n", err)
+						os.Exit(1)
+					}
+				} else {
+
+					fmt.Println("Backend folder not detected. Launching existing updater...")
+
+					updaterExe := filepath.Join(execDir, "updater.exe")
+					cmd := exec.Command(updaterExe)
+					cmd.Dir = filepath.Dir(updaterExe)
+					cmd.Stdout = os.Stdout
+					cmd.Stderr = os.Stderr
+					if err := cmd.Run(); err != nil {
+						fmt.Fprintf(os.Stderr, "Error launching updater: %v\n", err)
+						os.Exit(1)
+					}
+				}
+
+				os.Exit(0)
+			} else {
+				fmt.Println("Skipping update. Proceeding with the current version.")
+			}
+		} else {
+			fmt.Printf("You are using the latest version: %s\n", current)
+		}
+	}
 
 	// <--- ADJ --->
 
@@ -590,3 +695,23 @@ func getUDPFilter(addrs []net.IP, backendAddr net.IP, port uint16) string {
 
 	return fmt.Sprintf("(%s) and (%s) and (%s or (dst host %s))", udpPort, srcUdpAddrsStr, dstUdpAddrsStr, backendAddr)
 }
+
+type GitHubRelease struct {
+	TagName string `json:"tag_name"`
+}
+
+func getLatestVersionFromGitHub() (string, error) {
+	resp, err := http.Get("https://api.github.com/repos/HyperloopUPV-H8/software/releases/latest")
+	if err != nil {
+		return "", fmt.Errorf("unable to connect to the internet: %w", err)
+	}
+	defer resp.Body.Close()
+
+	var release GitHubRelease
+	if err := json.NewDecoder(resp.Body).Decode(&release); err != nil {
+		return "", fmt.Errorf("error decoding GitHub response: %w", err)
+	}
+
+	version := strings.TrimPrefix(release.TagName, "v")
+	return version, nil
+}

backend/go.mod

@@ -7,9 +7,11 @@ require (
 	github.com/google/gopacket v1.1.19
 	github.com/google/uuid v1.3.0
 	github.com/gorilla/websocket v1.5.0
+	github.com/hashicorp/go-version v1.7.0
 	github.com/jmaralo/sntp v0.0.0-20240116111937-45a0a3419272
 	github.com/pelletier/go-toml/v2 v2.0.7
 	github.com/pin/tftp/v3 v3.0.0
+	github.com/pkg/browser v0.0.0-20240102092130-5ac0b6a4141c
 	github.com/rs/zerolog v1.29.0
 	golang.org/x/exp v0.0.0-20230522175609-2e198f4a06a1
 )
@@ -29,7 +31,6 @@ require (
 	github.com/mattn/go-colorable v0.1.13 // indirect
 	github.com/mattn/go-isatty v0.0.17 // indirect
 	github.com/pjbgf/sha1cd v0.3.0 // indirect
-	github.com/pkg/browser v0.0.0-20240102092130-5ac0b6a4141c // indirect
 	github.com/pkg/errors v0.9.1 // indirect
 	github.com/sergi/go-diff v1.3.2-0.20230802210424-5b0b94c5c0d3 // indirect
 	github.com/skeema/knownhosts v1.2.2 // indirect

backend/go.sum

@@ -46,6 +46,8 @@ github.com/google/uuid v1.3.0 h1:t6JiXgmwXMjEs8VusXIJk2BXHsn+wx8BZdTaoZ5fu7I=
 github.com/google/uuid v1.3.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
 github.com/gorilla/websocket v1.5.0 h1:PPwGk2jz7EePpoHN/+ClbZu8SPxiqlu12wZP/3sWmnc=
 github.com/gorilla/websocket v1.5.0/go.mod h1:YR8l580nyteQvAITg2hZ9XVh4b55+EU/adAjf1fMHhE=
+github.com/hashicorp/go-version v1.7.0 h1:5tqGy27NaOTB8yJKUZELlFAS/LTKJkrmONwQKeRZfjY=
+github.com/hashicorp/go-version v1.7.0/go.mod h1:fltr4n8CU8Ke44wwGCBoEymUuxUHl09ZGVZPK5anwXA=
 github.com/jbenet/go-context v0.0.0-20150711004518-d14ea06fba99 h1:BQSFePA1RWJOlocH6Fxy8MmwDt+yVQYULKfN0RoTN8A=
 github.com/jbenet/go-context v0.0.0-20150711004518-d14ea06fba99/go.mod h1:1lJo3i6rXxKeerYnT8Nvf0QmHCRC1n8sfWVwXF2Frvo=
 github.com/jmaralo/sntp v0.0.0-20240116111937-45a0a3419272 h1:dtQzdBn2P781UxyDPZd1tv4QE29ffpnmJ15qBkXHypY=

go.work

@@ -1,8 +1,9 @@
-go 1.21.3
+go 1.23.1
 
 use (
 	./backend
-	./packet-sender
 	./scripts
 	./state-order-tester
+	./updater
+	./packet-sender
 )

updater/go.mod

@@ -0,0 +1,3 @@
+module updater
+
+go 1.23.1