roulette/cmd/web.go

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

package cmd

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

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

func generatePageHtml(w http.ResponseWriter, r http.Request, paths []string) error {
	fileList, err := getFileList(paths)
	if err != nil {
		return err
	}

	fileName, filePath, err := pickFile(fileList)
	if err != nil {
		http.NotFound(w, &r)
		return nil
	}

	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 += 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 {
	request := r.RequestURI

	filePath, err := url.QueryUnescape(request)
	if err != nil {
		return err
	}

	var matchesPrefix = false

	for i := 0; i < len(paths); i++ {
		if strings.HasPrefix(filePath, paths[i]) {
			matchesPrefix = true
		}
	}

	if matchesPrefix == false {
		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.Now().Sub(startTime).Round(time.Microsecond))
	}

	return nil
}

func servePageHandler(paths []string) http.HandlerFunc {
	return func(w http.ResponseWriter, r *http.Request) {
		if r.RequestURI == "/" {
			err := generatePageHtml(w, *r, paths)
			if err != nil {
				log.Fatal(err)
			}
		} else {
			err := serveStaticFile(w, *r, paths)
			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)
	}

	http.HandleFunc("/", servePageHandler(paths))
	http.HandleFunc("/favicon.ico", doNothing)

	port := strconv.Itoa(Port)

	log.Fatal(http.ListenAndServe(":"+port, nil))
}
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"`
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"`

No longer picking random files from slice, shuffle slice once instead. Return 404 not found if no valid image files are found. 2022-09-09 22:21:03 +00:00			`func generatePageHtml(w http.ResponseWriter, r http.Request, paths []string) error {`
Added file not found response for invalid files 2022-09-08 20:57:52 +00:00			`fileList, err := getFileList(paths)`
			`if err != nil {`
			`return err`
			`}`

No longer picking random files from slice, shuffle slice once instead. Return 404 not found if no valid image files are found. 2022-09-09 22:21:03 +00:00			`fileName, filePath, err := pickFile(fileList)`
			`if err != nil {`
			`http.NotFound(w, &r)`
			`return nil`
			`}`
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="`
			`htmlBody += filePath`
			htmlBody += `"></img></a>
Re-added missing HTML footer 2022-09-09 00:11:07 +00:00			`</body>`
			</html>`

Added file not found response for invalid files 2022-09-08 20:57:52 +00:00			`_, err = io.WriteString(w, htmlBody)`
			`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 {`
			`request := r.RequestURI`
Added file not found response for invalid files 2022-09-08 20:57:52 +00:00
			`filePath, err := url.QueryUnescape(request)`
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			`var matchesPrefix = false`

			`for i := 0; i < len(paths); i++ {`
			`if strings.HasPrefix(filePath, paths[i]) {`
			`matchesPrefix = true`
			`}`
			`}`

			`if matchesPrefix == false {`
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 {`
Fixed timestamp formatting 2022-09-12 01:48:52 +00:00			`fmt.Printf(" (Finished in %v)\n", time.Now().Sub(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			`}`

Reached first working version 2022-09-08 20:30:51 +00:00			`func servePageHandler(paths []string) http.HandlerFunc {`
Initial commit 2022-09-08 15:12:06 +00:00			`return func(w http.ResponseWriter, r *http.Request) {`
Reached first working version 2022-09-08 20:30:51 +00:00			`if r.RequestURI == "/" {`
No longer picking random files from slice, shuffle slice once instead. Return 404 not found if no valid image files are found. 2022-09-09 22:21:03 +00:00			`err := generatePageHtml(w, *r, paths)`
Reached first working version 2022-09-08 20:30:51 +00:00			`if err != nil {`
			`log.Fatal(err)`
			`}`
Added file not found response for invalid files 2022-09-08 20:57:52 +00:00			`} else {`
Removed unused functions 2022-09-09 19:14:54 +00:00			`err := serveStaticFile(w, *r, paths)`
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			`}`
			`}`
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
			`http.HandleFunc("/", servePageHandler(paths))`
Initial commit 2022-09-08 15:12:06 +00:00			`http.HandleFunc("/favicon.ico", doNothing)`

			`port := strconv.Itoa(Port)`

			`log.Fatal(http.ListenAndServe(":"+port, nil))`
			`}`