Mailspring/packages/nylas-core/scheduler-utils.js

const ACCOUNTS_UNCLAIMED = 'accounts:unclaimed';
const ACCOUNTS_CLAIMED_PREFIX = 'accounts:id-';
const ACCOUNTS_FOR = (id) => `${ACCOUNTS_CLAIMED_PREFIX}${id}`;

const CONNECTION_COUNT_FOR = (id) => `connections:${id}`

const HEARTBEAT_FOR = (id) => `heartbeat:${id}`;
const HEARTBEAT_EXPIRES = 30; // 2 min in prod?

const CLAIM_DURATION = 10 * 60 * 1000; // 2 hours on prod?

const PubsubConnector = require('./pubsub-connector');
const MessageTypes = require('./message-types')

const forEachAccountList = (forEachCallback) => {
  const client = PubsubConnector.broadcastClient();
  return Promise.each(client.keysAsync(`accounts:*`), (key) => {
    const processId = key.replace('accounts:', '');
    return client.lrangeAsync(key, 0, 20000).then((foundIds) =>
      forEachCallback(processId, foundIds)
    )
  });
}

const assignPolicy = (accountId, policy) => {
  console.log(`Changing policy for ${accountId} to ${JSON.stringify(policy)}`)
  const DatabaseConnector = require('./database-connector');
  DatabaseConnector.forShared().then(({Account}) => {
    Account.find({where: {id: accountId}}).then((account) => {
      account.syncPolicy = policy;
      account.save()
    })
  });
}

const checkIfAccountIsActive = (accountId) => {
  const client = PubsubConnector.broadcastClient();
  const key = CONNECTION_COUNT_FOR(accountId);
  return client.getAsync(key).then((val) => val && val > 0)
}

const notifyAccountIsActive = (accountId) => {
  const client = PubsubConnector.broadcastClient();
  const key = CONNECTION_COUNT_FOR(accountId);
  client.incrAsync(key).then((val) => {
    client.expireAsync(key, 15 * 60 * 1000); // 15 min
    if (val === 1) {
      PubsubConnector.notify({
        accountId: accountId,
        type: MessageTypes.ACCOUNT_UPDATED
      });
    }
  });
}

module.exports = {
  ACCOUNTS_UNCLAIMED,
  ACCOUNTS_CLAIMED_PREFIX,
  ACCOUNTS_FOR,
  HEARTBEAT_FOR,
  HEARTBEAT_EXPIRES,
  CLAIM_DURATION,

  assignPolicy,
  forEachAccountList,
  notifyAccountIsActive,
  checkIfAccountIsActive,
}