Moving repo to bmci-labs

arduino-ide-extension/src/node/core-client-provider-impl.ts

@@ -0,0 +1,91 @@
+import { inject, injectable } from 'inversify';
+import * as grpc from 'grpc';
+import { ArduinoCoreClient } from './cli-protocol/commands_grpc_pb';
+import { InitResp, InitReq, Configuration, UpdateIndexReq, UpdateIndexResp } from './cli-protocol/commands_pb';
+import { WorkspaceServiceExt } from '../browser/workspace-service-ext';
+import { FileSystem } from '@theia/filesystem/lib/common';
+import URI from '@theia/core/lib/common/uri';
+import { CoreClientProvider, Client } from './core-client-provider';
+
+@injectable()
+export class CoreClientProviderImpl implements CoreClientProvider {
+
+    @inject(FileSystem)
+    protected readonly fileSystem: FileSystem;
+
+    @inject(WorkspaceServiceExt)
+    protected readonly workspaceServiceExt: WorkspaceServiceExt;
+
+    protected clients = new Map<string, Client>();
+
+    async getClient(workspaceRootOrResourceUri?: string): Promise<Client> {
+        const roots = await this.workspaceServiceExt.roots();
+        if (!workspaceRootOrResourceUri) {
+            return this.getOrCreateClient(roots[0]);
+        }
+        const root = roots
+            .sort((left, right) => right.length - left.length) // Longest "paths" first
+            .map(uri => new URI(uri))
+            .find(uri => uri.isEqualOrParent(new URI(workspaceRootOrResourceUri)));
+        if (!root) {
+            console.warn(`Could not retrieve the container workspace root for URI: ${workspaceRootOrResourceUri}.`);
+            console.warn(`Falling back to ${roots[0]}`);
+            return this.getOrCreateClient(roots[0]);
+        }
+        return this.getOrCreateClient(root.toString());
+    }
+
+    protected async getOrCreateClient(rootUri: string): Promise<Client> {
+        const existing = this.clients.get(rootUri);
+        if (existing) {
+            console.debug(`Reusing existing client for ${rootUri}.`);
+            return existing;
+        }
+
+        console.info(` >>> Creating and caching a new client for ${rootUri}...`);
+        const client = new ArduinoCoreClient('localhost:50051', grpc.credentials.createInsecure());
+        const config = new Configuration();
+        const path = await this.fileSystem.getFsPath(rootUri);
+        if (!path) {
+            throw new Error(`Could not resolve file-system path of URI: ${rootUri}.`);
+        }
+        config.setSketchbookdir(path);
+        config.setDatadir(path);
+        config.setDownloadsdir(path);
+
+        const initReq = new InitReq();
+        initReq.setConfiguration(config);
+        const initResp = await new Promise<InitResp>((resolve, reject) => client.init(initReq, (err, resp) => (!!err ? reject : resolve)(!!err ? err : resp)));
+        const instance = initResp.getInstance();
+        if (!instance) {
+            throw new Error(`Could not retrieve instance from the initialize response.`);
+        }
+        const updateReq = new UpdateIndexReq();
+        updateReq.setInstance(instance);
+        const updateResp = client.updateIndex(updateReq);
+        updateResp.on('data', (o: UpdateIndexResp) => {
+            const progress = o.getDownloadProgress();
+            if (progress) {
+                if (progress.getCompleted()) {
+                    console.log(`Download${progress.getFile() ? ` of ${progress.getFile()}` : ''} completed.`);
+                } else {
+                    console.log(`Downloading${progress.getFile() ? ` ${progress.getFile()}:` : ''} ${progress.getDownloaded()}.`);
+                }
+            }
+        });
+        // TODO: revisit this!!!
+        // `updateResp.on('data'` is called only when running, for instance, `compile`. It does not run eagerly.
+        // await new Promise<void>((resolve, reject) => {
+        //     updateResp.on('close', resolve);
+        //     updateResp.on('error', reject);
+        // });
+
+        const result = {
+            client,
+            instance
+        }
+        this.clients.set(rootUri, result);
+        console.info(` <<< New client has been successfully created and cached for ${rootUri}.`);
+        return result;
+    }
+}