From a2f02212297f8861a82526a24dc75df35bc0c06b Mon Sep 17 00:00:00 2001
From: Rakantor <rakantor.dev@gmail.com>
Date: Thu, 3 Nov 2022 15:32:55 +0100
Subject: [PATCH] Add permission levels for students/tutors/admins

---
 pages/help.vue  | 7 +------
 pages/index.vue | 8 ++++++++
 2 files changed, 9 insertions(+), 6 deletions(-)

diff --git a/pages/help.vue b/pages/help.vue
index 930cd79..46d8ec0 100644
--- a/pages/help.vue
+++ b/pages/help.vue
@@ -16,16 +16,11 @@
 </template>
 
 <script>
-import { demoAccounts } from '~/components/DemoInfoDialog.vue'
-
 export default {
   name: 'HelpPage',
   layout ({ $auth }) {
     // Ref: https://firebase.google.com/docs/reference/js/v8/firebase.auth.Auth#currentuser
-    // return $auth.currentUser.emailVerified ? 'default' : 'unverified'
-
-    // TODO: For demo purposes only. Delete in production.
-    return $auth.currentUser.emailVerified || demoAccounts.includes($auth.currentUser.email) ? 'default' : 'unverified'
+    return $auth.currentUser.emailVerified ? 'default' : 'unverified'
   },
 }
 </script>
diff --git a/pages/index.vue b/pages/index.vue
index 1828fd5..bd58c62 100644
--- a/pages/index.vue
+++ b/pages/index.vue
@@ -18,12 +18,20 @@ export default {
     // whenever the user's sign-in state changes.
     onAuthStateChanged(this.$auth, (user) => {
       this.$store.commit('initFirebase')
+
       if (user) {
         // User is signed in; get user data from db, then redirect to main page (dashboard)
         this.getUser()
+
+        // Get user claims
+        // Ref: https://firebase.google.com/docs/auth/admin/custom-claims
+        user.getIdTokenResult().then((idTokenResult) => {
+          this.$store.commit('setIdTokenResult', idTokenResult)
+        })
       } else {
         // User is signed out; redirect to login page
         this.$store.commit('setUser', null)
+        this.$store.commit('setIdTokenResult', null)
         this.$router.push({ name: 'login' })
       }
     })