JavaScript 获取二级域名的方法

Question

有个需求，将 cookie 设置到我们的二级域名上，在不同的项目都可以公用。

准备

1. 一个完整的域名由二个或二个以上部分组成，各部分之间用英文的句号 . 来分隔；
2. 在设置 cookie 时，如省略 domain 参数，那么它默认设置为当前域名；
3. domain 参数可以设置父域名以及当前域名。不能设置为其他域名，包括子域名也不行；
4. cookie 的作用域是 domain 本身以及 domain 下的所有子域名。(这里忽略 path 参数限制，当做 /)

思路

如何获取二级域名，以 shortcuts.sspai.com 为例：

接着我们对 URL 进行拆分，按先后顺序对 com、sspai.com、shortcuts.sspai.com （以此类推）设置 cookie。倘若能设置 cookie 则说明域名是合法的，此时停止继续往下执行，直接返回结果 sspai.com，同时删掉该 cookie。

其他

1. 需要注意的是，baidu.com 应属于二级域名。
2. 关于顶级域名、二级域名、三级域名等划分如理解有偏差，自行搜索。可参考
3. 当前采用部分 ES6 语法，若需兼容 ES5 自行修改即可。
4. 还有一些比较实用的 JS 方法，比如中文转拼音、身份证号码校验等。

实现

/**
 * 获取当前 URL 二级域名
 * 如果当前是 IP 地址，则直接返回 IP Address
 */
function getSubdomain() {
  try {
    let subdomain = ''
    const key = `mh_${Math.random()}`
    const expiredDate = new Date(0)
    const { domain } = document
    const domainList = domain.split('.')

    const reg = new RegExp(`(^|;)\\s*${key}=12345`)
    const ipAddressReg = /^(\d{1,2}|1\d\d|2[0-4]\d|25[0-5])\.(\d{1,2}|1\d\d|2[0-4]\d|25[0-5])\.(\d{1,2}|1\d\d|2[0-4]\d|25[0-5])\.(\d{1,2}|1\d\d|2[0-4]\d|25[0-5])$/

    // 若为 IP 地址、localhost，则直接返回
    if (ipAddressReg.test(domain) || domain === 'localhost') {
      return domain
    }

    const urlItems = []
    urlItems.unshift(domainList.pop())

    while (domainList.length) {
      urlItems.unshift(domainList.pop())
      subdomain = urlItems.join('.')

      const cookie = `${key}=12345;domain=.${subdomain}`
      document.cookie = cookie

      if (reg.test(document.cookie)) {
        document.cookie = `${cookie};expires=${expiredDate}`
        break
      }
    }

    return subdomain || document.domain
  } catch (e) {
    return document.domain
  }
}

参考

• 前端 JS 如何获取主域名(根域名)