Server status

cmd/main.go

@@ -12,23 +12,30 @@ func main() {
 	token := ""
 	botChannel := ""
 	serverStatusMessageID := ""
+	username := ""
+	password := ""
+	serverName := ""
 
 	for _, ele := range arg {
-		if strings.Index(ele, "-t ") == 0 {
-			token = ele[3:]
-		} else if strings.Index(ele, "-c ") == 0 {
-			botChannel = ele[3:]
+		if strings.Index(ele, "--token ") == 0 {
+			token = ele[8:]
+		} else if strings.Index(ele, "--channel ") == 0 {
+			botChannel = ele[10:]
 		} else if strings.Index(ele, "--serverStatusMessageId ") == 0 {
 			serverStatusMessageID = ele[24:]
+		} else if strings.Index(ele, "--username ") == 0 {
+			username = ele[11:]
+		} else if strings.Index(ele, "--password ") == 0 {
+			password = ele[11:]
+		} else if strings.Index(ele, "--serverName ") == 0 {
+			serverName = ele[13:]
 		}
 	}
-	if token == "" {
-		log.Fatal("Missing \"-t \" parameter")
-	} else if botChannel == "" {
-		log.Fatal("Missing \"-c \" parameter")
+	if token == "" || botChannel == "" || serverStatusMessageID == "" || username == "" || password == "" || serverName == "" {
+		log.Fatal("Missing parameter")
 	}
 
-	err := m.RunBot(token, botChannel, serverStatusMessageID)
+	err := m.RunBot(token, botChannel, serverStatusMessageID, username, password, serverName)
 	if err != nil {
 		log.Fatal(err)
 	}

dcskellerdiscord.go

@@ -1,43 +1,130 @@
 package dcskellerdiscordgo
 
 import (
+	"encoding/json"
+	"errors"
+	"io/ioutil"
+	"net/http"
 	"strconv"
+	"strings"
 	"time"
 
 	"github.com/bwmarrin/discordgo"
 )
 
+type dcsServer struct {
+	ID          string `json:"ID"`
+	NAME        string `json:"NAME"`
+	IPADDRESS   string `json:"IP_ADDRESS"`
+	PORT        string `json:"PORT"`
+	MISSIONNAME string `json:"MISSION_NAME"`
+	MISSIONTIME string `json:"MISSION_TIME"`
+	PLAYERS     string `json:"PLAYERS"`
+	PLAYERSMAX  string `json:"PLAYERS_MAX"`
+	PASSWORD    string `json:"PASSWORD"`
+	URLTODETAIL string `json:"URL_TO_DETAIL"`
+}
+
+type dcsServerList struct {
+	SERVERSMAXCOUNT int         `json:"SERVERS_MAX_COUNT"`
+	SERVERSMAXDATE  string      `json:"SERVERS_MAX_DATE"`
+	PLAYERSCOUNT    int         `json:"PLAYERS_COUNT"`
+	MYSERVERS       []dcsServer `json:"MY_SERVERS"`
+	SERVERS         []struct {
+		NAME                 string `json:"NAME"`
+		IPADDRESS            string `json:"IP_ADDRESS"`
+		PORT                 string `json:"PORT"`
+		MISSIONNAME          string `json:"MISSION_NAME"`
+		MISSIONTIME          string `json:"MISSION_TIME"`
+		PLAYERS              string `json:"PLAYERS"`
+		PLAYERSMAX           string `json:"PLAYERS_MAX"`
+		PASSWORD             string `json:"PASSWORD"`
+		DESCRIPTION          string `json:"DESCRIPTION"`
+		UALIAS0              string `json:"UALIAS_0"`
+		MISSIONTIMEFORMATTED string `json:"MISSION_TIME_FORMATTED"`
+	} `json:"SERVERS"`
+}
+
+func getServerStatus(username string, password string, serverName string) (dcsServer, error) {
+	client := &http.Client{}
+
+	url := "https://www.digitalcombatsimulator.com/en/personal/server/?ajax=y&_=" + strconv.FormatInt(time.Now().UTC().Unix(), 10)
+	req, err := http.NewRequest("GET", url, nil)
+	if err != nil {
+		return dcsServer{}, err
+	}
+
+	req.SetBasicAuth(username, password)
+	resp, err := client.Do(req)
+	if err != nil {
+		return dcsServer{}, err
+	}
+
+	bodyBytes, err := ioutil.ReadAll(resp.Body)
+	body := string(bodyBytes)
+	jsonStart := strings.Index(body, "{")
+	serverList := body[jsonStart:]
+	serverStatus := &dcsServerList{}
+	err = json.Unmarshal([]byte(serverList), serverStatus)
+	if err != nil || body == "" {
+		return dcsServer{}, err
+	}
+
+	for _, server := range serverStatus.MYSERVERS {
+		if server.NAME == serverName {
+			return server, nil
+		}
+	}
+	return dcsServer{}, errors.New("Server not found")
+}
+
 // RunBot starts the dcs kellergeschwader discord bot
-func RunBot(token string, botChannel string, serverStatusMessageID string) error {
+func RunBot(token string, botChannel string, serverStatusMessageID string, username string, password string, serverName string) error {
 	session, err := discordgo.New("Bot " + token)
 	if err != nil {
 		return err
 	}
 
 	colorOnline := 3388721   //33b531
 	colorOffline := 11878449 //b54031
+	serverStatus, err := getServerStatus(username, password, serverName)
 	serverOnline := true
-	playerOnline := 0
+
+	if err != nil {
+		if err.Error() == "Server not found" {
+			serverOnline = false
+		} else {
+			return err
+		}
+	}
 
 	embedMessage := discordgo.MessageEmbed{}
 	embedMessage.Title = "Server Status"
 	embedMessage.Thumbnail = &discordgo.MessageEmbedThumbnail{
 		URL: "https://upload.wikimedia.org/wikipedia/commons/thumb/f/ff/F16_drawing.svg/320px-F16_drawing.svg.png",
 	}
-	embedMessage.Timestamp = time.Now().Format("2006-01-02T15:04:05-0700")
-	embedMessage.Footer = &discordgo.MessageEmbedFooter{
-		Text: "Last update",
-	}
 
 	if serverOnline == true {
-		embedMessage.Description += "Online\n"
+		playersOnline, err := strconv.Atoi(serverStatus.PLAYERS)
+		if err != nil {
+			return err
+		}
+		playersOnline--
+
 		embedMessage.Color = colorOnline
+		embedMessage.Description += "**Online**\n"
+		embedMessage.Description += "IP address: **" + serverStatus.IPADDRESS + ":" + serverStatus.PORT + "**\n"
+		embedMessage.Description += "Mission: **" + serverStatus.MISSIONNAME + "**\n"
+		embedMessage.Description += "Players online: **" + strconv.Itoa(playersOnline) + "**"
 	} else {
-		embedMessage.Description += "Offline\n"
 		embedMessage.Color = colorOffline
+		embedMessage.Description += "**Offline**\n"
 	}
 
-	embedMessage.Description += strconv.Itoa(playerOnline) + " players online"
+	embedMessage.Timestamp = time.Now().UTC().Format("2006-01-02T15:04:05-0700")
+	embedMessage.Footer = &discordgo.MessageEmbedFooter{
+		Text: "Last update",
+	}
 
 	session.ChannelMessageEditEmbed(botChannel, serverStatusMessageID, &embedMessage)
 	return nil

go.mod

@@ -2,4 +2,6 @@ module dcskellerdiscordgo
 
 go 1.16
 
-require github.com/bwmarrin/discordgo v0.23.2
+require (
+	github.com/bwmarrin/discordgo v0.23.2
+)

go.sum

@@ -4,3 +4,19 @@ github.com/gorilla/websocket v1.4.0 h1:WDFjx/TMzVgy9VdMMQi2K2Emtwi2QcUQsztZ/zLaH
 github.com/gorilla/websocket v1.4.0/go.mod h1:E7qHFY5m1UJ88s3WnNqhKjPHQ0heANvMoAMk2YaljkQ=
 golang.org/x/crypto v0.0.0-20181030102418-4d3f4d9ffa16 h1:y6ce7gCWtnH+m3dCjzQ1PCuwl28DDIc3VNnvY29DlIA=
 golang.org/x/crypto v0.0.0-20181030102418-4d3f4d9ffa16/go.mod h1:6SG95UA2DQfeDnfUPMdvaQW0Q7yPrPDi9nlGo2tz2b4=
+golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
+golang.org/x/crypto v0.0.0-20200622213623-75b288015ac9/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto=
+golang.org/x/net v0.0.0-20190404232315-eb5bcb51f2a3/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg=
+golang.org/x/net v0.0.0-20200904194848-62affa334b73/go.mod h1:/O7V0waA8r7cgGh81Ro3o1hOxt32SMVPicZroKQ2sZA=
+golang.org/x/net v0.0.0-20210226172049-e18ecbb05110 h1:qWPm9rbaAMKs8Bq/9LRpbMqxWRVUAQwMI9fVrssnTfw=
+golang.org/x/net v0.0.0-20210226172049-e18ecbb05110/go.mod h1:m0MpNAwzfU5UDzcl9v0D8zg8gWTRqZa9RBIspLL5mdg=
+golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
+golang.org/x/sys v0.0.0-20190412213103-97732733099d/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
+golang.org/x/sys v0.0.0-20200323222414-85ca7c5b95cd/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
+golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
+golang.org/x/sys v0.0.0-20210301091718-77cc2087c03b h1:kHlr0tATeLRMEiZJu5CknOw/E8V6h69sXXQFGoPtjcc=
+golang.org/x/sys v0.0.0-20210301091718-77cc2087c03b/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
+golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo=
+golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
+golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
+golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=