package main import ( "encoding/json" "flag" "html/template" "log" "math/rand" "net/http" "os" "strconv" "strings" "time" "github.com/sirupsen/logrus" "github.com/spf13/viper" ) var images []string var logger = logrus.New() var port string // Okay I admit, this is bad. Just a workaround for now, until I figure out a clean // way of displaying all images in a grid. var tmpl = template.Must(template.New("index").Parse(` Image Gallery

Image Gallery

{{range $index, $img := .Images}} Image {{$index}} {{end}}
`)) func init() { // Log as JSON instead of the default ASCII formatter logger.SetFormatter(&logrus.JSONFormatter{}) // Output to stdout (or any other output you prefer) logger.SetOutput(os.Stdout) // Set the log level (info, warning, error, etc.) logger.SetLevel(logrus.InfoLevel) } func main() { viper.SetConfigName("config") // name of config file (without extension) viper.SetConfigType("yaml") // REQUIRED if the config file does not have the extension in the name viper.AddConfigPath(".") // path to look for the config file in if err := viper.ReadInConfig(); err != nil { log.Fatalf("Error reading configuration file: %v", err) } port = viper.GetString("server.port") flag.Parse() images = getImages() // Add request logging middleware mux := http.NewServeMux() mux.HandleFunc("/", homeHandler) mux.HandleFunc("/api/id", idHandler) mux.HandleFunc("/api/list", listHandler) mux.HandleFunc("/api/random", randomHandler) mux.HandleFunc("/api/", func(w http.ResponseWriter, r *http.Request) { http.Error(w, "Invalid API path", http.StatusNotFound) }) // Wrap the mux with the logging middleware http.Handle("/", logRequest(mux)) log.Println("Server started at port", port) log.Fatal(http.ListenAndServe(":"+port, nil)) } func getImages() []string { files, err := os.ReadDir("images/") if err != nil { logger.WithError(err).Fatal("Error reading images directory") } if len(files) == 0 { logger.Warn("No images found in the images directory") } var images []string for _, file := range files { images = append(images, file.Name()) logger.Info("Loaded image:", file.Name()) } return images } func homeHandler(w http.ResponseWriter, r *http.Request) { w.Header().Set("Content-Type", "text/html") tmpl.Execute(w, struct { Images []string }{Images: images}) } func idHandler(w http.ResponseWriter, r *http.Request) { id := r.URL.Query().Get("id") if id == "" { http.Error(w, "Missing id", http.StatusBadRequest) return } i, err := strconv.Atoi(id) if err != nil || i < 0 || i >= len(images) { http.Error(w, "Invalid id", http.StatusBadRequest) return } imagePath := "images/" + images[i] if !isValidImagePath(imagePath) { http.Error(w, "Invalid image path", http.StatusBadRequest) return } http.ServeFile(w, r, imagePath) } func isValidImagePath(path string) bool { if !strings.HasPrefix(path, "images/") { return false } return true } func listHandler(w http.ResponseWriter, r *http.Request) { imageList := []map[string]string{} for i := range images { imageInfo := map[string]string{ "id": strconv.Itoa(i), "url": "/api/id?id=" + strconv.Itoa(i), } imageList = append(imageList, imageInfo) } jsonData, err := json.Marshal(imageList) if err != nil { http.Error(w, "Internal Server Error", http.StatusInternalServerError) return } w.Header().Set("Content-Type", "application/json") w.Write(jsonData) } func logRequest(next http.Handler) http.Handler { return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { start := time.Now() logger.Infof("Started %s %s", r.Method, r.URL.Path) next.ServeHTTP(w, r) duration := time.Since(start) logger.Infof("Completed %s %s in %v", r.Method, r.URL.Path, duration) }) } func randomHandler(w http.ResponseWriter, r *http.Request) { source := rand.NewSource(time.Now().UnixNano()) rand.New(source) i := rand.Intn(len(images)) http.Redirect(w, r, "/api/id?id="+strconv.Itoa(i), http.StatusSeeOther) }