造一个 js-cookie 轮子

简介: Cookie 相信大家都不陌生,但是很多时候我们都是这样:“啊,我这个地方要用 Cookie 了,怎么办?没事,装一个 npm 包嘛”,或者去 MDN 去抄一两个函数。没什么机会手写一个 js-cookie 的库,今天就带大家一起来写一个 js-cookie 的小库。

image.png


项目源码:github.com/Haixiang612…

预览链接:yanhaixiang.com/my-js-cooki…

参考轮子:www.npmjs.com/package/js-…


Cookie 相信大家都不陌生,但是很多时候我们都是这样:“啊,我这个地方要用 Cookie 了,怎么办?没事,装一个 npm 包嘛”,或者去 MDN 去抄一两个函数。没什么机会手写一个 js-cookie 的库,今天就带大家一起来写一个 js-cookie 的小库。


从零开始


首先,我们要摒弃所有所谓的“设计模式”,做一个最 Low 的版本:只有 get(key)set(key, value)del(key) 3 个 API。

通过对 MDN、菜鸟教程、掘金博客的大量阅读,很快就写出了最简单的 API。


get

document.cookie 长这样:a=1&b=2。将 document.cookie 字符串转化成 Object,在转化过程中判断是否存在对应的 key,如果有就返回对应的 value 即可。


function get(key: string): string | null {
  const cookiePairs = document.cookie ? document.cookie.split('; ') : []
  const cookieStore: Record<string, string> = {}
  cookiePairs.some(pair => {
    const [curtKey, ...curtValues] = pair.split('=')
    cookieStore[curtKey] = curtValues.join('=') // 有可能 value 存在 '='
    return curtKey === key // 如果相等时,就会 break
  })
  return key ? cookieStore[key] : null
}
复制代码


注意:cookie 的值有可能里会有 '=' 号,所以split('=') 后的,还要再 join('=') 一下变回原来的值。比如:a=123=456,join 后的 value 还是 123=456而不是 123


set

单纯 set 或 add 一个 cookie 更简单,只需要

document.cookie = `${key}=${value}`
复制代码


但其实,一个 cookie 不只有 key 和 value,还有 expires 过期时间以及 path 路径。一个完整的 set 应该长这样。

document.cookie = `${key}=${value}; expires=${expires}; path=${path}`
复制代码


当然,我们不希望 set 函数的入参变得很冗余,所以这里的入参设计为:key, value, attributes 3 个。其中,attributes 是个对象,里面为 cookie 的属性:expires, path。

interface Attributes {
  path: string; // Cookie 对应路径
  expires?: string | number | Date // Cookie 的过期时间,第N天过期
}
复制代码


为了提高扩展性,我们再造一个 defaultAttributes 作为默认参数传入。

const TWENTY_FOUR_HOURS = 864e5
const defaultAttributes: Attributes = {path: '/'}
function set(key: string, value: string, attributes = defaultAttributes): string | null {
  attributes = {...defaultAttributes, ...attributes}
  if (attributes.expires) {
    // 将过期天数转为 UTC string
    if (typeof attributes.expires === 'number') {
      attributes.expires = new Date(Date.now() + attributes.expires * TWENTY_FOUR_HOURS)
      attributes.expires = attributes.expires.toUTCString()
    }
  }
  // 获取 Cookie 其它属性的字符串形式,如 "; expires=1; path=/"
  const attrStr = Object.entries(attributes).reduce((prevStr, attrPair) => {
    const [attrKey, attrValue] = attrPair
    if (!attrValue) return prevStr
    prevStr += `; ${attrKey}`
    // attrValue 有可能为 truthy,所以要排除 true 值的情况
    if (attrValue === true) return prevStr
    // 排除 attrValue 存在 ";" 号的情况
    prevStr += `=${attrValue.split('; ')[0]}`
    return prevStr
  }, '')
  return document.cookie = `${key}=${value}${attrStr}`
}
复制代码


