From c76289d000438ac3686608fb8d4943b9b62554f2 Mon Sep 17 00:00:00 2001
From: Joel Griffith <joel.griffith@elastic.co>
Date: Tue, 8 Sep 2020 09:33:48 -0700
Subject: [PATCH] Adds back in custom images for reporting + tests (#76810)
 (#76823)

Co-authored-by: Elastic Machine <elasticmachine@users.noreply.github.com>
---
 .../plugins/reporting/server/plugin.test.ts   |  9 ++++++
 x-pack/plugins/reporting/server/plugin.ts     | 31 +++++++++++++++++--
 2 files changed, 38 insertions(+), 2 deletions(-)

diff --git a/x-pack/plugins/reporting/server/plugin.test.ts b/x-pack/plugins/reporting/server/plugin.test.ts
index 420fa8347cdeb..e0d018869cef1 100644
--- a/x-pack/plugins/reporting/server/plugin.test.ts
+++ b/x-pack/plugins/reporting/server/plugin.test.ts
@@ -86,6 +86,15 @@ describe('Reporting Plugin', () => {
     expect(plugin.start(coreStart, pluginStart)).not.toHaveProperty('then');
   });
 
+  it('registers an advanced setting for PDF logos', async () => {
+    const plugin = new ReportingPlugin(initContext);
+    plugin.setup(coreSetup, pluginSetup);
+    expect(coreSetup.uiSettings.register).toHaveBeenCalled();
+    expect(coreSetup.uiSettings.register.mock.calls[0][0]).toHaveProperty(
+      'xpackReporting:customPdfLogo'
+    );
+  });
+
   it('logs start issues', async () => {
     const plugin = new ReportingPlugin(initContext);
     // @ts-ignore overloading error logger
diff --git a/x-pack/plugins/reporting/server/plugin.ts b/x-pack/plugins/reporting/server/plugin.ts
index 20e22c2db00e3..8c0e352aa06c5 100644
--- a/x-pack/plugins/reporting/server/plugin.ts
+++ b/x-pack/plugins/reporting/server/plugin.ts
@@ -4,16 +4,21 @@
  * you may not use this file except in compliance with the Elastic License.
  */
 
+import { schema } from '@kbn/config-schema';
+import { i18n } from '@kbn/i18n';
 import { CoreSetup, CoreStart, Plugin, PluginInitializerContext } from 'src/core/server';
+import { PLUGIN_ID, UI_SETTINGS_CUSTOM_PDF_LOGO } from '../common/constants';
 import { ReportingCore } from './';
 import { initializeBrowserDriverFactory } from './browsers';
 import { buildConfig, ReportingConfigType } from './config';
-import { createQueueFactory, LevelLogger, runValidations, ReportingStore } from './lib';
+import { createQueueFactory, LevelLogger, ReportingStore, runValidations } from './lib';
 import { registerRoutes } from './routes';
 import { setFieldFormats } from './services';
 import { ReportingSetup, ReportingSetupDeps, ReportingStart, ReportingStartDeps } from './types';
 import { registerReportingUsageCollector } from './usage';
 
+const kbToBase64Length = (kb: number) => Math.floor((kb * 1024 * 8) / 6);
+
 declare module 'src/core/server' {
   interface RequestHandlerContext {
     reporting?: ReportingStart | null;
@@ -34,7 +39,7 @@ export class ReportingPlugin
 
   public setup(core: CoreSetup, plugins: ReportingSetupDeps) {
     // prevent throwing errors in route handlers about async deps not being initialized
-    core.http.registerRouteHandlerContext('reporting', () => {
+    core.http.registerRouteHandlerContext(PLUGIN_ID, () => {
       if (this.reportingCore.pluginIsStarted()) {
         return {}; // ReportingStart contract
       } else {
@@ -42,6 +47,28 @@ export class ReportingPlugin
       }
     });
 
+    core.uiSettings.register({
+      [UI_SETTINGS_CUSTOM_PDF_LOGO]: {
+        name: i18n.translate('xpack.reporting.pdfFooterImageLabel', {
+          defaultMessage: 'PDF footer image',
+        }),
+        value: null,
+        description: i18n.translate('xpack.reporting.pdfFooterImageDescription', {
+          defaultMessage: `Custom image to use in the PDF's footer`,
+        }),
+        type: 'image',
+        schema: schema.nullable(schema.byteSize({ max: '200kb' })),
+        category: [PLUGIN_ID],
+        // Used client-side for size validation
+        validation: {
+          maxSize: {
+            length: kbToBase64Length(200),
+            description: '200 kB',
+          },
+        },
+      },
+    });
+
     const { elasticsearch, http } = core;
     const { licensing, security } = plugins;
     const { initializerContext: initContext, reportingCore } = this;