Implemented server sent events

middleware/Log.go

@@ -76,7 +76,7 @@ func logRequest(ctx *aero.Context, responseTime time.Duration) {
 
 	// Notify us about long requests.
 	// However ignore requests under /auth/ because those depend on 3rd party servers.
-	if responseTime >= 300*time.Millisecond && !strings.HasPrefix(ctx.URI(), "/auth/") && !strings.HasPrefix(ctx.URI(), "/sitemap/") {
+	if responseTime >= 300*time.Millisecond && !strings.HasPrefix(ctx.URI(), "/auth/") && !strings.HasPrefix(ctx.URI(), "/sitemap/") && !strings.HasPrefix(ctx.URI(), "/api/sse/") {
 		errorLog.Error("Long response time", nick, id, ip, responseTimeString, ctx.StatusCode, ctx.URI())
 	}
 }

pages/index/apiroutes/apiroutes.go

@@ -4,6 +4,7 @@ import (
 	"strings"
 
 	"github.com/animenotifier/notify.moe/pages/post"
+	"github.com/animenotifier/notify.moe/pages/sse"
 	"github.com/animenotifier/notify.moe/pages/thread"
 
 	"github.com/aerogo/aero"
@@ -42,6 +43,9 @@ func Register(l *layout.Layout, app *aero.Application) {
 	app.Get("/api/next/soundtrack", soundtrack.Next)
 	app.Get("/api/character/:id/ranking", character.Ranking)
 
+	// Live updates
+	app.Get("/api/sse/events", sse.Events)
+
 	// Thread
 	app.Get("/api/thread/:id/reply/ui", thread.ReplyUI)
 

pages/sse/sse.go

@@ -0,0 +1,42 @@
+package sse
+
+import (
+	"fmt"
+	"net/http"
+
+	"github.com/aerogo/aero"
+	"github.com/animenotifier/notify.moe/utils"
+)
+
+// Events streams server events to the client.
+func Events(ctx *aero.Context) string {
+	user := utils.GetUser(ctx)
+
+	if user == nil {
+		return ctx.Error(http.StatusUnauthorized, "Not logged in")
+	}
+
+	fmt.Println(user.Nick, "receiving live events")
+
+	events := make(chan *aero.Event)
+	disconnected := make(chan struct{})
+
+	go func() {
+		defer fmt.Println(user.Nick, "disconnected, stop sending events")
+
+		for {
+			select {
+			case <-disconnected:
+				close(events)
+				return
+
+				// case <-time.After(10 * time.Second):
+				// 	events <- &aero.Event{
+				// 		Name: "ping",
+				// 	}
+			}
+		}
+	}()
+
+	return ctx.EventStream(events, disconnected)
+}

scripts/AnimeNotifier.ts

@@ -9,6 +9,7 @@ import Analytics from "./Analytics"
 import SideBar from "./SideBar"
 import InfiniteScroller from "./InfiniteScroller"
 import ServiceWorkerManager from "./ServiceWorkerManager"
+import ServerEvents from "./ServerEvents"
 import { checkNewVersionDelayed } from "./NewVersionCheck"
 import { displayAiringDate, displayDate, displayTime } from "./DateView"
 import { findAll, canUseWebP, requestIdleCallback, swapElements, delay, findAllInside } from "./Utils"
@@ -35,6 +36,7 @@ export default class AnimeNotifier {
 	diffCompletedForCurrentPath: boolean
 	lastReloadContentPath: string
 	currentSoundTrackId: string
+	serverEvents: ServerEvents
 
 	constructor(app: Application) {
 		this.app = app
@@ -235,6 +237,11 @@ export default class AnimeNotifier {
 			window.resizeTo(finalWidth, finalHeight)
 		}
 
+		// Server sent events
+		if(this.user && EventSource) {
+			this.serverEvents = new ServerEvents()
+		}
+
 		// // Download popular anime titles for the search
 		// let response = await fetch("/api/popular/anime/titles/500")
 		// let titles = await response.json()

scripts/ServerEvents.ts

@@ -0,0 +1,26 @@
+class ServerEvent {
+	data: string
+}
+
+export default class ServerEvents {
+	supported: boolean
+	eventSource: EventSource
+
+	constructor() {
+		this.supported = ("EventSource" in window)
+
+		if(!this.supported) {
+			return
+		}
+
+		this.eventSource = new EventSource("/api/sse/events", {
+			withCredentials: true
+		})
+
+		this.eventSource.addEventListener("ping", (e: any) => this.ping(e))
+	}
+
+	ping(e: ServerEvent) {
+		console.log("sse: ping")
+	}
+}

utils/routetests/All.go

@@ -446,6 +446,7 @@ var routeTests = map[string][]string{
 	"/api/pushsubscriptions/:id/get/:item/:property": nil,
 	"/api/count/notifications/unseen":                nil,
 	"/api/mark/notifications/seen":                   nil,
+	"/api/sse/events":                                nil,
 	"/editor/kitsu/new/anime":                        nil,
 	"/paypal/success":                                nil,
 	"/paypal/cancel":                                 nil,