From b72320cc26c7b49eebf42d93765c4e63ecbd51e2 Mon Sep 17 00:00:00 2001
From: Joe Reuter <johannes.reuter@elastic.co>
Date: Mon, 20 Jul 2020 15:48:42 +0200
Subject: [PATCH] register graph usage (#72041) (#72300)

---
 x-pack/plugins/graph/server/lib/license_state.ts | 13 +++++++++++++
 x-pack/plugins/graph/server/plugin.ts            | 10 +++++++---
 x-pack/plugins/graph/server/routes/explore.ts    |  1 +
 x-pack/plugins/graph/server/routes/search.ts     |  1 +
 4 files changed, 22 insertions(+), 3 deletions(-)

diff --git a/x-pack/plugins/graph/server/lib/license_state.ts b/x-pack/plugins/graph/server/lib/license_state.ts
index d86cb5380a2e1..8d64c826d8fa1 100644
--- a/x-pack/plugins/graph/server/lib/license_state.ts
+++ b/x-pack/plugins/graph/server/lib/license_state.ts
@@ -7,6 +7,7 @@
 import Boom from 'boom';
 import { map } from 'rxjs/operators';
 import { Observable, Subscription } from 'rxjs';
+import { LicensingPluginStart } from '../../../licensing/server';
 import { ILicense } from '../../../licensing/common/types';
 import { checkLicense, GraphLicenseInformation } from '../../common/check_license';
 
@@ -14,6 +15,7 @@ export class LicenseState {
   private licenseInformation: GraphLicenseInformation = checkLicense(undefined);
   private subscription: Subscription | null = null;
   private observable: Observable<GraphLicenseInformation> | null = null;
+  private _notifyUsage: LicensingPluginStart['featureUsage']['notifyUsage'] | null = null;
 
   private updateInformation(licenseInformation: GraphLicenseInformation) {
     this.licenseInformation = licenseInformation;
@@ -24,6 +26,17 @@ export class LicenseState {
     this.subscription = this.observable.subscribe(this.updateInformation.bind(this));
   }
 
+  public setNotifyUsage(notifyUsage: LicensingPluginStart['featureUsage']['notifyUsage']) {
+    this._notifyUsage = notifyUsage;
+  }
+
+  // 'Graph' is the only allowed feature here at the moment, if this gets extended in the future, add to the union type
+  public notifyUsage(featureName: 'Graph') {
+    if (this._notifyUsage) {
+      this._notifyUsage(featureName);
+    }
+  }
+
   public stop() {
     if (this.subscription) {
       this.subscription.unsubscribe();
diff --git a/x-pack/plugins/graph/server/plugin.ts b/x-pack/plugins/graph/server/plugin.ts
index 141d5d0ea8db4..b2b825fa4683b 100644
--- a/x-pack/plugins/graph/server/plugin.ts
+++ b/x-pack/plugins/graph/server/plugin.ts
@@ -5,8 +5,8 @@
  */
 
 import { i18n } from '@kbn/i18n';
-import { Plugin, CoreSetup } from 'src/core/server';
-import { LicensingPluginSetup } from '../../licensing/server';
+import { Plugin, CoreSetup, CoreStart } from 'src/core/server';
+import { LicensingPluginSetup, LicensingPluginStart } from '../../licensing/server';
 import { LicenseState } from './lib/license_state';
 import { registerSearchRoute } from './routes/search';
 import { registerExploreRoute } from './routes/explore';
@@ -34,6 +34,7 @@ export class GraphPlugin implements Plugin {
     licenseState.start(licensing.license$);
     this.licenseState = licenseState;
     core.savedObjects.registerType(graphWorkspace);
+    licensing.featureUsage.register('Graph', 'platinum');
 
     if (home) {
       registerSampleData(home.sampleData, licenseState);
@@ -79,7 +80,10 @@ export class GraphPlugin implements Plugin {
     registerExploreRoute({ licenseState, router });
   }
 
-  public start() {}
+  public start(core: CoreStart, { licensing }: { licensing: LicensingPluginStart }) {
+    this.licenseState!.setNotifyUsage(licensing.featureUsage.notifyUsage);
+  }
+
   public stop() {
     if (this.licenseState) {
       this.licenseState.stop();
diff --git a/x-pack/plugins/graph/server/routes/explore.ts b/x-pack/plugins/graph/server/routes/explore.ts
index b0b8cf14ff699..c436fbd1c79af 100644
--- a/x-pack/plugins/graph/server/routes/explore.ts
+++ b/x-pack/plugins/graph/server/routes/explore.ts
@@ -42,6 +42,7 @@ export function registerExploreRoute({
         response
       ) => {
         verifyApiAccess(licenseState);
+        licenseState.notifyUsage('Graph');
         try {
           return response.ok({
             body: {
diff --git a/x-pack/plugins/graph/server/routes/search.ts b/x-pack/plugins/graph/server/routes/search.ts
index 645e6b520013f..e1d430eeb311a 100644
--- a/x-pack/plugins/graph/server/routes/search.ts
+++ b/x-pack/plugins/graph/server/routes/search.ts
@@ -42,6 +42,7 @@ export function registerSearchRoute({
         response
       ) => {
         verifyApiAccess(licenseState);
+        licenseState.notifyUsage('Graph');
         const includeFrozen = await uiSettings.get<boolean>(UI_SETTINGS.SEARCH_INCLUDE_FROZEN);
         try {
           return response.ok({