128 lines
3.0 KiB
Go
128 lines
3.0 KiB
Go
// vi:ts=4:sts=4:sw=4:noet:tw=72
|
|
|
|
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"log"
|
|
"net/http"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/davidbyttow/govips/pkg/vips"
|
|
)
|
|
|
|
var (
|
|
imgServer = flag.String("s", "https://images.example.com/", "URL prefix")
|
|
listenAddr = flag.String("l", "0.0.0.0:8080", "Listen address")
|
|
sizesFilter = flag.String("f", "", "Allowed sizes")
|
|
)
|
|
|
|
func init() {
|
|
flag.Parse()
|
|
}
|
|
|
|
func resizeHandler(w http.ResponseWriter, r *http.Request) {
|
|
// Get the query parameters from the request URL
|
|
path := r.URL.EscapedPath()
|
|
parm := strings.Split(path, "/")
|
|
if len(parm) < 4 {
|
|
writeError(w, "", "", "error in arguments", http.StatusBadRequest)
|
|
return
|
|
}
|
|
|
|
method := parm[1]
|
|
args := parm[2]
|
|
imgPath := strings.Join(parm[3:], "/")
|
|
|
|
// Start fetching the image from the given url
|
|
resp, err := http.Get(*imgServer + imgPath)
|
|
if err != nil {
|
|
writeError(w, imgPath, method, "failed to fetch image", http.StatusNotFound)
|
|
return
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
switch method {
|
|
case "resize":
|
|
allowed := strings.Split(*sizesFilter, ",")
|
|
forbidden := true
|
|
if allowed[0] == "*" {
|
|
forbidden = false
|
|
} else {
|
|
for _, v := range allowed {
|
|
if size == v {
|
|
forbidden = false
|
|
break
|
|
}
|
|
}
|
|
}
|
|
if forbidden {
|
|
writeError(w, imgPath, method, "size is not allowed", http.StatusForbidden)
|
|
}
|
|
|
|
size := strings.Split(args, "x")
|
|
if len(size) < 2 {
|
|
writeError(w, imgPath, method, "url, width, and height are required", http.StatusBadRequest)
|
|
return
|
|
}
|
|
width := size[0]
|
|
height := size[1]
|
|
|
|
// Validate that all three required fields are present
|
|
if imgPath == "" || width == "" || height == "" {
|
|
writeError(w, imgPath, method, "url, width, and height are required", http.StatusBadRequest)
|
|
return
|
|
}
|
|
|
|
// Convert width and height to integers
|
|
iw, errW := strconv.Atoi(width)
|
|
ih, errH := strconv.Atoi(height)
|
|
if errW != nil || errH != nil {
|
|
writeError(w, imgPath, method, "width and height must be integers", http.StatusBadRequest)
|
|
return
|
|
}
|
|
|
|
// Ensure that a valid response was given
|
|
if resp.StatusCode/100 != 2 {
|
|
writeError(w, imgPath, method, fmt.Sprintf("failed to fetch image (%v)", resp.StatusCode), http.StatusBadRequest)
|
|
return
|
|
}
|
|
|
|
// govips returns the output of the image as a []byte object. We don't need
|
|
// it since we are directly piping it to the ResponseWriter
|
|
_, _, err = vips.NewTransform().
|
|
Load(resp.Body).
|
|
ResizeStrategy(vips.ResizeStrategyAuto).
|
|
Resize(iw, ih).
|
|
Output(w).
|
|
Apply()
|
|
|
|
if err != nil {
|
|
writeError(w, imgPath, method, "failed to resize", http.StatusInternalServerError)
|
|
return
|
|
}
|
|
case "thumbnail":
|
|
|
|
default:
|
|
writeError(w, imgPath, method, "unknown method", http.StatusBadRequest)
|
|
return
|
|
}
|
|
|
|
}
|
|
|
|
func writeError(w http.ResponseWriter, imgPath, method, err string, status int) {
|
|
w.WriteHeader(status)
|
|
w.Write([]byte(fmt.Sprintf("Error [%s %s]: %s", method, imgPath, err)))
|
|
}
|
|
|
|
func main() {
|
|
// Start vips with the default configuration
|
|
vips.Startup(nil)
|
|
defer vips.Shutdown()
|
|
|
|
http.HandleFunc("/", resizeHandler)
|
|
log.Fatal(http.ListenAndServe(*listenAddr, nil))
|
|
}
|