Add featureUsage API to licensing context provider

Author: joshdover

x-pack/plugins/licensing/server/licensing_route_handler_context.test.ts

@@ -8,14 +8,15 @@ import { BehaviorSubject } from 'rxjs';
 import { licenseMock } from '../common/licensing.mock';
 
 import { createRouteHandlerContext } from './licensing_route_handler_context';
+import { featureUsageMock } from './services/feature_usage_service.mock';
 
 describe('createRouteHandlerContext', () => {
   it('returns a function providing the last license value', async () => {
     const firstLicense = licenseMock.createLicense();
     const secondLicense = licenseMock.createLicense();
     const license$ = new BehaviorSubject(firstLicense);
 
-    const routeHandler = createRouteHandlerContext(license$);
+    const routeHandler = createRouteHandlerContext(license$, featureUsageMock.createStart());
 
     const firstCtx = await routeHandler({} as any, {} as any, {} as any);
     license$.next(secondLicense);
@@ -24,4 +25,14 @@ describe('createRouteHandlerContext', () => {
     expect(firstCtx.license).toBe(firstLicense);
     expect(secondCtx.license).toBe(secondLicense);
   });
+
+  it('returns a the feature usage API', async () => {
+    const license$ = new BehaviorSubject(licenseMock.createLicense());
+    const featureUsage = featureUsageMock.createStart();
+
+    const routeHandler = createRouteHandlerContext(license$, featureUsage);
+    const ctx = await routeHandler({} as any, {} as any, {} as any);
+
+    expect(ctx.featureUsage).toBe(featureUsage);
+  });
 });

x-pack/plugins/licensing/server/licensing_route_handler_context.ts

@@ -9,16 +9,21 @@ import { Observable } from 'rxjs';
 import { take } from 'rxjs/operators';
 
 import { ILicense } from '../common/types';
+import { FeatureUsageServiceStart } from './services';
 
 /**
  * Create a route handler context for access to Kibana license information.
  * @param license$ An observable of a License instance.
  * @public
  */
 export function createRouteHandlerContext(
-  license$: Observable<ILicense>
+  license$: Observable<ILicense>,
+  featureUsage: FeatureUsageServiceStart
 ): IContextProvider<RequestHandler<any, any, any>, 'licensing'> {
   return async function licensingRouteHandlerContext() {
-    return { license: await license$.pipe(take(1)).toPromise() };
+    return {
+      license: await license$.pipe(take(1)).toPromise(),
+      featureUsage,
+    };
   };
 }

x-pack/plugins/licensing/server/plugin.ts

@@ -128,7 +128,10 @@ export class LicensingPlugin implements Plugin<LicensingPluginSetup, LicensingPl
       pollingFrequency.asMilliseconds()
     );
 
-    core.http.registerRouteHandlerContext('licensing', createRouteHandlerContext(license$));
+    core.http.registerRouteHandlerContext(
+      'licensing',
+      createRouteHandlerContext(license$, this.featureUsage.start())
+    );
 
     registerRoutes(core.http.createRouter(), core.getStartServices);
     core.http.registerOnPreResponse(createOnPreResponseHandler(refresh, license$));

x-pack/plugins/licensing/server/types.ts

@@ -42,6 +42,7 @@ export interface RawLicense {
 
 declare module 'src/core/server' {
   interface RequestHandlerContext {
+    featureUsage: FeatureUsageServiceStart;
     licensing: {
       license: ILicense;
     };