Merge branch 'servermon' of dan/cwtch into master

app/bots/servermon/main.go

@@ -0,0 +1,84 @@
+package main
+
+import (
+	"cwtch.im/cwtch/peer"
+	"cwtch.im/cwtch/peer/connections"
+	"errors"
+	"fmt"
+	"git.openprivacy.ca/openprivacy/libricochet-go/connectivity"
+	"io/ioutil"
+	"log"
+	"os"
+	"time"
+)
+
+func waitForPeerServerConnection(peer peer.CwtchPeer, server string) error {
+	for {
+		servers := peer.GetServers()
+		state, ok := servers[server]
+		if ok {
+			if state == connections.FAILED {
+				return errors.New("Connection to server " + server + " failed!")
+			}
+			if state != connections.AUTHENTICATED {
+				fmt.Printf("peer %v waiting to authenticate with server %v, current state: %v\n", peer.GetProfile().Onion, server, connections.ConnectionStateName[state])
+				time.Sleep(time.Second * 10)
+				continue
+			}
+		} else {
+			return errors.New("peer server connections should have entry for server but do not")
+		}
+		break
+	}
+	return nil
+}
+
+func main() {
+	log.SetOutput(ioutil.Discard)
+	if len(os.Args) != 2 {
+		fmt.Printf("Usage: ./servermon SERVER_ADDRESS\n")
+		os.Exit(1)
+	}
+
+	serverAddr := os.Args[1]
+
+	acn, err := connectivity.StartTor(".", "")
+	if err != nil {
+		fmt.Printf("Could not start tor: %v\n", err)
+		os.Exit(1)
+	}
+
+	botPeer := peer.NewCwtchPeer("servermon")
+	botPeer.Init(acn)
+
+	fmt.Printf("Connecting to %v...\n", serverAddr)
+	botPeer.JoinServer(serverAddr)
+	groupID, _, err := botPeer.StartGroup(serverAddr)
+	if err != nil {
+		fmt.Printf("Error creating group on server %v: %v\n", serverAddr, err)
+		os.Exit(1)
+	}
+
+	err = waitForPeerServerConnection(botPeer, serverAddr)
+	if err != nil {
+		fmt.Printf("Could not connect to server %v: %v\n", serverAddr, err)
+		os.Exit(1)
+	}
+
+	timeout := 1 * time.Second
+	timeElapsed := 0 * time.Second
+	for {
+		err := botPeer.SendMessageToGroup(groupID, timeout.String())
+		if err != nil {
+			fmt.Printf("Sent to group on server %v failed at interval %v of total %v with: %v\n", serverAddr, timeout, timeElapsed, err)
+			os.Exit(1)
+		} else {
+			fmt.Printf("Successfully sent message to %v at interval %v of total %v\n", serverAddr, timeout, timeElapsed)
+		}
+		time.Sleep(timeout)
+		timeElapsed += timeout
+		if timeout < 2*time.Minute {
+			timeout = timeout * 2
+		}
+	}
+}