snapshot of https://github.com/go-gitea/gitea.git (main, 2026-09-15; blobs > 131072B omitted)

services/websocket/events.go

@@ -0,0 +1,53 @@
+// Copyright 2026 The Gitea Authors. All rights reserved.
+// SPDX-License-Identifier: MIT
+
+package websocket
+
+import (
+	"bytes"
+
+	"gitea.dev/modules/json"
+	"gitea.dev/modules/setting"
+	"gitea.dev/services/pubsub"
+)
+
+// Wire contract with web_src/js/user-events.sharedworker.ts — keep in sync.
+const (
+	EventNotificationCount = "notification-count"
+	EventStopwatches       = "stopwatches"
+	EventLogout            = "logout"
+)
+
+type UserEventMessage[T any] struct {
+	EventType string `json:"eventType"`
+	EventData T      `json:"eventData"`
+}
+
+func publishUserEvent(userID int64, eventType string, eventData any) {
+	if pubsub.DefaultBroker == nil {
+		return
+	}
+	b := MakeUserEventMessage(eventType, eventData)
+	if b == nil {
+		return
+	}
+	pubsub.DefaultBroker.Publish(pubsub.UserTopic(userID), b)
+}
+
+func MakeUserEventMessage(eventType string, eventData any) []byte {
+	buf := &bytes.Buffer{}
+	buf.WriteString(eventType)
+	buf.WriteByte('\n')
+	err := json.MarshalWrite(buf, &UserEventMessage[any]{EventType: eventType, EventData: eventData})
+	payloadBytes := bytes.TrimSuffix(buf.Bytes(), []byte("\n")) // json v1 adds extra "\n" but we don't want it
+	if err != nil {
+		setting.PanicInDevOrTesting("websocket: marshal event: %v", err)
+		return nil
+	}
+	return payloadBytes
+}
+
+func ExtractUserEventMessage(b []byte) (string, []byte) {
+	eventType, eventDataBytes, _ := bytes.Cut(b, []byte("\n"))
+	return string(eventType), eventDataBytes
+}

services/websocket/logout_publisher.go

@@ -0,0 +1,12 @@
+// Copyright 2026 The Gitea Authors. All rights reserved.
+// SPDX-License-Identifier: MIT
+
+package websocket
+
+type LogoutEventData struct {
+	SessionID string `json:"sessionID,omitempty"`
+}
+
+func PublishLogout(userID int64, sessionID string) {
+	publishUserEvent(userID, EventLogout, LogoutEventData{SessionID: sessionID})
+}

services/websocket/notification_notifier.go

@@ -0,0 +1,39 @@
+// Copyright 2026 The Gitea Authors. All rights reserved.
+// SPDX-License-Identifier: MIT
+
+package websocket
+
+import (
+	"context"
+
+	activities_model "gitea.dev/models/activities"
+	"gitea.dev/models/db"
+	"gitea.dev/modules/log"
+	notify_service "gitea.dev/services/notify"
+	"gitea.dev/services/pubsub"
+)
+
+type notificationCountEventData struct {
+	Count int64 `json:"count"`
+}
+
+type wsNotifier struct {
+	notify_service.NullNotifier
+}
+
+var _ notify_service.Notifier = &wsNotifier{}
+
+func (n *wsNotifier) NotificationCountChange(ctx context.Context, userID int64) {
+	if !pubsub.DefaultBroker.HasTopicSubscribers(pubsub.UserTopic(userID)) {
+		return
+	}
+	count, err := db.Count[activities_model.Notification](ctx, activities_model.FindNotificationOptions{
+		UserID: userID,
+		Status: []activities_model.NotificationStatus{activities_model.NotificationStatusUnread},
+	})
+	if err != nil {
+		log.Error("websocket: count notifications for user %d: %v", userID, err)
+		return
+	}
+	publishUserEvent(userID, EventNotificationCount, notificationCountEventData{Count: count})
+}

services/websocket/notifier.go

@@ -0,0 +1,18 @@
+// Copyright 2026 The Gitea Authors. All rights reserved.
+// SPDX-License-Identifier: MIT
+
+package websocket
+
+import (
+	notify_service "gitea.dev/services/notify"
+	"gitea.dev/services/pubsub"
+)
+
+func Init() error {
+	// the pubsub broker must be ready before the notifier starts publishing to it
+	if err := pubsub.Init(); err != nil {
+		return err
+	}
+	notify_service.RegisterNotifier(&wsNotifier{})
+	return nil
+}

services/websocket/stopwatch_notifier.go

@@ -0,0 +1,37 @@
+// Copyright 2026 The Gitea Authors. All rights reserved.
+// SPDX-License-Identifier: MIT
+
+package websocket
+
+import (
+	"context"
+
+	"gitea.dev/models/db"
+	issues_model "gitea.dev/models/issues"
+	user_model "gitea.dev/models/user"
+	"gitea.dev/modules/log"
+	"gitea.dev/modules/util"
+	"gitea.dev/services/convert"
+	"gitea.dev/services/pubsub"
+)
+
+func (n *wsNotifier) StopwatchChanged(ctx context.Context, user *user_model.User) {
+	if !pubsub.DefaultBroker.HasTopicSubscribers(pubsub.UserTopic(user.ID)) {
+		return
+	}
+
+	sws, err := issues_model.GetUserStopwatches(ctx, user.ID, db.ListOptions{})
+	if err != nil {
+		log.Error("websocket: GetUserStopwatches %d: %v", user.ID, err)
+		return
+	}
+
+	apiStopWatches, err := convert.ToStopWatches(ctx, user, sws)
+	if err != nil {
+		if !issues_model.IsErrIssueNotExist(err) {
+			log.Error("websocket: ToStopWatches: %v", err)
+		}
+		return
+	}
+	publishUserEvent(user.ID, EventStopwatches, util.SliceNilAsEmpty(apiStopWatches))
+}