js-cookie源码学习

这篇文章最初发表在我自己搭建的站点js-cookie库源码学习

最近在做项目的时候，前端登录功能要做一个记住密码的功能。但开发用的框架中没有实现这个功能，所以我就想自己实现这个功能。实现起来其实很简单，就是每次用户在登录页面点击登录时，把用户输入的用户名和密码保存到cookie中就可以了，当用户再登录时，再从cookie中获取用户名和密码填充到表单中就可以了。当然，也可以选择保存在localStorage中，不过本文主要是想分析下用到的js-cookie这个轻量级的js库。

Document.cookie这个API比较"简陋"。这里先简单的介绍下Document.cookie这个API，主要是写入、读取、删除三个操作。

// 1. 写入cookie, 会追加到当前的cookie中
document.cookie = "username=abc"
// 2. 读取cookie，可以用正则匹配或者字符串解析查找的方法来读取
var usename = /username=(S+);/.exec(document.cookie)[1] // 这里只是简单的匹配了下，实际开发中要考虑很多情况
// 3. 删除cookie，设置某个cookie过期即可
document.cookie = "username=; expires=Thu, 01 Jan 1970 00:00:00 GMT"
// 4. 判断是否有某个cookie，其实和读取差不多
/username=(S+);/.test(document.cookie) // true有，false没有

可以看到原生的Document.cookie写入基本靠拼字符串，读取和判断是否有要靠正则或者字符串解析，删除则也是要拼字符串。这个操作不太方便，而且和我们平常用的API不太一样，比如我们常用的Map：

var map = new Map();
map.set(key, value);
map.get(key);
map.has(key);
map.delete(key);

它的api就是对"键值对"这种数据结构进行操作。Document.cookie其实也是key=value这种键值对结构，但是它没有提供像map这样的API接口，这显然不符合我们的“直觉”和使用习惯。

js-cookie库使用起来很简单，支持向浏览器中写入、读取、删除cookie三种操作，API简单也符合我们的使用习惯。那么它是如何实现这三个操作的呢？这就要分析它的源码了。
注意：这里分析的是它的releases-v2.2.0版本。github上的master分支源码已经修改，主要是把set和get方法重构了，拆分成了两个多带带的方法，releases-v2.2.0版本set和get方法是写在一起的。下面简单分析下它的源码实现：set和get。remove是set的一种特殊情况，增加expires属性即可，这里就不细说了。

