lobby/ctl/main.go

package main

import (
	"flag"
	"fmt"
	"os"
	"strings"

	"github.com/rosti-cz/server_lobby/client"
	"github.com/rosti-cz/server_lobby/server"
)

func Usage() {
	flag.Usage()
	fmt.Println("")
	fmt.Println("Commands:")
	fmt.Println("  discovery                      returns discovery packet of the server where the client is connected to")
	fmt.Println("  discoveries                    returns list of all registered discovery packets")
	fmt.Println("  labels add LABEL [LABEL] ...   adds new runtime labels")
	fmt.Println("  labels del LABEL [LABEL] ...   deletes runtime labels")
}

func main() {
	config := GetConfig()

	// Setup flags
	proto := flag.String("proto", "", "Select HTTP or HTTPS protocol")
	host := flag.String("host", "", "Hostname or IP address of lobby daemon")
	port := flag.Uint("port", 0, "Port of lobby daemon")
	token := flag.String("token", "", "Token needed to communicate lobby daemon, if empty auth is disabled")

	flag.Parse()

	// Replace empty values from flags by values from environment variables
	if *proto == "" {
		proto = &config.Proto
	}
	if *host == "" {
		host = &config.Host
	}
	if *port == 0 {
		port = &config.Port
	}
	if *token == "" {
		token = &config.Token
	}

	// Validation
	if *proto != "http" && *proto != "https" {
		fmt.Println("Protocol can be only http or https")
	}

	// Setup lobby client library
	client := client.LobbyClient{
		Proto: strings.ToLower(*proto),
		Host:  *host,
		Port:  *port,
		Token: *token,
	}

	// Process rest of the arguments
	if len(flag.Args()) == 0 {
		Usage()
		os.Exit(0)
	}

	switch flag.Args()[0] {
	case "discoveries":
		discoveries, err := client.GetDiscoveries()
		if err != nil {
			fmt.Println(err)
		}
		printDiscoveries(discoveries)
	case "discovery":
		discovery, err := client.GetDiscovery()
		if err != nil {
			fmt.Println(err)
		}
		printDiscovery(discovery)
	case "labels":
		if len(flag.Args()) < 3 {
			fmt.Println("ERROR: not enough arguments for labels command")
			fmt.Println("")
			Usage()
			os.Exit(0)
		}

		labels := server.Labels{}
		labelsString := flag.Args()[2:]
		for _, labelString := range labelsString {
			labels = append(labels, server.Label(labelString))
		}

		if flag.Args()[1] == "add" {
			err := client.AddLabels(labels)
			if err != nil {
				fmt.Printf("ERROR: %v\n", err)
				os.Exit(2)
			}
		} else if flag.Args()[1] == "del" {
			err := client.DeleteLabels(labels)
			if err != nil {
				fmt.Printf("ERROR: %v\n", err)
				os.Exit(2)
			}
		} else {
			fmt.Printf("ERROR: wrong labels subcommand\n\n")
			Usage()
			os.Exit(2)
		}

	default:
		Usage()
		os.Exit(0)
	}

}