solve issue medialize#338

fishead · Aug 30, 2017 · 34cb8ea · 34cb8ea
1 parent 61cd727
commit 34cb8ea
Show file tree

Hide file tree

Showing 3 changed files with 55 additions and 15 deletions.
diff --git a/src/URI.fragmentQuery.js b/src/URI.fragmentQuery.js
@@ -19,6 +19,10 @@
 // uri.toString() === "http://example.org/#?bar=foo&name=value";
 // uri.removeFragment("name");
 // uri.toString() === "http://example.org/#?bar=foo";
+// uri.setFragment("name", "value1");
+// uri.toString() === "http://example.org/#?bar=foo&name=value1";
+// uri.setFragment("name", "value2");
+// uri.toString() === "http://example.org/#?bar=foo&name=value2";
 
 (function (root, factory) {
   'use strict';
@@ -57,12 +61,12 @@
   p.fragment = function(v, build) {
     var prefix = this._parts.fragmentPrefix;
     var fragment = this._parts.fragment || '';
-  
+
     if (v === true) {
       if (fragment.substring(0, prefix.length) !== prefix) {
         return {};
       }
-    
+
       return URI.parseQuery(fragment.substring(prefix.length));
     } else if (v !== undefined && typeof v !== 'string') {
       this._parts.fragment = prefix + URI.buildQuery(v);
@@ -96,9 +100,22 @@
     this.build(!build);
     return this;
   };
+  p.setFragment = function(name, value, build) {
+    var prefix = this._parts.fragmentPrefix;
+    var data = URI.parseQuery((this._parts.fragment || '').substring(prefix.length));
+    URI.setQuery(data, name, value);
+    this._parts.fragment = prefix + URI.buildQuery(data);
+    if (typeof name !== 'string') {
+      build = value;
+    }
+
+    this.build(!build);
+    return this;
+  };
   p.addHash = p.addFragment;
   p.removeHash = p.removeFragment;
+  p.setHash = p.setFragment;
 
   // extending existing object rather than defining something new
   return URI;
-}));
+}));
diff --git a/src/URI.js b/src/URI.js
@@ -780,6 +780,21 @@
       throw new TypeError('URI.addQuery() accepts an object, string as the name parameter');
     }
   };
+
+  URI.setQuery = function(data, name, value) {
+    if (typeof name === 'object') {
+      for (var key in name) {
+        if (hasOwn.call(name, key)) {
+          URI.setQuery(data, key, name[key]);
+        }
+      }
+    } else if (typeof name === 'string') {
+      data[name] = value === undefined ? null : value;
+    } else {
+      throw new TypeError('URI.setQuery() accepts an object, string as the name parameter');
+    }
+  };
+
   URI.removeQuery = function(data, name, value) {
     var i, length, key;
 

diff --git a/test/test_fragmentQuery.js b/test/test_fragmentQuery.js
@@ -5,55 +5,63 @@
   module('URI.fragmentQuery');
   test('storing query-data in fragment', function() {
     var u = URI('http://example.org');
-  
+
     deepEqual(u.fragment(true), {}, 'empty map for missing fragment');
 
     u = URI('http://example.org/#');
     deepEqual(u.fragment(true), {}, 'empty map for empty fragment');
-  
+
     u = URI('http://example.org/#?hello=world');
     deepEqual(u.fragment(true), {hello: 'world'}, 'reading data object');
-  
+
     u.fragment({bar: 'foo'});
     deepEqual(u.fragment(true), {bar: 'foo'}, 'setting data object');
     equal(u.toString(), 'http://example.org/#?bar=foo', 'setting data object serialized');
 
     u.addFragment('name', 'value');
     deepEqual(u.fragment(true), {bar: 'foo', name: 'value'}, 'adding value');
     equal(u.toString(), 'http://example.org/#?bar=foo&name=value', 'adding value serialized');
-  
+
     u.removeFragment('bar');
     deepEqual(u.fragment(true), {name: 'value'}, 'removing value bar');
     equal(u.toString(), 'http://example.org/#?name=value', 'removing value bar serialized');
-  
+
     u.removeFragment('name');
     deepEqual(u.fragment(true), {}, 'removing value name');
     equal(u.toString(), 'http://example.org/#?', 'removing value name serialized');
+
+    u.setFragment('name', 'value1');
+    deepEqual(u.fragment(true), {name: 'value1'}, 'setting name to value1');
+    equal(u.toString(), 'http://example.org/#?name=value1', 'setting name to value1 serialized');
+
+    u.setFragment('name', 'value2');
+    deepEqual(u.fragment(true), {name: 'value2'}, 'setting name to value2');
+    equal(u.toString(), 'http://example.org/#?name=value2', 'setting name to value2 serialized');
   });
   test('fragmentPrefix', function() {
     var u;
-  
+
     URI.fragmentPrefix = '!';
     u = URI('http://example.org');
     equal(u._parts.fragmentPrefix, '!', 'init using global property');
-  
+
     u.fragment('#?hello=world');
     equal(u.fragment(), '?hello=world', 'unparsed ?');
     deepEqual(u.fragment(true), {}, 'parsing ? prefix');
-  
+
     u.fragment('#!hello=world');
     equal(u.fragment(), '!hello=world', 'unparsed !');
     deepEqual(u.fragment(true), {hello: 'world'}, 'parsing ! prefix');
-  
+
     u.fragmentPrefix('§');
     equal(u.fragment(), '!hello=world', 'unparsed §');
     deepEqual(u.fragment(true), {}, 'parsing § prefix');
-  
+
     u.fragment('#§hello=world');
     equal(u.fragment(), '§hello=world', 'unparsed §');
     deepEqual(u.fragment(true), {hello: 'world'}, 'parsing § prefix');
-  
+
     URI.fragmentPrefix = '?';
   });
 
-})();
+})();