59 lines
1.5 KiB
Go
59 lines
1.5 KiB
Go
package main
|
|
|
|
import (
|
|
"encoding/json"
|
|
"net/http"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
)
|
|
|
|
const staticDir = "static"
|
|
|
|
// move FileData and AsciiEntry here if you want, or leave in main.go
|
|
func file_handler(w http.ResponseWriter, r *http.Request) {
|
|
data := FileData{
|
|
ImageNames: []string{},
|
|
SpicyImageNames: []string{},
|
|
AsciiFiles: []AsciiEntry{},
|
|
}
|
|
|
|
dirs, err := os.ReadDir(staticDir)
|
|
if err != nil {
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
|
return
|
|
}
|
|
|
|
for _, dir := range dirs {
|
|
dirName := dir.Name()
|
|
|
|
if strings.EqualFold(dirName, "images") {
|
|
|
|
files, _ := os.ReadDir(filepath.Join(staticDir, dirName))
|
|
for _, file := range files {
|
|
fileName := file.Name()
|
|
data.ImageNames = append(data.ImageNames, filepath.Join(staticDir, dirName, fileName))
|
|
}
|
|
} else if strings.EqualFold(dirName, "spicy") {
|
|
|
|
files, _ := os.ReadDir(filepath.Join(staticDir, dirName))
|
|
for _, file := range files {
|
|
fileName := file.Name()
|
|
data.SpicyImageNames = append(data.SpicyImageNames, filepath.Join(staticDir, dirName, fileName))
|
|
}
|
|
} else if strings.EqualFold(dirName, "ascii_art") {
|
|
|
|
files, _ := os.ReadDir(filepath.Join(staticDir, dirName))
|
|
for _, file := range files {
|
|
fileName := file.Name()
|
|
data.AsciiFiles = append(data.AsciiFiles,
|
|
AsciiEntry{Name: filepath.Join(staticDir, dirName, fileName), FontSize: 12},
|
|
)
|
|
}
|
|
}
|
|
}
|
|
|
|
w.Header().Set("Content-Type", "application/json")
|
|
json.NewEncoder(w).Encode(data)
|
|
}
|