package main import ( "encoding/json" "html/template" "image" "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 title string var header string var port string // Cache for image list, it should expire every 10 minutes // but until it does, images should load faster in the frontend. var cachedImages struct { images []string expiry time.Time } // 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(` {{.Title}}

{{.Header}}

{{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) // Load config 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") images = getImages() // Load site settings (title and header) title = viper.GetString("site.title") header = viper.GetString("site.header") } func main() { // 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 getCachedImages() []string { if time.Now().After(cachedImages.expiry) { cachedImages.images = getImages() cachedImages.expiry = time.Now().Add(10 * time.Minute) } return cachedImages.images } 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 { Title string Header string Images []string }{ Title: title, Header: header, Images: getCachedImages(), }) } 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]interface{}{} for i := range getCachedImages() { imageInfo := map[string]interface{}{ "id": strconv.Itoa(i), "url": "/api/id?id=" + strconv.Itoa(i), "filename": images[i], "size": getImageSize("images/" + images[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 getImageSize(path string) map[string]interface{} { file, err := os.Open(path) if err != nil { logger.WithError(err).Error("Error opening file for size") return nil } defer file.Close() // Decode image to get dimensions (JPEG/PNG only) img, _, err := image.Decode(file) if err != nil { logger.WithError(err).Error("Error decoding image") return nil } // Get file info for size fileInfo, err := file.Stat() if err != nil { logger.WithError(err).Error("Error getting file info") return nil } return map[string]interface{}{ "width": img.Bounds().Dx(), "height": img.Bounds().Dy(), "size": fileInfo.Size(), } } 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) { rand.New(rand.NewSource(time.Now().UnixNano())) i := rand.Intn(len(images)) http.Redirect(w, r, "/api/id?id="+strconv.Itoa(i), http.StatusSeeOther) }