63 lines
1.4 KiB
Go
63 lines
1.4 KiB
Go
package api
|
|
|
|
import (
|
|
"log"
|
|
"net/http"
|
|
"net/http/cookiejar"
|
|
|
|
"github.com/google/uuid"
|
|
"github.com/gorilla/mux"
|
|
"golang.org/x/net/publicsuffix"
|
|
)
|
|
|
|
type ApiApplication interface {
|
|
NewRouter() *mux.Router
|
|
Visit(w http.ResponseWriter, r *http.Request)
|
|
}
|
|
|
|
func NewApiApp() (ApiApplication, error) {
|
|
cookiejar, err := cookiejar.New(&cookiejar.Options{PublicSuffixList: publicsuffix.List})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &apiapp{
|
|
cj: cookiejar,
|
|
}, nil
|
|
}
|
|
|
|
type apiapp struct {
|
|
router mux.Router
|
|
cj *cookiejar.Jar
|
|
}
|
|
|
|
func (a *apiapp) NewRouter() *mux.Router {
|
|
router := mux.NewRouter()
|
|
|
|
router.HandleFunc("/visit", a.Visit)
|
|
|
|
return router
|
|
}
|
|
|
|
// Called when someone visits any page of the website
|
|
// Calls for the spyware cookie and sets it if it doesn't yet exist
|
|
func (a *apiapp) Visit(w http.ResponseWriter, r *http.Request) {
|
|
log.Printf("Visit called\n")
|
|
|
|
c, err := r.Cookie("spyware")
|
|
if err != nil && err != http.ErrNoCookie {
|
|
log.Printf("Error: %s\n", err.Error())
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
|
} else if err == http.ErrNoCookie {
|
|
// Create cookie and send it
|
|
log.Printf("No cookie sent by client, sending cookie to them!\n")
|
|
c = &http.Cookie{Name: "spyware", Value: uuid.NewString(), MaxAge: 0}
|
|
http.SetCookie(w, c)
|
|
}
|
|
|
|
w.WriteHeader(200)
|
|
w.Write([]byte("yeet"))
|
|
|
|
log.Printf("Someone visited: %s\n", c.Value)
|
|
|
|
}
|