feat: add global page protected

LorexIQ · Nov 22, 2023 · e961187 · e961187
1 parent ae8359d
commit e961187
Show file tree

Hide file tree

Showing 6 changed files with 60 additions and 15 deletions.
diff --git a/playground/nuxt.config.ts b/playground/nuxt.config.ts
@@ -15,6 +15,9 @@ export default defineNuxtConfig({
     refreshToken: {
       enabled: true
     },
+    pages: {
+      protectAllPages: true
+    },
     endpoints: {
       signIn: { path: '/auth/', method: 'POST' },
       getMe: { path: 'users/me/', method: 'GET' },

diff --git a/playground/pages/error.vue b/playground/pages/error.vue
@@ -0,0 +1,11 @@
+<script setup lang="ts">
+
+</script>
+
+<template>
+error
+</template>
+
+<style scoped>
+
+</style>
diff --git a/playground/pages/login.vue b/playground/pages/login.vue
@@ -12,9 +12,9 @@ function testAuth() {
 }
 
 definePageMeta({
-  localAuth: {
-    unauthorizedOnly: true
-  }
+  // localAuth: {
+  //   unauthorizedOnly: true
+  // }
 })
 </script>
 

diff --git a/playground/pages/test.vue b/playground/pages/test.vue
@@ -0,0 +1,15 @@
+<script setup lang="ts">
+definePageMeta({
+  localAuth: {
+    authorizedOnly: true,
+  }
+})
+</script>
+
+<template>
+test
+</template>
+
+<style scoped>
+
+</style>
diff --git a/src/module.ts b/src/module.ts
@@ -88,6 +88,9 @@ interface ModuleOptionsPages {
   * get session data due to a timeout. Default: '/error'
   * */
   serverIsDown?: string;
+  /* Protecting all pages from guests. Default: false
+  * */
+  protectAllPages?: boolean
 }
 export interface ModuleOptions {
   /* Path to server. Default: ''
@@ -132,7 +135,9 @@ export default defineNuxtModule<ModuleOptions>({
     },
     pages: {
       auth: '/login',
-      defaultRedirect: '/'
+      defaultRedirect: '/',
+      serverIsDown: '/error',
+      protectAllPages: false,
     }
   },
   setup (options, nuxt) {

diff --git a/src/runtime/middleware/auth.ts b/src/runtime/middleware/auth.ts
@@ -1,9 +1,10 @@
-import { defineNuxtRouteMiddleware, navigateTo, useRuntimeConfig } from '#imports';
+import { defineNuxtRouteMiddleware, navigateTo } from '#imports';
 import useUtils from "../composables/useUtils";
-import { useLocalAuth } from "../composables/useLocalAuth";
+import { getContext } from "../helpers";
 
 type MiddlewareMeta = boolean | {
   unauthorizedOnly?: boolean;
+  authorizedOnly?: boolean;
 };
 
 declare module '#app/../pages/runtime/composables' {
@@ -12,27 +13,37 @@ declare module '#app/../pages/runtime/composables' {
   }
 }
 
-export default defineNuxtRouteMiddleware(to => {
+export default defineNuxtRouteMiddleware(async to => {
+  const { options, state: { meta} } = await getContext();
+  const { trimStartWithSymbol, trimWithSymbol } = useUtils();
+
   const metaAuth = typeof to.meta.localAuth === 'object' ?
     {
-      unauthorizedOnly: true,
+      unauthorizedOnly: false,
+      authorizedOnly: false,
       ...to.meta.localAuth
     } :
     to.meta.localAuth;
 
+  if (
+    !metaAuth &&
+    !options.pages.protectAllPages ||
+    [trimWithSymbol(options.pages.serverIsDown!, '/')].includes(trimWithSymbol(to.path, '/'))
+  ) return;
   if (metaAuth === false) return;
 
-  const options = useRuntimeConfig().public.localAuth;
-  const { trimStartWithSymbol } = useUtils();
-  const { meta } = useLocalAuth();
   const isAuthorized = meta.value.status === 'authorized';
   const isUnauthorizedOnly = typeof metaAuth === 'object' ? metaAuth.unauthorizedOnly : false;
+  const isAuthorizedOnly = typeof metaAuth === 'object' ? metaAuth.authorizedOnly : false;
 
-  if (isAuthorized && to.path === options.pages.auth && isUnauthorizedOnly) {
-    return navigateTo(`/${trimStartWithSymbol(options.pages.defaultRedirect, '/')}`);
+  if (isAuthorized && to.path !== options.pages.defaultRedirect && isUnauthorizedOnly) {
+    return navigateTo(`/${trimStartWithSymbol(options.pages.defaultRedirect!, '/')}`);
+  }
+  if (!isAuthorized && to.path !== options.pages.defaultRedirect && isAuthorizedOnly) {
+    return navigateTo(`/${trimStartWithSymbol(options.pages.defaultRedirect!, '/')}`);
   }
 
-  if (!isAuthorized && to.path !== options.pages.auth) {
-    return navigateTo(`/${trimStartWithSymbol(options.pages.auth, '/')}`);
+  if (!isAuthorized && to.path !== options.pages.auth && !isUnauthorizedOnly) {
+    return navigateTo(`/${trimStartWithSymbol(options.pages.auth!, '/')}`);
   }
 });