violentmonkey/src/injected/content/cmd-run.js

import bridge from './bridge';
import { sendCmd } from './util-content';
import { INJECT_CONTENT } from '../util';

const { runningIds } = bridge;
const resolvedPromise = promiseResolve();
let badgePromise;
let numBadgesSent = 0;
let bfCacheWired;

export function Run(id, realm) {
  runningIds::push(id);
  bridge.ids::push(id);
  if (realm === INJECT_CONTENT) {
    bridge.invokableIds::push(id);
  }
  if (!badgePromise) {
    badgePromise = resolvedPromise::then(throttledSetBadge);
  }
  if (!bfCacheWired) {
    bfCacheWired = true;
    window::on('pageshow', evt => {
      // isTrusted is `unforgeable` per DOM spec so we don't need to safeguard its getter
      if (evt.isTrusted && evt.persisted) {
        sendCmd('SetBadge', runningIds);
      }
    });
  }
}

function throttledSetBadge() {
  const num = runningIds.length;
  if (numBadgesSent < num) {
    numBadgesSent = num;
    return sendCmd('SetBadge', runningIds)::then(() => {
      badgePromise = throttledSetBadge();
    });
  }
}