Feature: Settings to enable telemetry for extension

package.json

@@ -170,6 +170,18 @@
           "fontCharacter": "\\e90c"
         }
       }
+    },
+    "configuration": {
+      "id": "pullflow",
+      "title": "Pullflow",
+      "properties": {
+        "pullflow.telemetry.enabled": {
+          "type": "boolean",
+          "default": true,
+         "markdownDescription": "Enable Pullflow to transmit product usage telemetry. \n\n_**Important:** To activate telemetry transmission, both this setting and the VS Code telemetry option must be enabled. Telemetry will not be sent if either of these settings is disabled._"
+
+        }
+      }
     }
   },
   "scripts": {

src/utils/initialize.ts

@@ -4,6 +4,7 @@ import {
   WindowState,
   commands,
   window,
+  workspace,
 } from 'vscode'
 import { Store } from './store'
 import { log } from './logger'
@@ -27,6 +28,9 @@ export const initialize = async ({
   statusBar: StatusBarItem
 }) => {
   log.info('initializing extension', module)
+
+  await initializeTelemetry(context)
+
   const errorCount = { count: 0 }
   await PullRequestState.update({
     context,
@@ -116,3 +120,29 @@ const setSpaceUsers = async ({
     spaceUsers: spaceUsers.spaceUsers,
   })
 }
+
+const extensionTelemetryFlag = () =>
+  workspace
+    .getConfiguration('pullflow.telemetry.enabled')
+    .get<boolean>('pullflow.telemetry.enabled', true)
+
+const vscodeTelemetryFlag = () =>
+  workspace.getConfiguration('telemetry').get<boolean>('enableTelemetry')
+
+const initializeTelemetry = async (context: ExtensionContext) => {
+  await Store.set(context, {
+    isTelemetryEnabled: vscodeTelemetryFlag() && extensionTelemetryFlag(),
+  })
+
+  const disposable = workspace.onDidChangeConfiguration(async (event) => {
+    if (
+      event.affectsConfiguration('telemetry.enableTelemetry') ||
+      event.affectsConfiguration('pullflow.telemetry.enabled')
+    ) {
+      await Store.set(context, {
+        isTelemetryEnabled: vscodeTelemetryFlag() && extensionTelemetryFlag(),
+      })
+    }
+  })
+  context.subscriptions.push(disposable)
+}

src/utils/trace.ts

@@ -1,3 +1,4 @@
+/* eslint-disable no-unused-vars */
 import { Span, Tracer, SpanKind, Attributes } from '@opentelemetry/api'
 import { OTLPTraceExporter } from '@opentelemetry/exporter-trace-otlp-http'
 import { Resource } from '@opentelemetry/resources'
@@ -10,13 +11,46 @@ import { ExtensionContext, extensions } from 'vscode'
 import { TraceAttributes } from './types'
 import { Store } from './store'
 import { AppConfig } from './appConfig'
+import { log } from './logger'
 
 const extensionInfo = extensions.getExtension('Pullflow.pullflow')?.packageJSON
 
-export class Trace {
+type FakeTracer = {}
+type FakeBasicTracerProvider = {}
+type FakeAttribute = {}
+
+export function instantiatePullflowTracer(context: ExtensionContext) {
+  const { isTelemetryEnabled } = Store.get(context)
+  if (isTelemetryEnabled) {
+    return new Trace(context)
+  } else {
+    return new FakeTrace(context)
+  }
+}
+class FakeTrace {
+  tracer: FakeTracer
+  provider: FakeBasicTracerProvider
+  defaultAttributes: FakeAttribute
+
+  constructor(_context: ExtensionContext) {
+    this.provider = {}
+    this.tracer = {}
+    this.defaultAttributes = {}
+  }
+
+  dispose(): void {}
+  start({ name, attributes }: { name: string; attributes?: TraceAttributes }) {
+    console.log({ name, attributes })
+  }
+  end({ attributes }: { attributes?: TraceAttributes }) {
+    console.log({ attributes })
+  }
+}
+class Trace {
   tracer: Tracer
   provider: BasicTracerProvider
   defaultAttributes: Attributes
+  span: Span | undefined
 
   constructor(context: ExtensionContext) {
     this.provider = new BasicTracerProvider({
@@ -34,43 +68,36 @@ export class Trace {
     this.tracer = this.provider.getTracer(extensionInfo.name)
     const { user } = Store.get(context)
     this.defaultAttributes = user || {}
+    this.span = undefined
   }
 
   dispose(): void {
     void this.provider.shutdown()
   }
 
-  start({
-    name,
-    attributes,
-  }: {
-    name: string
-    attributes?: TraceAttributes
-  }): Span {
-    const span = this.tracer.startSpan(name, {
+  start({ name, attributes }: { name: string; attributes?: TraceAttributes }) {
+    this.span = this.tracer.startSpan(name, {
       kind: SpanKind.INTERNAL,
       startTime: Date.now(),
     })
     if (attributes)
-      span.setAttributes({
+      this.span.setAttributes({
         ...attributes,
         ...this.defaultAttributes,
       })
-    return span
   }
 
-  end({
-    span,
-    attributes,
-  }: {
-    span: Span
-    attributes?: TraceAttributes
-  }): void {
+  end({ attributes }: { attributes?: TraceAttributes }): void {
+    if (this.span === undefined) {
+      log.warn('span is undefined', 'trace.ts')
+      return
+    }
     if (attributes)
-      span.setAttributes({
+      this.span.setAttributes({
         ...attributes,
         ...this.defaultAttributes,
       })
-    span.end(Date.now())
+    this.span.end(Date.now())
+    this.span = undefined
   }
 }

src/utils/types.ts

@@ -48,6 +48,7 @@ export type CacheObject = {
   keyStrokeCount?: number
   lastKeyStrokeTime?: number | null
   previousPresenceStatus?: PresenceStatus
+  isTelemetryEnabled?: boolean
 }
 export enum StatusBarState {
   Loading,

src/views/quickpicks/quickPick.ts

@@ -1,5 +1,5 @@
 import { ExtensionContext, QuickPickItem, window } from 'vscode'
-import { Trace } from '../../utils/trace'
+import { instantiatePullflowTracer } from '../../utils/trace'
 
 export const QuickPick = {
   create: <Type extends QuickPickItem>({
@@ -17,8 +17,8 @@ export const QuickPick = {
   }) => {
     const quickPick = window.createQuickPick<Type>()
 
-    const trace = new Trace(context)
-    const span = trace.start({
+    const trace = instantiatePullflowTracer(context)
+    trace.start({
       name: title,
     })
 
@@ -29,7 +29,6 @@ export const QuickPick = {
 
     quickPick.onDidHide(() => {
       trace.end({
-        span,
         attributes: {
           title,
         },
@@ -39,7 +38,6 @@ export const QuickPick = {
 
     quickPick.onDidAccept(() => {
       trace.end({
-        span,
         attributes: {
           title,
           selectedItem: quickPick.selectedItems[0]?.label,