From fa52e0126d44046d3bb8ff19296bf153de665f16 Mon Sep 17 00:00:00 2001
From: Ilya Kreymer <ikreymer@gmail.com>
Date: Tue, 15 Jul 2014 12:52:42 -0700
Subject: [PATCH] cookies: support client side rewriting of document.cooke ->
 WB_wombat_cookie to rewrite cookie path, if present

---
 pywb/rewrite/regex_rewriters.py |  1 +
 pywb/static/wombat.js           | 28 ++++++++++++++++++++++++++++
 2 files changed, 29 insertions(+)

diff --git a/pywb/rewrite/regex_rewriters.py b/pywb/rewrite/regex_rewriters.py
index c2359c1e..3f440eee 100644
--- a/pywb/rewrite/regex_rewriters.py
+++ b/pywb/rewrite/regex_rewriters.py
@@ -126,6 +126,7 @@ class JSLinkAndLocationRewriter(JSLinkOnlyRewriter):
              (r'(?<!/)\blocation\b', RegexRewriter.add_prefix(prefix), 0),
              (r'(?<=document\.)domain', RegexRewriter.add_prefix(prefix), 0),
              (r'(?<=document\.)referrer', RegexRewriter.add_prefix(prefix), 0),
+             (r'(?<=document\.)cookie', RegexRewriter.add_prefix(prefix), 0),
 
             #todo: move to mixin?
              (r'(?:[\s=(){]|^)(top)(?:[\s!=}()]|$)',
diff --git a/pywb/static/wombat.js b/pywb/static/wombat.js
index c765e683..9c6ba1c0 100644
--- a/pywb/static/wombat.js
+++ b/pywb/static/wombat.js
@@ -639,6 +639,31 @@ WB_wombat_init = (function() {
         }
     }
     
+    function init_cookies_override()
+    {
+        var cookie_path_regex = /\bPath=\'?\"?([^;'"\s]+)/i;
+        
+        var get_cookie = function() {
+            return document.cookie;
+        }
+        
+        var set_cookie = function(value) {
+            var matched = value.match(cookie_path_regex);
+                        
+            // if has cookie path, rewrite and replace
+            if (matched) {
+                var rewritten = rewrite_url(matched[1]);
+                value = value.replace(matched[1], rewritten);
+            }
+            
+            document.cookie = value;
+        }
+        
+        def_prop(document, "WB_wombat_cookie", document.cookie,
+                set_cookie,
+                get_cookie);
+    }
+    
     //============================================
     function init_write_override()
     {
@@ -747,6 +772,9 @@ WB_wombat_init = (function() {
         // Ajax
         init_ajax_rewrite();
         init_worker_override();
+        
+        // Cookies
+        init_cookies_override();
 
         // DOM
         init_dom_override();