@google/gemini-cli-core 0.0.1

package/dist/src/telemetry/sdk.test.js

@@ -0,0 +1,177 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
+import { initializeTelemetry, shutdownTelemetry } from './sdk.js';
+import { OTLPTraceExporter } from '@opentelemetry/exporter-trace-otlp-grpc';
+import { OTLPLogExporter } from '@opentelemetry/exporter-logs-otlp-grpc';
+import { OTLPMetricExporter } from '@opentelemetry/exporter-metrics-otlp-grpc';
+import { OTLPTraceExporter as OTLPTraceExporterHttp } from '@opentelemetry/exporter-trace-otlp-http';
+import { OTLPLogExporter as OTLPLogExporterHttp } from '@opentelemetry/exporter-logs-otlp-http';
+import { OTLPMetricExporter as OTLPMetricExporterHttp } from '@opentelemetry/exporter-metrics-otlp-http';
+import { NodeSDK } from '@opentelemetry/sdk-node';
+import { GcpTraceExporter, GcpLogExporter, GcpMetricExporter, } from './gcp-exporters.js';
+import { TelemetryTarget } from './index.js';
+vi.mock('@opentelemetry/exporter-trace-otlp-grpc');
+vi.mock('@opentelemetry/exporter-logs-otlp-grpc');
+vi.mock('@opentelemetry/exporter-metrics-otlp-grpc');
+vi.mock('@opentelemetry/exporter-trace-otlp-http');
+vi.mock('@opentelemetry/exporter-logs-otlp-http');
+vi.mock('@opentelemetry/exporter-metrics-otlp-http');
+vi.mock('@opentelemetry/sdk-node');
+vi.mock('./gcp-exporters.js');
+describe('Telemetry SDK', () => {
+    let mockConfig;
+    beforeEach(() => {
+        vi.clearAllMocks();
+        mockConfig = {
+            getTelemetryEnabled: () => true,
+            getTelemetryOtlpEndpoint: () => 'http://localhost:4317',
+            getTelemetryOtlpProtocol: () => 'grpc',
+            getTelemetryTarget: () => 'local',
+            getTelemetryUseCollector: () => false,
+            getTelemetryOutfile: () => undefined,
+            getDebugMode: () => false,
+            getSessionId: () => 'test-session',
+        };
+    });
+    afterEach(async () => {
+        await shutdownTelemetry(mockConfig);
+    });
+    it('should use gRPC exporters when protocol is grpc', () => {
+        initializeTelemetry(mockConfig);
+        expect(OTLPTraceExporter).toHaveBeenCalledWith({
+            url: 'http://localhost:4317',
+            compression: 'gzip',
+        });
+        expect(OTLPLogExporter).toHaveBeenCalledWith({
+            url: 'http://localhost:4317',
+            compression: 'gzip',
+        });
+        expect(OTLPMetricExporter).toHaveBeenCalledWith({
+            url: 'http://localhost:4317',
+            compression: 'gzip',
+        });
+        expect(NodeSDK.prototype.start).toHaveBeenCalled();
+    });
+    it('should use HTTP exporters when protocol is http', () => {
+        vi.spyOn(mockConfig, 'getTelemetryEnabled').mockReturnValue(true);
+        vi.spyOn(mockConfig, 'getTelemetryOtlpProtocol').mockReturnValue('http');
+        vi.spyOn(mockConfig, 'getTelemetryOtlpEndpoint').mockReturnValue('http://localhost:4318');
+        initializeTelemetry(mockConfig);
+        expect(OTLPTraceExporterHttp).toHaveBeenCalledWith({
+            url: 'http://localhost:4318/',
+        });
+        expect(OTLPLogExporterHttp).toHaveBeenCalledWith({
+            url: 'http://localhost:4318/',
+        });
+        expect(OTLPMetricExporterHttp).toHaveBeenCalledWith({
+            url: 'http://localhost:4318/',
+        });
+        expect(NodeSDK.prototype.start).toHaveBeenCalled();
+    });
+    it('should parse gRPC endpoint correctly', () => {
+        vi.spyOn(mockConfig, 'getTelemetryOtlpEndpoint').mockReturnValue('https://my-collector.com');
+        initializeTelemetry(mockConfig);
+        expect(OTLPTraceExporter).toHaveBeenCalledWith(expect.objectContaining({ url: 'https://my-collector.com' }));
+    });
+    it('should parse HTTP endpoint correctly', () => {
+        vi.spyOn(mockConfig, 'getTelemetryOtlpProtocol').mockReturnValue('http');
+        vi.spyOn(mockConfig, 'getTelemetryOtlpEndpoint').mockReturnValue('https://my-collector.com');
+        initializeTelemetry(mockConfig);
+        expect(OTLPTraceExporterHttp).toHaveBeenCalledWith(expect.objectContaining({ url: 'https://my-collector.com/' }));
+    });
+    it('should use direct GCP exporters when target is gcp, project ID is set, and useCollector is false', () => {
+        vi.spyOn(mockConfig, 'getTelemetryTarget').mockReturnValue(TelemetryTarget.GCP);
+        vi.spyOn(mockConfig, 'getTelemetryUseCollector').mockReturnValue(false);
+        vi.spyOn(mockConfig, 'getTelemetryOtlpEndpoint').mockReturnValue('');
+        const originalEnv = process.env['OTLP_GOOGLE_CLOUD_PROJECT'];
+        process.env['OTLP_GOOGLE_CLOUD_PROJECT'] = 'test-project';
+        try {
+            initializeTelemetry(mockConfig);
+            expect(GcpTraceExporter).toHaveBeenCalledWith('test-project');
+            expect(GcpLogExporter).toHaveBeenCalledWith('test-project');
+            expect(GcpMetricExporter).toHaveBeenCalledWith('test-project');
+            expect(NodeSDK.prototype.start).toHaveBeenCalled();
+        }
+        finally {
+            if (originalEnv) {
+                process.env['OTLP_GOOGLE_CLOUD_PROJECT'] = originalEnv;
+            }
+            else {
+                delete process.env['OTLP_GOOGLE_CLOUD_PROJECT'];
+            }
+        }
+    });
+    it('should use OTLP exporters when target is gcp but useCollector is true', () => {
+        vi.spyOn(mockConfig, 'getTelemetryTarget').mockReturnValue(TelemetryTarget.GCP);
+        vi.spyOn(mockConfig, 'getTelemetryUseCollector').mockReturnValue(true);
+        initializeTelemetry(mockConfig);
+        expect(OTLPTraceExporter).toHaveBeenCalledWith({
+            url: 'http://localhost:4317',
+            compression: 'gzip',
+        });
+        expect(OTLPLogExporter).toHaveBeenCalledWith({
+            url: 'http://localhost:4317',
+            compression: 'gzip',
+        });
+        expect(OTLPMetricExporter).toHaveBeenCalledWith({
+            url: 'http://localhost:4317',
+            compression: 'gzip',
+        });
+    });
+    it('should not use GCP exporters when project ID environment variable is not set', () => {
+        vi.spyOn(mockConfig, 'getTelemetryTarget').mockReturnValue(TelemetryTarget.GCP);
+        vi.spyOn(mockConfig, 'getTelemetryUseCollector').mockReturnValue(false);
+        vi.spyOn(mockConfig, 'getTelemetryOtlpEndpoint').mockReturnValue('');
+        const originalOtlpEnv = process.env['OTLP_GOOGLE_CLOUD_PROJECT'];
+        const originalGoogleEnv = process.env['GOOGLE_CLOUD_PROJECT'];
+        delete process.env['OTLP_GOOGLE_CLOUD_PROJECT'];
+        delete process.env['GOOGLE_CLOUD_PROJECT'];
+        try {
+            initializeTelemetry(mockConfig);
+            expect(GcpTraceExporter).not.toHaveBeenCalled();
+            expect(GcpLogExporter).not.toHaveBeenCalled();
+            expect(GcpMetricExporter).not.toHaveBeenCalled();
+            expect(NodeSDK.prototype.start).toHaveBeenCalled();
+        }
+        finally {
+            if (originalOtlpEnv) {
+                process.env['OTLP_GOOGLE_CLOUD_PROJECT'] = originalOtlpEnv;
+            }
+            if (originalGoogleEnv) {
+                process.env['GOOGLE_CLOUD_PROJECT'] = originalGoogleEnv;
+            }
+        }
+    });
+    it('should use GOOGLE_CLOUD_PROJECT as fallback when OTLP_GOOGLE_CLOUD_PROJECT is not set', () => {
+        vi.spyOn(mockConfig, 'getTelemetryTarget').mockReturnValue(TelemetryTarget.GCP);
+        vi.spyOn(mockConfig, 'getTelemetryUseCollector').mockReturnValue(false);
+        vi.spyOn(mockConfig, 'getTelemetryOtlpEndpoint').mockReturnValue('');
+        const originalOtlpEnv = process.env['OTLP_GOOGLE_CLOUD_PROJECT'];
+        const originalGoogleEnv = process.env['GOOGLE_CLOUD_PROJECT'];
+        delete process.env['OTLP_GOOGLE_CLOUD_PROJECT'];
+        process.env['GOOGLE_CLOUD_PROJECT'] = 'fallback-project';
+        try {
+            initializeTelemetry(mockConfig);
+            expect(GcpTraceExporter).toHaveBeenCalledWith('fallback-project');
+            expect(GcpLogExporter).toHaveBeenCalledWith('fallback-project');
+            expect(GcpMetricExporter).toHaveBeenCalledWith('fallback-project');
+            expect(NodeSDK.prototype.start).toHaveBeenCalled();
+        }
+        finally {
+            if (originalOtlpEnv) {
+                process.env['OTLP_GOOGLE_CLOUD_PROJECT'] = originalOtlpEnv;
+            }
+            if (originalGoogleEnv) {
+                process.env['GOOGLE_CLOUD_PROJECT'] = originalGoogleEnv;
+            }
+            else {
+                delete process.env['GOOGLE_CLOUD_PROJECT'];
+            }
+        }
+    });
+});
+//# sourceMappingURL=sdk.test.js.map

