github.com/Azareal/Gosora@v0.0.0-20210729070923-553e66b59003/common/errors.go

package common

import (
	"fmt"
	"log"
	"net/http"
	"runtime/debug"
	"strings"
	"sync"
	"sync/atomic"

	p "github.com/Azareal/Gosora/common/phrases"
)

type ErrorItem struct {
	error
	Stack []byte
}

// ! The errorBuffer uses o(n) memory, we should probably do something about that
// TODO: Use the errorBuffer variable to construct the system log in the Control Panel. Should we log errors caused by users too? Or just collect statistics on those or do nothing? Intercept recover()? Could we intercept the logger instead here? We might get too much information, if we intercept the logger, maybe make it part of the Debug page?
// ? - Should we pass Header / HeaderLite rather than forcing the errors to pull the global Header instance?
var errorBufferMutex sync.RWMutex
//var errorBuffer []ErrorItem
var ErrorCountSinceStartup int64

//var notfoundCountPerSecond int
//var nopermsCountPerSecond int

// A blank list to fill out that parameter in Page for routes which don't use it
var tList []interface{}

// WIP, a new system to propagate errors up from routes
type RouteError interface {
	Type() string
	Error() string
	Cause() string
	JSON() bool
	Handled() bool

	Wrap(string)
}

type RouteErrorImpl struct {
	userText string
	sysText  string
	system   bool
	json     bool
	handled  bool
}

func (err *RouteErrorImpl) Type() string {
	// System errors may contain sensitive information we don't want the user to see
	if err.system {
		return "system"
	}
	return "user"
}

func (err *RouteErrorImpl) Error() string {
	return err.userText
}

func (err *RouteErrorImpl) Cause() string {
	if err.sysText == "" {
		return err.Error()
	}
	return err.sysText
}

// Respond with JSON?
func (err *RouteErrorImpl) JSON() bool {
	return err.json
}

// Has this error been dealt with elsewhere?
func (err *RouteErrorImpl) Handled() bool {
	return err.handled
}

// Move the current error into the system error slot and add a new one to the user error slot to show the user
func (err *RouteErrorImpl) Wrap(userErr string) {
	err.sysText = err.userText
	err.userText = userErr
}

func HandledRouteError() RouteError {
	return &RouteErrorImpl{"", "", false, false, true}
}

func Error(errmsg string) RouteError {
	return &RouteErrorImpl{errmsg, "", false, false, false}
}

func FromError(err error) RouteError {
	return &RouteErrorImpl{err.Error(), "", false, false, false}
}

func ErrorJSQ(errmsg string, js bool) RouteError {
	return &RouteErrorImpl{errmsg, "", false, js, false}
}

func SysError(errmsg string) RouteError {
	return &RouteErrorImpl{errmsg, errmsg, true, false, false}
}

// LogError logs internal handler errors which can't be handled with InternalError() as a wrapper for log.Fatal(), we might do more with it in the future.
// TODO: Clean-up extra as a way of passing additional context
func LogError(err error, extra ...string) {
	LogWarning(err, extra...)
	ErrLogger.Fatal("")
}

func LogWarning(err error, extra ...string) {
	var esb strings.Builder
	for _, extraBit := range extra {
		esb.WriteString(extraBit)
		esb.WriteRune(10)
	}
	if err == nil {
		esb.WriteString("nil error found")
	} else {
		esb.WriteString(err.Error())
	}
	esb.WriteRune(10)
	errmsg := esb.String()

	errorBufferMutex.Lock()
	defer errorBufferMutex.Unlock()
	stack := debug.Stack() // debug.Stack() can't be executed concurrently, so we'll guard this with a mutex too
	Err(errmsg, string(stack))
	//errorBuffer = append(errorBuffer, ErrorItem{err, stack})
	atomic.AddInt64(&ErrorCountSinceStartup,1)
}

func errorHeader(w http.ResponseWriter, u *User, title string) *Header {
	h := DefaultHeader(w, u)
	h.Title = title
	h.Zone = "error"
	return h
}

