98 lines
2.3 KiB
Go
98 lines
2.3 KiB
Go
package main
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"os"
|
|
|
|
serveractions "manage-servers/server-actions"
|
|
"manage-servers/webserver"
|
|
)
|
|
|
|
func main() {
|
|
servers, err := loadServers("./config/servers.json")
|
|
if err != nil {
|
|
fmt.Printf("Error loading servers: %v\n", err)
|
|
return
|
|
}
|
|
|
|
if len(os.Args) < 2 {
|
|
printUsage()
|
|
return
|
|
}
|
|
|
|
command := os.Args[1]
|
|
switch command {
|
|
case "list":
|
|
listServers(servers)
|
|
case "wake":
|
|
if len(os.Args) < 3 {
|
|
fmt.Println("Please specify a server name to wake.")
|
|
printUsage()
|
|
return
|
|
}
|
|
serverName := os.Args[2]
|
|
serveractions.WakeServer(serverName, servers)
|
|
case "wakeall":
|
|
serveractions.WakeAllServers(servers)
|
|
case "status":
|
|
serveractions.CheckServersStatus(servers)
|
|
case "shutdown":
|
|
if len(os.Args) < 3 {
|
|
fmt.Println("Please specify a server name to shutdown.")
|
|
printUsage()
|
|
return
|
|
}
|
|
serverName := os.Args[2]
|
|
serveractions.ShutdownServer(serverName, servers)
|
|
case "reboot":
|
|
if len(os.Args) < 3 {
|
|
fmt.Println("Please specify a server name to reboot.")
|
|
printUsage()
|
|
return
|
|
}
|
|
serverName := os.Args[2]
|
|
serveractions.RebootServer(serverName, servers)
|
|
case "serve":
|
|
webserver.StartWebServer(servers)
|
|
default:
|
|
fmt.Printf("Unknown command: %s\n", command)
|
|
printUsage()
|
|
}
|
|
}
|
|
|
|
func printUsage() {
|
|
fmt.Println("Usage: go run . <command>")
|
|
fmt.Println("Commands:")
|
|
fmt.Println(" list - List all configured servers")
|
|
fmt.Println(" wake <server_name> - Wake a specific server")
|
|
fmt.Println(" wakeall - Wake all configured servers")
|
|
fmt.Println(" status - Check the status of all servers")
|
|
fmt.Println(" shutdown <server_name> - Shutdown a specific server")
|
|
fmt.Println(" reboot <server_name> - Reboot a specific server")
|
|
fmt.Println(" serve - Start a web server to manage servers")
|
|
}
|
|
|
|
func loadServers(filename string) ([]serveractions.Server, error) {
|
|
var servers []serveractions.Server
|
|
|
|
file, err := os.ReadFile(filename)
|
|
if err != nil {
|
|
fmt.Println("No servers configuration found, please create a servers.json file in the config folder.")
|
|
} else {
|
|
|
|
err = json.Unmarshal(file, &servers)
|
|
if err != nil {
|
|
return nil, err
|
|
}}
|
|
|
|
return servers, nil
|
|
}
|
|
|
|
func listServers(servers []serveractions.Server) {
|
|
fmt.Println("Configured Servers:")
|
|
for _, s := range servers {
|
|
fmt.Printf(" - %s (%s) - %s\n", s.Name, s.IP, s.Mac)
|
|
}
|
|
}
|