errors.go

package main

import (
	"fmt"
	"log"

	"github.com/5HT2B/heartbeat/templates"
	"github.com/ferluci/fast-realip"
	"github.com/valyala/fasthttp"
)

func errorPageHandler(ctx *fasthttp.RequestCtx, code int, message string, plaintext bool) {
	ctx.SetStatusCode(code)
	log.Printf("- Returned %v to %s - tried to connect with %s to %s",
		code, realip.FromRequest(ctx), ctx.Method(), ctx.Path())

	if plaintext {
		ctx.Response.Header.Set(fasthttp.HeaderContentType, "text/plain; charset=utf-8")
		_, _ = fmt.Fprintf(ctx, "%s\n", message)
	} else {
		p := &templates.ErrorPage{
			Message: message,
			Path:    ctx.Path(),
			Method:  ctx.Method(),
		}
		templates.WritePageTemplate(ctx, p)
	}
}

func ErrorBadRequest(ctx *fasthttp.RequestCtx, plaintext bool) {
	errorPageHandler(ctx, fasthttp.StatusBadRequest, "400 Bad Request", plaintext)
}

func ErrorForbidden(ctx *fasthttp.RequestCtx, plaintext bool) {
	errorPageHandler(ctx, fasthttp.StatusForbidden, "403 Forbidden", plaintext)
}

func ErrorNotFound(ctx *fasthttp.RequestCtx, plaintext bool) {
	errorPageHandler(ctx, fasthttp.StatusNotFound, "404 Not Found", plaintext)
}

func HandleInternalErr(ctx *fasthttp.RequestCtx, message string, err error) {
	errorPageHandler(ctx, fasthttp.StatusInternalServerError, "500 "+message+": "+err.Error(), true)
}

func HandleClientErr(ctx *fasthttp.RequestCtx, message string, err error) {
	errorPageHandler(ctx, fasthttp.StatusBadRequest, "400 "+message+": "+err.Error(), true)
}

func HandleSuccess(ctx *fasthttp.RequestCtx) {
	errorPageHandler(ctx, fasthttp.StatusOK, "200 OK", true)
}