Fix #68 not working with static plugin

package.json

@@ -43,6 +43,7 @@
   },
   "devDependencies": {
     "@elysiajs/cors": "^1.1.1",
+    "@elysiajs/static": "^1.1.1",
     "@elysiajs/stream": "^1.1.0",
     "@eslint/js": "^9.15.0",
     "@types/bun": "latest",

src/compression-stream.ts

@@ -16,18 +16,14 @@ export const CompressionStream = (
   let handler: Transform
 
   const zlibOptions: zlib.ZlibOptions = {
-    ...{
-      level: 6,
-    },
+    level: 6,
     ...options?.zlibOptions,
   }
   const brotliOptions: zlib.BrotliOptions = {
-    ...{
-      params: {
-        [zlib.constants.BROTLI_PARAM_MODE]: zlib.constants.BROTLI_MODE_TEXT,
-        [zlib.constants.BROTLI_PARAM_QUALITY]:
-          zlib.constants.BROTLI_DEFAULT_QUALITY,
-      },
+    params: {
+      [zlib.constants.BROTLI_PARAM_MODE]: zlib.constants.BROTLI_MODE_TEXT,
+      [zlib.constants.BROTLI_PARAM_QUALITY]:
+        zlib.constants.BROTLI_DEFAULT_QUALITY,
     },
     ...options?.brotliOptions,
   }

src/main.ts

@@ -36,17 +36,13 @@
   options?: CompressionOptions & LifeCycleOptions & CacheOptions,
 ) => {
   const zlibOptions: ZlibOptions = {
-    ...{
-      level: 6,
-    },
+    level: 6,
     ...options?.zlibOptions,
   }
   const brotliOptions: BrotliOptions = {
-    ...{
-      params: {
-        [constants.BROTLI_PARAM_MODE]: constants.BROTLI_MODE_GENERIC,
-        [constants.BROTLI_PARAM_QUALITY]: constants.BROTLI_DEFAULT_QUALITY,
-      },
+    params: {
+      [constants.BROTLI_PARAM_MODE]: constants.BROTLI_MODE_GENERIC,
+      [constants.BROTLI_PARAM_QUALITY]: constants.BROTLI_DEFAULT_QUALITY,
     },
     ...options?.brotliOptions,
   }
@@ -138,7 +134,7 @@
 
       contentType = resContentType ? resContentType : 'text/plain'
 
-      const buffer = await res.arrayBuffer()
+      const buffer = await res.clone().arrayBuffer()
       // Disable compression when buffer size is less than threshold
       if (buffer.byteLength < threshold) {
         return

src/types.ts

@@ -57,7 +57,7 @@ export type CompressionOptions = {
    * Defaults to `false`
    * @default false
    */
-  compressStream: boolean
+  compressStream?: boolean
 }
 
 export type LifeCycleOptions = {

tests/index.test.ts

@@ -1,10 +1,11 @@
 import { describe, expect, it } from 'bun:test'
 import zlib from 'node:zlib'
 import Elysia from 'elysia'
+import staticPlugin from '@elysiajs/static'
 import { Stream } from '@elysiajs/stream'
 import { cors } from '@elysiajs/cors'
 
-import { req, responseShort, jsonResponse } from './setup'
+import { req, responseShort, imageResponse, jsonResponse } from './setup'
 import compression from '../src'
 
 describe(`elysia-compress`, () => {
@@ -97,15 +98,40 @@ describe(`elysia-compress`, () => {
     expect(res.headers.get('vary')).toBe('accept-encoding')
   })
 
-  it('return correct image type', async () => {
+  it('return correct image', async () => {
     const app = new Elysia()
       .use(compression({ encodings: ['gzip'], threshold: 1 }))
-      .get('/', () => Bun.file('tests/waifu.png'))
+      .get('/', () => imageResponse)
 
     const res = await app.handle(req())
 
     expect(res.headers.get('Content-Type')).toBe('image/png')
     expect(res.headers.get('vary')).toBeNull()
+
+    const actualBody = await res.arrayBuffer()
+    const expectedBody = await imageResponse.arrayBuffer()
+    expect(actualBody).toEqual(expectedBody)
+  })
+
+  it('return correct image from static plugin', async () => {
+    const app = new Elysia()
+      .use(staticPlugin({ assets: 'tests/images', prefix: '' }))
+      .use(compression())
+
+    await app.modules
+
+    const res = await app.handle(
+      new Request('http://localhost/waifu.png', {
+        headers: { 'accept-encoding': 'br, deflate, gzip, zstd' },
+      }),
+    )
+
+    expect(res.headers.get('Content-Type')).toBe('image/png')
+    expect(res.headers.get('vary')).toBeNull()
+
+    const actualBody = await res.arrayBuffer()
+    const expectedBody = await imageResponse.arrayBuffer()
+    expect(actualBody).toEqual(expectedBody)
   })
 
   it('must be redirected to /not-found', async () => {

tests/setup.ts

@@ -15,3 +15,5 @@ Excepteur sint occaecat cupidatat non proident, sunt in culpa qui officia deseru
 export const responseLong = responseShort.repeat(100)
 
 export const jsonResponse = Bun.file('./tests/data.json')
+
+export const imageResponse = Bun.file('./tests/images/waifu.png')