From 68f1de576a577988b168cfc62e06b0c32f28374b Mon Sep 17 00:00:00 2001
From: Jason Rhodes <jason.matthew.rhodes@gmail.com>
Date: Wed, 24 Jun 2020 14:11:13 -0400
Subject: [PATCH] Makes usage collection methods available on start

---
 src/plugins/usage_collection/public/plugin.ts | 15 +++++++++++++--
 1 file changed, 13 insertions(+), 2 deletions(-)

diff --git a/src/plugins/usage_collection/public/plugin.ts b/src/plugins/usage_collection/public/plugin.ts
index cf2f6af1507c0..40f27f8269928 100644
--- a/src/plugins/usage_collection/public/plugin.ts
+++ b/src/plugins/usage_collection/public/plugin.ts
@@ -52,12 +52,17 @@ export interface UsageCollectionSetup {
   };
 }
 
+export interface UsageCollectionStart {
+  reportUiStats: Reporter['reportUiStats'];
+  METRIC_TYPE: typeof METRIC_TYPE;
+}
+
 export function isUnauthenticated(http: HttpSetup) {
   const { anonymousPaths } = http;
   return anonymousPaths.isAnonymous(window.location.pathname);
 }
 
-export class UsageCollectionPlugin implements Plugin<UsageCollectionSetup> {
+export class UsageCollectionPlugin implements Plugin<UsageCollectionSetup, UsageCollectionStart> {
   private readonly legacyAppId$ = new Subject<string>();
   private trackUserAgent: boolean = true;
   private reporter?: Reporter;
@@ -90,7 +95,7 @@ export class UsageCollectionPlugin implements Plugin<UsageCollectionSetup> {
 
   public start({ http, application }: CoreStart) {
     if (!this.reporter) {
-      return;
+      throw new Error('Usage collection reporter not set up correctly');
     }
 
     if (this.config.uiMetric.enabled && !isUnauthenticated(http)) {
@@ -100,7 +105,13 @@ export class UsageCollectionPlugin implements Plugin<UsageCollectionSetup> {
     if (this.trackUserAgent) {
       this.reporter.reportUserAgent('kibana');
     }
+
     reportApplicationUsage(merge(application.currentAppId$, this.legacyAppId$), this.reporter);
+
+    return {
+      reportUiStats: this.reporter.reportUiStats,
+      METRIC_TYPE,
+    };
   }
 
   public stop() {}