go-apns-server.go

// Copyright (c) 2016, CleverTap
// All rights reserved.
// Licensed under the New 3-Clause BSD License
//
// A mock server for the HTTP2 Apple push gateway

package main

import (
	"flag"
	"fmt"
	"io/ioutil"
	"log"
	"math/rand"
	"net/http"
	"strconv"
)

func main() {
	serverPort := flag.Int("port", 8443, "The port for the mock server to listen to")
	serverCert := flag.String("cert", "cert.pem", "The path to the certificate")
	serverKey := flag.String("key", "key.pem", "The path to the certificate key")

	flag.Parse()

	http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
		_, err := ioutil.ReadAll(r.Body)
		if err != nil {
			log.Println("Error:", err)
		}
		n := rand.Int31n(10)

		statusCode := 200

		var body string

		if n == 7 {
			statusCode = 400
			body = "{\"reason\": \"BadDeviceToken\"}"
		} else if n > 7 {
			statusCode = 410
			body = "{\"reason\": \"Unregistered\"}"
		}

		w.WriteHeader(statusCode)

		if statusCode != 200 {
			w.Header().Set("Content-Type", "application/json")
			fmt.Fprintf(w, body)
		}
	})

	log.Printf("Using certificate %#v with key %#v", *serverCert, *serverKey)
	log.Println("Starting server on port", *serverPort)
	log.Println("Press Ctrl+C to stop...")

	port := ":" + strconv.Itoa(*serverPort)
	log.Fatal(http.ListenAndServeTLS(port, *serverCert, *serverKey, nil))
}