From 6ec0bf4dd0c59fdfcacaaa36d3b7cb374da3e243 Mon Sep 17 00:00:00 2001
From: wolfee <adam.farkas@mongodb.com>
Date: Tue, 13 May 2025 11:26:13 +0200
Subject: [PATCH] SERVER-102895 Upgrade code-workspace for v5 (#34095)

Co-authored-by: Mathias Stearn <mathias@mongodb.com>
GitOrigin-RevId: 5762a4de82b908e357fe4e2b5fef4aaf8db0305f
---
 .../linux-virtual-workstation.code-workspace  | 41 ++++++++++++++-----
 bazel/mongo_src_rules.bzl                     |  5 ++-
 buildscripts/clangd_vscode.sh                 | 39 ++++++++++++++++++
 3 files changed, 74 insertions(+), 11 deletions(-)
 create mode 100755 buildscripts/clangd_vscode.sh

diff --git a/.vscode_defaults/linux-virtual-workstation.code-workspace b/.vscode_defaults/linux-virtual-workstation.code-workspace
index d2c16aea85d..3946b688aee 100644
--- a/.vscode_defaults/linux-virtual-workstation.code-workspace
+++ b/.vscode_defaults/linux-virtual-workstation.code-workspace
@@ -5,12 +5,9 @@
 		}
 	],
 	"settings": {
-		"clangd.arguments": [
-			"--query-driver=/opt/mongodbtoolchain/v4/bin/*",
-			"--header-insertion=never",
-		],
 		"clangd.checkUpdates": true,
-		"clang-format.executable": "/opt/mongodbtoolchain/v5/bin/clang-format",
+		"clangd.path": "${workspaceFolder}/buildscripts/clangd_vscode.sh",
+		"clang-format.executable": "${workspaceRoot}/bazel-mongo/external/mongo_toolchain_v5/v5/bin/clang-format",
 		"clang-tidy.executable": "buildscripts/clang_tidy_vscode.py",
 		"editor.codeActionsOnSave": {
 			"source.fixAll.eslint": "explicit"
@@ -24,11 +21,11 @@
 		"files.insertFinalNewline": true,
 		"js/ts.implicitProjectConfig.target": "ES2020",
 		"python.autoComplete.extraPaths": [
-			"/opt/mongodbtoolchain/v4/share/gcc-11.3.0/python",
+			"/opt/mongodbtoolchain/v5/share/gcc-14.2.0/python"
 		],
 		"python.defaultInterpreterPath": "python3-venv/bin/python",
 		"python.analysis.extraPaths": [
-			"/opt/mongodbtoolchain/v4/share/gcc-11.3.0/python",
+			"/opt/mongodbtoolchain/v5/share/gcc-14.2.0/python"
 		],
 		"mypy-type-checker.path": [
 			"${interpreter}",
@@ -87,6 +84,24 @@
 			"ms-vscode.cpptools"
 		]
 	},
+	"tasks": {
+		"version": "2.0.0",
+		"tasks": [
+			{
+				"label": "_install_pymongo",
+				"hide": true,
+				"type": "shell",
+				"command": "pip3 install pymongo",
+			},
+			{
+				"label": "_prelaunch_task",
+				"hide": true,
+				"dependsOn": [
+					"_install_pymongo"
+				]
+			}
+		]
+	},
 	"launch": {
 		"version": "0.2.0",
 		"configurations": [
@@ -101,6 +116,7 @@
 				"environment": [],
 				"externalConsole": false,
 				"MIMode": "gdb",
+				"preLaunchTask": "_prelaunch_task",
 				"setupCommands": [
 					{
 						"text": "set verbose",
@@ -144,7 +160,7 @@
 						"ignoreFailures": false
 					},
 				],
-				"miDebuggerPath": "/opt/mongodbtoolchain/v4/bin/gdb",
+				"miDebuggerPath": "${workspaceFolder}/bazel-mongo/external/gdb/v5/bin/gdb",
 			},
 			{
 				"name": "attach",
@@ -153,6 +169,7 @@
 				"program": "${workspaceFolder}/${input:runTargets}",
 				"processId": "${command:pickProcess}",
 				"MIMode": "gdb",
+				"preLaunchTask": "_prelaunch_task",
 				"setupCommands": [
 					{
 						"text": "set verbose",
@@ -196,7 +213,7 @@
 						"ignoreFailures": false
 					},
 				],
-				"miDebuggerPath": "/opt/mongodbtoolchain/v4/bin/gdb",
+				"miDebuggerPath": "${workspaceFolder}/bazel-mongo/external/gdb/v5/bin/gdb",
 			},
 			{
 				"name": "coredump",
@@ -206,7 +223,8 @@
 				"program": "${workspaceFolder}/${input:runTargets}",
 				"args": [],
 				"MIMode": "gdb",
-				"miDebuggerPath": "/opt/mongodbtoolchain/v4/bin/gdb",
+				"miDebuggerPath": "${workspaceFolder}/bazel-mongo/external/gdb/v5/bin/gdb",
+				"preLaunchTask": "_prelaunch_task",
 				"setupCommands": [
 					{
 						"text": "set verbose",
@@ -271,6 +289,9 @@
 				"args": {
 					"command": "bazel query 'attr(tags, \"mongo_binary\", //...)' union 'attr(tags, \"mongo_unittest\", //...)' | grep mongo | grep -v \"_with_debug\" | grep -v third_party | sed 's;//;bazel-bin/;g' | sed 's;:;/;g'",
 					"cwd": "${workspaceFolder}",
+					"taskId": "runTargets",
+					"rememberPrevious": true,
+					"warnOnStderr": false,
 				}
 			},
 		]
diff --git a/bazel/mongo_src_rules.bzl b/bazel/mongo_src_rules.bzl
index 923740edeeb..abe7d886442 100644
--- a/bazel/mongo_src_rules.bzl
+++ b/bazel/mongo_src_rules.bzl
@@ -810,7 +810,10 @@ def _mongo_cc_binary_and_test(
         "visibility": visibility,
         "testonly": testonly,
         "copts": copts,
-        "data": data + SANITIZER_DATA,
+        "data": data + SANITIZER_DATA + select({
+            "//bazel/platforms:use_mongo_toolchain": ["@gdb"],
+            "//conditions:default": [],
+        }),
         "tags": tags,
         "linkopts": linkopts + rpath_flags + select({
             "//bazel/config:thin_lto_enabled": ["-Wl,--threads=" + str(NUM_CPUS)],
diff --git a/buildscripts/clangd_vscode.sh b/buildscripts/clangd_vscode.sh
new file mode 100755
index 00000000000..a2bd5078826
--- /dev/null
+++ b/buildscripts/clangd_vscode.sh
@@ -0,0 +1,39 @@
+#!/usr/bin/env bash
+
+# Collect all passed arguments
+ARGS=("$@")
+
+# Ordered list of possible clangd locations
+CANDIDATES=(
+  "$(command -v custom-clangd)"
+  ".compiledb/compiledb-mongo/external/mongo_toolchain_v5/v5/bin/clangd"
+  "bazel-mongo/external/mongo_toolchain_v5/v5/bin/clangd"
+  "/opt/mongodbtoolchain/v5/bin/clangd"
+)
+
+# Find the first available clangd
+for CANDIDATE in "${CANDIDATES[@]}"; do
+  if [[ -x "$CANDIDATE" ]]; then
+    CLANGD="$CANDIDATE"
+    echo "[INFO] Using clangd at: $CLANGD" >&2
+    break
+  fi
+done
+
+# Fail if no clangd was found
+if [[ -z "$CLANGD" ]]; then
+  echo "[ERROR] clangd not found in any of the expected locations." >&2
+  exit 1
+fi
+
+FINAL_ARGS=(
+  "${ARGS[@]}"
+  "--query-driver=./**/*{clang,gcc,g++}*" # allow any clang or gcc binary in the repo
+  "--header-insertion=never"
+)
+
+# Log the full command (optional)
+echo "[INFO] Executing: $CLANGD ${FINAL_ARGS[*]}" >&2
+
+# Run clangd with the final arguments
+"$CLANGD" "${FINAL_ARGS[@]}"