roulette/cmd/web.go

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

package cmd

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

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

	fileName, filePath := pickFile(fileList)

	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>
`
	_, err = io.WriteString(w, htmlBody)
	if err != nil {
		return err
	}

	return nil
}

func statusNotFound(w http.ResponseWriter, filePath string) error {
	fmt.Println("Client requested non-existent file " + filePath + ".")

	w.WriteHeader(http.StatusNotFound)
	w.Header().Set("Content-Type", "txt/plain")
	htmlBody := "File not found."
	_, err := io.WriteString(w, htmlBody)
	if err != nil {
		return err
	}

	return nil
}

func serveStaticFile(w http.ResponseWriter, request string, paths []string) error {
	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 {
		err := statusNotFound(w, filePath)
		if err != nil {
			return err
		}

		return nil
	}

	_, err = os.Stat(filePath)
	if errors.Is(err, os.ErrNotExist) {
		err := statusNotFound(w, filePath)
		if err != nil {
			return err
		}

		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)

	return nil
}

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

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

func ServePage(args []string) {
	defer HandleExit()

	paths := normalizePaths(args)

	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			`"fmt"`
			`"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			`)`

Added file not found response for invalid files 2022-09-08 20:57:52 +00:00			`func generatePageHtml(w http.ResponseWriter, paths []string) error {`
			`fileList, err := getFileList(paths)`
			`if err != nil {`
			`return err`
			`}`

			`fileName, filePath := pickFile(fileList)`

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

Added file not found response for invalid files 2022-09-08 20:57:52 +00:00			`func statusNotFound(w http.ResponseWriter, filePath string) error {`
			`fmt.Println("Client requested non-existent file " + filePath + ".")`

			`w.WriteHeader(http.StatusNotFound)`
			`w.Header().Set("Content-Type", "txt/plain")`
			`htmlBody := "File not found."`
			`_, 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			`}`

Added file not found response for invalid files 2022-09-08 20:57:52 +00:00			`func serveStaticFile(w http.ResponseWriter, request string, paths []string) error {`
			`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 {`
			`err := statusNotFound(w, filePath)`
			`if err != nil {`
			`return err`
			`}`

			`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) {`
			`err := statusNotFound(w, filePath)`
			`if err != nil {`
			`return err`
			`}`

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

			`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 == "/" {`
Added file not found response for invalid files 2022-09-08 20:57:52 +00:00			`err := generatePageHtml(w, 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 {`
			`err := serveStaticFile(w, r.RequestURI, 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()`

Reached first working version 2022-09-08 20:30:51 +00:00			`paths := normalizePaths(args)`

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