2022-12-05 20:14:56 +00:00
|
|
|
import { getClientById } from 'wildebeest/backend/src/mastodon/client'
|
2023-01-17 14:14:58 +00:00
|
|
|
import { cors } from 'wildebeest/backend/src/utils/cors'
|
2023-01-11 14:32:06 +00:00
|
|
|
import { getVAPIDKeys } from 'wildebeest/backend/src/config'
|
|
|
|
import type { JWK } from 'wildebeest/backend/src/webpush/jwk'
|
2022-12-05 20:14:56 +00:00
|
|
|
import type { Actor } from 'wildebeest/backend/src/activitypub/actors'
|
|
|
|
import { createSubscription, getSubscription } from 'wildebeest/backend/src/mastodon/subscription'
|
|
|
|
import type { CreateRequest } from 'wildebeest/backend/src/mastodon/subscription'
|
|
|
|
import { ContextData } from 'wildebeest/backend/src/types/context'
|
2023-01-11 15:45:07 +00:00
|
|
|
import type { Env } from 'wildebeest/backend/src/types/env'
|
2022-12-05 20:14:56 +00:00
|
|
|
import * as errors from 'wildebeest/backend/src/errors'
|
2023-01-11 14:32:06 +00:00
|
|
|
import { VAPIDPublicKey } from 'wildebeest/backend/src/mastodon/subscription'
|
2023-02-22 16:09:43 +00:00
|
|
|
import { type Database, getDatabase } from 'wildebeest/backend/src/database'
|
2022-12-05 20:14:56 +00:00
|
|
|
|
|
|
|
export const onRequestGet: PagesFunction<Env, any, ContextData> = async ({ request, env, data }) => {
|
2023-02-27 17:37:03 +00:00
|
|
|
return handleGetRequest(await getDatabase(env), request, data.connectedActor, data.clientId, getVAPIDKeys(env))
|
2022-12-05 20:14:56 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
export const onRequestPost: PagesFunction<Env, any, ContextData> = async ({ request, env, data }) => {
|
2023-02-27 17:37:03 +00:00
|
|
|
return handlePostRequest(await getDatabase(env), request, data.connectedActor, data.clientId, getVAPIDKeys(env))
|
2022-12-05 20:14:56 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
const headers = {
|
2023-01-17 14:14:58 +00:00
|
|
|
...cors(),
|
2022-12-05 20:14:56 +00:00
|
|
|
'content-type': 'application/json; charset=utf-8',
|
|
|
|
}
|
|
|
|
|
2023-02-15 11:39:15 +00:00
|
|
|
export async function handleGetRequest(
|
2023-02-22 16:09:43 +00:00
|
|
|
db: Database,
|
2023-02-15 11:39:15 +00:00
|
|
|
request: Request,
|
|
|
|
connectedActor: Actor,
|
|
|
|
clientId: string,
|
|
|
|
vapidKeys: JWK
|
|
|
|
) {
|
2022-12-05 20:14:56 +00:00
|
|
|
const client = await getClientById(db, clientId)
|
|
|
|
if (client === null) {
|
|
|
|
return errors.clientUnknown()
|
|
|
|
}
|
|
|
|
|
|
|
|
const subscription = await getSubscription(db, connectedActor, client)
|
|
|
|
|
|
|
|
if (subscription === null) {
|
2023-02-15 15:36:04 +00:00
|
|
|
return errors.resourceNotFound('subscription', clientId)
|
2022-12-05 20:14:56 +00:00
|
|
|
}
|
|
|
|
|
2023-02-15 11:39:15 +00:00
|
|
|
const vapidKey = VAPIDPublicKey(vapidKeys)
|
|
|
|
|
2022-12-05 20:14:56 +00:00
|
|
|
const res = {
|
2023-02-15 11:39:15 +00:00
|
|
|
id: subscription.id,
|
2022-12-05 20:14:56 +00:00
|
|
|
endpoint: subscription.gateway.endpoint,
|
2023-02-15 11:39:15 +00:00
|
|
|
alerts: subscription.alerts,
|
|
|
|
policy: subscription.policy,
|
|
|
|
server_key: vapidKey,
|
2022-12-05 20:14:56 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
return new Response(JSON.stringify(res), { headers })
|
|
|
|
}
|
|
|
|
|
2023-01-11 14:32:06 +00:00
|
|
|
export async function handlePostRequest(
|
2023-02-22 16:09:43 +00:00
|
|
|
db: Database,
|
2023-01-11 14:32:06 +00:00
|
|
|
request: Request,
|
|
|
|
connectedActor: Actor,
|
|
|
|
clientId: string,
|
|
|
|
vapidKeys: JWK
|
|
|
|
) {
|
2022-12-05 20:14:56 +00:00
|
|
|
const client = await getClientById(db, clientId)
|
|
|
|
if (client === null) {
|
|
|
|
return errors.clientUnknown()
|
|
|
|
}
|
|
|
|
|
|
|
|
const data = await request.json<CreateRequest>()
|
|
|
|
|
|
|
|
let subscription = await getSubscription(db, connectedActor, client)
|
|
|
|
|
|
|
|
if (subscription === null) {
|
|
|
|
subscription = await createSubscription(db, connectedActor, client, data)
|
|
|
|
}
|
|
|
|
|
2023-01-11 14:32:06 +00:00
|
|
|
const vapidKey = VAPIDPublicKey(vapidKeys)
|
2022-12-05 20:14:56 +00:00
|
|
|
|
|
|
|
const res = {
|
2023-02-15 11:39:15 +00:00
|
|
|
id: subscription.id,
|
|
|
|
endpoint: subscription.gateway.endpoint,
|
|
|
|
alerts: subscription.alerts,
|
|
|
|
policy: subscription.policy,
|
2022-12-05 20:14:56 +00:00
|
|
|
server_key: vapidKey,
|
|
|
|
}
|
|
|
|
|
|
|
|
return new Response(JSON.stringify(res), { headers })
|
|
|
|
}
|