roulette/cmd/web.go

/*
Copyright © 2022 Seednode <seednode@seedno.de>
*/

package cmd

import (
	"errors"
	"fmt"
	"io"
	"log"
	"net/http"
	"net/url"
	"os"
	"path/filepath"
	"strconv"
	"strings"
	"time"
)

const LOGDATE string = "2006-01-02T15:04:05.000000000-07:00"

const PREFIX string = "/src"

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">
  <head>
    <style>img{max-width:100%;max-height:97vh;height:auto;}</style>
	<title>`
	htmlBody += fileName
	htmlBody += `</title>
  </head>
  <body>
    <a href="/"><img src="`
	htmlBody += PREFIX + filePath
	htmlBody += `"></img></a>
  </body>
</html>`

	_, err := io.WriteString(w, htmlBody)
	if err != nil {
		return err
	}

	return nil
}

func serveStaticFile(w http.ResponseWriter, r http.Request, paths []string) error {
	prefixedFilePath, err := url.QueryUnescape(r.RequestURI)
	if err != nil {
		return err
	}

	filePath := strings.TrimPrefix(prefixedFilePath, PREFIX)

	var matchesPrefix = false
	for i := 0; i < len(paths); i++ {
		if strings.HasPrefix(filePath, paths[i]) {
			matchesPrefix = true
		}
	}
	if !matchesPrefix {
		if Verbose {
			fmt.Printf("%v Failed to serve file outside specified path(s): %v", time.Now().Format(LOGDATE), filePath)
		}

		http.NotFound(w, &r)

		return nil
	}

	_, err = os.Stat(filePath)
	if errors.Is(err, os.ErrNotExist) {
		if Verbose {
			fmt.Printf("%v Failed to serve non-existent file: %v", time.Now().Format(LOGDATE), filePath)
		}

		http.NotFound(w, &r)

		return nil
	} else if !errors.Is(err, os.ErrNotExist) && err != nil {
		return err
	}

	var startTime time.Time
	if Verbose {
		startTime = time.Now()
		fmt.Printf("%v Serving file: %v", startTime.Format(LOGDATE), filePath)
	}

	buf, err := os.ReadFile(filePath)
	if err != nil {
		return err
	}

	w.Write(buf)

	if Verbose {
		fmt.Printf(" (Finished in %v)\n", time.Since(startTime).Round(time.Microsecond))
	}

	return nil
}

func serveStaticFileHandler(paths []string) http.HandlerFunc {
	return func(w http.ResponseWriter, r *http.Request) {
		err := serveStaticFile(w, *r, paths)
		if err != nil {
			log.Fatal(err)
		}
	}
}

func serveHtmlHandler(paths []string) http.HandlerFunc {
	return func(w http.ResponseWriter, r *http.Request) {
		refererUri := refererToUri(r.Referer())

		switch {
		case r.RequestURI == "/" && Successive && refererUri != "":
			query, err := url.QueryUnescape(refererUri)
			if err != nil {
				log.Fatal(err)
			}

			filePath, err := getNextFile(query)
			if err != nil {
				log.Fatal(err)
			}

			if filePath == "" {
				filePath, err = pickFile(paths)
				if err != nil {
					log.Fatal(err)
				}

				filePath, err = getFirstFile(filePath)
				if err != nil {
					log.Fatal(err)
				}
			}

			newUrl := r.URL.Host + filePath
			http.Redirect(w, r, newUrl, http.StatusSeeOther)
		case r.RequestURI == "/" && Successive && refererUri == "":
			filePath, err := pickFile(paths)
			if err != nil {
				log.Fatal(err)
			}

			filePath, err = getFirstFile(filePath)
			if err != nil {
				log.Fatal(err)
			}

			newUrl := r.URL.Host + filePath
			http.Redirect(w, r, newUrl, http.StatusSeeOther)
		case r.RequestURI == "/":
			filePath, err := pickFile(paths)
			if err != nil {
				log.Fatal(err)
			}

			newUrl := r.URL.Host + filePath
			http.Redirect(w, r, newUrl, http.StatusSeeOther)
		default:
			filePath, err := url.QueryUnescape(r.RequestURI)
			if err != nil {
				log.Fatal(err)
			}

			isImage, err := checkIfImage(filePath)
			if err != nil {
				log.Fatal(err)
			}
			if !isImage {
				http.NotFound(w, r)
			}

			err = serveHtml(w, *r, filePath)
			if err != nil {
				log.Fatal(err)
			}
		}
	}
}

