forked from albertba/Advertisement_Panel
36 lines
714 B
Go
36 lines
714 B
Go
package main
|
|
|
|
import (
|
|
"encoding/json"
|
|
"net/http"
|
|
"os"
|
|
"strings"
|
|
)
|
|
|
|
func main() {
|
|
http.HandleFunc("/files", func(w http.ResponseWriter, r *http.Request) {
|
|
files, err := os.ReadDir("./static")
|
|
if err != nil {
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
|
return
|
|
}
|
|
var names []string
|
|
for _, f := range files {
|
|
|
|
name := f.Name()
|
|
if strings.HasSuffix(name, ".jpg") || strings.HasSuffix(name, ".png") || strings.HasSuffix(name, ".gif") {
|
|
|
|
names = append(names, name)
|
|
}
|
|
}
|
|
w.Header().Set("Content-Type", "application/json")
|
|
|
|
json.NewEncoder(w).Encode(names)
|
|
})
|
|
|
|
fs := http.FileServer(http.Dir("./static"))
|
|
http.Handle("/", fs)
|
|
|
|
http.ListenAndServe(":8080", nil)
|
|
}
|