/* Copyright © 2023 Seednode */ package cmd import ( "errors" "fmt" "io" "net/http" "strings" "time" "github.com/yosssi/gohtml" ) var ( ErrInvalidFileCountRange = errors.New("maximum file count limit must be greater than or equal to minimum file count limit") ErrInvalidFileCountValue = errors.New("file count limits must be positive integers no greater than 2147483647") ErrInvalidPort = errors.New("listen port must be an integer between 1 and 65535 inclusive") ErrNoMediaFound = errors.New("no supported media formats found which match all criteria") ) func newErrorPage(title, body string) string { var htmlBody strings.Builder htmlBody.WriteString(``) htmlBody.WriteString(faviconHtml) htmlBody.WriteString(``) htmlBody.WriteString(fmt.Sprintf("%s", title)) htmlBody.WriteString(fmt.Sprintf("%s", body)) return htmlBody.String() } func notFound(w http.ResponseWriter, r *http.Request, path string) error { startTime := time.Now() if Verbose { fmt.Printf("%s | ERROR: Unavailable file %s requested by %s\n", startTime.Format(logDate), path, r.RemoteAddr, ) } w.WriteHeader(http.StatusNotFound) w.Header().Add("Content-Type", "text/html") _, err := io.WriteString(w, gohtml.Format(newErrorPage("Not Found", "404 Page not found"))) if err != nil { return err } return nil } func serverError(w http.ResponseWriter, r *http.Request, i interface{}) { startTime := time.Now() if Verbose { fmt.Printf("%s | ERROR: Invalid request for %s from %s\n", startTime.Format(logDate), r.URL.Path, r.RemoteAddr, ) } w.Header().Add("Content-Type", "text/html") io.WriteString(w, gohtml.Format(newErrorPage("Server Error", "An error has occurred. Please try again."))) } func serverErrorHandler() func(http.ResponseWriter, *http.Request, interface{}) { return serverError }