summaryrefslogtreecommitdiff
path: root/scripts/telemetry.js
diff options
context:
space:
mode:
Diffstat (limited to 'scripts/telemetry.js')
-rwxr-xr-xscripts/telemetry.js85
1 files changed, 85 insertions, 0 deletions
diff --git a/scripts/telemetry.js b/scripts/telemetry.js
new file mode 100755
index 00000000..9d441072
--- /dev/null
+++ b/scripts/telemetry.js
@@ -0,0 +1,85 @@
+#!/usr/bin/env node
+
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+import { execSync } from 'child_process';
+import { join } from 'path';
+import { existsSync, readFileSync } from 'fs';
+
+const projectRoot = join(import.meta.dirname, '..');
+
+const SETTINGS_DIRECTORY_NAME = '.gemini';
+const USER_SETTINGS_DIR = join(
+ process.env.HOME || process.env.USERPROFILE || process.env.HOMEPATH || '',
+ SETTINGS_DIRECTORY_NAME,
+);
+const USER_SETTINGS_PATH = join(USER_SETTINGS_DIR, 'settings.json');
+const WORKSPACE_SETTINGS_PATH = join(
+ projectRoot,
+ SETTINGS_DIRECTORY_NAME,
+ 'settings.json',
+);
+
+let settingsTarget = undefined;
+
+function loadSettingsValue(filePath) {
+ try {
+ if (existsSync(filePath)) {
+ const content = readFileSync(filePath, 'utf-8');
+ const jsonContent = content.replace(/\/\/[^\n]*/g, '');
+ const settings = JSON.parse(jsonContent);
+ return settings.telemetry?.target;
+ }
+ } catch (e) {
+ console.warn(
+ `⚠️ Warning: Could not parse settings file at ${filePath}: ${e.message}`,
+ );
+ }
+ return undefined;
+}
+
+settingsTarget = loadSettingsValue(WORKSPACE_SETTINGS_PATH);
+
+if (!settingsTarget) {
+ settingsTarget = loadSettingsValue(USER_SETTINGS_PATH);
+}
+
+let target = settingsTarget || 'local';
+const allowedTargets = ['local', 'gcp'];
+
+const targetArg = process.argv.find((arg) => arg.startsWith('--target='));
+if (targetArg) {
+ const potentialTarget = targetArg.split('=')[1];
+ if (allowedTargets.includes(potentialTarget)) {
+ target = potentialTarget;
+ console.log(`⚙️ Using command-line target: ${target}`);
+ } else {
+ console.error(
+ `🛑 Error: Invalid target '${potentialTarget}'. Allowed targets are: ${allowedTargets.join(', ')}.`,
+ );
+ process.exit(1);
+ }
+} else if (settingsTarget) {
+ console.log(
+ `⚙️ Using telemetry target from settings.json: ${settingsTarget}`,
+ );
+}
+
+const scriptPath = join(
+ projectRoot,
+ 'scripts',
+ target === 'gcp' ? 'telemetry_gcp.js' : 'local_telemetry.js',
+);
+
+try {
+ console.log(`🚀 Running telemetry script for target: ${target}.`);
+ execSync(`node ${scriptPath}`, { stdio: 'inherit', cwd: projectRoot });
+} catch (error) {
+ console.error(`🛑 Failed to run telemetry script for target: ${target}`);
+ console.error(error);
+ process.exit(1);
+}