handlers.go

package coresize

import (
	"fmt"
	"log"
	"net/http"
	"strconv"
	"time"

	"github.com/julienschmidt/httprouter"
	"github.com/kiasaki/batbelt/rest"
)

func (s *Server) handlePanic(w http.ResponseWriter, r *http.Request, err interface{}) {
	rest.SetInternalServerErrorResponse(w, err)
}

func (s *Server) handleNotFound(w http.ResponseWriter, r *http.Request) {
	rest.SetNotFoundResponse(w)
}

func (s *Server) handleIndex(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
	// return supported endpoints with url templates
	rest.SetOKResponse(w, map[string]interface{}{
		"v1": map[string]string{
			"root_url":  "/",
			"image_url": "/v1/i/{/file_name}{?file_hash,height,width,allign}",
		},
	})
}

func (s *Server) handleImage(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
	requestFilename := ps.ByName("filename")[1:]

	if cacheFile, ok := s.Cache.Get(requestFilename); ok {
		// extract url params
		height, _ := strconv.Atoi(r.URL.Query().Get("height"))
		width, _ := strconv.Atoi(r.URL.Query().Get("width"))
		align := r.URL.Query().Get("align")
		if align == "" {
			align = "cc"
		}

		// fetch and render, or, read from disk
		image, err := cacheFile.Image()
		w.Header().Set("Content-Type", cacheFile.FileType())
		w.Header().Set("Expires", time.Now().AddDate(1, 0, 0).Format(http.TimeFormat))
		w.Header().Set("ETag", fmt.Sprintf("%s,w=%d,h=%d,a=%s", image.Hash, width, height, align))
		err = image.Render(w, width, height, align)
		if err != nil {
			log.Printf("Error rendering %s: %s\n", requestFilename, err.Error())
			rest.SetInternalServerErrorResponse(w, err)
			return
		}
		return
	}
	http.Error(w, "File not found", http.StatusNotFound)
}

func (s *Server) handleReload(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
	cache := NewCache(s.Config)
	if err := cache.Setup(); err != nil {
		log.Printf("Error setting up cache: %s\n", err.Error())
		rest.SetInternalServerErrorResponse(w, err)
		return
	}
	s.Cache = cache
	rest.SetNoContentResponse(w)
}