notify.moe/scripts/ServerEvent/receiveServerEvents.ts
2019-11-18 11:04:13 +09:00

96 lines
2.2 KiB
TypeScript

import AnimeNotifier from "../AnimeNotifier"
import plural from "../Utils/plural"
import ServerEvent from "./ServerEvent"
const reconnectDelay = 3000
let supported: boolean
let eventSource: EventSource
let arn: AnimeNotifier
let etags: Map<string, string>
export default function receiveServerEvents(animeNotifier: AnimeNotifier) {
supported = ("EventSource" in window)
if(!supported) {
return
}
arn = animeNotifier
etags = new Map<string, string>()
connect()
}
function connect() {
if(eventSource) {
eventSource.close()
}
eventSource = new EventSource("/api/sse/events", {
withCredentials: true
})
eventSource.addEventListener("ping", (e: any) => ping(e))
eventSource.addEventListener("etag", (e: any) => etag(e))
eventSource.addEventListener("activity", (e: any) => activity(e))
eventSource.addEventListener("notificationCount", (e: any) => notificationCount(e))
eventSource.onerror = () => {
setTimeout(() => connect(), reconnectDelay)
}
}
function ping(_: ServerEvent) {
console.log("ping")
}
function etag(e: ServerEvent) {
const data = JSON.parse(e.data)
const oldETag = etags.get(data.url)
const newETag = data.etag
if(oldETag && newETag && oldETag !== newETag) {
arn.statusMessage.showInfo("A new version of the website is available. Please refresh the page.", -1)
}
etags.set(data.url, newETag)
}
function activity(e: ServerEvent) {
if(!location.pathname.startsWith("/activity")) {
return
}
const isFollowingUser = JSON.parse(e.data)
// If we're on the followed only feed and we receive an activity
// about a user we don't follow, ignore the message.
if(location.pathname.startsWith("/activity/followed") && !isFollowingUser) {
return
}
const button = document.getElementById("load-new-activities")
if(!button || !button.dataset.count) {
return
}
const buttonText = document.getElementById("load-new-activities-text")
if(!buttonText) {
return
}
const newCount = parseInt(button.dataset.count, 10) + 1
button.dataset.count = newCount.toString()
buttonText.textContent = plural(newCount, "new activity")
}
function notificationCount(e: ServerEvent) {
if(!arn.notificationManager) {
return
}
arn.notificationManager.setCounter(parseInt(e.data, 10))
}