70 lines
1.6 KiB
Go
70 lines
1.6 KiB
Go
package main
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"os"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/by-cx/lobby/server"
|
|
"github.com/fatih/color"
|
|
)
|
|
|
|
func printDiscovery(discovery server.Discovery) {
|
|
color.Yellow("Hostname:\n %s\n", discovery.Hostname)
|
|
|
|
if len(discovery.Labels) > 0 {
|
|
fmt.Printf("Labels:\n")
|
|
for _, label := range discovery.Labels {
|
|
fmt.Printf(" %s\n", label)
|
|
}
|
|
}
|
|
}
|
|
|
|
func colorLabel(label server.Label) string {
|
|
parts := strings.Split(label.String(), ":")
|
|
if len(parts) == 1 {
|
|
return color.GreenString(parts[0])
|
|
}
|
|
|
|
return color.GreenString(parts[0]) + ":" + color.MagentaString((strings.Join(parts[1:], ":")))
|
|
}
|
|
|
|
func printDiscoveries(discoveries []server.Discovery) {
|
|
maxHostnameWidth := 0
|
|
for _, discovery := range discoveries {
|
|
if len(discovery.Hostname) > maxHostnameWidth {
|
|
maxHostnameWidth = len(discovery.Hostname)
|
|
}
|
|
}
|
|
|
|
for _, discovery := range discoveries {
|
|
if len(discovery.Labels) == 0 {
|
|
// fmt.Println(discovery.Hostname)
|
|
color.Yellow(discovery.Hostname)
|
|
} else {
|
|
hostname := fmt.Sprintf("%"+strconv.Itoa(maxHostnameWidth)+"s", discovery.Hostname)
|
|
|
|
fmt.Printf("%s %s\n", color.YellowString(hostname), colorLabel(discovery.Labels[0]))
|
|
|
|
if len(discovery.Labels) > 1 {
|
|
for _, label := range discovery.Labels[1:] {
|
|
fmt.Printf("%"+strconv.Itoa(maxHostnameWidth+4)+"s%s\n", " ", colorLabel(label))
|
|
}
|
|
}
|
|
}
|
|
fmt.Println()
|
|
}
|
|
}
|
|
|
|
func printJSON(data interface{}) {
|
|
body, err := json.Marshal(data)
|
|
if err != nil {
|
|
fmt.Println("error occurred while formating the output into JSON:", err.Error())
|
|
os.Exit(3)
|
|
}
|
|
|
|
fmt.Println(string(body))
|
|
}
|