// TODO: Dump the request?
// InternalError is the main function for handling internal errors, while simultaneously printing out a page for the end-user to let them know that *something* has gone wrong
// ? - Add a user parameter?
// ! Do not call CustomError here or we might get an error loop
func InternalError(err error, w http.ResponseWriter, r *http.Request) RouteError {
	pi := ErrorPage{errorHeader(w, &GuestUser, p.GetErrorPhrase("internal_error_title")), p.GetErrorPhrase("internal_error_body")}
	handleErrorTemplate(w, r, pi, 500)
	LogError(err)
	return HandledRouteError()
}

// InternalErrorJSQ is the JSON "maybe" version of InternalError which can handle both JSON and normal requests
// ? - Add a user parameter?
func InternalErrorJSQ(err error, w http.ResponseWriter, r *http.Request, js bool) RouteError {
	if !js {
		return InternalError(err, w, r)
	}
	return InternalErrorJS(err, w, r)
}

// InternalErrorJS is the JSON version of InternalError on routes we know will only be requested via JSON. E.g. An API.
// ? - Add a user parameter?
func InternalErrorJS(err error, w http.ResponseWriter, r *http.Request) RouteError {
	w.WriteHeader(500)
	writeJsonError(p.GetErrorPhrase("internal_error_body"), w)
	LogError(err)
	return HandledRouteError()
}

// When the task system detects if the database is down, some database errors might slip by this
func DatabaseError(w http.ResponseWriter, r *http.Request) RouteError {
	pi := ErrorPage{errorHeader(w, &GuestUser, p.GetErrorPhrase("internal_error_title")), p.GetErrorPhrase("internal_error_body")}
	handleErrorTemplate(w, r, pi, 500)
	return HandledRouteError()
}

func InternalErrorXML(err error, w http.ResponseWriter, r *http.Request) RouteError {
	w.Header().Set("Content-Type", "application/xml")
	w.WriteHeader(500)
	w.Write([]byte(`<?xml version="1.0"encoding="UTF-8"?>
<error>` + p.GetErrorPhrase("internal_error_body") + `</error>`))
	LogError(err)
	return HandledRouteError()
}

// TODO: Stop killing the instance upon hitting an error with InternalError* and deprecate this
func SilentInternalErrorXML(err error, w http.ResponseWriter, r *http.Request) RouteError {
	w.Header().Set("Content-Type", "application/xml")
	w.WriteHeader(500)
	w.Write([]byte(`<?xml version="1.0"encoding="UTF-8"?>
<error>` + p.GetErrorPhrase("internal_error_body") + `</error>`))
	log.Print("InternalError: ", err)
	return HandledRouteError()
}

// ! Do not call CustomError here otherwise we might get an error loop
func PreError(errmsg string, w http.ResponseWriter, r *http.Request) RouteError {
	pi := ErrorPage{errorHeader(w, &GuestUser, p.GetErrorPhrase("error_title")), errmsg}
	handleErrorTemplate(w, r, pi, 500)
	return HandledRouteError()
}

func PreErrorJS(errmsg string, w http.ResponseWriter, r *http.Request) RouteError {
	w.WriteHeader(500)
	writeJsonError(errmsg, w)
	return HandledRouteError()
}

func PreErrorJSQ(errmsg string, w http.ResponseWriter, r *http.Request, js bool) RouteError {
	if !js {
		return PreError(errmsg, w, r)
	}
	return PreErrorJS(errmsg, w, r)
}

// LocalError is an error shown to the end-user when something goes wrong and it's not the software's fault
// TODO: Pass header in for this and similar errors instead of having to pass in both user and w? Would also allow for more stateful things, although this could be a problem
/*func LocalError(errmsg string, w http.ResponseWriter, r *http.Request, user *User) RouteError {
	w.WriteHeader(500)
	pi := ErrorPage{errorHeader(w, user, p.GetErrorPhrase("local_error_title")), errmsg}
	handleErrorTemplate(w, r, pi)
	return HandledRouteError()
}*/

