support directories in indexed db fsp

src/vs/workbench/services/log/browser/indexedDBLogProvider.ts

@@ -9,6 +9,7 @@ import { Disposable, IDisposable } from 'vs/base/common/lifecycle';
 import { Event, Emitter } from 'vs/base/common/event';
 import { VSBuffer } from 'vs/base/common/buffer';
 import { FileSystemError } from 'vs/workbench/api/common/extHostTypes';
+import { isEqualOrParent, joinPath } from 'vs/base/common/resources';
 
 const LOGS_OBJECT_STORE = 'logs';
 export const INDEXEDDB_LOG_SCHEME = 'vscode-logs-indexedbd';
@@ -41,7 +42,7 @@ export class IndexedDBLogProvider extends Disposable implements IFileSystemProvi
 					c(db);
 				}
 			};
-			request.onupgradeneeded = (e) => {
+			request.onupgradeneeded = () => {
 				const db = request.result;
 				if (!db.objectStoreNames.contains(LOGS_OBJECT_STORE)) {
 					db.createObjectStore(LOGS_OBJECT_STORE);
@@ -55,20 +56,7 @@ export class IndexedDBLogProvider extends Disposable implements IFileSystemProvi
 		return Disposable.None;
 	}
 
-	mkdir(resource: URI): Promise<void> {
-		return Promise.reject(new Error('Not Supported'));
-	}
-
-	rename(from: URI, to: URI, opts: FileOverwriteOptions): Promise<void> {
-		return Promise.reject(new Error('Not Supported'));
-	}
-
-	readdir(resource: URI): Promise<[string, FileType][]> {
-		return Promise.reject(new Error('Not Supported'));
-	}
-
-	delete(resource: URI, opts: FileDeleteOptions): Promise<void> {
-		return Promise.reject(new Error('Not Supported'));
+	async mkdir(resource: URI): Promise<void> {
 	}
 
 	async stat(resource: URI): Promise<IStat> {
@@ -81,33 +69,67 @@ export class IndexedDBLogProvider extends Disposable implements IFileSystemProvi
 				size: content.byteLength
 			};
 		} catch (e) {
+		}
+		const files = await this.readdir(resource);
+		if (files.length) {
 			return {
-				type: FileType.File,
+				type: FileType.Directory,
 				ctime: 0,
 				mtime: 0,
 				size: 0
 			};
 		}
+		return Promise.reject(new FileSystemError(resource, FileSystemProviderErrorCode.FileNotFound));
 	}
 
-	async readFile(resource: URI): Promise<Uint8Array> {
+	async readdir(resource: URI): Promise<[string, FileType][]> {
+		const hasKey = await this.hasKey(resource.path);
+		if (hasKey) {
+			return Promise.reject(new FileSystemError(resource, FileSystemProviderErrorCode.FileNotADirectory));
+		}
 		return new Promise(async (c, e) => {
 			const db = await this.database;
 			const transaction = db.transaction([LOGS_OBJECT_STORE]);
 			const objectStore = transaction.objectStore(LOGS_OBJECT_STORE);
-			const request = objectStore.get(resource.path);
+			const request = objectStore.getAllKeys();
 			request.onerror = () => e(request.error);
 			request.onsuccess = () => {
-				if (request.result) {
-					c(VSBuffer.fromString(request.result).buffer);
-				} else {
-					e(new FileSystemError(resource, FileSystemProviderErrorCode.FileNotFound));
+				const files: [string, FileType][] = [];
+				const resourceSegments = resource.path.split('/');
+				for (const key of <string[]>request.result) {
+					if (isEqualOrParent(URI.file(key).with({ scheme: INDEXEDDB_LOG_SCHEME }), resource, false)) {
+						const keySegments = key.split('/');
+						files.push([keySegments[resourceSegments.length], resourceSegments.length + 1 === keySegments.length ? FileType.File : FileType.Directory]);
+					}
 				}
+				c(files);
 			};
 		});
 	}
 
-	writeFile(resource: URI, content: Uint8Array, opts: FileWriteOptions): Promise<void> {
+	async readFile(resource: URI): Promise<Uint8Array> {
+		const hasKey = await this.hasKey(resource.path);
+		if (!hasKey) {
+			return Promise.reject(new FileSystemError(resource, FileSystemProviderErrorCode.FileNotFound));
+		}
+		return new Promise(async (c, e) => {
+			const db = await this.database;
+			const transaction = db.transaction([LOGS_OBJECT_STORE]);
+			const objectStore = transaction.objectStore(LOGS_OBJECT_STORE);
+			const request = objectStore.get(resource.path);
+			request.onerror = () => e(request.error);
+			request.onsuccess = () => c(VSBuffer.fromString(request.result || '').buffer);
+		});
+	}
+
+	async writeFile(resource: URI, content: Uint8Array, opts: FileWriteOptions): Promise<void> {
+		const hasKey = await this.hasKey(resource.path);
+		if (!hasKey) {
+			const files = await this.readdir(resource);
+			if (files.length) {
+				return Promise.reject(new FileSystemError(resource, FileSystemProviderErrorCode.FileIsADirectory));
+			}
+		}
 		return new Promise(async (c, e) => {
 			const db = await this.database;
 			const transaction = db.transaction([LOGS_OBJECT_STORE], 'readwrite');
@@ -122,4 +144,52 @@ export class IndexedDBLogProvider extends Disposable implements IFileSystemProvi
 		});
 	}
 
+	async delete(resource: URI, opts: FileDeleteOptions): Promise<void> {
+		const hasKey = await this.hasKey(resource.path);
+		if (hasKey) {
+			await this.deleteKey(resource.path);
+			return;
+		}
+
+		if (opts.recursive) {
+			const files = await this.readdir(resource);
+			await Promise.all(files.map(([key]) => this.delete(joinPath(resource, key), opts)));
+		}
+	}
+
+	rename(from: URI, to: URI, opts: FileOverwriteOptions): Promise<void> {
+		return Promise.reject(new Error('Not Supported'));
+	}
+
+	private hasKey(key: string): Promise<boolean> {
+		return new Promise<boolean>(async (c, e) => {
+			const db = await this.database;
+			const transaction = db.transaction([LOGS_OBJECT_STORE]);
+			const objectStore = transaction.objectStore(LOGS_OBJECT_STORE);
+			const request = objectStore.getKey(key);
+			request.onerror = () => e(request.error);
+			request.onsuccess = () => {
+				c(!!request.result);
+			};
+		});
+	}
+
+	private deleteKey(key: string): Promise<void> {
+		return new Promise(async (c, e) => {
+			const db = await this.database;
+			const transaction = db.transaction([LOGS_OBJECT_STORE], 'readwrite');
+			const objectStore = transaction.objectStore(LOGS_OBJECT_STORE);
+			const request = objectStore.delete(key);
+			request.onerror = () => e(request.error);
+			request.onsuccess = () => {
+				this.versions.delete(key);
+				this._onDidChangeFile.fire([{ resource: this.toResource(key), type: FileChangeType.DELETED }]);
+				c();
+			};
+		});
+	}
+
+	private toResource(key: string): URI {
+		return URI.file(key).with({ scheme: INDEXEDDB_LOG_SCHEME });
+	}
 }