ruby-grape/grape

# frozen_string_literal: true

module Grape
  class Cookies
    def initialize
      @cookies = {}
      @send_cookies = {}
    end

    def read(request)
      request.cookies.each do |name, value|
        @cookies[name.to_s] = value
      end
    end

    def write(header)
      @cookies.select { |key, _value| @send_cookies[key] == true }.each do |name, value|
        cookie_value = value.is_a?(Hash) ? value : { value: value }
        Rack::Utils.set_cookie_header! header, name, cookie_value
      end
    end

    def [](name)
      @cookies[name.to_s]
    end

    def []=(name, value)
      @cookies[name.to_s] = value
      @send_cookies[name.to_s] = true
    end

    def each(&block)
      @cookies.each(&block)
    end

    # see https://github.com/rack/rack/blob/main/lib/rack/utils.rb#L338-L340
    # rubocop:disable Layout/SpaceBeforeBrackets
    def delete(name, **opts)
      options = opts.merge(max_age: '0', value: '', expires: Time.at(0))
      self.[]=(name, options)
    end
    # rubocop:enable Layout/SpaceBeforeBrackets
  end
end