feat(photos): добавить пресеты кривой и починить переключение фото
- PCHIP для предпросмотра и серверного рендера - пресеты всех каналов в профиле, импорт и экспорт - мерж настроек auth без потери разделов - сохранение вкладки и правок текущего фото при переключении
This commit is contained in:
@@ -4,7 +4,10 @@ import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"io"
|
||||
"math"
|
||||
"net/http"
|
||||
"strings"
|
||||
"unicode/utf8"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
|
||||
@@ -181,6 +184,9 @@ const (
|
||||
maxLabelEntries = 500
|
||||
maxPreferenceKeyLen = 128
|
||||
maxPreferenceValueLen = 128
|
||||
maxCurvePresets = 100
|
||||
maxCurvePresetNameLen = 64
|
||||
maxCurvePresetPoints = 16
|
||||
)
|
||||
|
||||
type photoInfoPreferences struct {
|
||||
@@ -188,33 +194,102 @@ type photoInfoPreferences struct {
|
||||
Labels map[string]string `json:"labels,omitempty"`
|
||||
}
|
||||
|
||||
type curvePresetPoint struct {
|
||||
X float64 `json:"x"`
|
||||
Y float64 `json:"y"`
|
||||
}
|
||||
|
||||
type curvePresetChannels struct {
|
||||
RGB []curvePresetPoint `json:"rgb"`
|
||||
Red []curvePresetPoint `json:"red"`
|
||||
Green []curvePresetPoint `json:"green"`
|
||||
Blue []curvePresetPoint `json:"blue"`
|
||||
}
|
||||
|
||||
type curvePreset struct {
|
||||
Name string `json:"name"`
|
||||
Channels curvePresetChannels `json:"channels"`
|
||||
}
|
||||
|
||||
type preferencesPayload struct {
|
||||
PhotoInfo *photoInfoPreferences `json:"photo_info,omitempty"`
|
||||
PhotoInfo *photoInfoPreferences `json:"photo_info,omitempty"`
|
||||
CurvePresets *[]curvePreset `json:"curve_presets,omitempty"`
|
||||
}
|
||||
|
||||
// normalize trims preset names before they are validated and stored.
|
||||
func (p *preferencesPayload) normalize() {
|
||||
if p.CurvePresets == nil {
|
||||
return
|
||||
}
|
||||
for i := range *p.CurvePresets {
|
||||
(*p.CurvePresets)[i].Name = strings.TrimSpace((*p.CurvePresets)[i].Name)
|
||||
}
|
||||
}
|
||||
|
||||
func (p preferencesPayload) validate() error {
|
||||
pi := p.PhotoInfo
|
||||
if pi == nil {
|
||||
return nil
|
||||
}
|
||||
if len(pi.Visible) > maxVisibleAttrs {
|
||||
return fmt.Errorf("too many visible attributes")
|
||||
}
|
||||
for _, k := range pi.Visible {
|
||||
if k == "" || len(k) > maxPreferenceKeyLen {
|
||||
return fmt.Errorf("invalid visible attribute key")
|
||||
if pi != nil {
|
||||
if len(pi.Visible) > maxVisibleAttrs {
|
||||
return fmt.Errorf("too many visible attributes")
|
||||
}
|
||||
for _, k := range pi.Visible {
|
||||
if k == "" || len(k) > maxPreferenceKeyLen {
|
||||
return fmt.Errorf("invalid visible attribute key")
|
||||
}
|
||||
}
|
||||
if len(pi.Labels) > maxLabelEntries {
|
||||
return fmt.Errorf("too many label overrides")
|
||||
}
|
||||
for k, v := range pi.Labels {
|
||||
if k == "" || len(k) > maxPreferenceKeyLen {
|
||||
return fmt.Errorf("invalid label key")
|
||||
}
|
||||
if len(v) > maxPreferenceValueLen {
|
||||
return fmt.Errorf("label value too long")
|
||||
}
|
||||
}
|
||||
}
|
||||
if len(pi.Labels) > maxLabelEntries {
|
||||
return fmt.Errorf("too many label overrides")
|
||||
if p.CurvePresets != nil {
|
||||
return validateCurvePresets(*p.CurvePresets)
|
||||
}
|
||||
for k, v := range pi.Labels {
|
||||
if k == "" || len(k) > maxPreferenceKeyLen {
|
||||
return fmt.Errorf("invalid label key")
|
||||
return nil
|
||||
}
|
||||
|
||||
func validateCurvePresets(presets []curvePreset) error {
|
||||
if len(presets) > maxCurvePresets {
|
||||
return fmt.Errorf("too many curve presets")
|
||||
}
|
||||
names := make(map[string]struct{}, len(presets))
|
||||
for _, preset := range presets {
|
||||
if preset.Name == "" || utf8.RuneCountInString(preset.Name) > maxCurvePresetNameLen {
|
||||
return fmt.Errorf("invalid curve preset name")
|
||||
}
|
||||
if len(v) > maxPreferenceValueLen {
|
||||
return fmt.Errorf("label value too long")
|
||||
channels := []struct {
|
||||
name string
|
||||
points []curvePresetPoint
|
||||
}{
|
||||
{"rgb", preset.Channels.RGB},
|
||||
{"red", preset.Channels.Red},
|
||||
{"green", preset.Channels.Green},
|
||||
{"blue", preset.Channels.Blue},
|
||||
}
|
||||
for _, channel := range channels {
|
||||
if len(channel.points) < 2 || len(channel.points) > maxCurvePresetPoints {
|
||||
return fmt.Errorf("invalid curve preset points for %s", channel.name)
|
||||
}
|
||||
for _, pt := range channel.points {
|
||||
if math.IsNaN(pt.X) || math.IsNaN(pt.Y) || math.IsInf(pt.X, 0) || math.IsInf(pt.Y, 0) {
|
||||
return fmt.Errorf("invalid curve preset point")
|
||||
}
|
||||
if pt.X < 0 || pt.X > 1 || pt.Y < 0 || pt.Y > 1 {
|
||||
return fmt.Errorf("invalid curve preset point")
|
||||
}
|
||||
}
|
||||
}
|
||||
if _, dup := names[preset.Name]; dup {
|
||||
return fmt.Errorf("duplicate curve preset name")
|
||||
}
|
||||
names[preset.Name] = struct{}{}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
@@ -230,18 +305,40 @@ func (h *Handler) UpdatePreferences(c *gin.Context) {
|
||||
httpx.RespondError(c, httpx.BadRequest("invalid_preferences", "preferences must be a JSON object"))
|
||||
return
|
||||
}
|
||||
payload.normalize()
|
||||
if err := payload.validate(); err != nil {
|
||||
httpx.RespondError(c, httpx.BadRequest("invalid_preferences", err.Error()))
|
||||
return
|
||||
}
|
||||
raw, err := json.Marshal(payload)
|
||||
|
||||
ctx := c.Request.Context()
|
||||
u, err := h.svc.store.GetUserByID(ctx, UserID(c))
|
||||
if err != nil {
|
||||
httpx.RespondError(c, httpx.BadRequest("invalid_preferences", "could not encode preferences"))
|
||||
httpx.RespondError(c, httpx.Unauthorized("unauthorized", "user not found"))
|
||||
return
|
||||
}
|
||||
if err := h.svc.store.UpdatePreferences(c.Request.Context(), UserID(c), raw); err != nil {
|
||||
httpx.RespondError(c, err)
|
||||
return
|
||||
// Merge the submitted sections into the stored blob so saving one section
|
||||
// never clears the others. Unknown keys are preserved.
|
||||
merged := make(map[string]any, len(u.Preferences)+2)
|
||||
for k, v := range u.Preferences {
|
||||
merged[k] = v
|
||||
}
|
||||
c.JSON(http.StatusOK, gin.H{"preferences": payload})
|
||||
if payload.PhotoInfo != nil {
|
||||
merged["photo_info"] = payload.PhotoInfo
|
||||
}
|
||||
if payload.CurvePresets != nil {
|
||||
merged["curve_presets"] = *payload.CurvePresets
|
||||
}
|
||||
if payload.PhotoInfo != nil || payload.CurvePresets != nil {
|
||||
raw, err := json.Marshal(merged)
|
||||
if err != nil {
|
||||
httpx.RespondError(c, httpx.BadRequest("invalid_preferences", "could not encode preferences"))
|
||||
return
|
||||
}
|
||||
if err := h.svc.store.UpdatePreferences(ctx, UserID(c), raw); err != nil {
|
||||
httpx.RespondError(c, err)
|
||||
return
|
||||
}
|
||||
}
|
||||
c.JSON(http.StatusOK, gin.H{"preferences": merged})
|
||||
}
|
||||
|
||||
@@ -5,6 +5,7 @@ import (
|
||||
"encoding/json"
|
||||
"net/http"
|
||||
"net/http/httptest"
|
||||
"strings"
|
||||
"testing"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
@@ -30,6 +31,7 @@ func newTestRouter(t *testing.T) *gin.Engine {
|
||||
authed.PATCH("/auth/me", h.UpdateMe)
|
||||
authed.POST("/auth/me/password", h.ChangePassword)
|
||||
authed.POST("/auth/me/identifiers", h.AddIdentifier)
|
||||
authed.PATCH("/auth/me/preferences", h.UpdatePreferences)
|
||||
authed.DELETE("/auth/me", h.DeleteMe)
|
||||
return r
|
||||
}
|
||||
@@ -213,3 +215,215 @@ func TestDeleteMeHandler(t *testing.T) {
|
||||
t.Fatalf("me after delete status = %d", rec.Code)
|
||||
}
|
||||
}
|
||||
|
||||
func curvePresetPayloadWith(name, channel string, points []map[string]float64) map[string]any {
|
||||
linear := func() []map[string]float64 {
|
||||
return []map[string]float64{
|
||||
{"x": 0, "y": 0},
|
||||
{"x": 1, "y": 1},
|
||||
}
|
||||
}
|
||||
channels := map[string]any{
|
||||
"rgb": linear(),
|
||||
"red": linear(),
|
||||
"green": linear(),
|
||||
"blue": linear(),
|
||||
}
|
||||
channels[channel] = points
|
||||
return map[string]any{"name": name, "channels": channels}
|
||||
}
|
||||
|
||||
func curvePresetPayload(name string) map[string]any {
|
||||
return curvePresetPayloadWith(name, "rgb", []map[string]float64{
|
||||
{"x": 0, "y": 0},
|
||||
{"x": 1, "y": 0.8},
|
||||
})
|
||||
}
|
||||
|
||||
func decodePreferences(t *testing.T, rec *httptest.ResponseRecorder) map[string]any {
|
||||
t.Helper()
|
||||
var body struct {
|
||||
Preferences map[string]any `json:"preferences"`
|
||||
}
|
||||
if err := json.Unmarshal(rec.Body.Bytes(), &body); err != nil {
|
||||
t.Fatalf("decode preferences %s: %v", rec.Body.String(), err)
|
||||
}
|
||||
return body.Preferences
|
||||
}
|
||||
|
||||
func TestUpdatePreferencesMergesSections(t *testing.T) {
|
||||
r := newTestRouter(t)
|
||||
reg := doJSON(r, http.MethodPost, "/auth/register", "", map[string]string{
|
||||
"email": "prefs@test.local", "password": "password123", "name": "Prefs",
|
||||
})
|
||||
token := decodeAuth(t, reg).Tokens.AccessToken
|
||||
|
||||
rec := doJSON(r, http.MethodPatch, "/auth/me/preferences", token, map[string]any{
|
||||
"photo_info": map[string]any{
|
||||
"visible": []string{"camera"},
|
||||
"labels": map[string]string{"camera": "Камера"},
|
||||
},
|
||||
})
|
||||
if rec.Code != http.StatusOK {
|
||||
t.Fatalf("save photo_info status = %d body = %s", rec.Code, rec.Body.String())
|
||||
}
|
||||
|
||||
rec = doJSON(r, http.MethodPatch, "/auth/me/preferences", token, map[string]any{
|
||||
"curve_presets": []any{curvePresetPayload("Мой")},
|
||||
})
|
||||
if rec.Code != http.StatusOK {
|
||||
t.Fatalf("save presets status = %d body = %s", rec.Code, rec.Body.String())
|
||||
}
|
||||
prefs := decodePreferences(t, rec)
|
||||
if _, ok := prefs["photo_info"]; !ok {
|
||||
t.Fatalf("photo_info lost after saving presets: %+v", prefs)
|
||||
}
|
||||
presets, ok := prefs["curve_presets"].([]any)
|
||||
if !ok || len(presets) != 1 {
|
||||
t.Fatalf("presets = %#v, want one", prefs["curve_presets"])
|
||||
}
|
||||
|
||||
// Saving photo_info again must keep the presets.
|
||||
rec = doJSON(r, http.MethodPatch, "/auth/me/preferences", token, map[string]any{
|
||||
"photo_info": map[string]any{"visible": []string{"iso"}},
|
||||
})
|
||||
prefs = decodePreferences(t, rec)
|
||||
if presets, ok := prefs["curve_presets"].([]any); !ok || len(presets) != 1 {
|
||||
t.Fatalf("presets lost after saving photo_info: %+v", prefs)
|
||||
}
|
||||
info, _ := prefs["photo_info"].(map[string]any)
|
||||
if visible, _ := info["visible"].([]any); len(visible) != 1 || visible[0] != "iso" {
|
||||
t.Fatalf("photo_info not updated: %+v", prefs["photo_info"])
|
||||
}
|
||||
|
||||
// The merged state is what GET /auth/me returns.
|
||||
rec = doJSON(r, http.MethodGet, "/auth/me", token, nil)
|
||||
var me struct {
|
||||
Preferences map[string]any `json:"preferences"`
|
||||
}
|
||||
if err := json.Unmarshal(rec.Body.Bytes(), &me); err != nil {
|
||||
t.Fatalf("decode me: %v", err)
|
||||
}
|
||||
if _, ok := me.Preferences["photo_info"]; !ok {
|
||||
t.Fatalf("me lost photo_info: %+v", me.Preferences)
|
||||
}
|
||||
if _, ok := me.Preferences["curve_presets"]; !ok {
|
||||
t.Fatalf("me lost curve_presets: %+v", me.Preferences)
|
||||
}
|
||||
|
||||
// An empty preset list clears only presets.
|
||||
rec = doJSON(r, http.MethodPatch, "/auth/me/preferences", token, map[string]any{
|
||||
"curve_presets": []any{},
|
||||
})
|
||||
prefs = decodePreferences(t, rec)
|
||||
if presets, ok := prefs["curve_presets"].([]any); !ok || len(presets) != 0 {
|
||||
t.Fatalf("presets not cleared: %+v", prefs["curve_presets"])
|
||||
}
|
||||
if _, ok := prefs["photo_info"]; !ok {
|
||||
t.Fatalf("photo_info lost after clearing presets: %+v", prefs)
|
||||
}
|
||||
|
||||
// Names are stored without surrounding whitespace.
|
||||
rec = doJSON(r, http.MethodPatch, "/auth/me/preferences", token, map[string]any{
|
||||
"curve_presets": []any{curvePresetPayload(" Мой ")},
|
||||
})
|
||||
prefs = decodePreferences(t, rec)
|
||||
presets = prefs["curve_presets"].([]any)
|
||||
first, _ := presets[0].(map[string]any)
|
||||
if first["name"] != "Мой" {
|
||||
t.Fatalf("preset name = %#v, want trimmed", first["name"])
|
||||
}
|
||||
}
|
||||
|
||||
func TestPreferencesArePrivate(t *testing.T) {
|
||||
r := newTestRouter(t)
|
||||
first := decodeAuth(t, doJSON(r, http.MethodPost, "/auth/register", "", map[string]string{
|
||||
"email": "private1@test.local", "password": "password123", "name": "One",
|
||||
}))
|
||||
second := decodeAuth(t, doJSON(r, http.MethodPost, "/auth/register", "", map[string]string{
|
||||
"email": "private2@test.local", "password": "password123", "name": "Two",
|
||||
}))
|
||||
|
||||
rec := doJSON(r, http.MethodPatch, "/auth/me/preferences", first.Tokens.AccessToken, map[string]any{
|
||||
"curve_presets": []any{curvePresetPayload("Секрет")},
|
||||
})
|
||||
if rec.Code != http.StatusOK {
|
||||
t.Fatalf("save status = %d body = %s", rec.Code, rec.Body.String())
|
||||
}
|
||||
|
||||
rec = doJSON(r, http.MethodGet, "/auth/me", second.Tokens.AccessToken, nil)
|
||||
var me struct {
|
||||
Preferences map[string]any `json:"preferences"`
|
||||
}
|
||||
if err := json.Unmarshal(rec.Body.Bytes(), &me); err != nil {
|
||||
t.Fatalf("decode me: %v", err)
|
||||
}
|
||||
if _, ok := me.Preferences["curve_presets"]; ok {
|
||||
t.Fatalf("other user sees presets: %+v", me.Preferences)
|
||||
}
|
||||
}
|
||||
|
||||
func TestUpdatePreferencesRejectsInvalidCurvePresets(t *testing.T) {
|
||||
r := newTestRouter(t)
|
||||
reg := doJSON(r, http.MethodPost, "/auth/register", "", map[string]string{
|
||||
"email": "invalidprefs@test.local", "password": "password123", "name": "Invalid",
|
||||
})
|
||||
token := decodeAuth(t, reg).Tokens.AccessToken
|
||||
|
||||
rec := doJSON(r, http.MethodPatch, "/auth/me/preferences", token, map[string]any{
|
||||
"curve_presets": []any{curvePresetPayload("Хороший")},
|
||||
})
|
||||
if rec.Code != http.StatusOK {
|
||||
t.Fatalf("seed status = %d body = %s", rec.Code, rec.Body.String())
|
||||
}
|
||||
|
||||
manyPoints := make([]map[string]float64, 17)
|
||||
for i := range manyPoints {
|
||||
manyPoints[i] = map[string]float64{"x": float64(i) / 16, "y": float64(i) / 16}
|
||||
}
|
||||
tooMany := make([]any, 101)
|
||||
for i := range tooMany {
|
||||
tooMany[i] = curvePresetPayload("preset-" + string(rune('a'+i%26)) + strings.Repeat("x", i/26))
|
||||
}
|
||||
|
||||
missingChannel := curvePresetPayload("bad")
|
||||
delete(missingChannel["channels"].(map[string]any), "blue")
|
||||
|
||||
cases := []struct {
|
||||
name string
|
||||
presets []any
|
||||
}{
|
||||
{"too long name", []any{curvePresetPayload(strings.Repeat("a", 65))}},
|
||||
{"missing channel", []any{missingChannel}},
|
||||
{"one point", []any{curvePresetPayloadWith("one", "red", []map[string]float64{
|
||||
{"x": 0, "y": 0},
|
||||
})}},
|
||||
{"too many points", []any{curvePresetPayloadWith("many", "rgb", manyPoints)}},
|
||||
{"x out of range", []any{curvePresetPayloadWith("range", "blue", []map[string]float64{
|
||||
{"x": 1.1, "y": 0},
|
||||
{"x": 1, "y": 1},
|
||||
})}},
|
||||
{"duplicate name", []any{curvePresetPayload("dup"), curvePresetPayload("dup")}},
|
||||
{"too many presets", tooMany},
|
||||
}
|
||||
for _, tc := range cases {
|
||||
rec := doJSON(r, http.MethodPatch, "/auth/me/preferences", token, map[string]any{
|
||||
"curve_presets": tc.presets,
|
||||
})
|
||||
if rec.Code != http.StatusBadRequest {
|
||||
t.Fatalf("%s status = %d body = %s", tc.name, rec.Code, rec.Body.String())
|
||||
}
|
||||
}
|
||||
|
||||
rec = doJSON(r, http.MethodGet, "/auth/me", token, nil)
|
||||
var me struct {
|
||||
Preferences map[string]any `json:"preferences"`
|
||||
}
|
||||
if err := json.Unmarshal(rec.Body.Bytes(), &me); err != nil {
|
||||
t.Fatalf("decode me: %v", err)
|
||||
}
|
||||
presets, _ := me.Preferences["curve_presets"].([]any)
|
||||
if len(presets) != 1 {
|
||||
t.Fatalf("invalid requests changed presets: %+v", me.Preferences)
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user