Add selectTeamFortress2Folder command

cooolbros · Feb 22, 2025 · 9213e8f · 9213e8f
1 parent a152365
commit 9213e8f
Show file tree

Hide file tree

Showing 4 changed files with 48 additions and 16 deletions.
diff --git a/apps/extension/browser/client/src/extension.ts b/apps/extension/browser/client/src/extension.ts
@@ -2,6 +2,7 @@ import { Client, VSCodeVDFLanguageIDSchema, VSCodeVDFLanguageNameSchema, type VS
 import { copyKeyValuePath } from "client/commands/copyKeyValuePath"
 import { importPopfileTemplates } from "client/commands/importPopfileTemplates"
 import { JSONToVDF } from "client/commands/JSONToVDF"
+import { selectTeamFortress2Folder } from "client/commands/selectTeamFortress2Folder"
 import { showReferences } from "client/commands/showReferences"
 import { VDFToJSON } from "client/commands/VDFToJSON"
 import { onDidChangeActiveTextEditor } from "client/decorations"
@@ -17,7 +18,7 @@ export function activate(context: ExtensionContext): void {
 	// https://code.visualstudio.com/api/references/vscode-api
 
 	// Commands
-
+	context.subscriptions.push(commands.registerCommand("vscode-vdf.selectTeamFortress2Folder", selectTeamFortress2Folder))
 	context.subscriptions.push(commands.registerTextEditorCommand("vscode-vdf.copyKeyValuePath", copyKeyValuePath))
 	context.subscriptions.push(commands.registerTextEditorCommand("vscode-vdf.importPopfileTemplates", importPopfileTemplates))
 	context.subscriptions.push(commands.registerTextEditorCommand("vscode-vdf.JSONToVDF", JSONToVDF))

diff --git a/apps/extension/desktop/client/src/extension.ts b/apps/extension/desktop/client/src/extension.ts
@@ -5,6 +5,7 @@ import { VDFToJSON } from "client/commands/VDFToJSON"
 import { copyKeyValuePath } from "client/commands/copyKeyValuePath"
 import { extractVPKFileToWorkspace } from "client/commands/extractVPKFileToWorkspace"
 import { importPopfileTemplates } from "client/commands/importPopfileTemplates"
+import { selectTeamFortress2Folder } from "client/commands/selectTeamFortress2Folder"
 import { showReferences } from "client/commands/showReferences"
 import { onDidChangeActiveTextEditor } from "client/decorations"
 import { join } from "path"
@@ -20,7 +21,7 @@ export function activate(context: ExtensionContext): void {
 	// https://code.visualstudio.com/api/references/vscode-api
 
 	// Commands
-
+	context.subscriptions.push(commands.registerCommand("vscode-vdf.selectTeamFortress2Folder", selectTeamFortress2Folder))
 	context.subscriptions.push(commands.registerTextEditorCommand("vscode-vdf.copyKeyValuePath", copyKeyValuePath))
 	context.subscriptions.push(commands.registerTextEditorCommand("vscode-vdf.extractVPKFileToWorkspace", extractVPKFileToWorkspace))
 	context.subscriptions.push(commands.registerTextEditorCommand("vscode-vdf.importPopfileTemplates", importPopfileTemplates))
@@ -53,6 +54,13 @@ export function activate(context: ExtensionContext): void {
 		const serverModule = context.asAbsolutePath(join("apps/extension/desktop/servers/dist", `${languageId}.js`))
 		const name = VSCodeVDFLanguageNameSchema.shape[languageId].value
 
+		const options = {
+			execArgv: [
+				"--nolazy",
+				`--inspect=${6000 + Object.keys(VSCodeVDFLanguageNameSchema.shape).indexOf(languageId)}`
+			]
+		}
+
 		const client = languageClients[languageId] = new Client<LanguageClient>(
 			languageClients,
 			startServer,
@@ -65,23 +73,13 @@ export function activate(context: ExtensionContext): void {
 						module: serverModule,
 						transport: TransportKind.ipc,
 						...(process.env.NODE_ENV != "production" && {
-							options: {
-								execArgv: [
-									"--nolazy",
-									`--inspect=${6000 + Object.keys(VSCodeVDFLanguageNameSchema.shape).indexOf(languageId)}`
-								]
-							}
+							options: options
 						})
 					},
 					debug: {
 						module: serverModule,
 						transport: TransportKind.ipc,
-						options: {
-							execArgv: [
-								"--nolazy",
-								"--inspect=6009"
-							]
-						}
+						options: options
 					}
 				} satisfies ServerOptions,
 				{

diff --git a/package.json b/package.json
@@ -82,6 +82,11 @@
                 "title": "Restart VMT Language Server",
                 "command": "vscode-vdf.restartVMTLanguageServer"
             },
+            {
+                "category": "VDF",
+                "command": "vscode-vdf.selectTeamFortress2Folder",
+                "title": "Select Team Fortress 2 folder"
+            },
             {
                 "category": "VDF",
                 "title": "Show References",
@@ -112,8 +117,11 @@
                     },
                     "vscode-vdf.teamFortress2Folder": {
                         "type": "string",
-                        "markdownDescription": "Path to `Team Fortress 2` folder",
-                        "default": "C:/Program Files (x86)/Steam/steamapps/common/Team Fortress 2"
+                        "markdownDescription": "Path to `Team Fortress 2` folder.\n\n[Select Folder](command:vscode-vdf.selectTeamFortress2Folder)",
+                        "default": "C:/Program Files (x86)/Steam/steamapps/common/Team Fortress 2",
+                        "examples": [
+                            "~/.local/share/Steam/steamapps/common/Team Fortress 2"
+                        ]
                     },
                     "vscode-vdf.updateDiagnosticsEvent": {
                         "type": "string",

diff --git a/packages/client/src/commands/selectTeamFortress2Folder.ts b/packages/client/src/commands/selectTeamFortress2Folder.ts
@@ -0,0 +1,25 @@
+import { commands, Uri, window, workspace } from "vscode"
+
+export async function selectTeamFortress2Folder() {
+	const result = await window.showOpenDialog({
+		canSelectFiles: false,
+		canSelectFolders: true,
+		canSelectMany: false,
+	})
+
+	if (result && result.length) {
+		const uri = result[0]
+
+		const exists = await workspace.fs.stat(Uri.joinPath(uri, "tf/gameinfo.txt")).then(() => true, () => false)
+		if (!exists) {
+			window.showErrorMessage(`Invalid Team Fortress 2 folder: "${uri.fsPath}"`)
+			return
+		}
+
+		const path = uri.fsPath.replaceAll("\\", "/")
+		workspace.getConfiguration("vscode-vdf").update("teamFortress2Folder", path, true)
+
+		// Open settings UI to unfocus "Select Folder" link and refresh UI
+		commands.executeCommand("workbench.action.openSettings2")
+	}
+}