page.rb in digger-0.1.8

- old
+ new

@@ -2,10 +2,11 @@
 require 'json'
 require 'ostruct'
 require 'set'
 require 'kconv'
 require 'uri'
+require 'http/cookie'
 
 # https://github.com/taganaka/polipus/blob/master/lib/polipus/page.rb
 module Digger
   class Page
     attr_reader :url
@@ -99,10 +100,14 @@
 
     def jsonp
       @jsonp ||= JSON.parse body.match(/^[^(]+?\((.+)\)[^)]*$/)[1]
     end
 
+    def cookies
+      @cookies ||= (headers['set-cookie'] || []).flat_map { |c| ::HTTP::Cookie.parse(c, url) }
+    end
+
     #
     # Discard links, a next call of page.links will return an empty array
     #
     def discard_links!
       @links = []
@@ -271,6 +276,6 @@
     def self.from_json(json)
       hash = JSON.parse json
       from_hash hash
     end
   end
-end
\ No newline at end of file
+end