refactor(ide): Improve IDE detection discovery (#6765)

packages/core/src/ide/detect-ide.ts

@@ -7,12 +7,13 @@
 export enum DetectedIde {
   Devin = 'devin',
   Replit = 'replit',
-  VSCode = 'vscode',
   Cursor = 'cursor',
   CloudShell = 'cloudshell',
   Codespaces = 'codespaces',
   FirebaseStudio = 'firebasestudio',
   Trae = 'trae',
+  VSCode = 'vscode',
+  VSCodeFork = 'vscodefork',
 }
 
 export interface IdeInfo {
@@ -29,10 +30,6 @@ export function getIdeInfo(ide: DetectedIde): IdeInfo {
       return {
         displayName: 'Replit',
       };
-    case DetectedIde.VSCode:
-      return {
-        displayName: 'VS Code',
-      };
     case DetectedIde.Cursor:
       return {
         displayName: 'Cursor',
@@ -53,6 +50,14 @@ export function getIdeInfo(ide: DetectedIde): IdeInfo {
       return {
         displayName: 'Trae',
       };
+    case DetectedIde.VSCode:
+      return {
+        displayName: 'VS Code',
+      };
+    case DetectedIde.VSCodeFork:
+      return {
+        displayName: 'IDE',
+      };
     default: {
       // This ensures that if a new IDE is added to the enum, we get a compile-time error.
       const exhaustiveCheck: never = ide;
@@ -61,11 +66,7 @@ export function getIdeInfo(ide: DetectedIde): IdeInfo {
   }
 }
 
-export function detectIde(): DetectedIde | undefined {
-  // Only VSCode-based integrations are currently supported.
-  if (process.env['TERM_PROGRAM'] !== 'vscode') {
-    return undefined;
-  }
+export function detectIdeFromEnv(): DetectedIde {
   if (process.env['__COG_BASHRC_SOURCED']) {
     return DetectedIde.Devin;
   }
@@ -89,3 +90,32 @@ export function detectIde(): DetectedIde | undefined {
   }
   return DetectedIde.VSCode;
 }
+
+function verifyVSCode(
+  ide: DetectedIde,
+  ideProcessInfo: {
+    pid: number;
+    command: string;
+  },
+): DetectedIde {
+  if (ide !== DetectedIde.VSCode) {
+    return ide;
+  }
+  if (ideProcessInfo.command.toLowerCase().includes('code')) {
+    return DetectedIde.VSCode;
+  }
+  return DetectedIde.VSCodeFork;
+}
+
+export function detectIde(ideProcessInfo: {
+  pid: number;
+  command: string;
+}): DetectedIde | undefined {
+  // Only VSCode-based integrations are currently supported.
+  if (process.env['TERM_PROGRAM'] !== 'vscode') {
+    return undefined;
+  }
+
+  const ide = detectIdeFromEnv();
+  return verifyVSCode(ide, ideProcessInfo);
+}