SeanOMik
/
roulette
mirror of https://git.seedno.de/seednode/roulette.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

serveHtmlHandler could potentially return an empty response in default case

This commit is contained in:
Seednode 2022-09-18 13:17:29 -05:00
parent f3280a6b6c
commit 9e05a12fd7
2 changed files with 10 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
cmd/version.go
View File

@ -10,7 +10,7 @@ import (
"github.com/spf13/cobra" "github.com/spf13/cobra"
) )
var Version = "0.8.0" var Version = "0.8.1"
func init() { func init() {
rootCmd.AddCommand(versionCmd) rootCmd.AddCommand(versionCmd)

16
cmd/web.go
View File

@ -136,12 +136,12 @@ func serveHtmlHandler(paths []string) http.HandlerFunc {
switch { switch {
case r.RequestURI == "/" && Successive && refererUri != "": case r.RequestURI == "/" && Successive && refererUri != "":
f, err := url.QueryUnescape(refererUri) query, err := url.QueryUnescape(refererUri)
if err != nil { if err != nil {
log.Fatal(err) log.Fatal(err)
} }
filePath, err = getNextFile(f) filePath, err = getNextFile(query)
if err != nil { if err != nil {
log.Fatal(err) log.Fatal(err)
} }
@ -189,14 +189,16 @@ func serveHtmlHandler(paths []string) http.HandlerFunc {
isImage, err := checkIfImage(filePath) isImage, err := checkIfImage(filePath)
if err != nil { if err != nil {
log.Fatal(err)
}
if !isImage {
http.NotFound(w, r) http.NotFound(w, r)
} }
if isImage { err = serveHtml(w, *r, filePath)
err := serveHtml(w, *r, filePath) if err != nil {
if err != nil { log.Fatal(err)
log.Fatal(err)
}
} }
} }
} }