function api (key, value, attributes) {
  // set 主要逻辑
  if (arguments.length > 1) {
  attributes = extend({
    path: "/"
  }, api.defaults, attributes);

  if (typeof attributes.expires === "number") {
    var expires = new Date();
    expires.setMilliseconds(expires.getMilliseconds() + attributes.expires * 864e+5);
    attributes.expires = expires;
  }

  // We"re using "expires" because "max-age" is not supported by IE
  attributes.expires = attributes.expires ? attributes.expires.toUTCString() : "";

  try {
    // 这里需要注意，用户有可能要保存json格式的cookie值，所以这里需要转化下
    result = JSON.stringify(value);
    if (/^[{[]/.test(result)) {
      // 如果转化成了合法的json，则将value重新赋值为json字符串，如果不含有{或[，则不是json字符串，也就不会走这个分支
      value = result;
    }
  } catch (e) {}

  // 这里为什么要把value先编码再解码呢？，下面的key也是，不过key要解码的unicode值少些
  if (!converter.write) {
    // 内置的编码转换
    // ["#", "$", "&", "+", ":", "<", ">", "=", "/", "?", "@", "[", "]", "^", "`", "{", "}", "|"]
    value = encodeURIComponent(String(value))
      .replace(/%(23|24|26|2B|3A|3C|3E|3D|2F|3F|40|5B|5D|5E|60|7B|7D|7C)/g, decodeURIComponent);
  } else {
    value = converter.write(value, key);
  }

  // 先编码
  key = encodeURIComponent(String(key));
  // ["#", "$", "&", "+", "^", "`", "|"]
  // 再通过字符串替换的方式解码
  key = key.replace(/%(23|24|26|2B|5E|60|7C)/g, decodeURIComponent);
  // ECMA-262 standard规范已经不推荐用这个escape函数了  https://developer.mozilla.org/en-US/docs/Web/JavaScript/Reference/Global_Objects/escape
  // 而且 ()也不会被编码，所以感觉下面的这句是没有必要的
  key = key.replace(/[()]/g, escape);

  // 拼接其它的cookies属性值
  var stringifiedAttributes = "";

  for (var attributeName in attributes) {
    if (!attributes[attributeName]) {
      continue;
    }
    stringifiedAttributes += "; " + attributeName;
    if (attributes[attributeName] === true) {
      continue;
    }
    stringifiedAttributes += "=" + attributes[attributeName];
  }
  return (document.cookie = key + "=" + value + stringifiedAttributes);
  }
}

set方法在写入cookie时会先对cookie值encodeURIComponent然后再decodeURIComponent，这样可以保证存储再cookie中的值始终是不编码的可读性性好的字符串。

function api (key, value, attributes) {
  // Read

  if (!key) {
    result = {};
  }

  // To prevent the for loop in the first place assign an empty array
  // in case there are no cookies at all. Also prevents odd result when
  // calling "get()"
  var cookies = document.cookie ? document.cookie.split("; ") : [];
  var rdecode = /(%[0-9A-Z]{2})+/g;
  var i = 0;

  for (; i < cookies.length; i++) {
    var parts = cookies[i].split("=");
    var cookie = parts.slice(1).join("=");

    if (!this.json && cookie.charAt(0) === """) {
      // 没看懂为什么需要这个if分支
      cookie = cookie.slice(1, -1); // 去掉了 " "
    }

    try {
      var name = parts[0].replace(rdecode, decodeURIComponent);
      cookie = converter.read ?
        converter.read(cookie, name) : converter(cookie, name) ||
        cookie.replace(rdecode, decodeURIComponent);

      if (this.json) {
        try {
          cookie = JSON.parse(cookie);
        } catch (e) {}
      }

      if (key === name) {
        result = cookie;
        break;
      }

      if (!key) {
        result[name] = cookie;
      }
    } catch (e) {}
  }

  return result;
}

通过上面的get、set以及下面的代码可以看到整个库其实返回的就是 function api() {}这个函数，
set、get、remove的逻辑都写在api这个函数里了，所以api这个函数看起来很长，而且由于其中涉及到许多细节处理导致逻辑也比较复杂，那么库的作者应该是意识到了这一点，所以在github的master分支上的代码已经把get和set拆分开了，感兴趣的可以去看看js-cookie@master。

function init(converter) {
  function api (key, value, attributes) {
      api.set = api; // Cookie.set(key, value, attributes)
      api.get = function (key) { // Cookie.get(key)
        return api.call(api, key);
      };
      api.getJSON = function () { // Cookie.getJSON(key)
        return api.apply({
          json: true
        }, [].slice.call(arguments));
      };
      api.defaults = {};

      api.remove = function (key, attributes) { // Cookie.remove(key, attributes)
        api(key, "", extend(attributes, {
          expires: -1
        }));
      };

      api.withConverter = init; // 支持自定义的编码方式

      // 返回的是 function api() {}
      return api;
  }
  // 返回的是 function api() {}
  return init(function () {});
}

js-cookie声称支持AMD规范和CommonJS规范，那么它是如何支持这两种规范的呢？

;(function (factory) {
    var registeredInModuleLoader = false;
    if (typeof define === "function" && define.amd) { // AMD
        define(factory);
        registeredInModuleLoader = true;
    }
    if (typeof exports === "object") { // CommonJS
        module.exports = factory();
        registeredInModuleLoader = true;
    }
    if (!registeredInModuleLoader) { // 浏览器
        var OldCookies = window.Cookies;
        var api = window.Cookies = factory();
        api.noConflict = function () {
            window.Cookies = OldCookies;
            return api;
        };
    }
}(function () {
  // factory 逻辑...

  // 返回的是 function api() {}
    return init(function () {});
});

其实也很简单，现在很多js库基本都是这么实现的。

js-cookie库的源码很简单，但是它的v2.2.0-release版本将set和get逻辑都写在一个方法的方式其实是不提倡这么做的。这么做需要用if分支来做很多过滤，会导致代码逻辑较复杂。上面也看到，暴露接口时，它用到了function.call()和function.apply()，以及代码内部用到的已经不推荐使用的arguments，这都使代码的可读性降低了。不过它里面对于cookie的处理过程仍然值得参考。其实MDN上也提供了一个轻量的处理cookie的库一个完整支持unicode的cookie读取/写入器，有时间也可以看看。

MDN: Document.cookie
GitHub: js-cookie
AMD 异步模块定义
CommonJS