iceshrimp/packages/client/src/components/MkPushNotificationAllowButton.vue

<template>
	<MkButton
		v-if="supported && !pushRegistrationInServer"
		type="button"
		primary
		:gradate="gradate"
		:rounded="rounded"
		:inline="inline"
		:autofocus="autofocus"
		:wait="wait"
		:full="full"
		@click="subscribe"
	>
		{{ i18n.ts.subscribePushNotification }}
	</MkButton>
	<MkButton
		v-else-if="
			!showOnlyToRegister &&
			($i ? pushRegistrationInServer : pushSubscription)
		"
		type="button"
		:primary="false"
		:gradate="gradate"
		:rounded="rounded"
		:inline="inline"
		:autofocus="autofocus"
		:wait="wait"
		:full="full"
		@click="unsubscribe"
	>
		{{ i18n.ts.unsubscribePushNotification }}
	</MkButton>
	<MkButton
		v-else-if="$i && pushRegistrationInServer"
		disabled
		:rounded="rounded"
		:inline="inline"
		:wait="wait"
		:full="full"
	>
		{{ i18n.ts.pushNotificationAlreadySubscribed }}
	</MkButton>
	<MkButton
		v-else-if="!supported"
		disabled
		:rounded="rounded"
		:inline="inline"
		:wait="wait"
		:full="full"
	>
		{{ i18n.ts.pushNotificationNotSupported }}
	</MkButton>
</template>

<script setup lang="ts">
import { $i, getAccounts } from "@/account";
import MkButton from "@/components/MkButton.vue";
import { instance } from "@/instance";
import { api, apiWithDialog, promiseDialog } from "@/os";
import { i18n } from "@/i18n";

defineProps<{
	primary?: boolean;
	gradate?: boolean;
	rounded?: boolean;
	inline?: boolean;
	link?: boolean;
	to?: string;
	autofocus?: boolean;
	wait?: boolean;
	danger?: boolean;
	full?: boolean;
	showOnlyToRegister?: boolean;
}>();

// ServiceWorker registration
let registration = $ref<ServiceWorkerRegistration | undefined>();
// If this browser supports push notification
let supported = $ref(false);
// If this browser has already subscribed to push notification
let pushSubscription = $ref<PushSubscription | null>(null);
let pushRegistrationInServer = $ref<
	| {
			state?: string;
			key?: string;
			userId: string;
			endpoint: string;
			sendReadMessage: boolean;
	  }
	| undefined
>();

function subscribe() {
	if (!registration || !supported || !instance.swPublickey) return;

	// SEE: https://developer.mozilla.org/en-US/docs/Web/API/PushManager/subscribe#Parameters
	return promiseDialog(
		registration.pushManager
			.subscribe({
				userVisibleOnly: true,
				applicationServerKey: urlBase64ToUint8Array(
					instance.swPublickey
				),
			})
			.then(
				async (subscription) => {
					pushSubscription = subscription;

					// Register
					pushRegistrationInServer = await api("sw/register", {
						endpoint: subscription.endpoint,
						auth: encode(subscription.getKey("auth")),
						publickey: encode(subscription.getKey("p256dh")),
					});
				},
				async (err) => {
					// When subscribe failed
					// 通知が許可されていなかったとき
					if (err?.name === "NotAllowedError") {
						console.info(
							"User denied the notification permission request."
						);
						return;
					}

					// 違うapplicationServerKey (または gcm_sender_id)のサブスクリプションが
					// 既に存在していることが原因でエラーになった可能性があるので、
					// そのサブスクリプションを解除しておく
					// （これは実行されなさそうだけど、おまじない的に古い実装から残してある）
					await unsubscribe();
				}
			),
		null,
		null
	);
}

async function unsubscribe() {
	if (!pushSubscription) return;

	const endpoint = pushSubscription.endpoint;
	const accounts = await getAccounts();

	pushRegistrationInServer = undefined;

	if ($i && accounts.length >= 2) {
		apiWithDialog("sw/unregister", {
			i: $i.token,
			endpoint,
		});
	} else {
		pushSubscription.unsubscribe();
		apiWithDialog("sw/unregister", {
			endpoint,
		});
		pushSubscription = null;
	}
}

function encode(buffer: ArrayBuffer | null) {
	return btoa(String.fromCharCode.apply(null, new Uint8Array(buffer)));
}

/**
 * Convert the URL safe base64 string to a Uint8Array
 * @param base64String base64 string
 */
function urlBase64ToUint8Array(base64String: string): Uint8Array {
	const padding = "=".repeat((4 - (base64String.length % 4)) % 4);
	const base64 = (base64String + padding)
		.replace(/-/g, "+")
		.replace(/_/g, "/");

	const rawData = window.atob(base64);
	const outputArray = new Uint8Array(rawData.length);

	for (let i = 0; i < rawData.length; ++i) {
		outputArray[i] = rawData.charCodeAt(i);
	}
	return outputArray;
}

if (navigator.serviceWorker == null) {
	// TODO: よしなに？
} else {
	navigator.serviceWorker.ready.then(async (swr) => {
		registration = swr;

		pushSubscription = await registration.pushManager.getSubscription();

		if (instance.swPublickey && "PushManager" in window && $i && $i.token) {
			supported = true;

			if (pushSubscription) {
				const res = await api("sw/show-registration", {
					endpoint: pushSubscription.endpoint,
				});

				if (res) {
					pushRegistrationInServer = res;
				}
			}
		}
	});
}

defineExpose({
	pushRegistrationInServer: $$(pushRegistrationInServer),
});
</script>