forked from wgroeneveld/go-jamming
48 lines
1.1 KiB
Go
48 lines
1.1 KiB
Go
package pictures
|
|
|
|
import (
|
|
"brainbaking.com/go-jamming/app/mf"
|
|
"brainbaking.com/go-jamming/common"
|
|
"brainbaking.com/go-jamming/db"
|
|
"brainbaking.com/go-jamming/rest"
|
|
_ "embed"
|
|
"github.com/gorilla/mux"
|
|
"github.com/rs/zerolog/log"
|
|
"net/http"
|
|
)
|
|
|
|
//go:embed anonymous.jpg
|
|
var anonymous []byte
|
|
|
|
func init() {
|
|
if anonymous == nil {
|
|
log.Fatal().Msg("embedded anonymous image missing?")
|
|
}
|
|
}
|
|
|
|
// Handle handles picture GET calls.
|
|
// It does not validate the picture query as it's part of a composite key anyway.
|
|
func Handle(repo db.MentionRepo) http.HandlerFunc {
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
|
picDomain := mux.Vars(r)["picture"]
|
|
if picDomain == mf.Anonymous || common.Includes(rest.SiloDomains, picDomain) {
|
|
servePicture(w, anonymous)
|
|
return
|
|
}
|
|
|
|
picData := repo.GetPicture(picDomain)
|
|
if picData == nil {
|
|
http.NotFound(w, r)
|
|
return
|
|
}
|
|
servePicture(w, picData)
|
|
}
|
|
}
|
|
|
|
// servePicture writes an OK and raw bytes.
|
|
// For some reason, headers - although they should be there - aren't needed?
|
|
func servePicture(w http.ResponseWriter, bytes []byte) {
|
|
w.WriteHeader(http.StatusOK)
|
|
w.Write(bytes)
|
|
}
|