feat: trailing slash middleware handles HEAD request

Author: sushichan044

src/middleware/trailing-slash/index.test.ts

@@ -45,6 +45,46 @@ describe('Resolve trailing slash', () => {
       expect(loc.pathname).toBe('/the/example/endpoint/without/trailing/slash')
       expect(loc.searchParams.get('exampleParam')).toBe('1')
     })
+
+    it('should handle HEAD request for root path correctly', async () => {
+      const resp = await app.request('/', { method: 'HEAD' })
+
+      expect(resp).not.toBeNull()
+      expect(resp.status).toBe(200)
+    })
+
+    it('should handle HEAD request for path without trailing slash correctly', async () => {
+      const resp = await app.request('/the/example/endpoint/without/trailing/slash', {
+        method: 'HEAD',
+      })
+
+      expect(resp).not.toBeNull()
+      expect(resp.status).toBe(200)
+    })
+
+    it('should handle HEAD request for path with trailing slash correctly', async () => {
+      const resp = await app.request('/the/example/endpoint/without/trailing/slash/', {
+        method: 'HEAD',
+      })
+      const loc = new URL(resp.headers.get('location')!)
+
+      expect(resp).not.toBeNull()
+      expect(resp.status).toBe(301)
+      expect(loc.pathname).toBe('/the/example/endpoint/without/trailing/slash')
+    })
+
+    it('should preserve query parameters when redirecting HEAD requests', async () => {
+      const resp = await app.request(
+        '/the/example/endpoint/without/trailing/slash/?exampleParam=1',
+        { method: 'HEAD' }
+      )
+      const loc = new URL(resp.headers.get('location')!)
+
+      expect(resp).not.toBeNull()
+      expect(resp.status).toBe(301)
+      expect(loc.pathname).toBe('/the/example/endpoint/without/trailing/slash')
+      expect(loc.searchParams.get('exampleParam')).toBe('1')
+    })
   })
 
   describe('appendTrailingSlash middleware', () => {
@@ -100,5 +140,51 @@ describe('Resolve trailing slash', () => {
       expect(loc.pathname).toBe('/the/example/endpoint/with/trailing/slash/')
       expect(loc.searchParams.get('exampleParam')).toBe('1')
     })
+
+    it('should handle HEAD request for root path correctly', async () => {
+      const resp = await app.request('/', { method: 'HEAD' })
+
+      expect(resp).not.toBeNull()
+      expect(resp.status).toBe(200)
+    })
+
+    it('should handle HEAD request for file-like paths correctly', async () => {
+      const resp = await app.request('/the/example/simulate/a.file', { method: 'HEAD' })
+
+      expect(resp).not.toBeNull()
+      expect(resp.status).toBe(200)
+    })
+
+    it('should handle HEAD request for path with trailing slash correctly', async () => {
+      const resp = await app.request('/the/example/endpoint/with/trailing/slash/', {
+        method: 'HEAD',
+      })
+
+      expect(resp).not.toBeNull()
+      expect(resp.status).toBe(200)
+    })
+
+    it('should redirect HEAD request for path without trailing slash to one with it', async () => {
+      const resp = await app.request('/the/example/endpoint/with/trailing/slash', {
+        method: 'HEAD',
+      })
+      const loc = new URL(resp.headers.get('location')!)
+
+      expect(resp).not.toBeNull()
+      expect(resp.status).toBe(301)
+      expect(loc.pathname).toBe('/the/example/endpoint/with/trailing/slash/')
+    })
+
+    it('should preserve query parameters when redirecting HEAD requests', async () => {
+      const resp = await app.request('/the/example/endpoint/with/trailing/slash?exampleParam=1', {
+        method: 'HEAD',
+      })
+      const loc = new URL(resp.headers.get('location')!)
+
+      expect(resp).not.toBeNull()
+      expect(resp.status).toBe(301)
+      expect(loc.pathname).toBe('/the/example/endpoint/with/trailing/slash/')
+      expect(loc.searchParams.get('exampleParam')).toBe('1')
+    })
   })
 })

src/middleware/trailing-slash/index.ts

@@ -26,7 +26,7 @@ export const trimTrailingSlash = (): MiddlewareHandler => {
 
     if (
       c.res.status === 404 &&
-      c.req.method === 'GET' &&
+      (c.req.method === 'GET' || c.req.method === 'HEAD') &&
       c.req.path !== '/' &&
       c.req.path.at(-1) === '/'
     ) {
@@ -57,7 +57,11 @@ export const appendTrailingSlash = (): MiddlewareHandler => {
   return async function appendTrailingSlash(c, next) {
     await next()
 
-    if (c.res.status === 404 && c.req.method === 'GET' && c.req.path.at(-1) !== '/') {
+    if (
+      c.res.status === 404 &&
+      (c.req.method === 'GET' || c.req.method === 'HEAD') &&
+      c.req.path.at(-1) !== '/'
+    ) {
       const url = new URL(c.req.url)
       url.pathname += '/'