import { getClientById } from 'wildebeest/backend/src/mastodon/client'
import { cors } from 'wildebeest/backend/src/utils/cors'
import { getVAPIDKeys } from 'wildebeest/backend/src/config'
import type { JWK } from 'wildebeest/backend/src/webpush/jwk'
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'
import type { Env } from 'wildebeest/backend/src/types/env'
import * as errors from 'wildebeest/backend/src/errors'
import { VAPIDPublicKey } from 'wildebeest/backend/src/mastodon/subscription'

export const onRequestGet: PagesFunction<Env, any, ContextData> = async ({ request, env, data }) => {
	return handleGetRequest(env.DATABASE, request, data.connectedActor, data.clientId)
}

export const onRequestPost: PagesFunction<Env, any, ContextData> = async ({ request, env, data }) => {
	return handlePostRequest(env.DATABASE, request, data.connectedActor, data.clientId, getVAPIDKeys(env))
}

const headers = {
	...cors(),
	'content-type': 'application/json; charset=utf-8',
}

export async function handleGetRequest(db: D1Database, request: Request, connectedActor: Actor, clientId: string) {
	const client = await getClientById(db, clientId)
	if (client === null) {
		return errors.clientUnknown()
	}

	const subscription = await getSubscription(db, connectedActor, client)

	if (subscription === null) {
		return new Response('', { status: 404 })
	}

	const res = {
		id: 4,
		endpoint: subscription.gateway.endpoint,
		alerts: {
			follow: true,
			favourite: true,
			reblog: true,
			mention: true,
			poll: true,
		},
		policy: 'all',

		// FIXME: stub value
		server_key: 'TODO',
	}

	return new Response(JSON.stringify(res), { headers })
}

export async function handlePostRequest(
	db: D1Database,
	request: Request,
	connectedActor: Actor,
	clientId: string,
	vapidKeys: JWK
) {
	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)
	}

	const vapidKey = VAPIDPublicKey(vapidKeys)

	const res = {
		id: 4,
		endpoint: data.subscription.endpoint,
		alerts: {
			follow: true,
			favourite: true,
			reblog: true,
			mention: true,
			poll: true,
		},
		policy: 'all',
		server_key: vapidKey,
	}

	return new Response(JSON.stringify(res), { headers })
}