func doNothing(http.ResponseWriter, *http.Request) {}

func ServePage(args []string) {
	paths, err := normalizePaths(args)
	if err != nil {
		log.Fatal(err)
	}

	for _, i := range paths {
		fmt.Println("Paths: " + i)
	}

	http.HandleFunc("/", serveHtmlHandler(paths))
	http.Handle(PREFIX+"/", http.StripPrefix(PREFIX, serveStaticFileHandler(paths)))
	http.HandleFunc("/favicon.ico", doNothing)

	port := strconv.Itoa(Port)

	err = http.ListenAndServe(":"+port, nil)
	if err != nil {
		log.Fatal(err)
	}
}
Initial commit 2022-09-08 15:12:06 +00:00			`/*`
			`Copyright © 2022 Seednode <seednode@seedno.de>`
			`*/`

			`package cmd`

			`import (`
Added file not found response for invalid files 2022-09-08 20:57:52 +00:00			`"errors"`
Replaced log.Print with fmt.Printf to allow finer control of newlines 2022-09-10 23:03:04 +00:00			`"fmt"`
Initial commit 2022-09-08 15:12:06 +00:00			`"io"`
			`"log"`
			`"net/http"`
Reached first working version 2022-09-08 20:30:51 +00:00			`"net/url"`
First attempt 2022-09-08 15:57:59 +00:00			`"os"`
Reworked to allow use of back button to return to previous results, via HTTP redirects 2022-09-16 18:45:33 +00:00			`"path/filepath"`
Initial commit 2022-09-08 15:12:06 +00:00			`"strconv"`
Added file not found response for invalid files 2022-09-08 20:57:52 +00:00			`"strings"`
Replaced log.Print with fmt.Printf to allow finer control of newlines 2022-09-10 23:03:04 +00:00			`"time"`
Initial commit 2022-09-08 15:12:06 +00:00			`)`

Replaced log.Print with fmt.Printf to allow finer control of newlines 2022-09-10 23:03:04 +00:00			`const LOGDATE string = "2006-01-02T15:04:05.000000000-07:00"`

Reworked to allow use of back button to return to previous results, via HTTP redirects 2022-09-16 18:45:33 +00:00			`const PREFIX string = "/src"`
Added file not found response for invalid files 2022-09-08 20:57:52 +00:00
Added option to display sequentially numbered files instead of randomly choosing, when possible 2022-09-17 16:07:37 +00:00			`func refererToUri(referer string) string {`
			`parts := strings.SplitAfterN(referer, "/", 4)`

			`if len(parts) < 4 {`
			`return ""`
			`}`

			`return "/" + parts[3]`
			`}`

Renamed functions for consistency 2022-09-16 19:45:54 +00:00			`func serveHtml(w http.ResponseWriter, r http.Request, filePath string) error {`
Reworked to allow use of back button to return to previous results, via HTTP redirects 2022-09-16 18:45:33 +00:00			`fileName := filepath.Base(filePath)`
Added file not found response for invalid files 2022-09-08 20:57:52 +00:00
			`w.Header().Add("Content-Type", "text/html")`

			htmlBody := `<html lang="en">
Initial commit 2022-09-08 15:12:06 +00:00			`<head>`
Reached first working version 2022-09-08 20:30:51 +00:00			`<style>img{max-width:100%;max-height:97vh;height:auto;}</style>`
			<title>`
Added file not found response for invalid files 2022-09-08 20:57:52 +00:00			`htmlBody += fileName`
			htmlBody += `</title>
Initial commit 2022-09-08 15:12:06 +00:00			`</head>`
			`<body>`
Added file not found response for invalid files 2022-09-08 20:57:52 +00:00			<a href="/"><img src="`
Reworked to allow use of back button to return to previous results, via HTTP redirects 2022-09-16 18:45:33 +00:00			`htmlBody += PREFIX + filePath`
Added file not found response for invalid files 2022-09-08 20:57:52 +00:00			htmlBody += `"></img></a>
Re-added missing HTML footer 2022-09-09 00:11:07 +00:00			`</body>`
			</html>`

