From b6306c330f828e42ce4511ded258c49a8e311689 Mon Sep 17 00:00:00 2001
From: Akos Kitta <kittaakos@typefox.io>
Date: Fri, 2 Aug 2019 13:35:25 +0200
Subject: [PATCH] Update/download the `library_index.json` at start

Signed-off-by: Akos Kitta <kittaakos@typefox.io>
---
 .../src/node/core-client-provider-impl.ts     | 58 +++++++++++++++++--
 1 file changed, 53 insertions(+), 5 deletions(-)

diff --git a/arduino-ide-extension/src/node/core-client-provider-impl.ts b/arduino-ide-extension/src/node/core-client-provider-impl.ts
index 19bd9c47..f259259b 100644
--- a/arduino-ide-extension/src/node/core-client-provider-impl.ts
+++ b/arduino-ide-extension/src/node/core-client-provider-impl.ts
@@ -6,7 +6,9 @@ import {
     InitReq,
     Configuration,
     UpdateIndexReq,
-    UpdateIndexResp
+    UpdateIndexResp,
+    UpdateLibrariesIndexReq,
+    UpdateLibrariesIndexResp
 } from './cli-protocol/commands/commands_pb';
 import { WorkspaceServiceExt } from '../browser/workspace-service-ext';
 import { FileSystem } from '@theia/filesystem/lib/common';
@@ -111,20 +113,34 @@ export class CoreClientProviderImpl implements CoreClientProvider {
         }
 
         // in a separate promise, try and update the index
-        let succeeded = true;
+        let indexUpdateSucceeded = true;
         for (let i = 0; i < 10; i++) {
             try {
                 await this.updateIndex(client, instance);
-                succeeded = true;
+                indexUpdateSucceeded = true;
                 break;
             } catch (e) {
                 this.toolOutputService.publishNewOutput("daemon", `Error while updating index in attempt ${i}: ${e}`);
             }
         }
-        if (!succeeded) {
+        if (!indexUpdateSucceeded) {
             this.toolOutputService.publishNewOutput("daemon", `Was unable to update the index. Please restart to try again.`);
         }
 
+        let libIndexUpdateSucceeded = true;
+        for (let i = 0; i < 10; i++) {
+            try {
+                await this.updateLibraryIndex(client, instance);
+                libIndexUpdateSucceeded = true;
+                break;
+            } catch (e) {
+                this.toolOutputService.publishNewOutput("daemon", `Error while updating library index in attempt ${i}: ${e}`);
+            }
+        }
+        if (!libIndexUpdateSucceeded) {
+            this.toolOutputService.publishNewOutput("daemon", `Was unable to update the library index. Please restart to try again.`);
+        }
+
         const result = {
             client,
             instance
@@ -134,6 +150,38 @@ export class CoreClientProviderImpl implements CoreClientProvider {
         return result;
     }
 
+    protected async updateLibraryIndex(client: ArduinoCoreClient, instance: Instance): Promise<void> {
+        const req = new UpdateLibrariesIndexReq();
+        req.setInstance(instance);
+        const resp = client.updateLibrariesIndex(req);
+        let file: string | undefined;
+        resp.on('data', (data: UpdateLibrariesIndexResp) => {
+            const progress = data.getDownloadProgress();
+            if (progress) {
+                if (!file && progress.getFile()) {
+                    file = `${progress.getFile()}`;
+                }
+                if (progress.getCompleted()) {
+                    if (file) {
+                        if (/\s/.test(file)) {
+                            this.toolOutputService.publishNewOutput("daemon", `${file} completed.\n`);
+                        } else {
+                            this.toolOutputService.publishNewOutput("daemon", `Download of '${file}' completed.\n'`);
+                        }
+                    } else {
+                        this.toolOutputService.publishNewOutput("daemon", `The library index has been successfully updated.\n'`);
+                    }
+                    file = undefined;
+                }
+            }
+        });
+        await new Promise<void>((resolve, reject) => {
+            resp.on('error', reject);
+            resp.on('end', resolve);
+        });
+    }
+
+
     protected async updateIndex(client: ArduinoCoreClient, instance: Instance): Promise<void> {
         const updateReq = new UpdateIndexReq();
         updateReq.setInstance(instance);
@@ -165,4 +213,4 @@ export class CoreClientProviderImpl implements CoreClientProvider {
         });
     }
 
-}
\ No newline at end of file
+}