1 files changed, 96 insertions, 0 deletions
diff --git a/src/services/push/push.js b/src/services/push/push.js
new file mode 100644
index 00000000..4e4551bf
--- /dev/null
+++ b/src/services/push/push.js
@@ -0,0 +1,96 @@
+
+function urlBase64ToUint8Array (base64String) {
+  const padding = '='.repeat((4 - base64String.length % 4) % 4)
+  const base64 = (base64String + padding)
+    .replace(/-/g, '+')
+    .replace(/_/g, '/')
+
+  const rawData = window.atob(base64)
+  return Uint8Array.from([...rawData].map((char) => char.charCodeAt(0)))
+}
+
+function isPushSupported () {
+  return 'serviceWorker' in navigator && 'PushManager' in window
+}
+
+function registerServiceWorker () {
+  return navigator.serviceWorker.register('/static/sw.js')
+    .then(function (registration) {
+      console.log('Service worker successfully registered.')
+      return registration
+    })
+    .catch(function (err) {
+      console.error('Unable to register service worker.', err)
+    })
+}
+
+function askPermission () {
+  return new Promise(function (resolve, reject) {
+    if (!window.Notification) return resolve('Notifications disabled')
+
+    const permissionResult = window.Notification.requestPermission(function (result) {
+      resolve(result)
+    })
+
+    if (permissionResult) permissionResult.then(resolve, reject)
+  }).then(function (permissionResult) {
+    if (permissionResult !== 'granted') {
+      throw new Error('We weren\'t granted permission.')
+    }
+    return permissionResult
+  })
+}
+
+function subscribe (registration, store) {
+  const subscribeOptions = {
+    userVisibleOnly: true,
+    applicationServerKey: urlBase64ToUint8Array(store.rootState.instance.vapidPublicKey)
+  }
+  return registration.pushManager.subscribe(subscribeOptions)
+}
+
+function sendSubscriptionToBackEnd (subscription, store) {
+  return window.fetch('/api/v1/push/subscription/', {
+    method: 'POST',
+    headers: {
+      'Content-Type': 'application/json',
+      'Authorization': `Bearer ${store.rootState.oauth.token}`
+    },
+    body: JSON.stringify({
+      subscription,
+      data: {
+        alerts: {
+          follow: true,
+          favourite: true,
+          mention: true,
+          reblog: true
+        }
+      }
+    })
+  })
+    .then(function (response) {
+      if (!response.ok) {
+        throw new Error('Bad status code from server.')
+      }
+
+      return response.json()
+    })
+    .then(function (responseData) {
+      if (!responseData.id) {
+        throw new Error('Bad response from server.')
+      }
+      return responseData
+    })
+}
+
+export default function registerPushNotifications (store) {
+  if (isPushSupported()) {
+    registerServiceWorker()
+      .then(function (registration) {
+        return askPermission()
+          .then(() => subscribe(registration, store))
+          .then((subscription) => sendSubscriptionToBackEnd(subscription, store))
+          .catch((e) => console.warn(`Failed to setup Web Push Notifications: ${e.message}`))
+      })
+  }
+}