compressionmiddleware.ts

'use strict'
import { Initializer, api, action, log } from 'actionhero'
let compression = require('compression')

export class CompressMiddleware extends Initializer {
  constructor() {
    super()
    this.name = 'CompressMiddleware'
    this.loadPriority = 1000
    this.startPriority = 1000
    this.stopPriority = 1000
  }
  async initialize() {
    let compress = compression({})
    let compressMiddleware: action.ActionMiddleware = {
      global: false,
      name: 'compress',
      priority: 1010,
      preProcessor: async function (data) {
        if (data.connection.type === 'web') {
          data.toRender = false
          data.connection.rawConnection.res.setHeader('Content-Type', 'application/json')
          compress(data.connection.rawConnection.req, data.connection.rawConnection.res, () => {})
        }
      },
      postProcessor: async function (data) {
        if (data.connection.type === 'web') {
          data.connection.rawConnection.res.write(JSON.stringify(data.response))
          data.connection.rawConnection.res.end()
        }
      },
    }

    action.addMiddleware(compressMiddleware)
  }
  async start() {}
  async stop() {}
}