[WIP] Inject SDK

Author: yoannmoinet

packages/plugins/rum/src/index.ts

@@ -2,6 +2,7 @@
 // This product includes software developed at Datadog (https://www.datadoghq.com/).
 // Copyright 2019-Present Datadog, Inc.
 
+import { doRequest } from '@dd/core/helpers';
 import type { GlobalContext, GetPlugins, Logger } from '@dd/core/types';
 
 import { CONFIG_KEY, PLUGIN_NAME } from './constants';
@@ -20,6 +21,14 @@ export type types = {
     OptionsWithRum: OptionsWithRum;
 };
 
+type RumAppResponse = {
+    data: {
+        attributes: {
+            client_token: string;
+        };
+    };
+};
+
 export const getPlugins: GetPlugins<OptionsWithRum> = (
     opts: OptionsWithRum,
     context: GlobalContext,
@@ -28,9 +37,10 @@ export const getPlugins: GetPlugins<OptionsWithRum> = (
     // Verify configuration.
     const options = validateOptions(opts, log);
 
-    if (!options.sdk?.clientToken) {
-        // Fetch the client token from the API.
-    }
+    context.inject({
+        type: 'file',
+        value: 'https://www.datadoghq-browser-agent.com/us1/v5/datadog-rum.js',
+    });
 
     return [
         {
@@ -39,6 +49,52 @@ export const getPlugins: GetPlugins<OptionsWithRum> = (
             // Not supported by Rollup and ESBuild.
             // https://vitejs.dev/guide/api-plugin.html#plugin-ordering
             enforce: 'pre',
+            async buildStart() {
+                if (!options.sdk) {
+                    return;
+                }
+
+                if (!options.sdk.clientToken) {
+                    if (!context.auth?.apiKey || !context.auth?.appKey) {
+                        throw new Error(
+                            'Missing auth.apiKey and/or auth.appKey to fetch clientToken.',
+                        );
+                    }
+
+                    const sdkOpts = options.sdk;
+                    let clientToken: string;
+
+                    try {
+                        // Fetch the client token from the API.
+                        const appResponse = await doRequest<RumAppResponse>({
+                            url: `https://api.datadoghq.com/api/v2/rum/applications/${options.sdk.applicationId}`,
+                            type: 'json',
+                            auth: context.auth,
+                        });
+
+                        clientToken = appResponse.data?.attributes?.client_token;
+                    } catch (e: any) {
+                        // Could not fetch the clientToken.
+                        // Let's crash the build.
+                        throw new Error(`Could not fetch the clientToken: ${e.message}`);
+                    }
+
+                    // Still no clientToken.
+                    if (!clientToken) {
+                        throw new Error('Missing clientToken in the API response.');
+                    }
+
+                    console.log('INJECTING');
+                    // Inject the initialization code.
+                    context.inject({
+                        type: 'code',
+                        value: `DD_RUM.init(${JSON.stringify({
+                            clientToken,
+                            ...sdkOpts,
+                        })});`,
+                    });
+                }
+            },
         },
     ];
 };

packages/tests/src/plugins/rum/index.test.ts

@@ -0,0 +1,30 @@
+// Unless explicitly stated otherwise all files in this repository are licensed under the MIT License.
+// This product includes software developed at Datadog (https://www.datadoghq.com/).
+// Copyright 2019-Present Datadog, Inc.
+
+import { runBundlers } from '@dd/tests/_jest/helpers/runBundlers';
+import type { CleanupFn } from '@dd/tests/_jest/helpers/types';
+
+describe('RUM Plugin', () => {
+    const cleanups: CleanupFn[] = [];
+
+    afterAll(async () => {
+        await Promise.all(cleanups.map((cleanup) => cleanup()));
+    });
+
+    test('Should get the clientToken.', async () => {
+        cleanups.push(
+            await runBundlers({
+                auth: {
+                    apiKey: process.env.DD_API_KEY,
+                    appKey: process.env.DD_APP_KEY,
+                },
+                rum: {
+                    sdk: {
+                        applicationId: '54caaabe-a702-4c07-89d8-8b7a064089aa',
+                    },
+                },
+            }),
+        );
+    });
+});