Reworked to allow use of back button to return to previous results, via HTTP redirects 2022-09-16 18:45:33 +00:00			`_, err := io.WriteString(w, htmlBody)`
Added file not found response for invalid files 2022-09-08 20:57:52 +00:00			`if err != nil {`
			`return err`
			`}`
Initial commit 2022-09-08 15:12:06 +00:00
Added file not found response for invalid files 2022-09-08 20:57:52 +00:00			`return nil`
Initial commit 2022-09-08 15:12:06 +00:00			`}`

Removed unused functions 2022-09-09 19:14:54 +00:00			`func serveStaticFile(w http.ResponseWriter, r http.Request, paths []string) error {`
Rearranged some variable initializations 2022-09-18 18:32:58 +00:00			`prefixedFilePath, err := url.QueryUnescape(r.RequestURI)`
Initial commit 2022-09-08 15:12:06 +00:00			`if err != nil {`
Added file not found response for invalid files 2022-09-08 20:57:52 +00:00			`return err`
Initial commit 2022-09-08 15:12:06 +00:00			`}`

Reworked to allow use of back button to return to previous results, via HTTP redirects 2022-09-16 18:45:33 +00:00			`filePath := strings.TrimPrefix(prefixedFilePath, PREFIX)`

Added file not found response for invalid files 2022-09-08 20:57:52 +00:00			`var matchesPrefix = false`
			`for i := 0; i < len(paths); i++ {`
			`if strings.HasPrefix(filePath, paths[i]) {`
			`matchesPrefix = true`
			`}`
			`}`
Use raw strings instead of quoted strings for regexp.MustCompile, replaced time.Now().Sub with time.Since, fixed clobbered filePath value, per staticcheck 2022-09-22 02:24:05 +00:00			`if !matchesPrefix {`
Added option for -v\|--verbose, to log file accesses to stdout 2022-09-10 21:17:55 +00:00			`if Verbose {`
Replaced log.Print with fmt.Printf to allow finer control of newlines 2022-09-10 23:03:04 +00:00			`fmt.Printf("%v Failed to serve file outside specified path(s): %v", time.Now().Format(LOGDATE), filePath)`
Added option for -v\|--verbose, to log file accesses to stdout 2022-09-10 21:17:55 +00:00			`}`

Removed unused functions 2022-09-09 19:14:54 +00:00			`http.NotFound(w, &r)`
Added file not found response for invalid files 2022-09-08 20:57:52 +00:00
			`return nil`
			`}`
Initial commit 2022-09-08 15:12:06 +00:00
Added file not found response for invalid files 2022-09-08 20:57:52 +00:00			`_, err = os.Stat(filePath)`
			`if errors.Is(err, os.ErrNotExist) {`
Added option for -v\|--verbose, to log file accesses to stdout 2022-09-10 21:17:55 +00:00			`if Verbose {`
Replaced log.Print with fmt.Printf to allow finer control of newlines 2022-09-10 23:03:04 +00:00			`fmt.Printf("%v Failed to serve non-existent file: %v", time.Now().Format(LOGDATE), filePath)`
Added option for -v\|--verbose, to log file accesses to stdout 2022-09-10 21:17:55 +00:00			`}`

Removed unused functions 2022-09-09 19:14:54 +00:00			`http.NotFound(w, &r)`
Added file not found response for invalid files 2022-09-08 20:57:52 +00:00
			`return nil`
			`} else if !errors.Is(err, os.ErrNotExist) && err != nil {`
			`return err`
			`}`

