roulette/cmd/web.go

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

package cmd

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

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 {
			log.Println("Failed to serve file outside specified path(s): " + filePath)
		}

		http.NotFound(w, &r)

		return nil
	}

	_, err = os.Stat(filePath)
	if errors.Is(err, os.ErrNotExist) {
		if Verbose {
			log.Println("Failed to serve non-existent file: " + filePath)
		}

		http.NotFound(w, &r)

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

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

	w.Write(buf)

	if Verbose {
		log.Println("Served file: " + filePath)
	}

	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) {
	defer HandleExit()

	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"`
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"`
Initial commit 2022-09-08 15:12:06 +00: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 {`
			`log.Println("Failed to serve file outside specified path(s): " + filePath)`
			`}`

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 {`
			`log.Println("Failed to serve non-existent file: " + filePath)`
			`}`

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`
			`}`

			`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 {`
			`log.Println("Served file: " + filePath)`
			`}`

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) {`
Initial commit 2022-09-08 15:12:06 +00:00			`defer HandleExit()`

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))`
			`}`