上面的操作也非常简单。首先对 expires 做了转成 UTC 时间戳的处理,然后把 attributes 拍扁成一个 string,最后追加到 ${key}=${value} 后面。

这里可能有人会对这段神秘代号 864e5 感兴趣。这是 24 小时的毫秒值,具体可见 Stackoverflow


del

删除一个 cookie 一件再简单不过的事了。上面不是已经实现了 set 了么,我们把 expires 设置为 -1 天就好了。

/**
 * 删除某个 Cookie
 */
function del(key: string, attributes = defaultAttributes) {
  // 将 expires 减 1 天,Cookie 自动失败
  set(key, '', {...attributes, expires: -1})
}
复制代码


编码与解码


虽然没人要求 cookie 要做编码与解码,但是为了更 cookie 不受一些特殊字符的干扰,我们还要需要对 cookie 的值做编码与解码的工作。


这里普及一下:对于 cookie 的行为是有规范,从 RFC 2109RFC 2965 再到 RFC6265。有兴趣的可以看一看。好的,我知道你没有兴趣了。


咳咳,回到代码。这一步需要在 set 里做编码,在 get 里做解码。一般来说,习惯用 encodeURIComponent 和 decodeURIComponent 做编码和解码的工作。

function get(key: string): string | null {
  ...
  cookiePairs.some(pair => {
    const [curtKey, ...curtValue] = pair.split('=')
    try {
      // 解码
      const decodeedValue = decodeURIComponent(curtValue.join('='))  // 有可能 value 存在 '='
      cookieStore[curtKey] = decodeedValue
    } catch (e) {}
    return curtKey === key // 如果相等时,就会 break
  })
  return key ? cookieStore[key] : null
}
function set(key: string, value: string, attributes = defaultAttributes): string | null {
  ...
  // 编码
  value = encodeURIComponent(value)
  ...
  return document.cookie = `${key}=${value}${attrStr}`
}
复制代码


so easy ~ 不过,上面的 encodeURIComponentdecodeURIComponent 有点像硬编码一样写在了代码里了,不妨抽象出来用 defaultConverter 来封装编码和解码两个操作。

export interface Converter {
  encode: (text: string) => string // 编码
  decode: (text: string) => string // 解码
}
// 默认 Cookie 值的转换器
export const defaultConverter: Converter = {
  encode(text: string) {
    return text.replace(ASCII_HEX_REGEXP, encodeURIComponent)
  },
  decode(text: string) {
    return text.replace(ASCII_HEX_REGEXP, decodeURIComponent)
  },
}
复制代码


set 和 get 函数将会更抽象了。

function get(key: string): string | null {
  ...
      // 解码
      const decodeedValue = defaultConverter.decode(curtValue.join('='))  // 有可能 value 存在 '='
  ...
}
function set(key: string, value: string, attributes = defaultAttributes): string | null {
  ...
  // 编码
  value = defaultConverter.encode(value)
  ...
}
复制代码


配置中心


上面只是“我们觉得”习惯上会用 encodeURIComponentdecodeURIComponent 来编码和解码。别人可能会用别的编码和解码函数来完成,因此需要提供一个配置中心给开发者。一次配置,以后都会按照初始设置来 setget


像下面的例子,初始时设定 expires 为 1 天,以后直接 set(xxx, yyy) 设置 Cookie 过期时间都是 1 天后。

// 初始配置
Cookies.atributes = { expires: 1 }
Cookies.converter = {
  encode(text: string) {
    return "hello"
  },
  decode(text: string) {
    return "world"
  },
}
Cookies.set('aaa', 111) // 过期时间为 1 天,值 aaa="hello"
Cookies.set('bbb', 222) // 过期时间为 1 天,值 bbb="hello"
Cookies.get('aaa') // "world"
Cookies.get('bbb') // "world"
复制代码


要实现上面的效果,我们需要首先提供一个初始配置中心的入口,然后暴露配置中心。而且还需要将 attributes 和 converter 配置存下来。

