feat: preload fuze and full files tree (conwnet#239)

Author: conwnet

extensions/github1s/src/helpers/async.ts

@@ -0,0 +1,31 @@
+/**
+ * @file async related helpers
+ */
+
+// below code is comes from:
+//https://github.com/microsoft/vscode/blob/a3415e669a8f3879c290af5616a8ed45dd0534af/src/vs/base/common/async.ts#L344
+export class Barrier {
+	private _isOpen: boolean;
+	private _promise: Promise<boolean>;
+	private _completePromise!: (v: boolean) => void;
+
+	constructor() {
+		this._isOpen = false;
+		this._promise = new Promise<boolean>((c, e) => {
+			this._completePromise = c;
+		});
+	}
+
+	isOpen(): boolean {
+		return this._isOpen;
+	}
+
+	open(): void {
+		this._isOpen = true;
+		this._completePromise(true);
+	}
+
+	wait(): Promise<boolean> {
+		return this._promise;
+	}
+}

extensions/github1s/src/listeners/router/explorer.ts

@@ -7,6 +7,7 @@ import * as vscode from 'vscode';
 import {
 	changedFileDecorationProvider,
 	submoduleDecorationProvider,
+	fileSearchProvider,
 } from '@/providers';
 import { RouterState } from '@/router/types';
 
@@ -28,5 +29,6 @@ export const explorerRouterListener = (
 
 		changedFileDecorationProvider.updateDecorations();
 		submoduleDecorationProvider.updateDecorations();
+		fileSearchProvider.loadFuzeForCurrentAuthority();
 	}
 };

extensions/github1s/src/providers/fileSearchProvider.ts

@@ -26,12 +26,23 @@ export class GitHub1sFileSearchProvider
 	private readonly disposable: Disposable;
 	private fuseMap: Map<string, Fuse<GithubRESTEntry>> = new Map();
 
-	constructor(private fsProvider: GitHub1sFileSystemProvider) {}
+	constructor(private fsProvider: GitHub1sFileSystemProvider) {
+		// Preload the fuze for better `ctrl/command + p` experience.
+		// Once we have loaded the fuze, it will also populate the files into
+		// fileSystemProvider's cache. So after that, you don't have to send
+		// a request when you open the new directory in explorer late
+		this.loadFuzeForCurrentAuthority();
+	}
 
 	dispose() {
 		this.disposable?.dispose();
 	}
 
+	// load the fuze for current authority
+	async loadFuzeForCurrentAuthority() {
+		return this.getFuse(await router.getAuthority());
+	}
+
 	/**
 	 * getFuse for fuzzy file search,
 	 * it maybe take longer time, so we just run it in backend

extensions/github1s/src/router/index.ts

@@ -5,13 +5,15 @@
 
 import * as vscode from 'vscode';
 import { History, createMemoryHistory } from 'history';
+import { Barrier } from '@/helpers/async';
 import { parseGitHubUrl } from './parser';
 import { EventEmitter } from './events';
 import { RouterState } from './types';
 
 export class Router extends EventEmitter<RouterState> {
 	private static instance: Router;
 
+	private readyBarrier = new Barrier();
 	private _previousStatePromise: Promise<RouterState>;
 	private _currentStatePromise: Promise<RouterState>;
 	public history: History = createMemoryHistory();
@@ -31,6 +33,7 @@ export class Router extends EventEmitter<RouterState> {
 		this.history.replace(targetPath);
 		this._currentStatePromise = parseGitHubUrl(targetPath);
 		this._previousStatePromise = this._currentStatePromise;
+		this.readyBarrier.open();
 
 		this.history.listen(async ({ location }) => {
 			const targetPath = `${location.pathname}${location.search}${location.hash}`;
@@ -51,7 +54,7 @@ export class Router extends EventEmitter<RouterState> {
 
 	// get the routerState for current url
 	public async getState(): Promise<RouterState> {
-		return this._currentStatePromise;
+		return this.readyBarrier.wait().then(() => this._currentStatePromise);
 	}
 
 	// compute the file URI authority of current routerState