From 734ef1c5549cbe4d79f60a7aa74d0e4e2fbccd52 Mon Sep 17 00:00:00 2001
From: Inesh Bose <56732164+ineshbose@users.noreply.github.com>
Date: Sat, 27 Jan 2024 16:30:03 +0000
Subject: [PATCH] fix(viewer): handle redirect and callback await, thank you
 danielroe :)

---
 src/viewer.ts | 48 ++++++++++++++++++++++++++++++++----------------
 1 file changed, 32 insertions(+), 16 deletions(-)

diff --git a/src/viewer.ts b/src/viewer.ts
index dc86cb90..61857b7d 100644
--- a/src/viewer.ts
+++ b/src/viewer.ts
@@ -8,25 +8,41 @@ import type { TWConfig, ViewerConfig } from './types'
 
 export const setupViewer = async (twConfig: Partial<TWConfig>, config: ViewerConfig, nuxt = useNuxt()) => {
   const route = joinURL(nuxt.options.app?.baseURL, config.endpoint)
-  // @ts-ignore
-  const createServer = await import('tailwind-config-viewer/server/index.js').then(r => r.default || r) as any
-  const routerPrefix = isNuxt3() ? route : undefined
-
-  const _viewerDevMiddleware = createServer({ tailwindConfigProvider: () => twConfig, routerPrefix }).asMiddleware()
-  const viewerDevMiddleware = eventHandler((event) => {
-    const withoutSlash = withoutTrailingSlash(route)
-    if (event.node?.req.url === withoutSlash || event.req.url === withoutSlash) {
-      return sendRedirect(event, route, 301)
-    }
-    _viewerDevMiddleware(event.node?.req || event.req, event.node?.res || event.res)
-  })
+  const [routeWithSlash, routeWithoutSlash] = [withTrailingSlash(route), withoutTrailingSlash(route)]
 
-  if (isNuxt3()) { addDevServerHandler({ route, handler: viewerDevMiddleware }) }
-  // @ts-ignore
-  if (isNuxt2()) { nuxt.options.serverMiddleware.push({ route, handler: (req, res) => viewerDevMiddleware(new H3Event(req, res)) }) }
+  // @ts-expect-error untyped package export
+  const viewerServer = (await import('tailwind-config-viewer/server/index.js').then(r => r.default || r))({ tailwindConfigProvider: () => twConfig }).asMiddleware()
+  const viewerDevMiddleware = eventHandler(event => viewerServer(event.node?.req || event.req, event.node?.res || event.res))
+
+  if (isNuxt3()) {
+    addDevServerHandler({
+      handler: eventHandler(event => {
+        if (event.path === routeWithoutSlash) {
+          return sendRedirect(event, routeWithSlash, 301)
+        }
+      })
+    })
+    addDevServerHandler({ route, handler: viewerDevMiddleware })
+  }
+
+  if (isNuxt2()) {
+    // @ts-expect-error untyped nuxt2 property
+    nuxt.options.serverMiddleware.push(
+      // @ts-expect-error untyped handler parameters
+      (req, res, next) => {
+        if (req.url === routeWithoutSlash) {
+          return sendRedirect(new H3Event(req, res), routeWithSlash, 301)
+        }
+
+        next();
+      },
+      // @ts-expect-error untyped handler parameters
+      { route, handler: (req, res) => viewerDevMiddleware(new H3Event(req, res)) }
+    )
+  }
 
   nuxt.hook('listen', (_, listener) => {
-    const viewerUrl = `${cleanDoubleSlashes(joinURL(withoutTrailingSlash(listener.url), config.endpoint))}`
+    const viewerUrl = cleanDoubleSlashes(joinURL(listener.url, config.endpoint))
     logger.info(`Tailwind Viewer: ${underline(yellow(withTrailingSlash(viewerUrl)))}`)
   })
 }