封裝一個加密的storage工具類

/***
 * title: storage.js 
 * Desc: 對本地存儲進行封裝, 命名規(guī)范、設置過期時間、安全加密
 */

import CryptoJS from 'crypto-js';

// 十六位十六進制數(shù)作為密鑰
const SECRET_KEY = CryptoJS.enc.Utf8.parse('3333e6e143439161');
// 十六位十六進制數(shù)作為密鑰偏移量
const SECRET_IV = CryptoJS.enc.Utf8.parse('e3bbe7e3ba84431a');

// 類型 window.localStorage,window.sessionStorage,
const config = {
  type: 'sessionStorage', // 本地存儲類型 localStorage sessionStorage
  prefix: 'xxx_0.0.1', // 名稱前綴 建議:項目名 + 項目版本
  expire: 0, //過期時間 單位:秒
  isEncrypt: false // 默認加密 為了調試方便, 開發(fā)過程中可以不加密
};

// 判斷是否支持 Storage
export const isSupStorage = () => {
  if (!window) {
    throw new Error('當前環(huán)境非瀏覽器,無法消費全局window實例。');
  }
  if (!window.localStorage) {
    throw new Error('當前環(huán)境非無法使用localStorage');
  }
  if (!window.sessionStorage) {
    throw new Error('當前環(huán)境非無法使用sessionStorage');
  }

  return typeof Storage !== 'undefined' ? true : false;
};

// 設置 setStorage
export const setStorage = (key, value, expire = 0) => {
  if (value === '' || value === null || value === undefined) {
    value = null;
  }

  if (isNaN(expire) || expire < 0) throw new Error('Expire must be a number');

  expire = (expire ? expire : config.expire) * 1000;
  let data = {
    value: value, // 存儲值
    time: Date.now(), //存值時間戳
    expire: expire // 過期時間
  };
  const encryptString = config.isEncrypt
    ? encrypt(JSON.stringify(data))
    : JSON.stringify(data);
  window[config.type].setItem(autoAddPrefix(key), encryptString);
};

// 獲取 getStorage
export const getStorage = key => {
  let value = null;
  key = autoAddPrefix(key);
  // key 不存在判斷
  if (
    !window[config.type].getItem(key) ||
    JSON.stringify(window[config.type].getItem(key)) === 'null'
  ) {
    return null;
  }

  // 優(yōu)化 持續(xù)使用中續(xù)期
  const storage = config.isEncrypt
    ? JSON.parse(decrypt(window[config.type].getItem(key)))
    : JSON.parse(window[config.type].getItem(key));
  const nowTime = Date.now();
  // 過期刪除
  if (storage.expire && storage.expire < nowTime - storage.time) {
    removeStorage(key);
    return null;
  } else {
    // // 未過期期間被調用 則自動續(xù)期 進行保活
    // setStorage(autoRemovePrefix(key), storage.value);
    if (isJson(storage.value)) {
      value = JSON.parse(storage.value);
    } else {
      value = storage.value;
    }
    return value;
  }
};

// 是否存在 hasStorage
export const hasStorage = key => {
  key = autoAddPrefix(key);
  let arr = getStorageAll().filter(item => {
    return item.key === key;
  });
  return arr.length ? true : false;
};

// 獲取所有key
export const getStorageKeys = () => {
  let items = getStorageAll();
  let keys = [];
  for (let index = 0; index < items.length; index++) {
    keys.push(items[index].key);
  }
  return keys;
};

// 根據(jù)索引獲取key
export const getStorageForIndex = index => {
  return window[config.type].key(index);
};

// 獲取localStorage長度
export const getStorageLength = () => {
  return window[config.type].length;
};

