mirror of
https://github.com/J3vb/OwnCord.git
synced 2026-09-03 03:50:00 +03:00
J3vbandClaude Opus 4.8
4fc21cb372
Make server failures debuggable without leaking secrets: - configurable stdout log level (config.yaml logging.level + OWNCORD_LOGGING_LEVEL) - preserve the DB cause in ErrInternal wraps; log auth-DB failures distinctly from bad tokens; log the previously-silent expired-session cleanup goroutine - route HTTP handler panics through slog (was chi stderr-only, invisible to the admin log stream) - stackutil: argument-free panic stacks so key/token bytes never reach the admin ring buffer / SSE; slog.LogValuer redaction on VoiceConfig/GitHubConfig/ GIFConfig/Config and db.User/db.Session - logctx: req_id/trace_id correlation on ...Context log calls Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
75 lines
2.3 KiB
Go
75 lines
2.3 KiB
Go
package ws
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"log/slog"
|
|
|
|
"github.com/owncord/server/stackutil"
|
|
)
|
|
|
|
// handlerV2Entry pairs a V2 handler with its domain-specific dependency struct.
|
|
type handlerV2Entry struct {
|
|
handler HandlerV2
|
|
deps any // concrete deps struct for this handler's domain
|
|
}
|
|
|
|
// HandlerRegistry maps message type strings to their typed V2 handlers.
|
|
// It is not safe for concurrent use after initialization; all RegisterV2
|
|
// calls must happen before any DispatchV2 calls.
|
|
type HandlerRegistry struct {
|
|
handlersV2 map[string]handlerV2Entry
|
|
}
|
|
|
|
// NewHandlerRegistry creates an empty handler registry.
|
|
func NewHandlerRegistry() *HandlerRegistry {
|
|
return &HandlerRegistry{
|
|
handlersV2: make(map[string]handlerV2Entry),
|
|
}
|
|
}
|
|
|
|
// RegisterV2 registers a V2 handler for the given command type.
|
|
// PANICS if cmdType is already registered (duplicate guard).
|
|
func (r *HandlerRegistry) RegisterV2(cmdType string, handler HandlerV2, deps any) {
|
|
if _, exists := r.handlersV2[cmdType]; exists {
|
|
panic(fmt.Sprintf("RegisterV2: cmdType %q already registered", cmdType))
|
|
}
|
|
r.handlersV2[cmdType] = handlerV2Entry{handler: handler, deps: deps}
|
|
}
|
|
|
|
// DispatchV2 looks up a V2 handler and calls it.
|
|
// Returns (result, true) if found, (Result{}, false) if not.
|
|
// Recovers from panics (e.g. bad type assertions on cmd or deps) to prevent
|
|
// a single malformed message from crashing the entire server.
|
|
func (r *HandlerRegistry) DispatchV2(ctx context.Context, cmd Command, info ClientInfo) (result Result, ok bool) {
|
|
entry, found := r.handlersV2[cmd.Type()]
|
|
if !found {
|
|
return Result{}, false
|
|
}
|
|
defer func() {
|
|
if rec := recover(); rec != nil {
|
|
// stackutil.Capture omits argument values, which for E2EE
|
|
// handlers can include encrypted key material.
|
|
slog.Error("DispatchV2 panic recovered",
|
|
"type", cmd.Type(),
|
|
"user_id", info.UserID,
|
|
"panic", rec,
|
|
"stack", stackutil.Capture(),
|
|
)
|
|
result = Result{Error: ClientError{Code: ErrCodeInternal, Message: "internal error"}}
|
|
ok = true
|
|
}
|
|
}()
|
|
return entry.handler(ctx, cmd, info, entry.deps), true
|
|
}
|
|
|
|
// RegisteredV2Types returns all V2-registered message types (unordered).
|
|
// Intended for testing and diagnostics.
|
|
func (r *HandlerRegistry) RegisteredV2Types() []string {
|
|
types := make([]string, 0, len(r.handlersV2))
|
|
for t := range r.handlersV2 {
|
|
types = append(types, t)
|
|
}
|
|
return types
|
|
}
|