mirror of
https://github.com/ckaczor/azuredatastudio.git
synced 2026-02-06 01:25:38 -05:00
Merge from vscode a348d103d1256a06a2c9b3f9b406298a9fef6898 (#15681)
* Merge from vscode a348d103d1256a06a2c9b3f9b406298a9fef6898 * Fixes and cleanup * Distro * Fix hygiene yarn * delete no yarn lock changes file * Fix hygiene * Fix layer check * Fix CI * Skip lib checks * Remove tests deleted in vs code * Fix tests * Distro * Fix tests and add removed extension point * Skip failing notebook tests for now * Disable broken tests and cleanup build folder * Update yarn.lock and fix smoke tests * Bump sqlite * fix contributed actions and file spacing * Fix user data path * Update yarn.locks Co-authored-by: ADS Merger <karlb@microsoft.com>
This commit is contained in:
@@ -5,14 +5,8 @@
|
||||
|
||||
import { session, WebContents, webContents } from 'electron';
|
||||
import { Disposable } from 'vs/base/common/lifecycle';
|
||||
import { URI } from 'vs/base/common/uri';
|
||||
import { IFileService } from 'vs/platform/files/common/files';
|
||||
import { ILogService } from 'vs/platform/log/common/log';
|
||||
import { ITunnelService } from 'vs/platform/remote/common/tunnel';
|
||||
import { IRequestService } from 'vs/platform/request/common/request';
|
||||
import { webviewPartitionId } from 'vs/platform/webview/common/resourceLoader';
|
||||
import { IWebviewManagerService, RegisterWebviewMetadata, WebviewManagerDidLoadResourceResponse, WebviewWebContentsId, WebviewWindowId } from 'vs/platform/webview/common/webviewManagerService';
|
||||
import { WebviewPortMappingProvider } from 'vs/platform/webview/electron-main/webviewPortMappingProvider';
|
||||
import { IWebviewManagerService, webviewPartitionId, WebviewWebContentsId, WebviewWindowId } from 'vs/platform/webview/common/webviewManagerService';
|
||||
import { WebviewProtocolProvider } from 'vs/platform/webview/electron-main/webviewProtocolProvider';
|
||||
import { IWindowsMainService } from 'vs/platform/windows/electron-main/windows';
|
||||
|
||||
@@ -20,64 +14,24 @@ export class WebviewMainService extends Disposable implements IWebviewManagerSer
|
||||
|
||||
declare readonly _serviceBrand: undefined;
|
||||
|
||||
private readonly protocolProvider: WebviewProtocolProvider;
|
||||
private readonly portMappingProvider: WebviewPortMappingProvider;
|
||||
|
||||
constructor(
|
||||
@IFileService fileService: IFileService,
|
||||
@ILogService logService: ILogService,
|
||||
@IRequestService requestService: IRequestService,
|
||||
@ITunnelService tunnelService: ITunnelService,
|
||||
@IWindowsMainService private readonly windowsMainService: IWindowsMainService,
|
||||
) {
|
||||
super();
|
||||
this.protocolProvider = this._register(new WebviewProtocolProvider(fileService, logService, requestService, windowsMainService));
|
||||
this.portMappingProvider = this._register(new WebviewPortMappingProvider(tunnelService));
|
||||
this._register(new WebviewProtocolProvider());
|
||||
|
||||
const sess = session.fromPartition(webviewPartitionId);
|
||||
sess.setPermissionRequestHandler((webContents, permission /* 'media' | 'geolocation' | 'notifications' | 'midiSysex' | 'pointerLock' | 'fullscreen' | 'openExternal' */, callback) => {
|
||||
sess.setPermissionRequestHandler((_webContents, permission, callback) => {
|
||||
if (permission === 'clipboard-read') {
|
||||
return callback(true);
|
||||
}
|
||||
|
||||
return callback(false);
|
||||
});
|
||||
|
||||
sess.setPermissionCheckHandler((webContents, permission /* 'media' */) => {
|
||||
return false;
|
||||
});
|
||||
}
|
||||
|
||||
public async registerWebview(id: string, windowId: number, metadata: RegisterWebviewMetadata): Promise<void> {
|
||||
const extensionLocation = metadata.extensionLocation ? URI.from(metadata.extensionLocation) : undefined;
|
||||
|
||||
this.protocolProvider.registerWebview(id, {
|
||||
...metadata,
|
||||
windowId: windowId,
|
||||
extensionLocation,
|
||||
localResourceRoots: metadata.localResourceRoots.map(x => URI.from(x))
|
||||
});
|
||||
|
||||
this.portMappingProvider.registerWebview(id, {
|
||||
extensionLocation,
|
||||
mappings: metadata.portMappings,
|
||||
resolvedAuthority: metadata.remoteConnectionData,
|
||||
});
|
||||
}
|
||||
|
||||
public async unregisterWebview(id: string): Promise<void> {
|
||||
this.protocolProvider.unregisterWebview(id);
|
||||
this.portMappingProvider.unregisterWebview(id);
|
||||
}
|
||||
|
||||
public async updateWebviewMetadata(id: string, metaDataDelta: Partial<RegisterWebviewMetadata>): Promise<void> {
|
||||
const extensionLocation = metaDataDelta.extensionLocation ? URI.from(metaDataDelta.extensionLocation) : undefined;
|
||||
|
||||
this.protocolProvider.updateWebviewMetadata(id, {
|
||||
...metaDataDelta,
|
||||
extensionLocation,
|
||||
localResourceRoots: metaDataDelta.localResourceRoots?.map(x => URI.from(x)),
|
||||
});
|
||||
|
||||
this.portMappingProvider.updateWebviewMetadata(id, {
|
||||
...metaDataDelta,
|
||||
extensionLocation,
|
||||
sess.setPermissionCheckHandler((_webContents, permission /* 'media' */) => {
|
||||
return permission === 'clipboard-read';
|
||||
});
|
||||
}
|
||||
|
||||
@@ -87,7 +41,7 @@ export class WebviewMainService extends Disposable implements IWebviewManagerSer
|
||||
if (typeof (id as WebviewWindowId).windowId === 'number') {
|
||||
const { windowId } = (id as WebviewWindowId);
|
||||
const window = this.windowsMainService.getWindowById(windowId);
|
||||
if (!window) {
|
||||
if (!window?.win) {
|
||||
throw new Error(`Invalid windowId: ${windowId}`);
|
||||
}
|
||||
contents = window.win.webContents;
|
||||
@@ -103,8 +57,4 @@ export class WebviewMainService extends Disposable implements IWebviewManagerSer
|
||||
contents.setIgnoreMenuShortcuts(enabled);
|
||||
}
|
||||
}
|
||||
|
||||
public async didLoadResource(requestId: number, response: WebviewManagerDidLoadResourceResponse): Promise<void> {
|
||||
this.protocolProvider.didLoadResource(requestId, response);
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user