[APM] Fix Synthtrace script (#133620)

packages/elastic-apm-synthtrace/bin/synthtrace

@@ -14,4 +14,4 @@ require('@babel/register')({
   presets: [['@babel/preset-env', { targets: { node: 'current' } }], '@babel/preset-typescript'],
 });
 
-require('../src/scripts/run_synthtrace');
+require('../src/scripts/run_synthtrace').runSynthtrace();

packages/elastic-apm-synthtrace/package.json

@@ -3,6 +3,9 @@
   "version": "0.1.0",
   "description": "Elastic APM trace data generator",
   "license": "SSPL-1.0 OR Elastic License 2.0",
+  "bin": {
+    "synthtrace": "./bin/synthtrace"
+  },
   "main": "./target_node/index.js",
   "private": true
 }

packages/elastic-apm-synthtrace/src/index.ts

@@ -6,6 +6,7 @@
  * Side Public License, v 1.
  */
 
+export { runSynthtrace } from './scripts/run_synthtrace';
 export { timerange } from './lib/timerange';
 export { apm } from './lib/apm';
 export { stackMonitoring } from './lib/stack_monitoring';

packages/elastic-apm-synthtrace/src/scripts/run_synthtrace.ts

@@ -124,7 +124,8 @@ function options(y: Argv) {
 
 export type RunCliFlags = ReturnType<typeof options>['argv'];
 
-yargs(process.argv.slice(2))
+export function runSynthtrace() {
+  yargs(process.argv.slice(2))
     .command(
       '*',
       'Generate data and index into Elasticsearch',
@@ -238,3 +239,4 @@ yargs(process.argv.slice(2))
       }
     )
     .parse();
+}

packages/elastic-apm-synthtrace/src/scripts/utils/parse_run_cli_flags.ts

@@ -20,15 +20,18 @@ function getParsedFile(flags: RunCliFlags) {
     throw new Error('Please specify a scenario to run');
   }
 
-  const result = [
+  const filepath = [
     path.resolve(parsedFile),
     path.resolve(`${parsedFile}.ts`),
     path.resolve(__dirname, '../../scenarios', parsedFile),
     path.resolve(__dirname, '../../scenarios', `${parsedFile}.ts`),
-  ].find((filepath) => existsSync(filepath));
+    path.resolve(__dirname, '../../scenarios', `${parsedFile}.js`),
+  ].find((p) => existsSync(p));
 
-  if (result) {
-    return result;
+  if (filepath) {
+    // eslint-disable-next-line no-console
+    console.log(`Loading scenario from ${filepath}`);
+    return filepath;
   }
 
   throw new Error(`Could not find scenario file: "${parsedFile}"`);

scripts/synthtrace.js

@@ -6,5 +6,11 @@
  * Side Public License, v 1.
  */
 
-require('../src/setup_node_env/node_version_validator');
-require('@elastic/apm-synthtrace/bin/synthtrace');
+require('../src/setup_node_env');
+
+// We have to import directly from package since scenarios and worker.js are imported dynamically,
+// If we import the package (require('@elastic/apm-synthtrace')) the program will be executed on the compiled files, and thus we need to
+// compile scenarios with `yarn kbn bootstrap` every time scenario changes.
+
+// eslint-disable-next-line @kbn/imports/uniform_imports
+require('../packages/elastic-apm-synthtrace/src/index').runSynthtrace();