From 64f2fa012ee327a90ed7328089d09616425b390d Mon Sep 17 00:00:00 2001
From: Viswamedha Nalabotu <vxn217@student.bham.ac.uk>
Date: Wed, 18 Mar 2026 01:08:28 +0000
Subject: [PATCH] Ensured users are logged in for using invites

---
 site/src/router/index.ts        |  1 +
 site/src/views/LoginView.vue    | 10 ++++++++++
 site/src/views/RegisterView.vue | 10 ++++++++++
 3 files changed, 21 insertions(+)

diff --git a/site/src/router/index.ts b/site/src/router/index.ts
index 0169373..41a806c 100644
--- a/site/src/router/index.ts
+++ b/site/src/router/index.ts
@@ -54,6 +54,7 @@ const router = createRouter({
             path: '/invite/:inviteUuid',
             name: 'invite-accept',
             component: () => import('../views/InviteAccept.vue'),
+            meta: { requiresAuth: true },
         },
         {
             path: '/agents',
diff --git a/site/src/views/LoginView.vue b/site/src/views/LoginView.vue
index 787cd69..22307cd 100644
--- a/site/src/views/LoginView.vue
+++ b/site/src/views/LoginView.vue
@@ -97,6 +97,10 @@ onMounted(async () => {
                     />
                 </Form.Item>
                 <Button type="primary" html-type="submit" block :loading="loading">Login</Button>
+                <div class="auth-switch">
+                    Don't have an account?
+                    <a @click="router.push({ path: '/register', query: route.query })">Register</a>
+                </div>
             </Form>
         </Card>
     </div>
@@ -114,4 +118,10 @@ onMounted(async () => {
     max-width: 400px;
     width: 100%;
 }
+.auth-switch {
+    margin-top: 1rem;
+    text-align: center;
+    font-size: 0.875rem;
+    color: #6b7280;
+}
 </style>
diff --git a/site/src/views/RegisterView.vue b/site/src/views/RegisterView.vue
index 836fc54..898abd1 100644
--- a/site/src/views/RegisterView.vue
+++ b/site/src/views/RegisterView.vue
@@ -164,6 +164,10 @@ onMounted(async () => {
                 </Form.Item>
 
                 <Button type="primary" html-type="submit" block :loading="loading">Register</Button>
+                <div class="auth-switch">
+                    Already have an account?
+                    <a @click="router.push({ path: '/login', query: route.query })">Login</a>
+                </div>
             </Form>
         </Card>
     </div>
@@ -175,4 +179,10 @@ onMounted(async () => {
     margin: 0 auto;
     padding: 1rem;
 }
+.auth-switch {
+    margin-top: 1rem;
+    text-align: center;
+    font-size: 0.875rem;
+    color: #6b7280;
+}
 </style>