Display time spent displaying image when running in verbose mode 2022-09-12 00:01:56 +00:00			`var startTime time.Time`
Replaced log.Print with fmt.Printf to allow finer control of newlines 2022-09-10 23:03:04 +00:00			`if Verbose {`
Display time spent displaying image when running in verbose mode 2022-09-12 00:01:56 +00:00			`startTime = time.Now()`
Simplified timestamp functions 2022-09-12 01:05:55 +00:00			`fmt.Printf("%v Serving file: %v", startTime.Format(LOGDATE), filePath)`
Replaced log.Print with fmt.Printf to allow finer control of newlines 2022-09-10 23:03:04 +00:00			`}`

Added file not found response for invalid files 2022-09-08 20:57:52 +00:00			`buf, err := os.ReadFile(filePath)`
Initial commit 2022-09-08 15:12:06 +00:00			`if err != nil {`
Added file not found response for invalid files 2022-09-08 20:57:52 +00:00			`return err`
Initial commit 2022-09-08 15:12:06 +00:00			`}`
Added file not found response for invalid files 2022-09-08 20:57:52 +00:00
			`w.Write(buf)`

Added option for -v\|--verbose, to log file accesses to stdout 2022-09-10 21:17:55 +00:00			`if Verbose {`
Use raw strings instead of quoted strings for regexp.MustCompile, replaced time.Now().Sub with time.Since, fixed clobbered filePath value, per staticcheck 2022-09-22 02:24:05 +00:00			`fmt.Printf(" (Finished in %v)\n", time.Since(startTime).Round(time.Microsecond))`
Added option for -v\|--verbose, to log file accesses to stdout 2022-09-10 21:17:55 +00:00			`}`

Added file not found response for invalid files 2022-09-08 20:57:52 +00:00			`return nil`
Initial commit 2022-09-08 15:12:06 +00:00			`}`

Reworked to allow use of back button to return to previous results, via HTTP redirects 2022-09-16 18:45:33 +00:00			`func serveStaticFileHandler(paths []string) http.HandlerFunc {`
			`return func(w http.ResponseWriter, r *http.Request) {`
			`err := serveStaticFile(w, *r, paths)`
			`if err != nil {`
			`log.Fatal(err)`
			`}`
			`}`
			`}`

Renamed functions for consistency 2022-09-16 19:45:54 +00:00			`func serveHtmlHandler(paths []string) http.HandlerFunc {`
Initial commit 2022-09-08 15:12:06 +00:00			`return func(w http.ResponseWriter, r *http.Request) {`
Display first image in directory on first load if --successive is enabled 2022-09-18 18:07:46 +00:00			`refererUri := refererToUri(r.Referer())`

			`switch {`
			`case r.RequestURI == "/" && Successive && refererUri != "":`
serveHtmlHandler could potentially return an empty response in default case 2022-09-18 18:17:29 +00:00			`query, err := url.QueryUnescape(refererUri)`
Display first image in directory on first load if --successive is enabled 2022-09-18 18:07:46 +00:00			`if err != nil {`
			`log.Fatal(err)`
			`}`

Rearranged some variable initializations 2022-09-18 18:32:58 +00:00			`filePath, err := getNextFile(query)`
Display first image in directory on first load if --successive is enabled 2022-09-18 18:07:46 +00:00			`if err != nil {`
			`log.Fatal(err)`
Added option to display sequentially numbered files instead of randomly choosing, when possible 2022-09-17 16:07:37 +00:00			`}`

			`if filePath == "" {`
Use raw strings instead of quoted strings for regexp.MustCompile, replaced time.Now().Sub with time.Since, fixed clobbered filePath value, per staticcheck 2022-09-22 02:24:05 +00:00			`filePath, err = pickFile(paths)`
Added option to display sequentially numbered files instead of randomly choosing, when possible 2022-09-17 16:07:37 +00:00			`if err != nil {`
			`log.Fatal(err)`
			`}`
Display first image in directory on first load if --successive is enabled 2022-09-18 18:07:46 +00:00
			`filePath, err = getFirstFile(filePath)`
			`if err != nil {`
			`log.Fatal(err)`
			`}`
			`}`

			`newUrl := r.URL.Host + filePath`
			`http.Redirect(w, r, newUrl, http.StatusSeeOther)`
			`case r.RequestURI == "/" && Successive && refererUri == "":`
