From 84b916973a2c3945e12134e3848be590e06a2aed Mon Sep 17 00:00:00 2001 From: Wanjohi Ryan <71614375+wanjohiryan@users.noreply.github.com> Date: Wed, 26 Jun 2024 23:19:06 +0300 Subject: [PATCH] feat: Add `viper` --- cmd/root.go | 171 ++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 171 insertions(+) diff --git a/cmd/root.go b/cmd/root.go index 653ef68..6b7a4f8 100644 --- a/cmd/root.go +++ b/cmd/root.go @@ -7,8 +7,12 @@ import ( _ "embed" "fmt" "os" + "os/exec" + "path/filepath" + "runtime" "strings" "sync" + "time" "github.com/nestriness/cli/pkg/specs" @@ -16,6 +20,7 @@ import ( "github.com/charmbracelet/lipgloss/table" "github.com/muesli/termenv" "github.com/spf13/cobra" + "github.com/spf13/viper" ) //go:embed nestri.ascii @@ -76,6 +81,172 @@ var neoFetchCmd = &cobra.Command{ }, } +// this is the "nestri run" subcommand, takes no arguments for now +var runCmd = &cobra.Command{ + Use: "run", + Short: "Run a game using nestri", + Args: cobra.MaximumNArgs(1), + RunE: func(cmd *cobra.Command, args []string) error { + if runtime.GOOS != "linux" { + //make sure os is linux + fmt.Println("This command is only supported on Linux.") + return nil + } + + var game string + if len(args) > 0 { + game = args[0] + viper.Set("game", game) + viper.WriteConfig() + } else { + game = viper.GetString("game") + if filepath.Ext(game) != ".exe" { + return fmt.Errorf("Make sure the game is a .exe") + } + if game == "" { + return fmt.Errorf("no game specified and no previous game selected") + } + } + + fmt.Printf("Running game: %s\n\n", game) + // if gpu > 0 { + // fmt.Print("Using gpu %s\n", gpu) + // } + // if hdr { + // fmt.Println("Enabling HDR mode") + // } + + //get linux version + versionCmd := exec.Command("grep", "VERSION", "/etc/os-release") + versionOutput, err := versionCmd.CombinedOutput() + if err != nil { + return fmt.Errorf("error getting linux version:") + } + fmt.Printf("Linux version:\n%s\n", string(versionOutput)) + + //Step 1: change to games dir + fmt.Println("changing to game dir.") //this is a temp command for debug as well as leads to a hardcoded dir + + HomeDir, err := os.UserHomeDir() + if err != nil { + return fmt.Errorf("error getting home directory %v\n", err) + } + + err = os.Chdir(fmt.Sprintf("%s/game", HomeDir)) + if err != nil { + return fmt.Errorf("error changing directory: %v\n", err) + } + //verify we are in game dir + dir, err := os.Getwd() + if err != nil { + return fmt.Errorf("error getting current directory: %v\n", err) + } + fmt.Printf("Current directory: %s\n\n", dir) + + //list games dir + listDir := exec.Command("ls", "-la", ".") + listDirOutput, err := listDir.CombinedOutput() + if err != nil { + fmt.Errorf("error listing games: %v\n") + } + fmt.Printf("List of Games: \n%s\n", listDirOutput) + + //step 2: Generate a Session ID + //generate id + SID := exec.Command("bash", "-c", "head /dev/urandom | LC_ALL=C tr -dc 'a-zA-Z0-9' | head -c 16") + + //save output to variable + output, err := SID.Output() + if err != nil { + fmt.Errorf("Error generating Session ID: %v\n", err) + } + sessionID := strings.TrimSpace(string(output)) + fmt.Printf("Your Session ID is: %s\n\n", sessionID) + + //step 3: Launch netris server + fmt.Println("Installing Netris/Launching Netris Server\n") + checkRunning := exec.Command("sudo", "docker", "ps", "-q", "-f", "name=netris") + containerId, err := checkRunning.Output() + if err != nil { + return fmt.Errorf("error checking running Docker container: %v", err) + } + + if len(containerId) == 0 { + checkExisting := exec.Command("sudo", "docker", "ps", "-aq", "-f", "name=netris") + containerId, err = checkExisting.Output() + if err != nil { + return fmt.Errorf("error checking for existing docker container: %v", err) + } + + if len(containerId) == 0 { + installCmd := exec.Command( + "sudo", "docker", "run", "-d", "--gpus", "all", "--device=/dev/dri", + "--name", "netris", "-it", "--entrypoint", "/bin/bash", + "-e", fmt.Sprintf("SESSION_ID=%s", sessionID), + "-v", fmt.Sprintf("%s:/game", dir), "-p", "8080:8080/udp", + "--cap-add=SYS_NICE", "--cap-add=SYS_ADMIN", "ghcr.io/netrisdotme/netris/server:nightly", + ) + installCmd.Stdout = os.Stdout + installCmd.Stderr = os.Stderr + + if err := installCmd.Run(); err != nil { + return fmt.Errorf("error running docker command: %v", err) + } + } else { + startContainer := exec.Command("sudo", "docker", "start", "netris") + startContainer.Stdout = os.Stdout + startContainer.Stderr = os.Stderr + + if err := startContainer.Run(); err != nil { + return fmt.Errorf("error starting existing Docker container: %v", err) + } + } + } + + //main part of step 4: + //start netris server + + fmt.Println("starting netris server\n\n") + checkFileCmd := exec.Command("sudo", "docker", "exec", "netris", "ls", "-la", "/tmp") + output, err = checkFileCmd.Output() + if err != nil { + return fmt.Errorf("error checking /tmp dir in docker container: %v\n", err) + } + + if !strings.Contains(string(output), ".X11-unix") { + startupCmd := exec.Command("sudo", "docker", "exec", "netris", "/etc/startup.sh", ">", "/dev/null", "&") + startupCmd.Stdout = os.Stdout + startupCmd.Stderr = os.Stderr + + if err := startupCmd.Run(); err != nil { + return fmt.Errorf("error running startup command: %v\n", err) + } + + for { + time.Sleep(7 * time.Minute) + output, err := checkFileCmd.Output() + if err != nil { + return fmt.Errorf("error checking /tmp directory in container: %v\n", err) + } + if strings.Contains(string(output), ".X11-unix") { + break + } + } + } + + gameCmd := fmt.Sprintf("netris-proton -pr %s", game) + execCmd := exec.Command("sudo", "docker", "exec", "netris", gameCmd) + execCmd.Stdout = os.Stdout + execCmd.Stderr = os.Stderr + + if err := execCmd.Run(); err != nil { + return fmt.Errorf("error executing game command in docker container: %v\n", err) + } + + return nil + }, +} + // Execute adds all child commands to the root command and sets flags appropriately. // This is called by main.main(). It only needs to happen once to the rootCmd. func Execute() {