roulette/cmd/web.go

253 lines
4.9 KiB
Go
Raw Normal View History

2022-09-08 15:12:06 +00:00
/*
Copyright © 2022 Seednode <seednode@seedno.de>
*/
package cmd
import (
"fmt"
2022-09-08 15:12:06 +00:00
"io"
"net/http"
2022-09-08 20:30:51 +00:00
"net/url"
2022-09-08 15:57:59 +00:00
"os"
"path/filepath"
2022-09-08 15:12:06 +00:00
"strconv"
"strings"
"time"
2022-09-08 15:12:06 +00:00
)
type appHandler func(http.ResponseWriter, *http.Request) error
func (fn appHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
if err := fn(w, r); err != nil {
http.Error(w, err.Error(), 500)
}
}
const LOGDATE string = "2006-01-02T15:04:05.000-07:00"
const PREFIX string = "/src"
func stripQueryParam(inUrl string) (string, error) {
url, err := url.Parse(inUrl)
if err != nil {
return "", err
}
url.RawQuery = ""
return url.String(), nil
}
func refererToUri(referer string) string {
parts := strings.SplitAfterN(referer, "/", 4)
if len(parts) < 4 {
return ""
}
return "/" + parts[3]
}
func serveHtml(w http.ResponseWriter, r *http.Request, filePath string) error {
fileName := filepath.Base(filePath)
w.Header().Add("Content-Type", "text/html")
htmlBody := `<html lang="en">
2022-09-08 15:12:06 +00:00
<head>
2022-09-08 20:30:51 +00:00
<style>img{max-width:100%;max-height:97vh;height:auto;}</style>
<title>`
htmlBody += fileName
htmlBody += `</title>
2022-09-08 15:12:06 +00:00
</head>
<body>`
htmlBody += fmt.Sprintf(`<a href="/?f=%v&s=%v"><img src="`, r.URL.Query().Get("f"), r.URL.Query().Get("s"))
htmlBody += PREFIX + filePath
htmlBody += `"></img></a>
2022-09-09 00:11:07 +00:00
</body>
</html>`
_, err := io.WriteString(w, htmlBody)
if err != nil {
return err
}
2022-09-08 15:12:06 +00:00
return nil
2022-09-08 15:12:06 +00:00
}
func serveStaticFile(w http.ResponseWriter, r *http.Request, paths []string) error {
strippedUrl, err := stripQueryParam(r.URL.Path)
2022-09-08 15:12:06 +00:00
if err != nil {
return err
2022-09-08 15:12:06 +00:00
}
prefixedFilePath, err := url.QueryUnescape(strippedUrl)
if err != nil {
return err
}
filePath := strings.TrimPrefix(prefixedFilePath, PREFIX)
if !pathIsValid(filePath, paths) {
http.NotFound(w, r)
}
2022-09-08 15:12:06 +00:00
exists, err := fileExists(filePath)
if err != nil {
return err
}
if !exists {
http.NotFound(w, r)
return nil
}
var startTime time.Time
if Verbose {
startTime = time.Now()
2022-09-12 01:05:55 +00:00
fmt.Printf("%v Serving file: %v", startTime.Format(LOGDATE), filePath)
}
buf, err := os.ReadFile(filePath)
2022-09-08 15:12:06 +00:00
if err != nil {
return err
2022-09-08 15:12:06 +00:00
}
w.Write(buf)
if Verbose {
fmt.Printf(" (Finished in %v)\n", time.Since(startTime).Round(time.Microsecond))
}
return nil
2022-09-08 15:12:06 +00:00
}
func serveStaticFileHandler(paths []string) appHandler {
return func(w http.ResponseWriter, r *http.Request) error {
err := serveStaticFile(w, r, paths)
if err != nil {
return err
}
return nil
}
}
func serveHtmlHandler(paths []string) appHandler {
return func(w http.ResponseWriter, r *http.Request) error {
refererUri, err := stripQueryParam(refererToUri(r.Referer()))
if err != nil {
return err
}
filter := r.URL.Query().Get("f")
successive := r.URL.Query().Get("s")
switch {
case r.URL.Path == "/" && successive == "true" && refererUri != "":
query, err := url.QueryUnescape(refererUri)
if err != nil {
return err
}
filePath, err := getNextFile(query)
if err != nil {
return err
}
if filePath == "" {
filePath, err = pickFile(paths, filter, successive)
switch {
case err != nil && err == ErrNoImagesFound:
http.NotFound(w, r)
return nil
case err != nil:
return err
}
filePath, err = getFirstFile(filePath)
if err != nil {
return err
}
}
newUrl := fmt.Sprintf("%v%v?f=%v&s=%v", r.URL.Host, filePath, filter, successive)
http.Redirect(w, r, newUrl, http.StatusSeeOther)
case r.URL.Path == "/" && successive == "true" && refererUri == "":
filePath, err := pickFile(paths, filter, successive)
if err != nil && err == ErrNoImagesFound {
http.NotFound(w, r)
return nil
} else if err != nil {
return err
}
filePath, err = getFirstFile(filePath)
if err != nil {
return err
}
newUrl := fmt.Sprintf("%v%v?f=%v&s=%v", r.URL.Host, filePath, filter, successive)
http.Redirect(w, r, newUrl, http.StatusSeeOther)
case r.URL.Path == "/":
filePath, err := pickFile(paths, filter, successive)
if err != nil && err == ErrNoImagesFound {
http.NotFound(w, r)
return nil
} else if err != nil {
return err
2022-09-08 20:30:51 +00:00
}
newUrl := fmt.Sprintf("%v%v?f=%v&s=%v", r.URL.Host, filePath, filter, successive)
http.Redirect(w, r, newUrl, http.StatusSeeOther)
default:
filePath := r.URL.Path
image, err := isImage(filePath)
if err != nil {
return err
}
if !image {
http.NotFound(w, r)
return nil
}
err = serveHtml(w, r, filePath)
if err != nil {
return err
}
2022-09-08 20:30:51 +00:00
}
return nil
2022-09-08 15:12:06 +00:00
}
}
func doNothing(http.ResponseWriter, *http.Request) {}
func ServePage(args []string) error {
paths, err := normalizePaths(args)
if err != nil {
return err
}
2022-09-08 20:30:51 +00:00
for _, i := range paths {
fmt.Println("Paths: " + i)
}
http.Handle("/", serveHtmlHandler(paths))
http.Handle(PREFIX+"/", http.StripPrefix(PREFIX, serveStaticFileHandler(paths)))
2022-09-08 15:12:06 +00:00
http.HandleFunc("/favicon.ico", doNothing)
err = http.ListenAndServe(":"+strconv.Itoa(Port), nil)
2022-09-16 19:45:54 +00:00
if err != nil {
return err
2022-09-16 19:45:54 +00:00
}
return nil
2022-09-08 15:12:06 +00:00
}