metamask-extension/app/scripts/lib/segment.js

import Analytics from 'analytics-node'

const isDevOrTestEnvironment = Boolean(
  process.env.METAMASK_DEBUG || process.env.IN_TEST,
)
const SEGMENT_WRITE_KEY = process.env.SEGMENT_WRITE_KEY ?? null
const SEGMENT_LEGACY_WRITE_KEY = process.env.SEGMENT_LEGACY_WRITE_KEY ?? null
const SEGMENT_HOST = process.env.SEGMENT_HOST ?? null

// flushAt controls how many events are sent to segment at once. Segment will
// hold onto a queue of events until it hits this number, then it sends them as
// a batch. This setting defaults to 20, but in development we likely want to
// see events in real time for debugging, so this is set to 1 to disable the
// queueing mechanism.
const SEGMENT_FLUSH_AT =
  process.env.METAMASK_ENVIRONMENT === 'production' ? undefined : 1

// flushInterval controls how frequently the queue is flushed to segment.
// This happens regardless of the size of the queue. The default setting is
// 10,000ms (10 seconds). This default is rather high, though thankfully
// using the background process as our event handler means we don't have to
// deal with short lived sessions that happen faster than the interval
// e.g confirmations. This is set to 5,000ms (5 seconds) arbitrarily with the
// intent of having a value less than 10 seconds.
const SEGMENT_FLUSH_INTERVAL = 5000

/**
 * Creates a mock segment module for usage in test environments. This is used
 * when building the application in test mode to catch event calls and prevent
 * them from being sent to segment. It is also used in unit tests to mock and
 * spy on the methods to ensure proper behavior
 * @param {number} flushAt - number of events to queue before sending to segment
 * @param {number} flushInterval - ms interval to flush queue and send to segment
 * @returns {SegmentInterface}
 */
export const createSegmentMock = (
  flushAt = SEGMENT_FLUSH_AT,
  flushInterval = SEGMENT_FLUSH_INTERVAL,
) => {
  const segmentMock = {
    // Internal queue to keep track of events and properly mimic segment's
    // queueing behavior.
    queue: [],

    /**
     * Used to immediately send all queued events and reset the queue to zero.
     * For our purposes this simply triggers the callback method registered with
     * the event.
     */
    flush() {
      segmentMock.queue.forEach(([_, callback]) => {
        callback()
      })
      segmentMock.queue = []
    },

    /**
     * Track an event and add it to the queue. If the queue size reaches the
     * flushAt threshold, flush the queue.
     */
    track(payload, callback = () => undefined) {
      segmentMock.queue.push([payload, callback])

      if (segmentMock.queue.length >= flushAt) {
        segmentMock.flush()
      }
    },

    /**
     * A true NOOP, these methods are either not used or do not await callback
     * and therefore require no functionality.
     */
    page() {
      // noop
    },
    identify() {
      // noop
    },
  }
  // Mimic the flushInterval behavior with an interval
  setInterval(segmentMock.flush, flushInterval)
  return segmentMock
}

export const segment =
  !SEGMENT_WRITE_KEY || (isDevOrTestEnvironment && !SEGMENT_HOST)
    ? createSegmentMock(SEGMENT_FLUSH_AT, SEGMENT_FLUSH_INTERVAL)
    : new Analytics(SEGMENT_WRITE_KEY, {
        host: SEGMENT_HOST,
        flushAt: SEGMENT_FLUSH_AT,
        flushInterval: SEGMENT_FLUSH_INTERVAL,
      })

export const segmentLegacy =
  !SEGMENT_LEGACY_WRITE_KEY || (isDevOrTestEnvironment && !SEGMENT_HOST)
    ? createSegmentMock(SEGMENT_FLUSH_AT, SEGMENT_FLUSH_INTERVAL)
    : new Analytics(SEGMENT_LEGACY_WRITE_KEY, {
        host: SEGMENT_HOST,
        flushAt: SEGMENT_FLUSH_AT,
        flushInterval: SEGMENT_FLUSH_INTERVAL,
      })