Merge pull request #21 from Corentin-cott/CobraCommandCLI

Cobra command cli

Author: Corentin-cott

cmd/daemon/main.go

@@ -3,17 +3,101 @@ package main
 import (
 	"fmt"
 	"log"
+	"os"
+	"strconv"
 	"time"
 
 	"github.com/Corentin-cott/ServeurSentinel/config"
 	"github.com/Corentin-cott/ServeurSentinel/internal/console"
 	"github.com/Corentin-cott/ServeurSentinel/internal/db"
 	periodic "github.com/Corentin-cott/ServeurSentinel/internal/events"
+	"github.com/Corentin-cott/ServeurSentinel/internal/tmux"
 	"github.com/Corentin-cott/ServeurSentinel/internal/triggers"
+	"github.com/spf13/cobra"
 )
 
 func main() {
-	fmt.Println(time.Now().Location())
+	var rootCmd = &cobra.Command{
+		Use:   "serversentinel",
+		Short: "ServerSentinel manages Minecraft and Palworld servers in tmux sessions.",
+	}
+
+	// Command: serversentinel start-server [id]
+	var startServerCmd = &cobra.Command{
+		Use:   "start-server [id]",
+		Short: "Starts a game server by its ID",
+		Args:  cobra.ExactArgs(1),
+		Run: func(cmd *cobra.Command, args []string) {
+			serverID := args[0]
+			fmt.Printf("Starting server with ID: %s\n", serverID)
+			commandStartStopServerWithID(serverID, "start")
+		},
+	}
+
+	// Command: serversentinel stop-server [id]
+	var stopServerCmd = &cobra.Command{
+		Use:   "stop-server [id]",
+		Short: "Stops a game server by its ID",
+		Args:  cobra.ExactArgs(1),
+		Run: func(cmd *cobra.Command, args []string) {
+			serverID := args[0]
+			fmt.Printf("Stopping server with ID: %s\n", serverID)
+			commandStartStopServerWithID(serverID, "stop")
+		},
+	}
+
+	// Command: serversentinel check-server
+	var checkServerCmd = &cobra.Command{
+		Use:   "check-server",
+		Short: "Check and start servers that are supposed to be running",
+		Args:  cobra.NoArgs,
+		Run: func(cmd *cobra.Command, args []string) {
+			// Because it's a CLI command, we need to load the configuration file
+			err := config.LoadConfig("/opt/serversentinel/config.json")
+			if err != nil {
+				log.Fatalf("FATAL ERROR LOADING CONFIG JSON FILE: %v", err)
+				return
+			}
+
+			// Initialize the connection to the database
+			err = db.ConnectToDatabase()
+			if err != nil {
+				log.Fatalf("FATAL ERROR TESTING DATABASE CONNECTION: %v", err)
+				return
+			}
+
+			// Check if the right tmux servers are running
+			fmt.Printf("Checking and starting servers that are supposed to be running\n")
+			message, err := tmux.CheckRunningServers()
+			if err != nil {
+				log.Fatalf("FATAL ERROR CHECKING RUNNING SERVERS: %v", err)
+				return
+			}
+			fmt.Println(message)
+		},
+	}
+
+	// Command: serversentinel daemon
+	var daemonCmd = &cobra.Command{
+		Use:   "daemon",
+		Short: "Runs ServerSentinel in daemon mode",
+		Run:   runDaemon,
+	}
+
+	// Add commands to root
+	rootCmd.AddCommand(daemonCmd)
+	rootCmd.AddCommand(startServerCmd)
+	rootCmd.AddCommand(stopServerCmd)
+	rootCmd.AddCommand(checkServerCmd)
+
+	// Execute CLI
+	if err := rootCmd.Execute(); err != nil {
+		fmt.Println(err)
+		os.Exit(1)
+	}
+}
+
+func runDaemon(cmd *cobra.Command, args []string) {
 	fmt.Println("Starting the Server Sentinel daemon (" + time.Now().Format("02/01/2006 15:04:05") + ") ...")
 
 	// Load the configuration file
@@ -64,3 +148,74 @@ func main() {
 
 	fmt.Println("♦ Server Sentinel daemon stopped.")
 }
+
+// Function to start a server by its ID. This function is use in the CLI command "start-server"
+func commandStartStopServerWithID(serverID string, action string) {
+	// Action can only be "start" or "stop"
+	if action != "start" && action != "stop" {
+		log.Fatalf("FATAL ERROR: INVALID ACTION: %s", action)
+		return
+	}
+
+	// Check if the server ID is a number, and if so, convert it to an integer
+	serverIDInt, err := strconv.Atoi(serverID)
+	if err != nil {
+		log.Fatalf("FATAL ERROR: SERVER ID IS NOT A NUMBER: %v", err)
+		return
+	}
+
+	// Because it's a CLI command, we need to load the configuration file
+	err = config.LoadConfig("/opt/serversentinel/config.json")
+	if err != nil {
+		log.Fatalf("FATAL ERROR LOADING CONFIG JSON FILE: %v", err)
+		return
+	}
+
+	// Initialize the connection to the database
+	err = db.ConnectToDatabase()
+	if err != nil {
+		log.Fatalf("FATAL ERROR TESTING DATABASE CONNECTION: %v", err)
+	}
+
+	// Now we check if the server exists in the database
+	server, err := db.GetServerById(serverIDInt)
+	if err != nil {
+		log.Fatalf("FATAL ERROR GETTING SERVER BY ID: %v", err)
+		return
+	}
+
+	// If the action is "stop", we stop the server
+	if action == "stop" {
+		// Stop the server
+		err = tmux.StopServerTmux(server.Nom)
+		if err != nil {
+			log.Fatalf("FATAL ERROR STOPPING SERVER: %v", err)
+			return
+		}
+		return
+	}
+
+	// If the action is "start", we first check if the server is primary or not
+	if server.ID == 1 {
+		err = db.SetPrimaryServerId(server.ID)
+		if err != nil {
+			log.Fatalf("FATAL ERROR SETTING PRIMARY SERVER ID: %v", err)
+			return
+		}
+	} else {
+		err = db.SetSecondaryServerId(server.ID)
+		if err != nil {
+			log.Fatalf("FATAL ERROR SETTING SECONDARY SERVER ID: %v", err)
+			return
+		}
+	}
+
+	// Start the server
+	message, err := tmux.CheckRunningServers()
+	if err != nil {
+		log.Fatalf("FATAL ERROR CHECKING RUNNING SERVERS: %v", err)
+		return
+	}
+
+	fmt.Println(message)
+}

go.mod

@@ -4,4 +4,9 @@ go 1.22.2
 
 require github.com/go-sql-driver/mysql v1.8.1
 
-require filippo.io/edwards25519 v1.1.0 // indirect
+require (
+	filippo.io/edwards25519 v1.1.0 // indirect
+	github.com/inconshreveable/mousetrap v1.1.0 // indirect
+	github.com/spf13/cobra v1.9.1 // direct
+	github.com/spf13/pflag v1.0.6 // indirect
+)

go.sum

@@ -1,4 +1,14 @@
 filippo.io/edwards25519 v1.1.0 h1:FNf4tywRC1HmFuKW5xopWpigGjJKiJSV0Cqo0cJWDaA=
 filippo.io/edwards25519 v1.1.0/go.mod h1:BxyFTGdWcka3PhytdK4V28tE5sGfRvvvRV7EaN4VDT4=
+github.com/cpuguy83/go-md2man/v2 v2.0.6/go.mod h1:oOW0eioCTA6cOiMLiUPZOpcVxMig6NIQQ7OS05n1F4g=
 github.com/go-sql-driver/mysql v1.8.1 h1:LedoTUt/eveggdHS9qUFC1EFSa8bU2+1pZjSRpvNJ1Y=
 github.com/go-sql-driver/mysql v1.8.1/go.mod h1:wEBSXgmK//2ZFJyE+qWnIsVGmvmEKlqwuVSjsCm7DZg=
+github.com/inconshreveable/mousetrap v1.1.0 h1:wN+x4NVGpMsO7ErUn/mUI3vEoE6Jt13X2s0bqwp9tc8=
+github.com/inconshreveable/mousetrap v1.1.0/go.mod h1:vpF70FUmC8bwa3OWnCshd2FqLfsEA9PFc4w1p2J65bw=
+github.com/russross/blackfriday/v2 v2.1.0/go.mod h1:+Rmxgy9KzJVeS9/2gXHxylqXiyQDYRxCVz55jmeOWTM=
+github.com/spf13/cobra v1.9.1 h1:CXSaggrXdbHK9CF+8ywj8Amf7PBRmPCOJugH954Nnlo=
+github.com/spf13/cobra v1.9.1/go.mod h1:nDyEzZ8ogv936Cinf6g1RU9MRY64Ir93oCnqb9wxYW0=
+github.com/spf13/pflag v1.0.6 h1:jFzHGLGAlb3ruxLB8MhbI6A8+AQX/2eW4qeyNZXNp2o=
+github.com/spf13/pflag v1.0.6/go.mod h1:McXfInJRrz4CZXVZOBLb0bTZqETkiAhM9Iw0y3An2Bg=
+gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
+gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=

internal/db/db_controller.go

@@ -130,6 +130,28 @@ func GetSecondaryServerId() int {
 	return serverID
 }
 
+// Setter to set the primary server
+func SetPrimaryServerId(serverID int) error {
+	query := "UPDATE serveurs_parameters SET id_serv_primaire = ?"
+	_, err := db.Exec(query, serverID)
+	if err != nil {
+		return fmt.Errorf("FAILED TO SET PRIMARY SERVER: %v", err)
+	}
+
+	return nil
+}
+
+// Setter to set the secondary server
+func SetSecondaryServerId(serverID int) error {
+	query := "UPDATE serveurs_parameters SET id_serv_secondaire = ?"
+	_, err := db.Exec(query, serverID)
+	if err != nil {
+		return fmt.Errorf("FAILED TO SET SECONDARY SERVER: %v", err)
+	}
+
+	return nil
+}
+
 // Getter to get all the server informations
 func GetServerById(serverID int) (models.Server, error) {
 	query := "SELECT * FROM serveurs WHERE id = ?"