Add single telemetry support

Author: Suchita Doshi

lib/gather/gather-single-telemetry.js

@@ -0,0 +1,45 @@
+const puppeteer = require('puppeteer');
+const { setTelemetry, getTelemetry } = require('../utils/telemetry');
+
+const DEFAULT_PUPPETEER_ARGS = { ignoreHTTPSErrors: true };
+
+module.exports = async function gatherSingleTelemetry(url, gatherFn, ...args) {
+  const browser = await puppeteer.launch(DEFAULT_PUPPETEER_ARGS);
+  const page = await browser.newPage();
+
+  await page.goto(url);
+
+  await page.exposeFunction('logErrorInNodeProcess', message => {
+    console.error(message); // eslint-disable-line no-console
+  });
+
+  const telemetry = await bridgeEvaluate(
+    page,
+    async (gFn, ...supportFns) => {
+      supportFns.forEach(fn => {
+        this[fn.name] = fn;
+      });
+      let telemetry = {};
+
+      telemetry = await gFn(...supportFns);
+      return telemetry;
+    },
+    gatherFn,
+    ...args
+  );
+
+  // get the current telemetry and add the `single-telemetry` key to it.
+  const currentTelemetry = getTelemetry();
+  currentTelemetry['single-telemetry'] = telemetry;
+  setTelemetry(currentTelemetry);
+  await browser.close();
+
+  async function bridgeEvaluate(page, fn, ...rawArgs) {
+    const args = await Promise.all(
+      rawArgs.map(arg => {
+        return typeof arg === 'function' ? page.evaluateHandle(`(${arg.toString()})`) : arg;
+      })
+    );
+    return page.evaluate(fn, ...args);
+  }
+};

lib/gather/gather-single-telemetry.test.js

@@ -0,0 +1,51 @@
+const startApp = require('../../test/helpers/start-app');
+const gatherSingleTelemetry = require('./gather-single-telemetry');
+const { getTelemetry } = require('../utils/telemetry');
+const APP_TIMEOUT = 100000;
+
+function resolverWithoutArgs() {
+  return { foo: 'bar' };
+}
+
+function resolverWithArgs(lookupNames) {
+  if (lookupNames) {
+    return lookupNames.map(item => {
+      const lookupSplit = item.split(':');
+      return { name: lookupSplit[1], type: lookupSplit[0] };
+    });
+  }
+}
+
+describe('Gather single telemetry', () => {
+  let app;
+  let localAppPath = './test/fixtures/classic-app';
+  beforeAll(async () => {
+    app = await startApp(localAppPath);
+    console.log(`Spawned PID: ${app.emberServe.pid}`);
+  }, APP_TIMEOUT);
+
+  test('can determine base single telemetry case', async () => {
+    await gatherSingleTelemetry('http://localhost:4200', resolverWithoutArgs);
+    let telemetry = getTelemetry();
+    expect(telemetry['single-telemetry']).toEqual({
+      foo: 'bar',
+    });
+  });
+
+  test('can determine single telemetry with arguments passed', async () => {
+    const lookupNames = ['component:foo', 'helper:bar'];
+    await gatherSingleTelemetry('http://localhost:4200', resolverWithArgs, lookupNames);
+    let telemetry = getTelemetry();
+    expect(telemetry['single-telemetry']).toEqual([
+      { name: 'foo', type: 'component' },
+      { name: 'bar', type: 'helper' },
+    ]);
+  });
+
+  afterAll(async () => {
+    console.log(`Killing PID: ${app.emberServe.pid}`);
+    await app.emberServe.kill('SIGTERM', {
+      forceKillAfterTimeout: 200,
+    });
+  }, APP_TIMEOUT);
+});

lib/index.js

@@ -1,4 +1,5 @@
 const gatherTelemetryForUrl = require('./gather/gather-telemetry');
+const gatherSingleTelemetryForUrl = require('./gather/gather-single-telemetry.js');
 const { getModulePathFor } = require('./utils/get-module-path-for');
 const { getTelemetry, setTelemetry, getTelemetryFor } = require('./utils/telemetry');
 const analyzeEmberObject = require('./gather/analyze-ember-object');
@@ -7,6 +8,7 @@ module.exports = {
   getTelemetry,
   setTelemetry,
   gatherTelemetryForUrl,
+  gatherSingleTelemetryForUrl,
   getTelemetryFor,
   getModulePathFor,
   analyzeEmberObject,