Expose telemetry to frontend

containers · Jan 29, 2024 · 4cada1c · 4cada1c
1 parent c767595
commit 4cada1c
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 0 deletions.
diff --git a/packages/backend/src/studio-api-impl.ts b/packages/backend/src/studio-api-impl.ts
@@ -38,6 +38,7 @@ export class StudioApiImpl implements StudioAPI {
     private playgroundManager: PlayGroundManager,
     private catalogManager: CatalogManager,
     private modelsManager: ModelsManager,
+    private telemetry: podmanDesktopApi.TelemetryLogger,
   ) {}
 
   async ping(): Promise<string> {
@@ -111,4 +112,12 @@ export class StudioApiImpl implements StudioAPI {
   async deleteLocalModel(modelId: string): Promise<void> {
     await this.modelsManager.deleteLocalModel(modelId);
   }
+
+  async telemetryLogUsage(eventName: string, data?: Record<string, any | podmanDesktopApi.TelemetryTrustedValue>): Promise<void> {
+    this.telemetry.logUsage(eventName, data);
+  }
+
+  async telemetryLogError(eventName: string, data?: Record<string, any | podmanDesktopApi.TelemetryTrustedValue>): Promise<void> {
+    this.telemetry.logError(eventName, data);
+  }
 }
diff --git a/packages/backend/src/studio.ts b/packages/backend/src/studio.ts
@@ -128,6 +128,7 @@ export class Studio {
       this.playgroundManager,
       this.catalogManager,
       this.modelsManager,
+      this.telemetry,
     );
 
     await this.catalogManager.loadCatalog();

diff --git a/packages/shared/src/StudioAPI.ts b/packages/shared/src/StudioAPI.ts
@@ -3,6 +3,7 @@ import type { ModelInfo } from './models/IModelInfo';
 import type { QueryState } from './models/IPlaygroundQueryState';
 import type { Catalog } from './models/ICatalog';
 import type { PlaygroundState } from './models/IPlaygroundState';
+import type { TelemetryTrustedValue } from '@podman-desktop/api';
 
 export abstract class StudioAPI {
   abstract ping(): Promise<string>;
@@ -24,4 +25,9 @@ export abstract class StudioAPI {
   abstract askPlayground(modelId: string, prompt: string): Promise<number>;
   abstract getPlaygroundQueriesState(): Promise<QueryState[]>;
   abstract getPlaygroundsState(): Promise<PlaygroundState[]>;
+
+  // eslint-disable-next-line @typescript-eslint/no-explicit-any
+  abstract telemetryLogUsage(eventName: string, data?: Record<string, any | TelemetryTrustedValue>): Promise<void>;
+  // eslint-disable-next-line @typescript-eslint/no-explicit-any
+  abstract telemetryLogError(eventName: string, data?: Record<string, any | TelemetryTrustedValue>): Promise<void>;
 }