Added support for cookie management. Can now add and delete cookies - #…

…124
jonnnnyw · Nov 16, 2016 · 78887ba · 78887ba
1 parent 65564e8
commit 78887ba
Show file tree

Hide file tree

Showing 7 changed files with 338 additions and 30 deletions.
diff --git a/src/JonnyW/PhantomJs/Http/AbstractRequest.php b/src/JonnyW/PhantomJs/Http/AbstractRequest.php
@@ -36,6 +36,14 @@ abstract class AbstractRequest
      */
     protected $settings;
 
+    /**
+     * Cookies
+     *
+     * @var array
+     * @access protected
+     */
+    protected $cookies;
+
     /**
      * Request data
      *
@@ -118,6 +126,11 @@ public function __construct($url = null, $method = RequestInterface::METHOD_GET,
         $this->viewportWidth   = 0;
         $this->viewportHeight  = 0;
 
+        $this->cookies = array(
+            'add'    => array(),
+            'delete' => array()
+        );
+
         $this->setMethod($method);
         $this->setTimeout($timeout);
 
@@ -419,6 +432,63 @@ public function getSettings()
         return $this->settings;
     }
 
+    /**
+     * Add cookie.
+     *
+     * @access public
+     * @param  string                                 $name
+     * @param  mixed                                  $value
+     * @param  string                                 $path
+     * @param  string                                 $domain
+     * @param  bool                                   $httpOnly (default: true)
+     * @param  bool                                   $secure   (default: false)
+     * @param  int                                    $expires  (default: null)
+     * @return \JonnyW\PhantomJs\Http\AbstractRequest
+     */
+    public function addCookie($name, $value, $path, $domain, $httpOnly = true, $secure = false, $expires = null)
+    {
+        $filter = function ($value) {
+            return !is_null($value);
+        };
+
+        $this->cookies['add'][] = array_filter(array(
+            'name'     => $name,
+            'value'    => $value,
+            'path'     => $path,
+            'domain'   => $domain,
+            'httponly' => $httpOnly,
+            'secure'   => $secure,
+            'expires'  => $expires
+        ), $filter);
+
+        return $this;
+    }
+
+    /**
+     * Delete cookie.
+     *
+     * @access public
+     * @param  string                                 $name
+     * @return \JonnyW\PhantomJs\Http\AbstractRequest
+     */
+    public function deleteCookie($name)
+    {
+        $this->cookies['delete'][] = $name;
+
+        return $this;
+    }
+
+    /**
+     * Get cookies
+     *
+     * @access public
+     * @return array
+     */
+    public function getCookies()
+    {
+        return $this->cookies;
+    }
+
     /**
      * Set body styles
      *

diff --git a/src/JonnyW/PhantomJs/Http/RequestInterface.php b/src/JonnyW/PhantomJs/Http/RequestInterface.php
@@ -184,6 +184,22 @@ public function addHeaders(array $headers);
      */
     public function getHeaders();
 
+    /**
+     * Get settings
+     *
+     * @access public
+     * @return array|string
+     */
+    public function getSettings();
+
+    /**
+     * Get cookies
+     *
+     * @access public
+     * @return array|string
+     */
+    public function getCookies();
+
     /**
      * Set body styles
      *

diff --git a/src/JonnyW/PhantomJs/Resources/procedures/http_default.proc b/src/JonnyW/PhantomJs/Resources/procedures/http_default.proc
@@ -32,7 +32,6 @@ var page       = require('webpage').create(),
  */
 [[ engine.load('page_viewport_size') ]]
 
-
 /**
  * Define custom headers.
  */
@@ -43,6 +42,11 @@ var page       = require('webpage').create(),
  */
 [[ engine.load('page_settings') ]]
 
+/**
+ * Page cookies
+ */
+[[ engine.load('page_cookies') ]]
+
 /**
  * On resource timeout
  */

diff --git a/src/JonnyW/PhantomJs/Resources/procedures/http_lazy.proc b/src/JonnyW/PhantomJs/Resources/procedures/http_lazy.proc
@@ -34,7 +34,6 @@ var page       = require('webpage').create(),
  */
 [[ engine.load('page_viewport_size') ]]
 
-
 /**
  * Define custom headers.
  */
@@ -45,6 +44,11 @@ var page       = require('webpage').create(),
  */
 [[ engine.load('page_settings') ]]
 
+/**
+ * Page cookies
+ */
+[[ engine.load('page_cookies') ]]
+
 /**
  * On resource timeout
  */

diff --git a/src/JonnyW/PhantomJs/Resources/procedures/page_cookies.partial b/src/JonnyW/PhantomJs/Resources/procedures/page_cookies.partial
@@ -0,0 +1,23 @@
+
+var cookies = {{ input.getCookies()|json_encode() }};
+
+cookies.delete.forEach(function(name) {
+
+    if(name == '*') {
+        page.clearCookies();
+        debug.push(new Date().toISOString().slice(0, -5) + ' [INFO] PhantomJS - Deleted all cookies');
+    } else {
+
+        if(page.deleteCookie(name)) {
+            debug.push(new Date().toISOString().slice(0, -5) + ' [INFO] PhantomJS - Deleted cookie ' + name);
+        }
+    }
+});
+
+
+cookies.add.forEach(function(cookie) {
+
+    if(page.addCookie(cookie)) {
+        debug.push(new Date().toISOString().slice(0, -5) + ' [INFO] PhantomJS - Added cookie ' + cookie.name + '=' + cookie.value);
+    }
+});