index.js

const { v4: uuidv4 } = require('uuid');

function getInstance(prefix) {
  function generateUUID() {
    return prefix + uuidv4();
  }

  function isRef(value) {
    return typeof value === 'string' && value.startsWith(prefix);
  }

  return {
    parse: (jsonString) => {
      const refTargets = {};
      const replacers = [];

      const obj = JSON.parse(jsonString, function(key, value) {
        // 如果 key 是 @id 则把 this 放到引用池里备用
        if (key == '@id') {
          refTargets[value] = this;
          // 这里避免问题，直接不返回 @id 本身
          return undefined;
        }
        // 如果 value 是一个 ref 格式的字符串，则注册一个替换回调并原样返回，在后面再执行替换
        // 因为 ref 可能会先于它引用的对象被遍历到，所以要后置再替换
        if (isRef(value)) {
          replacers.push(() => {
            const refTarget = refTargets[value];
            if (refTarget == null) {
              throw new Error(`Can not find object for refId: ${value}`);
            }
            this[key] = refTarget;
          });
          return value;
        }
        // 除此之外的情况都直接返回
        return value;
      });
      // 执行前面注册的替换回调
      replacers.forEach(replacer => replacer());

      return obj;
    },
    stringify: (obj) => {
      const objIdMap = new Map()
      return JSON.stringify(obj, function(key, value) {
        // 只处理 object
        if (isObj(value)) {
          const objId = objIdMap.get(value);
          // 如果 object 没有 @id ，生成一个
          if (objId === undefined) {
            const id = generateUUID();
            objIdMap.set(value, id);
            return { ...value, '@id': id };
          }
          // 如果有，返回 @id 的值
          return objId;
        }
        // 其余情况直接返回
        return value;
      });
    }
  };
}

function isObj(obj) {
  return !Array.isArray(obj) && typeof obj === 'object' && obj != null;
}

const defaultInstance = getInstance('@id:');

module.exports = {
  parse: defaultInstance.parse,
  stringify: defaultInstance.stringify,
  getInstance
}