'use strict'
const LocalWebServer = require('../../')
const cacheControl = require('koa-cache-control')
const DefaultStack = require('local-web-server-default-stack')

class CacheControl extends DefaultStack {
  addAll () {
    return this.addLogging('dev')
      .addCacheControl()
      .addStatic()
      .addIndex()
  }
  addCacheControl () {
    this.add({
      optionDefinitions: {
        name: 'maxage', type: Number,
        description: 'The maxage to set on each response.'
      },
      middleware: function (options) {
        return cacheControl({ maxAge: options.maxage })
      }
    })
    return this
  }
}

module.exports = CacheControl