Use setTextDocumentLanguage to temporarily set language ID for CUDA sources, and headers (#7745)

Author: Colengms

Extension/src/LanguageServer/client.ts

@@ -449,6 +449,12 @@ interface GoToDirectiveInGroupParams {
     next: boolean;
 };
 
+interface SetTemporaryTextDocumentLanguageParams {
+    path: string;
+    isC: boolean;
+    isCuda: boolean;
+}
+
 // Requests
 const QueryCompilerDefaultsRequest: RequestType<QueryCompilerDefaultsParams, configs.CompilerDefaults, void, void> = new RequestType<QueryCompilerDefaultsParams, configs.CompilerDefaults, void, void>('cpptools/queryCompilerDefaults');
 const QueryTranslationUnitSourceRequest: RequestType<QueryTranslationUnitSourceParams, QueryTranslationUnitSourceResult, void, void> = new RequestType<QueryTranslationUnitSourceParams, QueryTranslationUnitSourceResult, void, void>('cpptools/queryTranslationUnitSource');
@@ -509,6 +515,7 @@ const ShowWarningNotification: NotificationType<ShowWarningParams, void> = new N
 const ReportTextDocumentLanguage: NotificationType<string, void> = new NotificationType<string, void>('cpptools/reportTextDocumentLanguage');
 const SemanticTokensChanged: NotificationType<string, void> = new NotificationType<string, void>('cpptools/semanticTokensChanged');
 const IntelliSenseSetupNotification: NotificationType<IntelliSenseSetup, void> = new NotificationType<IntelliSenseSetup, void>('cpptools/IntelliSenseSetup');
+const SetTemporaryTextDocumentLanguageNotification: NotificationType<SetTemporaryTextDocumentLanguageParams, void> = new NotificationType<SetTemporaryTextDocumentLanguageParams, void>('cpptools/setTemporaryTextDocumentLanguage');
 
 let failureMessageShown: boolean = false;
 
@@ -1971,6 +1978,7 @@ export class DefaultClient implements Client {
         this.languageClient.onNotification(ReportTextDocumentLanguage, (e) => this.setTextDocumentLanguage(e));
         this.languageClient.onNotification(SemanticTokensChanged, (e) => this.semanticTokensProvider?.invalidateFile(e));
         this.languageClient.onNotification(IntelliSenseSetupNotification, (e) => this.logIntellisenseSetupTime(e));
+        this.languageClient.onNotification(SetTemporaryTextDocumentLanguageNotification, (e) => this.setTemporaryTextDocumentLanguage(e));
         setupOutputHandlers();
     }
 
@@ -1984,6 +1992,14 @@ export class DefaultClient implements Client {
         }
     }
 
+    private async setTemporaryTextDocumentLanguage(params: SetTemporaryTextDocumentLanguageParams): Promise<void> {
+        const languageId: string = params.isC ? "c" : (params.isCuda ? "cuda-cpp" : "cpp");
+        const document: vscode.TextDocument = await vscode.workspace.openTextDocument(params.path);
+        if (!!document && document.languageId !== languageId) {
+            vscode.languages.setTextDocumentLanguage(document, languageId);
+        }
+    }
+
     private associations_for_did_change?: Set<string>;
 
     /**