grapesjs-plugins/grapesjs-storage-rate-limit/src/index.js

Rama main · réplica de lectura del repositorio original.

export default (editor, opts = {}) => {
  const options = { ...{
    // default options
    time: 1000, // ms
  },  ...opts };

  const storage = editor.StorageManager.getCurrentStorage();
  const storageOptions = editor.StorageManager.getCurrentOptions();

  editor.StorageManager.setCurrent('rate-limit')

  let isCoolingDown = false;
  let pendingSave = false;
  let latestData = null;

  // Prevent reload while cooling down and before saving
  window.addEventListener('beforeunload', (e) => {
    if (pendingSave) {
      e.preventDefault();
      e.returnValue = '';
    }
  });

  // Save the latest data once the cooldown period is over
  const doStore = async (data, isLater) => {
    // Start another cooldown period after saving
    startCooldown();
    // Save the data and trigger events
    try {
      if(isLater) {
        editor.trigger('storage:start', data)
        editor.trigger('storage:start:store', data)
      }
      // Save the data immediately
      await storage.store(data, storageOptions);
      if(isLater) {
        editor.trigger('storage:end')
        editor.trigger('storage:end:store', data)
        editor.trigger('storage:store', data)
      }
    } catch (err) {
      if(isLater) {
        editor.trigger('storage:error', err)
        editor.trigger('storage:error:store', err)
        editor.trigger('storage:end')
      } else {
        throw err;
      }
    }
  };

  // Start the cooldown period
  const startCooldown = () => {
    isCoolingDown = true;
    setTimeout(async () => {
      if (pendingSave) {
        // Reset the pending save flag
        pendingSave = false;
        // Save the latest data once cooldown is over
        const data = latestData;
        latestData = null;
        await doStore(data, true);
      } else {
        // Reset the cooldown flag
        isCoolingDown = false;
      }
    }, options.time);
  };

  // Add the rate-limit storage
  editor.Storage.add('rate-limit', {

    // Store the data
    // Adds a cooldown period before saving again
    async store(data) {
      if (isCoolingDown) {
        pendingSave = true;
        latestData = data;
      } else {
        await doStore(data, false);
      }
    },

    // Do nothing
    // Let the initial storage manager handle the load
    async load(options) {
      return storage.load(storageOptions);
    },
  });
};