let customAttributes: Attributes = defaultAttributes
let customConverter: Converter = defaultConverter
function get(key: string): string | null {
  ...
  const decodedValue = customConverter.decode(curtValue.join('='))  // 有可能 value 存在 '='
  ...
}
/**
 * 设置 Cookie key-val 对
 */
function set(key: string, value: string, attributes = defaultAttributes): string | null {
  attributes = {...customAttributes, ...attributes}
  ...
  value = customConverter.encode(value)
  ...
}
/**
 * 删除某个 Cookie
 */
function del(key: string, attributes = defaultAttributes) {
  // 将 expires 减 1 天,Cookie 自动失败
  set(key, '', {...attributes, expires: -1})
}
const Cookies = {
  get,
  set,
  del,
  attributes: customAttributes,
  converter: customConverter,
}
export default Cookies
复制代码


上面导出了一个函数,每次使用 attributes 的时候都会先和 customAttributes 合并,每次编码解码的时候会使用 customCoverter

上面这么实现在使用的时候会很麻烦,每次修改配置就要手动去做合并。

Cookies.attributes = {...Cookies.attributes, ...{ expires: 2 } }
复制代码


那我们想:好吧,暴露两个函数做合并呗。

...
function withAttributes(myAttributes: Attribute) {
  customAttributes = {...customAttributes, ...myAttributes}
}
function withAttributes(myConverter: Converter) {
  customConverter = {...customConverter, ...myConverter}
}
const Cookies = {
  get,
  set,
  del,
  withAttributes,
  withConverter
}
export default Cookies
复制代码


还有没有问题?有!把 customAttributescustomConverter 放到全局很容易造成污染。想象一下,这个项目很大,有人偷偷把 customAttributes 里的 expires 改成 3 天,下个要开发的人可能完全不知情。所以,把这配置项改为局部是十分有必要的。


直接给出实现:

function init(initConverter: Converter, initAttributes: Attributes) {
  function get(key: string): string | null {
    ...
    const decodeedValue = initConverter.decode(curtValue.join('='))
    ...
  }
  function set(key: string, value: string, attributes = customAttributes): string | null {
    ...
    attributes = {...initAttributes, ...attributes}
    value = initConverter.encode(value)
    ...
  }
  function del(key: string, attributes = customAttributes) {
    set(key, '', {...attributes, expires: -1})
  }
  function withConverter(customConverter: Converter) {
    return init({...this.converter, ...customConverter}, this.attributes)
  }
  function withAttributes(customAttributes: Attributes) {
    return init(this.converter, {...this.attributes, ...customAttributes})
  }
  return {
    get,
    set, 
    del, 
    attributes: initAttributes,
    converter: initConverter, 
    withAttributes,
    withConverter
  }
}
export default init(defaultConverter, defaultAttributes)
复制代码


上面把配置项存放到了生成对象的 attributesconverter 里了。调用 withConverterwithAttributes 的时候,再次调用 init 来创建 Cookies 对象。好处是 withXXX 后是一个全新的对象,不会造成全局污染。

const myCookies = Cookies.withAttributes({ expires: 2 }) // 新对象
attrCookies.set('aaa', 1) // 2 天后过期
Coookies.set('aaa', 1) // 没有过期时间
复制代码


把配置项“冻结”


上面的代码还有个问题,attributesconverter 还是被暴露了出来,万一哪个憨憨手抖改了,后面的接盘侠还是会傻眼。


这里可以用 Object.create 来生成对象,并在第 2 个参数里用 Object.freeze 把对象 atributesconverter“冻住”。

function init(initConverter: Converter, initAttributes: Attributes) {
  ...
  return Object.create(
    {get, set, del, withConverter, withAttributes},
    {
      converter: {value: Object.freeze(initConverter)}, // 被冻动了
      attributes: {value: Object.freeze(initAttributes)}, // 被冻动了
    }
  )
}
export default init(defaultConverter, defaultAttributes)
复制代码


关于 Object.create 第 2 个参数的内容可以看 Object.defineProperties,它的意义是描述对象属性,这里的描述就是“冻动”了。比如:

