Implement Go client template

Author: juho05

commands/new.go

@@ -42,26 +42,26 @@ func newServer() error {
 }
 
 func newClient() error {
-	domain, err := input.Input("Enter the domain of the game server:")
+	url, err := input.Input("Enter the URL of the game server:")
 	if err != nil {
 		return err
 	}
-	if strings.HasPrefix(domain, "http://") {
-		domain = strings.TrimPrefix(domain, "http://")
-	} else if strings.HasPrefix(domain, "https://") {
-		domain = strings.TrimPrefix(domain, "https://")
-	} else if strings.HasPrefix(domain, "ws://") {
-		domain = strings.TrimPrefix(domain, "ws://")
-	} else if strings.HasPrefix(domain, "wss://") {
-		domain = strings.TrimPrefix(domain, "wss://")
-	}
-	domain = strings.TrimSuffix(domain, "/")
-	ssl := isSSL(domain)
-	name, _, err := getCodeGameInfo(baseURL(domain, ssl))
+	if strings.HasPrefix(url, "http://") {
+		url = strings.TrimPrefix(url, "http://")
+	} else if strings.HasPrefix(url, "https://") {
+		url = strings.TrimPrefix(url, "https://")
+	} else if strings.HasPrefix(url, "ws://") {
+		url = strings.TrimPrefix(url, "ws://")
+	} else if strings.HasPrefix(url, "wss://") {
+		url = strings.TrimPrefix(url, "wss://")
+	}
+	url = strings.TrimSuffix(url, "/")
+	ssl := isSSL(url)
+	name, cgVersion, err := getCodeGameInfo(baseURL(url, ssl))
 	if err != nil {
 		return err
 	}
-	cgeVersion, err := getCGEVersion(baseURL(domain, ssl))
+	cgeVersion, err := getCGEVersion(baseURL(url, ssl))
 	if err != nil {
 		return err
 	}
@@ -89,13 +89,26 @@ func newClient() error {
 
 	eventsOutput := projectName
 	if language == "go" {
-		eventsOutput = filepath.Join(projectName, name)
+		eventsOutput = filepath.Join(projectName, strings.ReplaceAll(strings.ReplaceAll(name, "-", ""), "_", ""))
 	}
 
-	err = external.CGGenEvents(eventsOutput, baseURL(domain, ssl), cgeVersion, language)
+	err = external.CGGenEvents(eventsOutput, baseURL(url, ssl), cgeVersion, language)
+	if err != nil {
+		fmt.Fprintln(os.Stderr, "\x1b[31mFailed to generate event definitions:", err, "\x1b[0m")
+	}
+
+	switch language {
+	case "go":
+		err = newClientGo(projectName, url, cgVersion)
+	default:
+		return fmt.Errorf("Unsupported language: %s", language)
+	}
 	if err != nil {
 		return err
 	}
+
+	fmt.Printf("\x1b[32mSuccessfully created project in '%s'.\n\x1b[0m", projectName)
+
 	return nil
 }
 

commands/new_go.go

@@ -0,0 +1,75 @@
+package commands
+
+import (
+	"fmt"
+	"os"
+	"path/filepath"
+	"strings"
+
+	"github.com/code-game-project/codegame-cli/external"
+	"github.com/code-game-project/codegame-cli/input"
+	"github.com/code-game-project/codegame-cli/templates"
+)
+
+func newClientGo(projectName, serverURL, cgVersion string) error {
+	err := createGoTemplate(projectName, serverURL)
+	if err != nil {
+		return err
+	}
+
+	err = installGoLibrary(projectName, cgVersion)
+	if err != nil {
+		return err
+	}
+
+	err = external.ExecuteInDir(projectName, "go", "mod", "tidy")
+	if err != nil {
+		return err
+	}
+
+	return external.ExecuteInDir(projectName, "goimports", "-w", "main.go")
+}
+
+func createGoTemplate(projectName, serverURL string) error {
+	module, err := input.Input("Project module path:")
+	if err != nil {
+		return err
+	}
+
+	out, err := external.ExecuteInDirHidden(projectName, "go", "mod", "init", module)
+	if err != nil {
+		fmt.Println(out)
+		return err
+	}
+
+	return os.WriteFile(filepath.Join(projectName, "main.go"), []byte(strings.ReplaceAll(fmt.Sprintf(templates.Go, serverURL), "$", "%")), 0644)
+}
+
+func installGoLibrary(projectName, cgVersion string) error {
+	clientVersion := external.ClientVersionFromCGVersion("code-game-project", "go-client", cgVersion)
+
+	if clientVersion == "latest" {
+		out, err := external.ExecuteInDirHidden(projectName, "go", "get")
+		if err != nil {
+			fmt.Println(out)
+		}
+		return err
+	}
+
+	majorVersion := strings.Split(clientVersion, ".")[0]
+	tag, err := external.GithubTagFromVersion("code-game-project", "go-client", clientVersion)
+	if err != nil {
+		return err
+	}
+	path := "github.com/code-game-project/go-client/cg"
+	if majorVersion != "0" && majorVersion != "1" {
+		path = fmt.Sprintf("github.com/code-game-project/go-client/v%s/cg", majorVersion)
+	}
+	path += "@" + tag
+
+	out, err := external.ExecuteInDirHidden(projectName, "go", "get", path)
+	if err != nil {
+		fmt.Println(out)
+	}
+	return err
+}

external/cg_gen_events.go

@@ -23,7 +23,7 @@ func InstallCGGenEvents(cgeVersion string) error {
 		return nil
 	}
 
-	version, err := GithubTagFromMinorVersion("code-game-project", "cg-gen-events", cgeVersion)
+	version, err := GithubTagFromVersion("code-game-project", "cg-gen-events", cgeVersion)
 	if err != nil {
 		return err
 	}

external/external.go

@@ -4,11 +4,13 @@ import (
 	"encoding/json"
 	"errors"
 	"fmt"
+	"math"
 	"net/http"
 	"os"
 	"os/exec"
 	"os/user"
 	"runtime"
+	"strconv"
 	"strings"
 )
 
@@ -27,11 +29,24 @@ func Execute(programName string, args ...string) error {
 	return cmd.Run()
 }
 
-func ExecuteHidden(programName string, args ...string) (string, error) {
+func ExecuteInDir(workingDir, programName string, args ...string) error {
 	cmd := exec.Command(programName, args...)
+	cmd.Dir = workingDir
 	cmd.Stdin = os.Stdin
 	cmd.Stdout = os.Stdout
 	cmd.Stderr = os.Stderr
+	return cmd.Run()
+}
+
+func ExecuteHidden(programName string, args ...string) (string, error) {
+	cmd := exec.Command(programName, args...)
+	out, err := cmd.CombinedOutput()
+	return string(out), err
+}
+
+func ExecuteInDirHidden(workingDir, programName string, args ...string) (string, error) {
+	cmd := exec.Command(programName, args...)
+	cmd.Dir = workingDir
 	out, err := cmd.CombinedOutput()
 	return string(out), err
 }
@@ -64,7 +79,7 @@ func OpenBrowser(url string) error {
 	}
 }
 
-func GithubTagFromMinorVersion(owner, repo, version string) (string, error) {
+func GithubTagFromVersion(owner, repo, version string) (string, error) {
 	res, err := http.Get(fmt.Sprintf("https://api.github.com/repos/%s/%s/tags", owner, repo))
 	if err != nil {
 		return "", err
@@ -86,3 +101,90 @@ func GithubTagFromMinorVersion(owner, repo, version string) (string, error) {
 	}
 	return "", ErrTagNotFound
 }
+
+func ClientVersionFromCGVersion(owner, repo, cgVersion string) string {
+	res, err := http.Get(fmt.Sprintf("https://raw.githubusercontent.com/%s/%s/main/versions.json", owner, repo))
+	if err != nil {
+		fmt.Fprintf(os.Stderr, "\x1b[33mWARNING: Couldn't fetch versions.json. Using latest client library version.\n\x1b[0m")
+		return "latest"
+	}
+	defer res.Body.Close()
+
+	var versions map[string]string
+	err = json.NewDecoder(res.Body).Decode(&versions)
+	if err != nil {
+		fmt.Fprintf(os.Stderr, "\x1b[33mWARNING: Invalid versions.json. Using latest client library version.\n\x1b[0m")
+		return "latest"
+	}
+
+	// check exact match
+	if v, ok := versions[cgVersion]; ok {
+		return v
+	}
+
+	parts := strings.Split(cgVersion, ".")
+	if len(parts) < 2 {
+		fmt.Fprintf(os.Stderr, "\x1b[33mWARNING: Invalid versions.json. Using latest client library version.\n\x1b[0m")
+		return "latest"
+	}
+	major := parts[0]
+
+	// get all minor versions of the requested major version
+	compatibleMinorVersions := make([]int, 0)
+	for v := range versions {
+		clientParts := strings.Split(v, ".")
+		if len(clientParts) < 2 {
+			fmt.Fprintf(os.Stderr, "\x1b[33mWARNING: Invalid versions.json. Using latest client library version.\n\x1b[0m")
+			return "latest"
+		}
+		clientMajor := clientParts[0]
+		if major == clientMajor {
+			minor, err := strconv.Atoi(clientParts[1])
+			if err != nil {
+				fmt.Fprintf(os.Stderr, "\x1b[33mWARNING: Invalid versions.json. Using latest client library version.\n\x1b[0m")
+				return "latest"
+			}
+			compatibleMinorVersions = append(compatibleMinorVersions, minor)
+		}
+	}
+	if len(compatibleMinorVersions) == 0 {
+		fmt.Fprintf(os.Stderr, "\x1b[33mWARNING: No compatible client library version found. Using latest client library version.\n\x1b[0m")
+		return "latest"
+	}
+
+	minorStr := parts[1]
+	minor, err := strconv.Atoi(minorStr)
+	if err != nil {
+		fmt.Fprintf(os.Stderr, "\x1b[33mWARNING: Invalid versions.json. Using latest client library version.\n\x1b[0m")
+		return "latest"
+	}
+
+	// check closest minor version above requested
+	closestMinor := -1
+	for _, v := range compatibleMinorVersions {
+		if v > minor && float64(closestMinor-minor) > float64(v-minor) {
+			closestMinor = v
+		}
+	}
+	if closestMinor >= 0 {
+		v := fmt.Sprintf("%s.%d", major, minor)
+		fmt.Fprintf(os.Stderr, "\x1b[33mWARNING: No exact version match found. Using client library version %s.\n\x1b[0m", v)
+		return v
+	}
+
+	// check closest minor version below requested
+	closestMinor = math.MaxInt
+	for _, v := range compatibleMinorVersions {
+		if v < minor && float64(minor-closestMinor) > float64(minor-v) {
+			closestMinor = v
+		}
+	}
+	if closestMinor >= 0 {
+		v := fmt.Sprintf("%s.%d", major, minor)
+		fmt.Fprintf(os.Stderr, "\x1b[33mWARNING: No exact version match found. Using client library version %s.\n\x1b[0m", v)
+		return v
+	}
+
+	fmt.Fprintf(os.Stderr, "\x1b[33mWARNING: No compatible client library version found. Using latest client library version.\n\x1b[0m")
+	return "latest"
+}

main.go

@@ -44,7 +44,7 @@ func main() {
 		os.Exit(1)
 	}
 	if err != nil {
-		fmt.Fprintln(os.Stderr, err)
+		fmt.Fprint(os.Stderr, "\x1b[31m", err, "\n\x1b[0m")
 		os.Exit(1)
 	}
 }

templates/go.go

@@ -0,0 +1,10 @@
+package templates
+
+const Go = `package main
+
+func main() {
+	socket, err := cg.NewSocket("%s")
+	if err != nil {
+		log.Fatalf("failed to connect to server: $s", err)
+	}
+}`