class HTTP::CookieJar::YAMLSaver

def default_options
  {}
end

def load(io, jar)
  begin
    data = load_yaml(io)
  rescue ArgumentError => e
    case e.message
    when %r{\Aundefined class/module Mechanize::}
      # backward compatibility with Mechanize::Cookie
      begin
        io.rewind # hopefully
        yaml = io.read
        # a gross hack
        yaml.gsub!(%r{^(    [^ ].*:) !ruby/object:Mechanize::Cookie$}, "\\1")
        data = load_yaml(yaml)
      rescue Errno::ESPIPE
        @logger.warn "could not rewind the stream for conversion" if @logger
      rescue ArgumentError
      end
    end
  end
  case data
  when Array
    data.each { |cookie|
      jar.add(cookie)
    }
  when Hash
    # backward compatibility with Mechanize::Cookie
    data.each { |domain, paths|
      paths.each { |path, names|
        names.each { |cookie_name, cookie_hash|
          if cookie_hash.respond_to?(:ivars)
            # YAML::Object of Syck
            cookie_hash = cookie_hash.ivars
          end
          cookie = HTTP::Cookie.new({}.tap { |hash|
              cookie_hash.each_pair { |key, value|
                hash[key.to_sym] = value
              }
            })
          jar.add(cookie)
        }
      }
    }
  else
    @logger.warn "incompatible YAML cookie data discarded" if @logger
    return
  end
end

def load_yaml(yaml)
  YAML.safe_load(yaml, :permitted_classes => %w[Time HTTP::Cookie Mechanize::Cookie DomainName], :aliases => true)
end

def load_yaml(yaml)
  YAML.load(yaml)
end

def save(io, jar)
  YAML.dump(@session ? jar.to_a : jar.reject(&:session?), io)
end