package/dist/src/telemetry/sdk.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"sdk.test.js","sourceRoot":"","sources":["../../../src/telemetry/sdk.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AAEzE,OAAO,EAAE,mBAAmB,EAAE,iBAAiB,EAAE,MAAM,UAAU,CAAC;AAClE,OAAO,EAAE,iBAAiB,EAAE,MAAM,yCAAyC,CAAC;AAC5E,OAAO,EAAE,eAAe,EAAE,MAAM,wCAAwC,CAAC;AACzE,OAAO,EAAE,kBAAkB,EAAE,MAAM,2CAA2C,CAAC;AAC/E,OAAO,EAAE,iBAAiB,IAAI,qBAAqB,EAAE,MAAM,yCAAyC,CAAC;AACrG,OAAO,EAAE,eAAe,IAAI,mBAAmB,EAAE,MAAM,wCAAwC,CAAC;AAChG,OAAO,EAAE,kBAAkB,IAAI,sBAAsB,EAAE,MAAM,2CAA2C,CAAC;AACzG,OAAO,EAAE,OAAO,EAAE,MAAM,yBAAyB,CAAC;AAClD,OAAO,EACL,gBAAgB,EAChB,cAAc,EACd,iBAAiB,GAClB,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EAAE,eAAe,EAAE,MAAM,YAAY,CAAC;AAE7C,EAAE,CAAC,IAAI,CAAC,yCAAyC,CAAC,CAAC;AACnD,EAAE,CAAC,IAAI,CAAC,wCAAwC,CAAC,CAAC;AAClD,EAAE,CAAC,IAAI,CAAC,2CAA2C,CAAC,CAAC;AACrD,EAAE,CAAC,IAAI,CAAC,yCAAyC,CAAC,CAAC;AACnD,EAAE,CAAC,IAAI,CAAC,wCAAwC,CAAC,CAAC;AAClD,EAAE,CAAC,IAAI,CAAC,2CAA2C,CAAC,CAAC;AACrD,EAAE,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;AACnC,EAAE,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;AAE9B,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;IAC7B,IAAI,UAAkB,CAAC;IAEvB,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,UAAU,GAAG;YACX,mBAAmB,EAAE,GAAG,EAAE,CAAC,IAAI;YAC/B,wBAAwB,EAAE,GAAG,EAAE,CAAC,uBAAuB;YACvD,wBAAwB,EAAE,GAAG,EAAE,CAAC,MAAM;YACtC,kBAAkB,EAAE,GAAG,EAAE,CAAC,OAAO;YACjC,wBAAwB,EAAE,GAAG,EAAE,CAAC,KAAK;YACrC,mBAAmB,EAAE,GAAG,EAAE,CAAC,SAAS;YACpC,YAAY,EAAE,GAAG,EAAE,CAAC,KAAK;YACzB,YAAY,EAAE,GAAG,EAAE,CAAC,cAAc;SACd,CAAC;IACzB,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,KAAK,IAAI,EAAE;QACnB,MAAM,iBAAiB,CAAC,UAAU,CAAC,CAAC;IACtC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;QACzD,mBAAmB,CAAC,UAAU,CAAC,CAAC;QAEhC,MAAM,CAAC,iBAAiB,CAAC,CAAC,oBAAoB,CAAC;YAC7C,GAAG,EAAE,uBAAuB;YAC5B,WAAW,EAAE,MAAM;SACpB,CAAC,CAAC;QACH,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAAC;YAC3C,GAAG,EAAE,uBAAuB;YAC5B,WAAW,EAAE,MAAM;SACpB,CAAC,CAAC;QACH,MAAM,CAAC,kBAAkB,CAAC,CAAC,oBAAoB,CAAC;YAC9C,GAAG,EAAE,uBAAuB;YAC5B,WAAW,EAAE,MAAM;SACpB,CAAC,CAAC;QACH,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,gBAAgB,EAAE,CAAC;IACrD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;QACzD,EAAE,CAAC,KAAK,CAAC,UAAU,EAAE,qBAAqB,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;QAClE,EAAE,CAAC,KAAK,CAAC,UAAU,EAAE,0BAA0B,CAAC,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC;QACzE,EAAE,CAAC,KAAK,CAAC,UAAU,EAAE,0BAA0B,CAAC,CAAC,eAAe,CAC9D,uBAAuB,CACxB,CAAC;QAEF,mBAAmB,CAAC,UAAU,CAAC,CAAC;QAEhC,MAAM,CAAC,qBAAqB,CAAC,CAAC,oBAAoB,CAAC;YACjD,GAAG,EAAE,wBAAwB;SAC9B,CAAC,CAAC;QACH,MAAM,CAAC,mBAAmB,CAAC,CAAC,oBAAoB,CAAC;YAC/C,GAAG,EAAE,wBAAwB;SAC9B,CAAC,CAAC;QACH,MAAM,CAAC,sBAAsB,CAAC,CAAC,oBAAoB,CAAC;YAClD,GAAG,EAAE,wBAAwB;SAC9B,CAAC,CAAC;QACH,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,gBAAgB,EAAE,CAAC;IACrD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;QAC9C,EAAE,CAAC,KAAK,CAAC,UAAU,EAAE,0BAA0B,CAAC,CAAC,eAAe,CAC9D,0BAA0B,CAC3B,CAAC;QACF,mBAAmB,CAAC,UAAU,CAAC,CAAC;QAChC,MAAM,CAAC,iBAAiB,CAAC,CAAC,oBAAoB,CAC5C,MAAM,CAAC,gBAAgB,CAAC,EAAE,GAAG,EAAE,0BAA0B,EAAE,CAAC,CAC7D,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;QAC9C,EAAE,CAAC,KAAK,CAAC,UAAU,EAAE,0BAA0B,CAAC,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC;QACzE,EAAE,CAAC,KAAK,CAAC,UAAU,EAAE,0BAA0B,CAAC,CAAC,eAAe,CAC9D,0BAA0B,CAC3B,CAAC;QACF,mBAAmB,CAAC,UAAU,CAAC,CAAC;QAChC,MAAM,CAAC,qBAAqB,CAAC,CAAC,oBAAoB,CAChD,MAAM,CAAC,gBAAgB,CAAC,EAAE,GAAG,EAAE,2BAA2B,EAAE,CAAC,CAC9D,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kGAAkG,EAAE,GAAG,EAAE;QAC1G,EAAE,CAAC,KAAK,CAAC,UAAU,EAAE,oBAAoB,CAAC,CAAC,eAAe,CACxD,eAAe,CAAC,GAAG,CACpB,CAAC;QACF,EAAE,CAAC,KAAK,CAAC,UAAU,EAAE,0BAA0B,CAAC,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;QACxE,EAAE,CAAC,KAAK,CAAC,UAAU,EAAE,0BAA0B,CAAC,CAAC,eAAe,CAAC,EAAE,CAAC,CAAC;QAErE,MAAM,WAAW,GAAG,OAAO,CAAC,GAAG,CAAC,2BAA2B,CAAC,CAAC;QAC7D,OAAO,CAAC,GAAG,CAAC,2BAA2B,CAAC,GAAG,cAAc,CAAC;QAE1D,IAAI,CAAC;YACH,mBAAmB,CAAC,UAAU,CAAC,CAAC;YAEhC,MAAM,CAAC,gBAAgB,CAAC,CAAC,oBAAoB,CAAC,cAAc,CAAC,CAAC;YAC9D,MAAM,CAAC,cAAc,CAAC,CAAC,oBAAoB,CAAC,cAAc,CAAC,CAAC;YAC5D,MAAM,CAAC,iBAAiB,CAAC,CAAC,oBAAoB,CAAC,cAAc,CAAC,CAAC;YAC/D,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACrD,CAAC;gBAAS,CAAC;YACT,IAAI,WAAW,EAAE,CAAC;gBAChB,OAAO,CAAC,GAAG,CAAC,2BAA2B,CAAC,GAAG,WAAW,CAAC;YACzD,CAAC;iBAAM,CAAC;gBACN,OAAO,OAAO,CAAC,GAAG,CAAC,2BAA2B,CAAC,CAAC;YAClD,CAAC;QACH,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uEAAuE,EAAE,GAAG,EAAE;QAC/E,EAAE,CAAC,KAAK,CAAC,UAAU,EAAE,oBAAoB,CAAC,CAAC,eAAe,CACxD,eAAe,CAAC,GAAG,CACpB,CAAC;QACF,EAAE,CAAC,KAAK,CAAC,UAAU,EAAE,0BAA0B,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;QAEvE,mBAAmB,CAAC,UAAU,CAAC,CAAC;QAEhC,MAAM,CAAC,iBAAiB,CAAC,CAAC,oBAAoB,CAAC;YAC7C,GAAG,EAAE,uBAAuB;YAC5B,WAAW,EAAE,MAAM;SACpB,CAAC,CAAC;QACH,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAAC;YAC3C,GAAG,EAAE,uBAAuB;YAC5B,WAAW,EAAE,MAAM;SACpB,CAAC,CAAC;QACH,MAAM,CAAC,kBAAkB,CAAC,CAAC,oBAAoB,CAAC;YAC9C,GAAG,EAAE,uBAAuB;YAC5B,WAAW,EAAE,MAAM;SACpB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8EAA8E,EAAE,GAAG,EAAE;QACtF,EAAE,CAAC,KAAK,CAAC,UAAU,EAAE,oBAAoB,CAAC,CAAC,eAAe,CACxD,eAAe,CAAC,GAAG,CACpB,CAAC;QACF,EAAE,CAAC,KAAK,CAAC,UAAU,EAAE,0BAA0B,CAAC,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;QACxE,EAAE,CAAC,KAAK,CAAC,UAAU,EAAE,0BAA0B,CAAC,CAAC,eAAe,CAAC,EAAE,CAAC,CAAC;QAErE,MAAM,eAAe,GAAG,OAAO,CAAC,GAAG,CAAC,2BAA2B,CAAC,CAAC;QACjE,MAAM,iBAAiB,GAAG,OAAO,CAAC,GAAG,CAAC,sBAAsB,CAAC,CAAC;QAC9D,OAAO,OAAO,CAAC,GAAG,CAAC,2BAA2B,CAAC,CAAC;QAChD,OAAO,OAAO,CAAC,GAAG,CAAC,sBAAsB,CAAC,CAAC;QAE3C,IAAI,CAAC;YACH,mBAAmB,CAAC,UAAU,CAAC,CAAC;YAEhC,MAAM,CAAC,gBAAgB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;YAChD,MAAM,CAAC,cAAc,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;YAC9C,MAAM,CAAC,iBAAiB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;YACjD,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACrD,CAAC;gBAAS,CAAC;YACT,IAAI,eAAe,EAAE,CAAC;gBACpB,OAAO,CAAC,GAAG,CAAC,2BAA2B,CAAC,GAAG,eAAe,CAAC;YAC7D,CAAC;YACD,IAAI,iBAAiB,EAAE,CAAC;gBACtB,OAAO,CAAC,GAAG,CAAC,sBAAsB,CAAC,GAAG,iBAAiB,CAAC;YAC1D,CAAC;QACH,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uFAAuF,EAAE,GAAG,EAAE;QAC/F,EAAE,CAAC,KAAK,CAAC,UAAU,EAAE,oBAAoB,CAAC,CAAC,eAAe,CACxD,eAAe,CAAC,GAAG,CACpB,CAAC;QACF,EAAE,CAAC,KAAK,CAAC,UAAU,EAAE,0BAA0B,CAAC,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;QACxE,EAAE,CAAC,KAAK,CAAC,UAAU,EAAE,0BAA0B,CAAC,CAAC,eAAe,CAAC,EAAE,CAAC,CAAC;QAErE,MAAM,eAAe,GAAG,OAAO,CAAC,GAAG,CAAC,2BAA2B,CAAC,CAAC;QACjE,MAAM,iBAAiB,GAAG,OAAO,CAAC,GAAG,CAAC,sBAAsB,CAAC,CAAC;QAC9D,OAAO,OAAO,CAAC,GAAG,CAAC,2BAA2B,CAAC,CAAC;QAChD,OAAO,CAAC,GAAG,CAAC,sBAAsB,CAAC,GAAG,kBAAkB,CAAC;QAEzD,IAAI,CAAC;YACH,mBAAmB,CAAC,UAAU,CAAC,CAAC;YAEhC,MAAM,CAAC,gBAAgB,CAAC,CAAC,oBAAoB,CAAC,kBAAkB,CAAC,CAAC;YAClE,MAAM,CAAC,cAAc,CAAC,CAAC,oBAAoB,CAAC,kBAAkB,CAAC,CAAC;YAChE,MAAM,CAAC,iBAAiB,CAAC,CAAC,oBAAoB,CAAC,kBAAkB,CAAC,CAAC;YACnE,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACrD,CAAC;gBAAS,CAAC;YACT,IAAI,eAAe,EAAE,CAAC;gBACpB,OAAO,CAAC,GAAG,CAAC,2BAA2B,CAAC,GAAG,eAAe,CAAC;YAC7D,CAAC;YACD,IAAI,iBAAiB,EAAE,CAAC;gBACtB,OAAO,CAAC,GAAG,CAAC,sBAAsB,CAAC,GAAG,iBAAiB,CAAC;YAC1D,CAAC;iBAAM,CAAC;gBACN,OAAO,OAAO,CAAC,GAAG,CAAC,sBAAsB,CAAC,CAAC;YAC7C,CAAC;QACH,CAAC;IACH,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/telemetry/telemetry-utils.d.ts

@@ -0,0 +1,6 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export declare function getProgrammingLanguage(args: Record<string, unknown>): string | undefined;

package/dist/src/telemetry/telemetry-utils.js

@@ -0,0 +1,14 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { getLanguageFromFilePath } from '../utils/language-detection.js';
+export function getProgrammingLanguage(args) {
+    const filePath = args['file_path'] || args['path'] || args['absolute_path'];
+    if (typeof filePath === 'string') {
+        return getLanguageFromFilePath(filePath);
+    }
+    return undefined;
+}
+//# sourceMappingURL=telemetry-utils.js.map

package/dist/src/telemetry/telemetry-utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"telemetry-utils.js","sourceRoot":"","sources":["../../../src/telemetry/telemetry-utils.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,uBAAuB,EAAE,MAAM,gCAAgC,CAAC;AAEzE,MAAM,UAAU,sBAAsB,CACpC,IAA6B;IAE7B,MAAM,QAAQ,GAAG,IAAI,CAAC,WAAW,CAAC,IAAI,IAAI,CAAC,MAAM,CAAC,IAAI,IAAI,CAAC,eAAe,CAAC,CAAC;IAC5E,IAAI,OAAO,QAAQ,KAAK,QAAQ,EAAE,CAAC;QACjC,OAAO,uBAAuB,CAAC,QAAQ,CAAC,CAAC;IAC3C,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC"}

package/dist/src/telemetry/telemetry-utils.test.d.ts

@@ -0,0 +1,6 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export {};

package/dist/src/telemetry/telemetry-utils.test.js

@@ -0,0 +1,40 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect } from 'vitest';
+import { getProgrammingLanguage } from './telemetry-utils.js';
+describe('getProgrammingLanguage', () => {
+    it('should return the programming language when file_path is present', () => {
+        const args = { file_path: 'src/test.ts' };
+        const language = getProgrammingLanguage(args);
+        expect(language).toBe('TypeScript');
+    });
+    it('should return the programming language when absolute_path is present', () => {
+        const args = { absolute_path: 'src/test.py' };
+        const language = getProgrammingLanguage(args);
+        expect(language).toBe('Python');
+    });
+    it('should return the programming language when path is present', () => {
+        const args = { path: 'src/test.go' };
+        const language = getProgrammingLanguage(args);
+        expect(language).toBe('Go');
+    });
+    it('should return undefined when no file path is present', () => {
+        const args = {};
+        const language = getProgrammingLanguage(args);
+        expect(language).toBeUndefined();
+    });
+    it('should handle unknown file extensions gracefully', () => {
+        const args = { file_path: 'src/test.unknown' };
+        const language = getProgrammingLanguage(args);
+        expect(language).toBeUndefined();
+    });
+    it('should handle files with no extension', () => {
+        const args = { file_path: 'src/test' };
+        const language = getProgrammingLanguage(args);
+        expect(language).toBeUndefined();
+    });
+});
+//# sourceMappingURL=telemetry-utils.test.js.map

