mirror of
https://github.com/muety/wakapi.git
synced 2023-08-10 21:12:56 +03:00
99 lines
2.6 KiB
Go
99 lines
2.6 KiB
Go
package api
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/duke-git/lancet/v2/maputil"
|
|
"github.com/duke-git/lancet/v2/slice"
|
|
"github.com/gorilla/mux"
|
|
conf "github.com/muety/wakapi/config"
|
|
"github.com/muety/wakapi/models"
|
|
v1 "github.com/muety/wakapi/models/compat/shields/v1"
|
|
routeutils "github.com/muety/wakapi/routes/utils"
|
|
"github.com/muety/wakapi/services"
|
|
"github.com/muety/wakapi/utils"
|
|
"github.com/narqo/go-badge"
|
|
"github.com/patrickmn/go-cache"
|
|
"net/http"
|
|
"time"
|
|
)
|
|
|
|
type BadgeHandler struct {
|
|
config *conf.Config
|
|
cache *cache.Cache
|
|
userSrvc services.IUserService
|
|
summarySrvc services.ISummaryService
|
|
}
|
|
|
|
func NewBadgeHandler(userService services.IUserService, summaryService services.ISummaryService) *BadgeHandler {
|
|
return &BadgeHandler{
|
|
config: conf.Get(),
|
|
cache: cache.New(time.Hour, time.Hour),
|
|
userSrvc: userService,
|
|
summarySrvc: summaryService,
|
|
}
|
|
}
|
|
|
|
func (h *BadgeHandler) RegisterRoutes(router *mux.Router) {
|
|
r := router.PathPrefix("/badge/{user}").Subrouter()
|
|
r.Methods(http.MethodGet).HandlerFunc(h.Get)
|
|
}
|
|
|
|
func (h *BadgeHandler) Get(w http.ResponseWriter, r *http.Request) {
|
|
user, err := h.userSrvc.GetUserById(mux.Vars(r)["user"])
|
|
if err != nil {
|
|
w.WriteHeader(http.StatusNotFound)
|
|
return
|
|
}
|
|
|
|
interval, filters, err := routeutils.GetBadgeParams(r, user)
|
|
if err != nil {
|
|
w.WriteHeader(http.StatusForbidden)
|
|
w.Write([]byte(err.Error()))
|
|
return
|
|
}
|
|
|
|
cacheKey := fmt.Sprintf("%s_%v_%s", user.ID, *interval.Key, filters.Hash())
|
|
noCache := utils.IsNoCache(r, 1*time.Hour)
|
|
if cacheResult, ok := h.cache.Get(cacheKey); ok && !noCache {
|
|
respondSvg(w, cacheResult.([]byte))
|
|
return
|
|
}
|
|
|
|
params := &models.SummaryParams{
|
|
From: interval.Start,
|
|
To: interval.End,
|
|
User: user,
|
|
Filters: filters,
|
|
}
|
|
|
|
summary, err, status := routeutils.LoadUserSummaryByParams(h.summarySrvc, params)
|
|
if err != nil {
|
|
w.WriteHeader(status)
|
|
w.Write([]byte(err.Error()))
|
|
return
|
|
}
|
|
|
|
badgeData := v1.NewBadgeDataFrom(summary)
|
|
if customLabel := r.URL.Query().Get("label"); customLabel != "" {
|
|
badgeData.Label = customLabel
|
|
}
|
|
if customColor := r.URL.Query().Get("color"); customColor != "" {
|
|
badgeData.Color = customColor
|
|
}
|
|
|
|
if badgeData.Color[0:1] != "#" && !slice.Contain(maputil.Keys(badge.ColorScheme), badgeData.Color) {
|
|
badgeData.Color = "#" + badgeData.Color
|
|
}
|
|
|
|
badgeSvg, err := badge.RenderBytes(badgeData.Label, badgeData.Message, badge.Color(badgeData.Color))
|
|
h.cache.SetDefault(cacheKey, badgeSvg)
|
|
respondSvg(w, badgeSvg)
|
|
}
|
|
|
|
func respondSvg(w http.ResponseWriter, data []byte) {
|
|
w.Header().Set("Content-Type", "image/svg+xml")
|
|
w.Header().Set("Cache-Control", "max-age=3600")
|
|
w.WriteHeader(http.StatusOK)
|
|
w.Write(data)
|
|
}
|