message-pusher/client/main.go

package main

import (
	"flag"
	"github.com/gorilla/websocket"
	"log"
	"net/url"
	"os"
)

var (
	conn  = flag.String("url", "", "connection url")
	token = flag.String("token", "", "the access token")
)

type Verification struct {
	Prefix string `json:"prefix"`
	Token  string `json:"token"`
}

func main() {
	flag.Parse()
	connString := *conn
	if connString == "" {
		connString = os.Getenv("MESSAGE_PUSHER_URL")
	}
	if *token == "" {
		*token = os.Getenv("MESSAGE_PUSHER_TOKEN")
	}
	connUrl, err := url.Parse(connString)
	if err != nil {
		log.Fatal("Failed to parse connection url", err)
		return
	}
	scheme := "ws"
	if connUrl.Scheme == "https" {
		scheme = "wss"
	}
	u := url.URL{Scheme: scheme, Host: connUrl.Host, Path: "/"}
	for {
		log.Printf("Connecting to %s...\n", u.String())
		c, _, err := websocket.DefaultDialer.Dial(u.String(), nil)
		if err != nil {
			log.Fatal("Failed to connect to server:", err)
			return
		}
		log.Printf("Server connected.\n")
		verification := &Verification{
			Prefix: connUrl.Path[1:],
			Token:  *token,
		}
		err = c.WriteJSON(verification)
		if err != nil {
			log.Fatal(err.Error())
		}
		for {
			var message = new(Message)
			err = c.ReadJSON(message)
			if err != nil {
				log.Println("Error occurred when read message:", err)
				log.Println("Connection lost, retrying...")
				break
			} else {
				log.Println("New message arrived.")
				Notify(message)
			}
		}
	}
}