mirror of
https://github.com/h2non/imaginary.git
synced 2026-03-05 18:23:37 +01:00
62 lines
1.4 KiB
Go
62 lines
1.4 KiB
Go
package main
|
|
|
|
import (
|
|
"github.com/PuerkitoBio/throttled"
|
|
"net/http"
|
|
"os"
|
|
"strconv"
|
|
"time"
|
|
)
|
|
|
|
type ServerOptions struct {
|
|
Port int
|
|
CORS bool
|
|
Gzip bool
|
|
Address string
|
|
ApiKey string
|
|
Burst int
|
|
Concurrency int
|
|
}
|
|
|
|
func Server(o ServerOptions) error {
|
|
addr := o.Address + ":" + strconv.Itoa(o.Port)
|
|
handler := NewLog(NewServerMux(o), os.Stdout)
|
|
|
|
if o.Concurrency > 0 {
|
|
th := throttled.Interval(throttled.PerSec(o.Concurrency), o.Burst, &throttled.VaryBy{Method: true}, o.Burst)
|
|
handler = th.Throttle(handler)
|
|
}
|
|
|
|
server := &http.Server{
|
|
Addr: addr,
|
|
Handler: handler,
|
|
ReadTimeout: 60 * time.Second,
|
|
WriteTimeout: 60 * time.Second,
|
|
MaxHeaderBytes: 1 << 20,
|
|
}
|
|
|
|
return server.ListenAndServe()
|
|
}
|
|
|
|
func NewServerMux(o ServerOptions) http.Handler {
|
|
mux := http.NewServeMux()
|
|
|
|
image := ImageMiddleware(o)
|
|
mux.Handle("/", Middleware(indexController, o))
|
|
mux.Handle("/form", Middleware(formController, o))
|
|
mux.Handle("/resize", image(Resize))
|
|
mux.Handle("/enlarge", image(Enlarge))
|
|
mux.Handle("/extract", image(Extract))
|
|
mux.Handle("/crop", image(Crop))
|
|
mux.Handle("/rotate", image(Rotate))
|
|
mux.Handle("/flip", image(Flip))
|
|
mux.Handle("/flop", image(Flop))
|
|
mux.Handle("/thumbnail", image(Thumbnail))
|
|
mux.Handle("/zoom", image(Zoom))
|
|
mux.Handle("/convert", image(Convert))
|
|
mux.Handle("/watermark", image(Watermark))
|
|
mux.Handle("/info", image(Info))
|
|
|
|
return mux
|
|
}
|