func LocalError(errmsg string, w http.ResponseWriter, r *http.Request, u *User) RouteError {
	return SimpleError(errmsg, w, r, errorHeader(w, u, ""))
}

func LocalErrorf(errmsg string, w http.ResponseWriter, r *http.Request, u *User, params ...interface{}) RouteError {
	return LocalError(fmt.Sprintf(errmsg, params), w, r, u)
}

func SimpleError(errmsg string, w http.ResponseWriter, r *http.Request, h *Header) RouteError {
	if h == nil {
		h = errorHeader(w, &GuestUser, p.GetErrorPhrase("local_error_title"))
	} else {
		h.Title = p.GetErrorPhrase("local_error_title")
	}
	pi := ErrorPage{h, errmsg}
	handleErrorTemplate(w, r, pi, 500)
	return HandledRouteError()
}

func LocalErrorJSQ(errmsg string, w http.ResponseWriter, r *http.Request, u *User, js bool) RouteError {
	if !js {
		return SimpleError(errmsg, w, r, errorHeader(w, u, ""))
	}
	return LocalErrorJS(errmsg, w, r)
}

func LocalErrorJS(errmsg string, w http.ResponseWriter, r *http.Request) RouteError {
	w.WriteHeader(500)
	writeJsonError(errmsg, w)
	return HandledRouteError()
}

// TODO: We might want to centralise the error logic in the future and just return what the error handler needs to construct the response rather than handling it here
// NoPermissions is an error shown to the end-user when they try to access an area which they aren't authorised to access
func NoPermissions(w http.ResponseWriter, r *http.Request, u *User) RouteError {
	pi := ErrorPage{errorHeader(w, u, p.GetErrorPhrase("no_permissions_title")), p.GetErrorPhrase("no_permissions_body")}
	handleErrorTemplate(w, r, pi, 403)
	return HandledRouteError()
}

func NoPermissionsJSQ(w http.ResponseWriter, r *http.Request, u *User, js bool) RouteError {
	if !js {
		return NoPermissions(w, r, u)
	}
	return NoPermissionsJS(w, r, u)
}

func NoPermissionsJS(w http.ResponseWriter, r *http.Request, u *User) RouteError {
	w.WriteHeader(403)
	writeJsonError(p.GetErrorPhrase("no_permissions_body"), w)
	return HandledRouteError()
}

// ? - Is this actually used? Should it be used? A ban in Gosora should be more of a permission revocation to stop them posting rather than something which spits up an error page, right?
func Banned(w http.ResponseWriter, r *http.Request, u *User) RouteError {
	pi := ErrorPage{errorHeader(w, u, p.GetErrorPhrase("banned_title")), p.GetErrorPhrase("banned_body")}
	handleErrorTemplate(w, r, pi, 403)
	return HandledRouteError()
}

// nolint
// BannedJSQ is the version of the banned error page which handles both JavaScript requests and normal page loads
func BannedJSQ(w http.ResponseWriter, r *http.Request, user *User, js bool) RouteError {
	if !js {
		return Banned(w, r, user)
	}
	return BannedJS(w, r, user)
}

func BannedJS(w http.ResponseWriter, r *http.Request, u *User) RouteError {
	w.WriteHeader(403)
	writeJsonError(p.GetErrorPhrase("banned_body"), w)
	return HandledRouteError()
}

// nolint
func LoginRequiredJSQ(w http.ResponseWriter, r *http.Request, u *User, js bool) RouteError {
	if !js {
		return LoginRequired(w, r, u)
	}
	return LoginRequiredJS(w, r, u)
}

// ? - Where is this used? Should we use it more?
// LoginRequired is an error shown to the end-user when they try to access an area which requires them to login
func LoginRequired(w http.ResponseWriter, r *http.Request, u *User) RouteError {
	return CustomError(p.GetErrorPhrase("login_required_body"), 401, p.GetErrorPhrase("no_permissions_title"), w, r, nil, u)
}

// nolint
func LoginRequiredJS(w http.ResponseWriter, r *http.Request, u *User) RouteError {
	w.WriteHeader(401)
	writeJsonError(p.GetErrorPhrase("login_required_body"), w)
	return HandledRouteError()
}

