[Telemetry] Migrate ui_metric plugin to NP under usageCollection

packages/kbn-analytics/src/index.ts

@@ -20,3 +20,4 @@
 export { ReportHTTP, Reporter, ReporterConfig } from './reporter';
 export { UiStatsMetricType, METRIC_TYPE } from './metrics';
 export { Report, ReportManager } from './report';
+export { Storage } from './storage';

packages/kbn-analytics/src/report.ts

@@ -23,23 +23,25 @@ const REPORT_VERSION = 1;
 
 export interface Report {
   reportVersion: typeof REPORT_VERSION;
-  uiStatsMetrics: {
-    [key: string]: {
+  uiStatsMetrics?: Record<
+    string,
+    {
       key: string;
       appName: string;
       eventName: string;
       type: UiStatsMetricType;
       stats: Stats;
-    };
-  };
-  userAgent?: {
-    [key: string]: {
+    }
+  >;
+  userAgent?: Record<
+    string,
+    {
       userAgent: string;
       key: string;
       type: METRIC_TYPE.USER_AGENT;
       appName: string;
-    };
-  };
+    }
+  >;
 }
 
 export class ReportManager {
@@ -49,14 +51,15 @@ export class ReportManager {
     this.report = report || ReportManager.createReport();
   }
   static createReport(): Report {
-    return { reportVersion: REPORT_VERSION, uiStatsMetrics: {} };
+    return { reportVersion: REPORT_VERSION };
   }
   public clearReport() {
     this.report = ReportManager.createReport();
   }
   public isReportEmpty(): boolean {
-    const noUiStats = Object.keys(this.report.uiStatsMetrics).length === 0;
-    const noUserAgent = !this.report.userAgent || Object.keys(this.report.userAgent).length === 0;
+    const { uiStatsMetrics, userAgent } = this.report;
+    const noUiStats = !uiStatsMetrics || Object.keys(uiStatsMetrics).length === 0;
+    const noUserAgent = !userAgent || Object.keys(userAgent).length === 0;
     return noUiStats && noUserAgent;
   }
   private incrementStats(count: number, stats?: Stats): Stats {
@@ -113,14 +116,17 @@ export class ReportManager {
       case METRIC_TYPE.LOADED:
       case METRIC_TYPE.COUNT: {
         const { appName, type, eventName, count } = metric;
-        const existingStats = (report.uiStatsMetrics[key] || {}).stats;
-        this.report.uiStatsMetrics[key] = {
-          key,
-          appName,
-          eventName,
-          type,
-          stats: this.incrementStats(count, existingStats),
-        };
+        if (report.uiStatsMetrics) {
+          const existingStats = (report.uiStatsMetrics[key] || {}).stats;
+          this.report.uiStatsMetrics = this.report.uiStatsMetrics || {};
+          this.report.uiStatsMetrics[key] = {
+            key,
+            appName,
+            eventName,
+            type,
+            stats: this.incrementStats(count, existingStats),
+          };
+        }
         return;
       }
       default:

packages/kbn-analytics/src/storage.ts

@@ -19,7 +19,13 @@
 
 import { Report } from './report';
 
-export type Storage = Map<string, any>;
+export interface Storage<T = any, S = void> {
+  get: (key: string) => T | null;
+  set: (key: string, value: T) => S;
+  remove: (key: string) => T | null;
+  clear: () => void;
+}
+
 export class ReportStorageManager {
   storageKey: string;
   private storage?: Storage;

src/legacy/core_plugins/kibana/public/visualize/wizard/new_vis_modal.test.tsx

@@ -23,6 +23,7 @@ import { mountWithIntl } from 'test_utils/enzyme_helpers';
 import { VisType } from '../legacy_imports';
 import { TypesStart } from '../../../../visualizations/public/np_ready/public/types';
 
+jest.mock('ui/new_platform');
 jest.mock('../legacy_imports', () => ({
   State: () => null,
   AppState: () => null,

src/legacy/core_plugins/telemetry/public/services/telemetry_opt_in.ts

@@ -18,7 +18,6 @@
  */
 
 import moment from 'moment';
-import { setCanTrackUiMetrics } from 'ui/ui_metric';
 // @ts-ignore
 import { banners, toastNotifications } from 'ui/notify';
 import { npStart } from 'ui/new_platform';
@@ -69,8 +68,6 @@ export function TelemetryOptInProvider($injector: any, chrome: any, sendOptInSta
     'telemetryNotifyUserAboutOptInDefault'
   ) as boolean;
 
-  setCanTrackUiMetrics(currentOptInStatus);
-
   const provider = {
     getBannerId: () => bannerId,
     getOptInBannerNoticeId: () => optInBannerNoticeId,
@@ -116,7 +113,6 @@ export function TelemetryOptInProvider($injector: any, chrome: any, sendOptInSta
       if (!allowChangingOptInStatus) {
         return;
       }
-      setCanTrackUiMetrics(enabled);
       const $http = $injector.get('$http');
 
       try {

src/legacy/core_plugins/ui_metric/index.ts

@@ -18,36 +18,24 @@
  */
 
 import { resolve } from 'path';
-import JoiNamespace from 'joi';
-import { Server } from 'hapi';
 import { Legacy } from '../../../../kibana';
-import { registerUiMetricRoute } from './server/routes/api/ui_metric';
 
 // eslint-disable-next-line import/no-default-export
 export default function(kibana: any) {
   return new kibana.Plugin({
     id: 'ui_metric',
     require: ['kibana', 'elasticsearch'],
     publicDir: resolve(__dirname, 'public'),
-    config(Joi: typeof JoiNamespace) {
-      return Joi.object({
-        enabled: Joi.boolean().default(true),
-        debug: Joi.boolean().default(Joi.ref('$dev')),
-      }).default();
-    },
     uiExports: {
-      injectDefaultVars(server: Server) {
-        const config = server.config();
-        return {
-          uiMetricEnabled: config.get('ui_metric.enabled'),
-          debugUiMetric: config.get('ui_metric.debug'),
-        };
-      },
       mappings: require('./mappings.json'),
-      hacks: ['plugins/ui_metric/hacks/ui_metric_init'],
     },
     init(server: Legacy.Server) {
-      registerUiMetricRoute(server);
+      const { getSavedObjectsRepository } = server.savedObjects;
+      const { callWithInternalUser } = server.plugins.elasticsearch.getCluster('admin');
+      const internalRepository = getSavedObjectsRepository(callWithInternalUser);
+      const { usageCollection } = server.newPlatform.setup.plugins;
+
+      usageCollection.registerLegacySavedObjects(internalRepository);
     },
   });
 }

src/legacy/core_plugins/ui_metric/public/index.ts

@@ -17,5 +17,5 @@
  * under the License.
  */
 
-export { createUiStatsReporter, trackUserAgent } from './services/telemetry_analytics';
+export { createUiStatsReporter } from './services/telemetry_analytics';
 export { METRIC_TYPE, UiStatsMetricType } from '@kbn/analytics';

src/legacy/core_plugins/ui_metric/public/services/telemetry_analytics.ts

@@ -16,61 +16,9 @@
  * specific language governing permissions and limitations
  * under the License.
  */
+import { npSetup } from 'ui/new_platform';
 
-import { Reporter, UiStatsMetricType } from '@kbn/analytics';
-// @ts-ignore
-import { addSystemApiHeader } from 'ui/system_api';
-
-let telemetryReporter: Reporter;
-
-export const setTelemetryReporter = (aTelemetryReporter: Reporter): void => {
-  telemetryReporter = aTelemetryReporter;
-};
-
-export const getTelemetryReporter = () => {
-  return telemetryReporter;
-};
-
-export const createUiStatsReporter = (appName: string) => (
-  type: UiStatsMetricType,
-  eventNames: string | string[],
-  count?: number
-): void => {
-  if (telemetryReporter) {
-    return telemetryReporter.reportUiStats(appName, type, eventNames, count);
-  }
+export const createUiStatsReporter = (appName: string) => {
+  const { usageCollection } = npSetup.plugins;
+  return usageCollection.reportUiStats.bind(usageCollection, appName);
 };
-
-export const trackUserAgent = (appName: string) => {
-  if (telemetryReporter) {
-    return telemetryReporter.reportUserAgent(appName);
-  }
-};
-
-interface AnalyicsReporterConfig {
-  localStorage: any;
-  debug: boolean;
-  kfetch: any;
-}
-
-export function createAnalyticsReporter(config: AnalyicsReporterConfig) {
-  const { localStorage, debug, kfetch } = config;
-
-  return new Reporter({
-    debug,
-    storage: localStorage,
-    async http(report) {
-      const response = await kfetch({
-        method: 'POST',
-        pathname: '/api/telemetry/report',
-        body: JSON.stringify(report),
-        headers: addSystemApiHeader({}),
-      });
-
-      if (response.status !== 'ok') {
-        throw Error('Unable to store report.');
-      }
-      return response;
-    },
-  });
-}