gregbeech/xenon

require 'xenon/routing/route_directives'

module Xenon
  module Routing
    module HeaderDirectives
      include RouteDirectives

      def optional_header(name)
        optional_header(name, &proc)
      end

      def header(name)
        headers(name, &proc)
      end

      def optional_headers(*names)
        extract_request do |request|
          yield names.map { |name| request.header(name) }
        end
      end

      def headers(*names)
        optional_headers(*names) do |values|
          if values.all?
            yield *values
          else
            reject Rejection.new(:header, { required: names })
          end
        end
      end

      def respond_with_header(header)
        map_response -> r { r.copy(headers: r.headers.add(header)) } do
          yield
        end
      end

    end
  end
end