Rearranged some variable initializations 2022-09-18 18:32:58 +00:00			`filePath, err := pickFile(paths)`
Display first image in directory on first load if --successive is enabled 2022-09-18 18:07:46 +00:00			`if err != nil {`
			`log.Fatal(err)`
			`}`

			`filePath, err = getFirstFile(filePath)`
			`if err != nil {`
			`log.Fatal(err)`
			`}`

			`newUrl := r.URL.Host + filePath`
			`http.Redirect(w, r, newUrl, http.StatusSeeOther)`
			`case r.RequestURI == "/":`
Rearranged some variable initializations 2022-09-18 18:32:58 +00:00			`filePath, err := pickFile(paths)`
Display first image in directory on first load if --successive is enabled 2022-09-18 18:07:46 +00:00			`if err != nil {`
			`log.Fatal(err)`
Reached first working version 2022-09-08 20:30:51 +00:00			`}`
Reworked to allow use of back button to return to previous results, via HTTP redirects 2022-09-16 18:45:33 +00:00
			`newUrl := r.URL.Host + filePath`
			`http.Redirect(w, r, newUrl, http.StatusSeeOther)`
Display first image in directory on first load if --successive is enabled 2022-09-18 18:07:46 +00:00			`default:`
Reworked to allow use of back button to return to previous results, via HTTP redirects 2022-09-16 18:45:33 +00:00			`filePath, err := url.QueryUnescape(r.RequestURI)`
Reached first working version 2022-09-08 20:30:51 +00:00			`if err != nil {`
Added file not found response for invalid files 2022-09-08 20:57:52 +00:00			`log.Fatal(err)`
Reached first working version 2022-09-08 20:30:51 +00:00			`}`
Reworked to allow use of back button to return to previous results, via HTTP redirects 2022-09-16 18:45:33 +00:00
			`isImage, err := checkIfImage(filePath)`
			`if err != nil {`
serveHtmlHandler could potentially return an empty response in default case 2022-09-18 18:17:29 +00:00			`log.Fatal(err)`
			`}`
			`if !isImage {`
Reworked to allow use of back button to return to previous results, via HTTP redirects 2022-09-16 18:45:33 +00:00			`http.NotFound(w, r)`
			`}`

serveHtmlHandler could potentially return an empty response in default case 2022-09-18 18:17:29 +00:00			`err = serveHtml(w, *r, filePath)`
			`if err != nil {`
			`log.Fatal(err)`
Reworked to allow use of back button to return to previous results, via HTTP redirects 2022-09-16 18:45:33 +00:00			`}`
Reached first working version 2022-09-08 20:30:51 +00:00			`}`
Initial commit 2022-09-08 15:12:06 +00:00			`}`
			`}`

			`func doNothing(http.ResponseWriter, *http.Request) {}`

First attempt 2022-09-08 15:57:59 +00:00			`func ServePage(args []string) {`
Replaced panic with error return value in normalizePaths() 2022-09-09 19:28:57 +00:00			`paths, err := normalizePaths(args)`
			`if err != nil {`
			`log.Fatal(err)`
			`}`
Reached first working version 2022-09-08 20:30:51 +00:00
Reworked to allow use of back button to return to previous results, via HTTP redirects 2022-09-16 18:45:33 +00:00			`for _, i := range paths {`
			`fmt.Println("Paths: " + i)`
			`}`
Rearranged some variable initializations 2022-09-18 18:32:58 +00:00
Renamed functions for consistency 2022-09-16 19:45:54 +00:00			`http.HandleFunc("/", serveHtmlHandler(paths))`
Reworked to allow use of back button to return to previous results, via HTTP redirects 2022-09-16 18:45:33 +00:00			`http.Handle(PREFIX+"/", http.StripPrefix(PREFIX, serveStaticFileHandler(paths)))`
Initial commit 2022-09-08 15:12:06 +00:00			`http.HandleFunc("/favicon.ico", doNothing)`

			`port := strconv.Itoa(Port)`

Renamed functions for consistency 2022-09-16 19:45:54 +00:00			`err = http.ListenAndServe(":"+port, nil)`
			`if err != nil {`
			`log.Fatal(err)`
			`}`
Initial commit 2022-09-08 15:12:06 +00:00			`}`