api.go

package main

import (
	"log"
	"net/http"
)

type api struct {
	p *proxy
}

func newApi(p *proxy) *api {
	return &api{p: p}
}

// allowAction register a regex in whitelist
func (api *api) allowAction(w http.ResponseWriter, r *http.Request) {
	rgx := r.PostFormValue("regex")
	if rgx == "" {
		w.WriteHeader(http.StatusBadRequest)
		return
	}
	err := api.p.addToEndpointList(rgx, true)
	if err == nil {
		w.WriteHeader(http.StatusOK)
	} else {
		w.WriteHeader(http.StatusBadRequest)
		w.Write([]byte(err.Error()))
	}
}

// forbid action register a regex in blacklist
func (api *api) forbidAction(w http.ResponseWriter, r *http.Request) {
	rgx := r.PostFormValue("regex")
	if rgx == "" {
		w.WriteHeader(http.StatusBadRequest)
		return
	}
	err := api.p.addToEndpointList(rgx, false)
	if err == nil {
		w.WriteHeader(http.StatusOK)
	} else {
		w.WriteHeader(http.StatusBadRequest)
		w.Write([]byte(err.Error()))
	}
}

// run set handlers and launch api
func (api *api) run(port string) {
	http.HandleFunc("/allow", api.allowAction)
	http.HandleFunc("/forbid", api.forbidAction)
	log.Fatal(http.ListenAndServe(":"+port, nil))
}