Cookies.attributes = 1
console.log(Cookies.attributes) // 返回依然是 {path: '/'},不会变成 1
复制代码

到此,一个 js-cookie 库已经完美实现了!


总结


init 函数创建对象,对象里有以下函数


  1. get 函数:将 document.cookie 字符串转化为 Object 形式,转化过程中判断是否在存 key,如果有就返回对应 value
  2. set 函数:把 attributes stringify,然后追加到 key=value 后, document.cookie = ${key}=${value}${attrStr}
  3. del 函数:调用 set,把 expires 设置为 -1 天,cookie 直接过期被删除
  4. withAttributes:更新 attributes 配置,并返回全新 Cookie 对象
  5. withConverter:更新 converter 配置,并返回全新 Cookie 对象


为什么要用函数生成对象这么麻烦?因为要解决全局污染的问题。需要把 attributesconverter 两个配置存到函数参数里,并且通过 withAttributeswithConverter 调用 init 返回新 Cookie 对象。


为什么要冻动 attributesconverter,还是因为怕有憨憨把这两玩意改了。


最后


上面的代码其实就是 js-cookie 的核心代码了。

当然这个库里对一些特殊字符处理的代码没有过多提及,因为纠结这些过于细节的代码意义并不大。而且上面已经做了一些特殊字符处理了,已经涵盖大部分使用情况了。



相关文章
|
5月前
|
JavaScript iOS开发
js常用的代码有哪些 老司机带你来看看
js常用的一些代码分享给大家
30 4
|
JavaScript
开心档-软件开发入门之Vue.js 响应接口
开心档-软件开发入门之Vue.js 响应接口
56 0
|
中间件
一文带你弄懂nest.js文件上传
一文带你弄懂nest.js文件上传
|
前端开发 JavaScript
前端学习笔记202307学习笔记第六十五天-模拟面试-手写js方法4
前端学习笔记202307学习笔记第六十五天-模拟面试-手写js方法4
51 0
|
前端开发 JavaScript
前端学习笔记202307学习笔记第六十五天-模拟面试-手写js方法3
前端学习笔记202307学习笔记第六十五天-模拟面试-手写js方法3
41 0
|
前端开发 JavaScript
前端学习笔记202307学习笔记第六十五天-模拟面试-手写js方法5
前端学习笔记202307学习笔记第六十五天-模拟面试-手写js方法5
47 0
|
存储 JavaScript 前端开发
前端祖传三件套JavaScript的BOM的Cookie
在前端开发中,Cookie 是一个非常重要的概念。它可以用来存储用户的信息、状态和偏好设置等。JavaScript BOM 提供了内置的 document.cookie 属性和相关方法,可以让我们轻松地读取、创建、修改和删除 Cookie。在本文中,我们将深入探讨 JavaScript BOM 的 Cookie 功能,并介绍其中一些常用的属性和用法。
104 0
|
设计模式 JavaScript 前端开发
【再来亿遍 温故知新】—— 关于 JS 原型你必须要知道的二三
本瓜一向认为:学习不是一蹴而就的事情。一定是要求学习者对知识点进行反复咀嚼拿捏、不断打破重塑,长此以往,才以期达到融会贯通、为我所用的程度。所谓:温故知新,不亦乐乎?
|
JSON JavaScript 前端开发
JS案例:如何用300行原生js代码写出高大上的购物车
JS案例:如何用300行原生js代码写出高大上的购物车
278 0
JS案例:如何用300行原生js代码写出高大上的购物车
|
存储 JSON 前端开发
怎么防止同事用Evil.js的代码投毒
项目被发布到npm上后,引起了激烈的讨论,最终因为安全问题被npm官方移除,代码也闭源了 作为一个前端老司机,我肯定是反对这种行为,泄私愤有很多种方式,代码里下毒会被git log查到,万一违法了,还不如离职的时候给老板一个大逼兜来的解恨
197 0
怎么防止同事用Evil.js的代码投毒