example_ipfilter_test.go

package middleware_test

import (
	"fmt"
	"github.com/seb-ehm/middleware"
	"log"
	"net/http"
)

func ExampleIPFilter() {
	mux := http.NewServeMux()
	//Some handler that prints both to stdout and the http response
	handler := func(w http.ResponseWriter, r *http.Request) {
		fmt.Fprintln(w, "Hi!")
	}

	allowSomeIP := middleware.IPFilter([]string{"123.240.189.1/32"}, "")
	noLocalhost := middleware.New().Append(allowSomeIP)

	allowLocalhost := middleware.IPFilter([]string{"localhost"}, "")
	localhost := middleware.New().Append(allowLocalhost)

	allowLocalhostInHeader := middleware.IPFilter([]string{"localhost"}, "X-FORWARDED-FOR")
	localhostInHeader := middleware.New().Append(allowLocalhostInHeader)

	mux.Handle("/nolocalhost", noLocalhost.ApplyToFunc(handler))
	mux.Handle("/localhost", localhost.ApplyToFunc(handler))
	mux.Handle("/localhostinheader", localhostInHeader.ApplyToFunc(handler))

	go func() { log.Fatal(http.ListenAndServe("localhost:9192", mux)) }()

	_, status := GetWebsite("http://localhost:9192/nolocalhost")
	fmt.Println(status)
	content, _ := GetWebsite("http://localhost:9192/localhost")
	fmt.Println(content)
	content, _ = GetWebsiteWithHeader("http://localhost:9192/localhostinheader", "X-FORWARDED-FOR", "127.0.0.1")
	fmt.Println(content)
	// Output:
	// 403 Forbidden
	// Hi!
	//
	// Hi!
}