SQL Operations Studio Public Preview 1 (0.23) release source code

src/vs/workbench/parts/debug/browser/debugContentProvider.ts

@@ -0,0 +1,77 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the Source EULA. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+
+import uri from 'vs/base/common/uri';
+import { localize } from 'vs/nls';
+import { TPromise } from 'vs/base/common/winjs.base';
+import { guessMimeTypes, MIME_TEXT } from 'vs/base/common/mime';
+import { IModel } from 'vs/editor/common/editorCommon';
+import { IModelService } from 'vs/editor/common/services/modelService';
+import { IModeService } from 'vs/editor/common/services/modeService';
+import { ITextModelService, ITextModelContentProvider } from 'vs/editor/common/services/resolverService';
+import { IWorkbenchContribution } from 'vs/workbench/common/contributions';
+import { DEBUG_SCHEME, IDebugService, IProcess } from 'vs/workbench/parts/debug/common/debug';
+
+export class DebugContentProvider implements IWorkbenchContribution, ITextModelContentProvider {
+
+	constructor(
+		@ITextModelService textModelResolverService: ITextModelService,
+		@IDebugService private debugService: IDebugService,
+		@IModelService private modelService: IModelService,
+		@IModeService private modeService: IModeService
+	) {
+		textModelResolverService.registerTextModelContentProvider(DEBUG_SCHEME, this);
+	}
+
+	public getId(): string {
+		return 'debug.contentprovider';
+	}
+
+	public provideTextContent(resource: uri): TPromise<IModel> {
+
+		let process: IProcess;
+		if (resource.query) {
+			const keyvalues = resource.query.split('&');
+			for (let keyvalue of keyvalues) {
+				const pair = keyvalue.split('=');
+				if (pair.length === 2 && pair[0] === 'session') {
+					process = this.debugService.findProcessByUUID(decodeURIComponent(pair[1]));
+					break;
+				}
+			}
+		}
+
+		if (!process) {
+			// fallback: use focused process
+			process = this.debugService.getViewModel().focusedProcess;
+		}
+
+		if (!process) {
+			return TPromise.wrapError<IModel>(new Error(localize('unable', "Unable to resolve the resource without a debug session")));
+		}
+		const source = process.sources.get(resource.toString());
+		let rawSource: DebugProtocol.Source;
+		if (source) {
+			rawSource = source.raw;
+		} else {
+			// Remove debug: scheme
+			rawSource = { path: resource.with({ scheme: '', query: '' }).toString(true) };
+		}
+
+		return process.session.source({ sourceReference: source ? source.reference : undefined, source: rawSource }).then(response => {
+			const mime = response.body.mimeType || guessMimeTypes(resource.toString())[0];
+			const modePromise = this.modeService.getOrCreateMode(mime);
+			const model = this.modelService.createModel(response.body.content, modePromise, resource);
+
+			return model;
+		}, (err: DebugProtocol.ErrorResponse) => {
+			this.debugService.sourceIsNotAvailable(resource);
+			const modePromise = this.modeService.getOrCreateMode(MIME_TEXT);
+			const model = this.modelService.createModel(err.message, modePromise, resource);
+
+			return model;
+		});
+	}
+}