ckaczor/azuredatastudio
1
0
Fork 0
mirror of https://github.com/ckaczor/azuredatastudio.git synced 2026-01-27 01:25:36 -05:00
Code Issues Packages Projects Releases Wiki Activity

Toggle Trusted Button (#8569)

Browse Source 
* Toggle Trusted Button

* Persist Trusted information to Cache

* Trusted toggle unit test fixed

* Updated test comment
This commit is contained in:
rajeshka
2019-12-16 13:32:56 -08:00
committed by GitHub
parent 64e217eac2
commit 73b9849dd5
3 changed files with 26 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
src/sql/workbench/services/notebook/browser/notebookServiceImpl.ts
View File

@@ -115,6 +115,7 @@ export class NotebookService extends Disposable implements INotebookService {
private _themeParticipant: IDisposable;
private _overrideEditorThemeSetting: boolean;
private _trustedCacheQueue: URI[] = [];
private _unTrustedCacheQueue: URI[] = [];
private _updateTrustCacheScheduler: RunOnceScheduler;
constructor(
@@ -589,8 +590,12 @@ export class NotebookService extends Disposable implements INotebookService {
if (changeType === NotebookChangeType.Saved && firstIndex(this._trustedCacheQueue, uri => uri.toString() === notebookUriString) < 0) {
// Only save if it's trusted
let notebook = find(this.listNotebookEditors(), n => n.id === notebookUriString);
if (notebook && notebook.model.trustedMode) {
this._trustedCacheQueue.push(notebookUri);
if (notebook && notebook.model) {
if (notebook.model.trustedMode) {
this._trustedCacheQueue.push(notebookUri);
} else {
this._unTrustedCacheQueue.push(notebookUri);
}
this._updateTrustCacheScheduler.schedule();
}
}
@@ -625,7 +630,19 @@ export class NotebookService extends Disposable implements INotebookService {
};
}
}
this._trustedNotebooksMemento.saveMemento();
}
if (this._unTrustedCacheQueue.length > 0) {
// Copy out all items from the cache
let items = this._unTrustedCacheQueue;
this._unTrustedCacheQueue = [];
let trustedCache = this.trustedNotebooksMemento.trustedNotebooksCache;
//Remove the trusted intry from the cache
for (let i = 0; i < items.length; i++) {
if (trustedCache[items[i].toString()]) {
trustedCache[items[i].toString()] = null;
}
}
this._trustedNotebooksMemento.saveMemento();
}
} catch (err) {