module FastlyRails
  module CacheControlHeaders
    extend ActiveSupport::Concern

    included do
      before_filter :set_cache_control_headers, only: [:index, :show]
    end

    # Sets Cache-Control and Surrogate-Control headers
    # Surrogate-Control is stripped at the cache, Cache-Control persists (in case of other caches in front of fastly)
    # Defaults are:
    #  Cache-Control: 'public, no-cache, s-maxage: 30 days'
    #  Surrogate-Control: 'max-age: 30 days
    # custom config example: 
    #  {cache_control: 'public, no-cache, maxage=xyz', surrogate_control: 'max-age: blah'}
    def set_cache_control_headers(max_age = FastlyRails.configuration.max_age, opts = {})
      request.session_options[:skip] = true    # no cookies
      response.headers['Cache-Control'] = opts[:cache_control] || "public, no-cache"
      response.headers['Surrogate-Control'] = opts[:surrogate_control] || "max-age=#{max_age}"
    end


  end
end