// SecurityError is used whenever a session mismatch is found
// ? - Should we add JS and JSQ versions of this?
func SecurityError(w http.ResponseWriter, r *http.Request, u *User) RouteError {
	pi := ErrorPage{errorHeader(w, u, p.GetErrorPhrase("security_error_title")), p.GetErrorPhrase("security_error_body")}
	w.Header().Set("Content-Type", "text/html;charset=utf-8")
	w.WriteHeader(403)
	e := RenderTemplateAlias("error", "security_error", w, r, pi.Header, pi)
	if e != nil {
		LogError(e)
	}
	return HandledRouteError()
}

var microNotFoundBytes = []byte("file not found")
func MicroNotFound(w http.ResponseWriter, r *http.Request) RouteError {
	w.Header().Set("Content-Type", "text/html;charset=utf-8")
	w.WriteHeader(404)
	_, _ = w.Write(microNotFoundBytes)
	return HandledRouteError()
}

// NotFound is used when the requested page doesn't exist
// ? - Add a JSQ version of this?
// ? - Add a user parameter?
func NotFound(w http.ResponseWriter, r *http.Request, h *Header) RouteError {
	return CustomError(p.GetErrorPhrase("not_found_body"), 404, p.GetErrorPhrase("not_found_title"), w, r, h, &GuestUser)
}

// ? - Add a user parameter?
func NotFoundJS(w http.ResponseWriter, r *http.Request) RouteError {
	w.WriteHeader(404)
	writeJsonError(p.GetErrorPhrase("not_found_body"), w)
	return HandledRouteError()
}

func NotFoundJSQ(w http.ResponseWriter, r *http.Request, h *Header, js bool) RouteError {
	if js {
		return NotFoundJS(w, r)
	}
	if h == nil {
		h = DefaultHeader(w, &GuestUser)
	}
	return NotFound(w, r, h)
}

// CustomError lets us make custom error types which aren't covered by the generic functions above
func CustomError(errmsg string, errcode int, errtitle string, w http.ResponseWriter, r *http.Request, h *Header, u *User) (rerr RouteError) {
	if h == nil {
		h, rerr = UserCheck(w, r, u)
		if rerr != nil {
			h = errorHeader(w, u, errtitle)
		}
	}
	h.Title = errtitle
	h.Zone = "error"
	pi := ErrorPage{h, errmsg}
	handleErrorTemplate(w, r, pi, errcode)
	return HandledRouteError()
}

// CustomErrorJSQ is a version of CustomError which lets us handle both JSON and regular pages depending on how it's being accessed
func CustomErrorJSQ(errmsg string, errcode int, errtitle string, w http.ResponseWriter, r *http.Request, h *Header, u *User, js bool) RouteError {
	if !js {
		return CustomError(errmsg, errcode, errtitle, w, r, h, u)
	}
	return CustomErrorJS(errmsg, errcode, w, r, u)
}

// CustomErrorJS is the pure JSON version of CustomError
func CustomErrorJS(errmsg string, errcode int, w http.ResponseWriter, r *http.Request, u *User) RouteError {
	w.WriteHeader(errcode)
	writeJsonError(errmsg, w)
	return HandledRouteError()
}

// TODO: Should we optimise this by caching these json strings?
func writeJsonError(errmsg string, w http.ResponseWriter) {
	_, _ = w.Write([]byte(`{"errmsg":"` + strings.Replace(errmsg, "\"", "", -1) + `"}`))
}

func handleErrorTemplate(w http.ResponseWriter, r *http.Request, pi ErrorPage, errcode int) {
	w.Header().Set("Content-Type", "text/html;charset=utf-8")
	w.WriteHeader(errcode)
	err := RenderTemplateAlias("error", "error", w, r, pi.Header, pi)
	if err != nil {
		LogError(err)
	}
}

// Alias of routes.renderTemplate
var RenderTemplateAlias func(tmplName, hookName string, w http.ResponseWriter, r *http.Request, h *Header, pi interface{}) error