Adds lang service switch

Author: lewis-sanchez

package.json

@@ -1196,6 +1196,12 @@
           "default": false,
           "description": "%mssql.intelliSense.lowerCaseSuggestions%",
           "scope": "window"
+        },
+		"mssql.intelliSense.setQueryEditorLanguageToNone": {
+          "type": "boolean",
+          "default": false,
+          "description": "%mssql.intelliSense.setQueryEditorLanguageToNone%",
+          "scope": "window"
         },
         "mssql.persistQueryResultTabs": {
           "type": "boolean",

package.nls.json

@@ -101,6 +101,7 @@
 "mssql.enableQueryHistoryFeature":"Should Query History feature be enabled",
 "mssql.intelliSense.lowerCaseSuggestions":"Should IntelliSense suggestions be lowercase",
 "mssql.persistQueryResultTabs":"Should query result selections and scroll positions be saved when switching tabs (may impact performance)",
+"mssql.intelliSense.setQueryEditorLanguageToNone":"Should the editor language be set to None when a query editor is opened.",
 "mssql.queryHistoryLimit":"Number of query history entries to show in the Query History view",
 "mssql.createAzureFunction":"Create Azure Function with SQL binding",
 "mssql.query.maxXmlCharsToStore":"Maximum number of characters to store for each value in XML columns after running a query. Default value: 2,097,152. Valid value range: 1 to 2,147,483,647.",

src/constants/constants.ts

@@ -162,6 +162,7 @@ export const configQueryHistoryLimit = 'queryHistoryLimit';
 export const configEnableQueryHistoryCapture = 'enableQueryHistoryCapture';
 export const configEnableQueryHistoryFeature = 'enableQueryHistoryFeature';
 export const configEnableExperimentalFeatures = 'mssql.enableExperimentalFeatures';
+export const configIntelliSense = 'intelliSense';
 
 // ToolsService Constants
 export const serviceInstallingTo = 'Installing SQL tools service to';

src/controllers/connectionManager.ts

@@ -23,7 +23,7 @@ import { IAccount } from '../models/contracts/azure';
 import * as ConnectionContracts from '../models/contracts/connection';
 import { ClearPooledConnectionsRequest, ConnectionSummary } from '../models/contracts/connection';
 import * as LanguageServiceContracts from '../models/contracts/languageService';
-import { EncryptOptions, IConnectionProfile } from '../models/interfaces';
+import { EncryptOptions, IConnectionProfile, IntelliSenseConfig } from '../models/interfaces';
 import { PlatformInformation, Runtime } from '../models/platform';
 import * as Utils from '../models/utils';
 import { IPrompter, IQuestion, QuestionTypes } from '../prompts/question';
@@ -872,9 +872,22 @@ export default class ConnectionManager {
 
 				// Note: must call flavor changed before connecting, or the timer showing an animation doesn't occur
 				if (this.statusView) {
-					this.statusView.languageFlavorChanged(fileUri, Constants.mssqlProviderName);
+					const configuration = vscode.workspace.getConfiguration(Constants.extensionConfigSectionName);
+					const intelliSenseConfig = configuration.get<IntelliSenseConfig>(Constants.configIntelliSense);
+					const flavor = intelliSenseConfig.setQueryEditorLanguageToNone ? Constants.noneProviderName : Constants.mssqlProviderName;
+					this.statusView.languageFlavorChanged(fileUri, flavor);
 					this.statusView.connecting(fileUri, connectionCreds);
-					this.statusView.languageFlavorChanged(fileUri, Constants.mssqlProviderName);
+					this.statusView.languageFlavorChanged(fileUri, flavor);
+
+					// Notify the language service that the editor with the specified URI doesn't have a language flavor to avoid error squiggles
+					if (flavor === Constants.noneProviderName) {
+						SqlToolsServerClient.instance.sendNotification(LanguageServiceContracts.LanguageFlavorChangedNotification.type,
+							<LanguageServiceContracts.DidChangeLanguageFlavorParams>{
+								uri: fileUri,
+								language: 'sql',
+								flavor: Constants.noneProviderName
+							});
+					}
 				}
 				this.vscodeWrapper.logToOutputChannel(
 					LocalizedConstants.msgConnecting(connectionCreds.server, fileUri)

src/controllers/mainController.ts

@@ -13,7 +13,7 @@ import * as Constants from '../constants/constants';
 import * as LocalizedConstants from '../constants/locConstants';
 import SqlToolsServerClient from '../languageservice/serviceclient';
 import * as ConnInfo from '../models/connectionInfo';
-import { CompletionExtensionParams, CompletionExtLoadRequest, RebuildIntelliSenseNotification } from '../models/contracts/languageService';
+import { CompletionExtensionParams, CompletionExtLoadRequest, DidChangeLanguageFlavorParams, LanguageFlavorChangedNotification, RebuildIntelliSenseNotification } from '../models/contracts/languageService';
 import { ScriptOperation } from '../models/contracts/scripting/scriptingRequest';
 import { SqlOutputContentProvider } from '../models/sqlOutputContentProvider';
 import * as Utils from '../models/utils';
@@ -1288,8 +1288,20 @@ export default class MainController implements vscode.Disposable {
 		this._connectionMgr.onDidOpenTextDocument(doc);
 
 		if (doc && doc.languageId === Constants.languageId) {
+			const isLanguageSetToNone = this.configuration.get('mssql.intelliSense.setQueryEditorLanguageToNone');
+			const flavor = isLanguageSetToNone ? Constants.noneProviderName : Constants.mssqlProviderName;
 			// set encoding to false
-			this._statusview.languageFlavorChanged(doc.uri.toString(true), Constants.mssqlProviderName);
+			this._statusview.languageFlavorChanged(doc.uri.toString(true), flavor);
+
+			// Notify the language service that the editor with the specified URI doesn't have a language flavor to avoid error squiggles.
+			if (flavor === Constants.noneProviderName) {
+				SqlToolsServerClient.instance.sendNotification(LanguageFlavorChangedNotification.type,
+					<DidChangeLanguageFlavorParams>{
+						uri: doc.uri.toString(true),
+						language: 'sql',
+						flavor: Constants.noneProviderName
+					});
+			}
 		}
 
 		if (doc && doc.languageId === Constants.sqlPlanLanguageId) {

src/models/interfaces.ts

@@ -10,6 +10,10 @@ import * as vscodeMssql from 'vscode-mssql';
 import { AzureAuthType } from './contracts/azure';
 
 // interfaces
+export interface IntelliSenseConfig {
+	setQueryEditorLanguageToNone: boolean;
+}
+
 export enum ContentType {
 	Root = 0,
 	Messages = 1,