feat(edgedb): small improvements

Author: Tahul

src/runtime/api/auth/identity.ts

@@ -1,28 +1,44 @@
 import { defineEventHandler, deleteCookie, getCookie } from 'h3'
 import { useEdgeDb, useEdgeDbEnv } from '../../server'
 
-export default defineEventHandler(async (req) => {
+export default defineEventHandler(async (event) => {
   const { identityModel } = useEdgeDbEnv()
 
-  const token = getCookie(req, 'edgedb-auth-token')
+  const token = getCookie(event, 'edgedb-auth-token')
 
   if (!token) {
-    deleteCookie(req, 'edgedb-auth-token')
+    deleteCookie(event, 'edgedb-auth-token')
     return
   }
 
-  const client = useEdgeDb(req)
+  const client = useEdgeDb(event)
 
-  let identityTarget = await client.querySingle(`select global current_user;`)
+  try {
+    let identityTarget = await client.querySingle(`select global current_user;`)
 
-  if (!identityTarget && token) {
-    identityTarget = await client.query(`
+    if (!identityTarget && token) {
+      identityTarget = await client.query(`
       insert ${identityModel} {
         name := '',
         identity := global ext::auth::ClientTokenIdentity
       }
     `)
-  }
+    }
 
-  return identityTarget
+    return identityTarget
+  }
+  catch (err) {
+    setCookie(
+      event,
+      'edgedb-auth-token',
+      '',
+      {
+        httpOnly: true,
+        path: '/',
+        secure: true,
+        sameSite: true,
+        expires: new Date(0),
+      },
+    )
+  }
 })

src/runtime/api/auth/logout.ts

@@ -2,6 +2,7 @@ import { H3Error, defineEventHandler, getCookie, setCookie } from 'h3'
 
 export default defineEventHandler(async (event) => {
   const authToken = getCookie(event, 'edgedb-auth-token')
+
   if (!authToken) {
     const err = new H3Error('Not logged in')
     err.statusCode = 401