mirror of https://github.com/velour/catbase.git
27 lines
490 B
Go
27 lines
490 B
Go
|
package cowboy
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"net/http"
|
||
|
|
|
||
|
|
"github.com/go-chi/chi/v5"
|
||
|
|
)
|
||
|
|
|
||
|
|
func (p *Cowboy) registerWeb() {
|
||
|
|
r := chi.NewRouter()
|
||
|
|
r.HandleFunc("/img/{what}", p.handleImage)
|
||
|
|
p.b.RegisterWeb(r, "/cowboy")
|
||
|
|
}
|
||
|
|
|
||
|
|
func (p *Cowboy) handleImage(w http.ResponseWriter, r *http.Request) {
|
||
|
|
what := chi.URLParam(r, "what")
|
||
|
|
img, err := cowboy(p.c, what)
|
||
|
|
if err != nil {
|
||
|
|
w.WriteHeader(500)
|
||
|
|
fmt.Fprintf(w, "Error: %s", err)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
w.Header().Add("Content-Type", "image/png")
|
||
|
|
w.Write(img)
|
||
|
|
}
|