mirror of
https://github.com/nestriness/cli.git
synced 2025-12-11 11:55:35 +02:00
312 lines
8.2 KiB
Go
312 lines
8.2 KiB
Go
/*
|
|
Copyright © 2024 Nestri <>
|
|
*/
|
|
package cmd
|
|
|
|
import (
|
|
_ "embed"
|
|
"fmt"
|
|
"os"
|
|
"os/exec"
|
|
"path/filepath"
|
|
"runtime"
|
|
"strings"
|
|
"sync"
|
|
"time"
|
|
|
|
"github.com/nestriness/cli/pkg/specs"
|
|
|
|
"github.com/charmbracelet/lipgloss"
|
|
"github.com/charmbracelet/lipgloss/table"
|
|
"github.com/muesli/termenv"
|
|
"github.com/spf13/cobra"
|
|
"github.com/spf13/viper"
|
|
)
|
|
|
|
//go:embed nestri.ascii
|
|
var art string
|
|
|
|
// rootCmd represents the base command when called without any subcommands
|
|
var rootCmd = &cobra.Command{
|
|
Use: "nestri",
|
|
Short: "A CLI tool to manage your cloud gaming service",
|
|
Args: cobra.NoArgs,
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
return cmd.Help()
|
|
},
|
|
}
|
|
|
|
var neoFetchCmd = &cobra.Command{
|
|
Use: "neofetch",
|
|
Short: "Show important system information",
|
|
Args: cobra.NoArgs,
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
lipgloss.SetColorProfile(termenv.TrueColor)
|
|
|
|
// baseStyle := lipgloss.NewStyle().
|
|
// MarginTop(1).
|
|
// MarginRight(4).
|
|
// MarginBottom(1).
|
|
// MarginLeft(4)
|
|
|
|
var (
|
|
b strings.Builder
|
|
lines = strings.Split(art, "\n")
|
|
colors = []string{"#CC3D00", "#CC3D00"}
|
|
step = len(lines) / len(colors)
|
|
)
|
|
|
|
for i, l := range lines {
|
|
n := clamp(0, len(colors)-1, i/step)
|
|
b.WriteString(colorize(colors[n], l))
|
|
b.WriteRune('\n')
|
|
}
|
|
|
|
t := table.New().
|
|
Border(lipgloss.HiddenBorder()).BorderStyle(lipgloss.NewStyle().Width(3))
|
|
//TODO: show this specs
|
|
// info := &specs.Specs{}
|
|
// infoChan := make(chan specs.Specs, 1)
|
|
// var wg sync.WaitGroup
|
|
// wg.Add(1)
|
|
// go getSpecs(info, infoChan, &wg)
|
|
// wg.Wait()
|
|
// newInfo := <-infoChan
|
|
|
|
t.Row(b.String())
|
|
|
|
fmt.Print(t)
|
|
|
|
return nil
|
|
},
|
|
}
|
|
|
|
// 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() {
|
|
err := rootCmd.Execute()
|
|
if err != nil {
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
|
|
func init() {
|
|
// Here you will define your flags and configuration settings.
|
|
// Cobra supports persistent flags, which, if defined here,
|
|
// will be global for your application.
|
|
rootCmd.AddCommand(neoFetchCmd)
|
|
|
|
// rootCmd.PersistentFlags().StringVar(&cfgFile, "config", "", "config file (default is $HOME/.cli.yaml)")
|
|
|
|
// Cobra also supports local flags, which will only run
|
|
// when this action is called directly.
|
|
rootCmd.Flags().BoolP("toggle", "t", false, "Help message for toggle")
|
|
}
|
|
|
|
func colorize(c, s string) string {
|
|
return lipgloss.NewStyle().Foreground(lipgloss.Color(c)).Render(s)
|
|
}
|
|
|
|
func clamp(v, low, high int) int {
|
|
if high < low {
|
|
low, high = high, low
|
|
}
|
|
return min(high, max(low, v))
|
|
}
|
|
|
|
func min(a, b int) int {
|
|
if a < b {
|
|
return a
|
|
}
|
|
return b
|
|
}
|
|
|
|
func max(a, b int) int {
|
|
if a > b {
|
|
return a
|
|
}
|
|
return b
|
|
}
|
|
|
|
func getSpecs(info *specs.Specs, infoChan chan specs.Specs, wg *sync.WaitGroup) {
|
|
defer wg.Done()
|
|
sys := specs.New()
|
|
// info.Userhost = getUserHostname()
|
|
// info.OS = getOSName()
|
|
// info.Kernel = getKernelVersion()
|
|
// info.Uptime = getUptime()
|
|
// info.Shell = getShell()
|
|
// info.CPU = getCPUName()
|
|
// info.RAM = getMemStats()
|
|
info.GPU, _ = sys.GetGPUInfo()
|
|
// info.SystemArch, _ = getSystemArch()
|
|
// info.DiskUsage, _ = getDiskUsage()
|
|
infoChan <- *info
|
|
}
|