// 獲取全部 getAllStorage
export const getStorageAll = () => {
  const len = getStorageLength(); // 獲取長度
  let arr = []; // 定義數(shù)據(jù)集
  for (let i = 0; i < len; i++) {
    const key = window[config.type].key(i);
    // 獲取key 索引從0開始
    const getKey = autoRemovePrefix(key);
    // 獲取key對應的值
    const storage = config.isEncrypt
      ? JSON.parse(decrypt(window[config.type].getItem(key)))
      : JSON.parse(window[config.type].getItem(key));

    const nowTime = Date.now();
    if (storage.expire && nowTime - storage.time > storage.expire) {
      removeStorage(getKey);
    } else {
      let getVal = storage.value;
      // console.log(Object.prototype.toString.call(value));
      if (isJson(getVal)) {
        getVal = JSON.parse(getVal);
      }
      // 放進數(shù)組
      arr.push({ key: getKey, val: getVal });
    }
  }
  return arr;
};

// 刪除 removeStorage
export const removeStorage = key => {
  window[config.type].removeItem(autoAddPrefix(key));
};

// 清空 clearStorage
export const clearStorage = () => {
  window[config.type].clear();
};

// 判斷是否可用 JSON.parse
export const isJson = value => {
  if (Object.prototype.toString.call(value) === '[object String]') {
    try {
      const obj = JSON.parse(value);
      const objType = Object.prototype.toString.call(obj);
      return objType === '[object Object]' || objType === '[object Array]';
    } catch (e) {
      // console.log('error:' + value + '!!!' + e);
      return false;
    }
  }
  return false;
};

// 名稱前自動添加前綴
const autoAddPrefix = key => {
  const prefix = config.prefix ? config.prefix + '_' : '';
  return prefix + key;
};

// 移除已添加的前綴
const autoRemovePrefix = key => {
  const len = config.prefix ? config.prefix.length + 1 : '';
  return key.substr(len);
};

/**
 * 加密方法
 * @param data
 * @returns {string}
 */
const encrypt = data => {
  if (typeof data === 'object') {
    try {
      data = JSON.stringify(data);
    } catch (error) {
      console.log('encrypt error:', error);
    }
  }
  const dataHex = CryptoJS.enc.Utf8.parse(data);
  const encrypted = CryptoJS.AES.encrypt(dataHex, SECRET_KEY, {
    iv: SECRET_IV,
    mode: CryptoJS.mode.CBC,
    padding: CryptoJS.pad.Pkcs7
  });
  return encrypted.ciphertext.toString();
};

/**
 * 解密方法
 * @param data
 * @returns {string}
 */
const decrypt = data => {
  const encryptedHexStr = CryptoJS.enc.Hex.parse(data);
  const str = CryptoJS.enc.Base64.stringify(encryptedHexStr);
  const decrypt = CryptoJS.AES.decrypt(str, SECRET_KEY, {
    iv: SECRET_IV,
    mode: CryptoJS.mode.CBC,
    padding: CryptoJS.pad.Pkcs7
  });
  const decryptedStr = decrypt.toString(CryptoJS.enc.Utf8);
  return decryptedStr.toString();
};

export default {
  install(Vue) {
    // 掛載全局
    if (!Vue.$storage) {
      Vue.$storage = {
        set: setStorage,
        get: getStorage,
        getAll: getStorageAll,
        getLen: getStorageLength,
        isSub: isSupStorage,
        isJson: isJson,
        has: hasStorage,
        del: removeStorage,
        clear: clearStorage
      };
    } else {
      Vue.$storage.set = setStorage;
      Vue.$storage.get = getStorage;
      Vue.$storage.getAll = getStorageAll;
      Vue.$storage.getLen = getStorageLength;
      Vue.$storage.isSub = isSupStorage;
      Vue.$storage.isJson = isJson;
      Vue.$storage.has = hasStorage;
      Vue.$storage.del = removeStorage;
      Vue.$storage.clear = clearStorage;
    }
    Vue.mixin({
      created: function () {
        this.$storage = Vue.$storage;
      }
    });
  }
};

?著作權歸作者所有,轉載或內容合作請聯(lián)系作者
【社區(qū)內容提示】社區(qū)部分內容疑似由AI輔助生成,瀏覽時請結合常識與多方信息審慎甄別。
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發(fā)布,文章內容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務。

相關閱讀更多精彩內容

友情鏈接更多精彩內容