snapshot of https://github.com/ory/kratos.git (master, 2026-09-15; blobs > 131072B omitted)

continuity/container_reference_store.go

@@ -0,0 +1,58 @@
+// Copyright © 2023 Ory Corp
+// SPDX-License-Identifier: Apache-2.0
+
+package continuity
+
+import (
+	"context"
+	"net/http"
+
+	"github.com/gofrs/uuid"
+	"github.com/gorilla/sessions"
+	"github.com/pkg/errors"
+
+	"github.com/ory/kratos/x"
+)
+
+// ContainerReferenceStore abstracts where a continuity container's ID is
+// stored between Pause and Continue/Abort calls.
+type ContainerReferenceStore interface {
+	Store(ctx context.Context, w http.ResponseWriter, r *http.Request, name string, id uuid.UUID) error
+	Retrieve(ctx context.Context, w http.ResponseWriter, r *http.Request, name string) (uuid.UUID, error)
+	Clear(ctx context.Context, w http.ResponseWriter, r *http.Request, name string) error
+}
+
+// CookieReferenceStore stores the container ID in an HTTP cookie.
+type CookieReferenceStore struct {
+	cs sessions.StoreExact
+}
+
+func NewCookieReferenceStore(cs sessions.StoreExact) *CookieReferenceStore {
+	return &CookieReferenceStore{cs: cs}
+}
+
+func (s *CookieReferenceStore) Store(_ context.Context, w http.ResponseWriter, r *http.Request, name string, id uuid.UUID) error {
+	return x.SessionPersistValues(w, r, s.cs, CookieName, map[string]any{
+		name: id.String(),
+	})
+}
+
+func (s *CookieReferenceStore) Retrieve(_ context.Context, w http.ResponseWriter, r *http.Request, name string) (uuid.UUID, error) {
+	str, err := x.SessionGetString(r, s.cs, CookieName, name)
+	if err != nil {
+		_ = x.SessionUnsetKey(w, r, s.cs, CookieName, name)
+		return uuid.Nil, errors.WithStack(ErrNotResumable().WithDebugf("%+v", err))
+	}
+
+	sid, err := uuid.FromString(str)
+	if err != nil {
+		_ = x.SessionUnsetKey(w, r, s.cs, CookieName, name)
+		return uuid.Nil, errors.WithStack(ErrNotResumable().WithDebug("session id is not a valid uuid"))
+	}
+
+	return sid, nil
+}
+
+func (s *CookieReferenceStore) Clear(_ context.Context, w http.ResponseWriter, r *http.Request, name string) error {
+	return x.SessionUnsetKey(w, r, s.cs, CookieName, name)
+}