module Rack
  module OAuth2
    module Server
      module Rails
        module ResponseExt
          def redirect?
            ensure_finish do
              super
            end
          end

          def location
            ensure_finish do
              super
            end
          end

          def json
            ensure_finish do
              @body
            end
          end

          def headers
            ensure_finish do
              @headers
            end
          end

          def finish
            @finished = true
            super
          end

          private

          def finished?
            !!@finished
          end

          def ensure_finish
            @status, @headers, @body = finish unless finished?
            yield
          end
        end
      end
    end
  end
end