mirror of https://github.com/muety/wakapi.git
53 lines
1.3 KiB
Go
53 lines
1.3 KiB
Go
package utils
|
|
|
|
import (
|
|
"github.com/muety/wakapi/helpers"
|
|
"github.com/muety/wakapi/models"
|
|
"github.com/muety/wakapi/models/types"
|
|
"github.com/muety/wakapi/services"
|
|
"net/http"
|
|
"strings"
|
|
)
|
|
|
|
func LoadUserSummary(ss services.ISummaryService, r *http.Request) (*models.Summary, error, int) {
|
|
summaryParams, err := helpers.ParseSummaryParams(r)
|
|
if err != nil {
|
|
return nil, err, http.StatusBadRequest
|
|
}
|
|
return LoadUserSummaryByParams(ss, summaryParams)
|
|
}
|
|
|
|
func LoadUserSummaryByParams(ss services.ISummaryService, params *models.SummaryParams) (*models.Summary, error, int) {
|
|
var retrieveSummary types.SummaryRetriever = ss.Retrieve
|
|
if params.Recompute {
|
|
retrieveSummary = ss.Summarize
|
|
}
|
|
|
|
summary, err := ss.Aliased(
|
|
params.From,
|
|
params.To,
|
|
params.User,
|
|
retrieveSummary,
|
|
params.Filters,
|
|
params.Recompute,
|
|
)
|
|
if err != nil {
|
|
return nil, err, http.StatusInternalServerError
|
|
}
|
|
|
|
summary.FromTime = models.CustomTime(summary.FromTime.T().In(params.User.TZ()))
|
|
summary.ToTime = models.CustomTime(summary.ToTime.T().In(params.User.TZ()))
|
|
|
|
return summary, nil, http.StatusOK
|
|
}
|
|
|
|
func FilterColors(all map[string]string, haystack models.SummaryItems) map[string]string {
|
|
subset := make(map[string]string)
|
|
for _, item := range haystack {
|
|
if c, ok := all[strings.ToLower(item.Key)]; ok {
|
|
subset[strings.ToLower(item.Key)] = c
|
|
}
|
|
}
|
|
return subset
|
|
}
|