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

Reached first working version

This commit is contained in:
Seednode 2022-09-08 15:30:51 -05:00
parent d052fd6af9
commit 6f1c0c3943
2 changed files with 69 additions and 90 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
cmd/files.go
View File

@ -18,8 +18,13 @@ func getFiles(path string) ([]string, error) {
if info.IsDir() && p != path { if info.IsDir() && p != path {
return filepath.SkipDir return filepath.SkipDir
} else { } else {
paths = append(paths, p) absolutePath, err := filepath.Abs(p)
if err != nil {
return err
} }
paths = append(paths, absolutePath)
}
return err return err
}) })
if err != nil { if err != nil {
@ -34,7 +39,11 @@ func getFilesRecursive(path string) ([]string, error) {
err := filepath.WalkDir(path, func(p string, info os.DirEntry, err error) error { err := filepath.WalkDir(path, func(p string, info os.DirEntry, err error) error {
if !info.IsDir() { if !info.IsDir() {
paths = append(paths, p) absolutePath, err := filepath.Abs(p)
if err != nil {
return err
}
paths = append(paths, absolutePath)
} }
return err return err
}) })
@ -54,12 +63,14 @@ func getFileList(args []string) ([]string, error) {
if err != nil { if err != nil {
return nil, err return nil, err
} }
fileList = append(fileList, f...) fileList = append(fileList, f...)
} else { } else {
f, err := getFiles(args[i]) f, err := getFiles(args[i])
if err != nil { if err != nil {
return nil, err return nil, err
} }
fileList = append(fileList, f...) fileList = append(fileList, f...)
} }
} }
@ -68,10 +79,25 @@ func getFileList(args []string) ([]string, error) {
} }
func pickFile(fileList []string) (string, string) { func pickFile(fileList []string) (string, string) {
rand.Seed(time.Now().Unix()) rand.Seed(time.Now().UnixMicro())
filePath := fileList[rand.Intn(len(fileList))] filePath := fileList[rand.Intn(len(fileList))]
fileName := filepath.Base(filePath) fileName := filepath.Base(filePath)
return filePath, fileName return fileName, filePath
}
func normalizePaths(args []string) []string {
var paths []string
for i := 0; i < len(args); i++ {
absolutePath, err := filepath.Abs(args[i])
if err != nil {
panic(err)
}
paths = append(paths, absolutePath)
}
return paths
} }

123
cmd/web.go
View File

@ -9,15 +9,18 @@ import (
"io" "io"
"log" "log"
"net/http" "net/http"
"net/url"
"os" "os"
"strconv" "strconv"
"strings"
) )
func generateHeader() string { func generateHeader(fileName string) string {
htmlHeader := `<html> htmlHeader := `<html lang="en">
<head> <head>
<title>OUI Lookup</title> <style>img{max-width:100%;max-height:97vh;height:auto;}</style>
<title>`
htmlHeader += fileName
htmlHeader += `</title>
</head> </head>
<body> <body>
` `
@ -32,39 +35,26 @@ func generateFooter() string {
return htmlFooter return htmlFooter
} }
func generatePageRaw(w http.ResponseWriter, output []string) { func generatePageHtml(w http.ResponseWriter, paths []string, output []string) {
w.Header().Add("Content-Type", "text/plain") fileList, err := getFileList(paths)
for i := 0; i < len(output); i++ {
if Verbose {
fmt.Println(output[i])
}
_, err := io.WriteString(w, output[i]+"\n")
if err != nil { if err != nil {
fmt.Println(err) panic(err)
} }
}
}
func generatePageHtml(w http.ResponseWriter, output []string) { fileName, filePath := pickFile(fileList)
w.Header().Add("Content-Type", "text/html") w.Header().Add("Content-Type", "text/html")
_, err := io.WriteString(w, generateHeader()) _, err = io.WriteString(w, generateHeader(fileName))
if err != nil { if err != nil {
fmt.Println(err) fmt.Println(err)
} }
for i := 0; i < len(output); i++ { htmlBody := ` <a href="/"><img src="`
if Verbose { htmlBody += filePath
fmt.Println(output[i]) htmlBody += `"></img></a>
} `
_, err = io.WriteString(w, htmlBody)
_, err = io.WriteString(w, " "+output[i]+"<br />\n")
if err != nil {
fmt.Println(err)
}
}
_, err = io.WriteString(w, generateFooter()) _, err = io.WriteString(w, generateFooter())
if err != nil { if err != nil {
@ -72,74 +62,37 @@ func generatePageHtml(w http.ResponseWriter, output []string) {
} }
} }
func generatePageHelp(r *http.Request, w http.ResponseWriter) { func servePageHandler(paths []string) http.HandlerFunc {
w.Header().Add("Content-Type", "text/html")
_, err := io.WriteString(w, generateHeader())
if err != nil {
fmt.Println(err)
}
exampleHtmlUrl := fmt.Sprintf("http://%v/7C:0E:CE:FE:FE:FE,10:FE:ED:AB:AB:AB", r.Host)
exampleRawUrl := fmt.Sprintf("http://%v/raw/7C:0E:CE:FE:FE:FE,10:FE:ED:AB:AB:AB", r.Host)
exampleJsonUrl := fmt.Sprintf("http://%v/json/7C:0E:CE:FE:FE:FE,10:FE:ED:AB:AB:AB", r.Host)
help := fmt.Sprintf(" Provide one or more MAC addresses, separated by commas.<br />\n For example: <a href=%q>%v</a>", exampleHtmlUrl, exampleHtmlUrl)
help += fmt.Sprintf("<br /><br />\n Also supports plain text responses.<br />\n For example: <a href=%q>%v</a>", exampleRawUrl, exampleRawUrl)
help += fmt.Sprintf("<br /><br />\n And JSON responses, of course.<br />\n For example: <a href=%q>%v</a>", exampleJsonUrl, exampleJsonUrl)
_, err = io.WriteString(w, help)
if err != nil {
fmt.Println(err)
}
_, err = io.WriteString(w, generateFooter())
if err != nil {
fmt.Println(err)
}
}
func servePageHandler() http.HandlerFunc {
return func(w http.ResponseWriter, r *http.Request) { return func(w http.ResponseWriter, r *http.Request) {
var ouis []string
oui := r.RequestURI
if oui == "" {
generatePageHelp(r, w)
return
}
args := strings.Split(oui, ",")
for i := 0; i < len(args); i++ {
ouis = append(ouis, args[i])
}
var output []string var output []string
generatePageHtml(w, output) request := r.RequestURI
if r.RequestURI == "/" {
generatePageHtml(w, paths, output)
} else {
f, err := url.QueryUnescape(request)
if err != nil {
log.Fatal(err)
return
}
buf, err := os.ReadFile(f)
if err != nil {
panic(err)
}
w.Write(buf)
}
} }
} }
func doNothing(http.ResponseWriter, *http.Request) {} func doNothing(http.ResponseWriter, *http.Request) {}
func ServePage(args []string) { func ServePage(args []string) {
fileList, err := getFileList(args)
if err != nil {
panic(err)
}
fileName, filePath := pickFile(fileList)
fmt.Println(fileName)
fmt.Println(filePath)
os.Exit(0)
defer HandleExit() defer HandleExit()
http.HandleFunc("/", servePageHandler()) paths := normalizePaths(args)
http.HandleFunc("/", servePageHandler(paths))
http.HandleFunc("/favicon.ico", doNothing) http.HandleFunc("/favicon.ico", doNothing)
port := strconv.Itoa(Port) port := strconv.Itoa(Port)