mirror of
https://github.com/muety/wakapi.git
synced 2023-08-10 21:12:56 +03:00
52 lines
1.1 KiB
Go
52 lines
1.1 KiB
Go
package api
|
|
|
|
import (
|
|
"codeberg.org/Codeberg/avatars"
|
|
"github.com/gorilla/mux"
|
|
lru "github.com/hashicorp/golang-lru"
|
|
conf "github.com/muety/wakapi/config"
|
|
"github.com/muety/wakapi/utils"
|
|
"net/http"
|
|
"time"
|
|
)
|
|
|
|
type AvatarHandler struct {
|
|
config *conf.Config
|
|
cache *lru.Cache
|
|
}
|
|
|
|
func NewAvatarHandler() *AvatarHandler {
|
|
cache, err := lru.New(1 * 1000 * 64) // assuming an avatar is 1 kb, allocate up to 64 mb of memory for avatars cache
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
return &AvatarHandler{
|
|
config: conf.Get(),
|
|
cache: cache,
|
|
}
|
|
}
|
|
|
|
func (h *AvatarHandler) RegisterRoutes(router *mux.Router) {
|
|
r := router.PathPrefix("/avatar/{hash}.svg").Subrouter()
|
|
r.Path("").Methods(http.MethodGet).HandlerFunc(h.Get)
|
|
}
|
|
|
|
func (h *AvatarHandler) Get(w http.ResponseWriter, r *http.Request) {
|
|
hash := mux.Vars(r)["hash"]
|
|
|
|
if utils.IsNoCache(r, 1*time.Hour) {
|
|
h.cache.Remove(hash)
|
|
}
|
|
|
|
if !h.cache.Contains(hash) {
|
|
h.cache.Add(hash, avatars.MakeAvatar(hash))
|
|
}
|
|
data, _ := h.cache.Get(hash)
|
|
|
|
w.Header().Set("Content-Type", "image/svg+xml")
|
|
w.Header().Set("Cache-Control", "max-age=2592000")
|
|
w.WriteHeader(http.StatusOK)
|
|
w.Write([]byte(data.(string)))
|
|
}
|