package/dist/src/telemetry/telemetry-utils.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"telemetry-utils.test.js","sourceRoot":"","sources":["../../../src/telemetry/telemetry-utils.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,sBAAsB,EAAE,MAAM,sBAAsB,CAAC;AAE9D,QAAQ,CAAC,wBAAwB,EAAE,GAAG,EAAE;IACtC,EAAE,CAAC,kEAAkE,EAAE,GAAG,EAAE;QAC1E,MAAM,IAAI,GAAG,EAAE,SAAS,EAAE,aAAa,EAAE,CAAC;QAC1C,MAAM,QAAQ,GAAG,sBAAsB,CAAC,IAAI,CAAC,CAAC;QAC9C,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;IACtC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sEAAsE,EAAE,GAAG,EAAE;QAC9E,MAAM,IAAI,GAAG,EAAE,aAAa,EAAE,aAAa,EAAE,CAAC;QAC9C,MAAM,QAAQ,GAAG,sBAAsB,CAAC,IAAI,CAAC,CAAC;QAC9C,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAClC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6DAA6D,EAAE,GAAG,EAAE;QACrE,MAAM,IAAI,GAAG,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC;QACrC,MAAM,QAAQ,GAAG,sBAAsB,CAAC,IAAI,CAAC,CAAC;QAC9C,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC9B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;QAC9D,MAAM,IAAI,GAAG,EAAE,CAAC;QAChB,MAAM,QAAQ,GAAG,sBAAsB,CAAC,IAAI,CAAC,CAAC;QAC9C,MAAM,CAAC,QAAQ,CAAC,CAAC,aAAa,EAAE,CAAC;IACnC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;QAC1D,MAAM,IAAI,GAAG,EAAE,SAAS,EAAE,kBAAkB,EAAE,CAAC;QAC/C,MAAM,QAAQ,GAAG,sBAAsB,CAAC,IAAI,CAAC,CAAC;QAC9C,MAAM,CAAC,QAAQ,CAAC,CAAC,aAAa,EAAE,CAAC;IACnC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;QAC/C,MAAM,IAAI,GAAG,EAAE,SAAS,EAAE,UAAU,EAAE,CAAC;QACvC,MAAM,QAAQ,GAAG,sBAAsB,CAAC,IAAI,CAAC,CAAC;QAC9C,MAAM,CAAC,QAAQ,CAAC,CAAC,aAAa,EAAE,CAAC;IACnC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/telemetry/telemetry.test.d.ts

@@ -0,0 +1,6 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export {};

package/dist/src/telemetry/telemetry.test.js

@@ -0,0 +1,50 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
+import { initializeTelemetry, shutdownTelemetry, isTelemetrySdkInitialized, } from './sdk.js';
+import { Config } from '../config/config.js';
+import { NodeSDK } from '@opentelemetry/sdk-node';
+vi.mock('@opentelemetry/sdk-node');
+vi.mock('../config/config.js');
+describe('telemetry', () => {
+    let mockConfig;
+    let mockNodeSdk;
+    beforeEach(() => {
+        vi.resetAllMocks();
+        mockConfig = new Config({
+            sessionId: 'test-session-id',
+            model: 'test-model',
+            targetDir: '/test/dir',
+            debugMode: false,
+            cwd: '/test/dir',
+        });
+        vi.spyOn(mockConfig, 'getTelemetryEnabled').mockReturnValue(true);
+        vi.spyOn(mockConfig, 'getTelemetryOtlpEndpoint').mockReturnValue('http://localhost:4317');
+        vi.spyOn(mockConfig, 'getSessionId').mockReturnValue('test-session-id');
+        mockNodeSdk = {
+            start: vi.fn(),
+            shutdown: vi.fn().mockResolvedValue(undefined),
+        };
+        vi.mocked(NodeSDK).mockImplementation(() => mockNodeSdk);
+    });
+    afterEach(async () => {
+        // Ensure we shut down telemetry even if a test fails.
+        if (isTelemetrySdkInitialized()) {
+            await shutdownTelemetry(mockConfig);
+        }
+    });
+    it('should initialize the telemetry service', () => {
+        initializeTelemetry(mockConfig);
+        expect(NodeSDK).toHaveBeenCalled();
+        expect(mockNodeSdk.start).toHaveBeenCalled();
+    });
+    it('should shutdown the telemetry service', async () => {
+        initializeTelemetry(mockConfig);
+        await shutdownTelemetry(mockConfig);
+        expect(mockNodeSdk.shutdown).toHaveBeenCalled();
+    });
+});
+//# sourceMappingURL=telemetry.test.js.map

package/dist/src/telemetry/telemetry.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"telemetry.test.js","sourceRoot":"","sources":["../../../src/telemetry/telemetry.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EACL,mBAAmB,EACnB,iBAAiB,EACjB,yBAAyB,GAC1B,MAAM,UAAU,CAAC;AAClB,OAAO,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAC;AAC7C,OAAO,EAAE,OAAO,EAAE,MAAM,yBAAyB,CAAC;AAElD,EAAE,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;AACnC,EAAE,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;AAE/B,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;IACzB,IAAI,UAAkB,CAAC;IACvB,IAAI,WAAoB,CAAC;IAEzB,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;QAEnB,UAAU,GAAG,IAAI,MAAM,CAAC;YACtB,SAAS,EAAE,iBAAiB;YAC5B,KAAK,EAAE,YAAY;YACnB,SAAS,EAAE,WAAW;YACtB,SAAS,EAAE,KAAK;YAChB,GAAG,EAAE,WAAW;SACjB,CAAC,CAAC;QACH,EAAE,CAAC,KAAK,CAAC,UAAU,EAAE,qBAAqB,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;QAClE,EAAE,CAAC,KAAK,CAAC,UAAU,EAAE,0BAA0B,CAAC,CAAC,eAAe,CAC9D,uBAAuB,CACxB,CAAC;QACF,EAAE,CAAC,KAAK,CAAC,UAAU,EAAE,cAAc,CAAC,CAAC,eAAe,CAAC,iBAAiB,CAAC,CAAC;QACxE,WAAW,GAAG;YACZ,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;YACd,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;SACzB,CAAC;QACxB,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,WAAW,CAAC,CAAC;IAC3D,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,KAAK,IAAI,EAAE;QACnB,sDAAsD;QACtD,IAAI,yBAAyB,EAAE,EAAE,CAAC;YAChC,MAAM,iBAAiB,CAAC,UAAU,CAAC,CAAC;QACtC,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;QACjD,mBAAmB,CAAC,UAAU,CAAC,CAAC;QAChC,MAAM,CAAC,OAAO,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACnC,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,gBAAgB,EAAE,CAAC;IAC/C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;QACrD,mBAAmB,CAAC,UAAU,CAAC,CAAC;QAChC,MAAM,iBAAiB,CAAC,UAAU,CAAC,CAAC;QAEpC,MAAM,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC,gBAAgB,EAAE,CAAC;IAClD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/telemetry/tool-call-decision.d.ts

@@ -0,0 +1,13 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { ToolConfirmationOutcome } from '../tools/tools.js';
+export declare enum ToolCallDecision {
+    ACCEPT = "accept",
+    REJECT = "reject",
+    MODIFY = "modify",
+    AUTO_ACCEPT = "auto_accept"
+}
+export declare function getDecisionFromOutcome(outcome: ToolConfirmationOutcome): ToolCallDecision;

package/dist/src/telemetry/tool-call-decision.js

@@ -0,0 +1,29 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { ToolConfirmationOutcome } from '../tools/tools.js';
+export var ToolCallDecision;
+(function (ToolCallDecision) {
+    ToolCallDecision["ACCEPT"] = "accept";
+    ToolCallDecision["REJECT"] = "reject";
+    ToolCallDecision["MODIFY"] = "modify";
+    ToolCallDecision["AUTO_ACCEPT"] = "auto_accept";
+})(ToolCallDecision || (ToolCallDecision = {}));
+export function getDecisionFromOutcome(outcome) {
+    switch (outcome) {
+        case ToolConfirmationOutcome.ProceedOnce:
+            return ToolCallDecision.ACCEPT;
+        case ToolConfirmationOutcome.ProceedAlways:
+        case ToolConfirmationOutcome.ProceedAlwaysServer:
+        case ToolConfirmationOutcome.ProceedAlwaysTool:
+            return ToolCallDecision.AUTO_ACCEPT;
+        case ToolConfirmationOutcome.ModifyWithEditor:
+            return ToolCallDecision.MODIFY;
+        case ToolConfirmationOutcome.Cancel:
+        default:
+            return ToolCallDecision.REJECT;
+    }
+}
+//# sourceMappingURL=tool-call-decision.js.map

package/dist/src/telemetry/tool-call-decision.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool-call-decision.js","sourceRoot":"","sources":["../../../src/telemetry/tool-call-decision.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,uBAAuB,EAAE,MAAM,mBAAmB,CAAC;AAE5D,MAAM,CAAN,IAAY,gBAKX;AALD,WAAY,gBAAgB;IAC1B,qCAAiB,CAAA;IACjB,qCAAiB,CAAA;IACjB,qCAAiB,CAAA;IACjB,+CAA2B,CAAA;AAC7B,CAAC,EALW,gBAAgB,KAAhB,gBAAgB,QAK3B;AAED,MAAM,UAAU,sBAAsB,CACpC,OAAgC;IAEhC,QAAQ,OAAO,EAAE,CAAC;QAChB,KAAK,uBAAuB,CAAC,WAAW;YACtC,OAAO,gBAAgB,CAAC,MAAM,CAAC;QACjC,KAAK,uBAAuB,CAAC,aAAa,CAAC;QAC3C,KAAK,uBAAuB,CAAC,mBAAmB,CAAC;QACjD,KAAK,uBAAuB,CAAC,iBAAiB;YAC5C,OAAO,gBAAgB,CAAC,WAAW,CAAC;QACtC,KAAK,uBAAuB,CAAC,gBAAgB;YAC3C,OAAO,gBAAgB,CAAC,MAAM,CAAC;QACjC,KAAK,uBAAuB,CAAC,MAAM,CAAC;QACpC;YACE,OAAO,gBAAgB,CAAC,MAAM,CAAC;IACnC,CAAC;AACH,CAAC"}