Show a progress bar and allow the user to cancel

Author: davidwengier

l10n/bundle.l10n.json

@@ -36,6 +36,7 @@
   "Cancel": "Cancel",
   "Replace existing build and debug assets?": "Replace existing build and debug assets?",
   "Unable to generate assets to build and debug. {0}.": "Unable to generate assets to build and debug. {0}.",
+  "Downloading Razor Telemetry Package": "Downloading Razor Telemetry Package",
   "Cannot load Razor language server because the directory was not found: '{0}'": "Cannot load Razor language server because the directory was not found: '{0}'",
   "Could not find '{0}' in or above '{1}'.": "Could not find '{0}' in or above '{1}'.",
   "Invalid trace setting for Razor language server. Defaulting to '{0}'": "Invalid trace setting for Razor language server. Defaulting to '{0}'",

src/packageManager/downloadAndInstallPackages.ts

@@ -15,12 +15,14 @@ import { InstallationFailure, IntegrityCheckFailure } from '../omnisharp/logging
 import { mkdirpSync } from 'fs-extra';
 import { PackageInstallStart } from '../omnisharp/loggingEvents';
 import { DownloadValidator } from './isValidDownload';
+import { CancellationToken } from 'vscode';
 
 export async function downloadAndInstallPackages(
     packages: AbsolutePathPackage[],
     provider: NetworkSettingsProvider,
     eventStream: EventStream,
-    downloadValidator: DownloadValidator
+    downloadValidator: DownloadValidator,
+    token?: CancellationToken
 ): Promise<boolean> {
     eventStream.post(new PackageInstallStart());
     for (const pkg of packages) {
@@ -33,7 +35,14 @@ export async function downloadAndInstallPackages(
             while (willTryInstallingPackage()) {
                 count = count + 1;
                 installationStage = 'downloadPackage';
-                const buffer = await DownloadFile(pkg.description, eventStream, provider, pkg.url, pkg.fallbackUrl);
+                const buffer = await DownloadFile(
+                    pkg.description,
+                    eventStream,
+                    provider,
+                    pkg.url,
+                    pkg.fallbackUrl,
+                    token
+                );
                 if (downloadValidator(buffer, pkg.integrity, eventStream)) {
                     installationStage = 'installPackage';
                     await InstallZip(buffer, pkg.description, pkg.installPath, pkg.binaries, eventStream);

src/packageManager/fileDownloader.ts

@@ -17,18 +17,20 @@ import { NestedError } from '../nestedError';
 import { parse as parseUrl } from 'url';
 import { getProxyAgent } from './proxy';
 import { NetworkSettingsProvider } from '../networkSettings';
+import { CancellationToken } from 'vscode';
 
 export async function DownloadFile(
     description: string,
     eventStream: EventStream,
     networkSettingsProvider: NetworkSettingsProvider,
     url: string,
-    fallbackUrl?: string
+    fallbackUrl?: string,
+    token?: CancellationToken
 ): Promise<Buffer> {
     eventStream.post(new DownloadStart(description));
 
     try {
-        const buffer = await downloadFile(description, url, eventStream, networkSettingsProvider);
+        const buffer = await downloadFile(description, url, eventStream, networkSettingsProvider, token);
         eventStream.post(new DownloadSuccess(` Done!`));
         return buffer;
     } catch (primaryUrlError) {
@@ -54,7 +56,8 @@ async function downloadFile(
     description: string,
     urlString: string,
     eventStream: EventStream,
-    networkSettingsProvider: NetworkSettingsProvider
+    networkSettingsProvider: NetworkSettingsProvider,
+    token?: CancellationToken
 ): Promise<Buffer> {
     const url = parseUrl(urlString);
     const networkSettings = networkSettingsProvider();
@@ -66,12 +69,15 @@ async function downloadFile(
         agent: getProxyAgent(url, proxy, strictSSL),
         port: url.port,
         rejectUnauthorized: strictSSL,
-        timeout: 5 * 60 * 1000, // timeout is in milliseconds
     };
 
     const buffers: any[] = [];
 
     return new Promise<Buffer>((resolve, reject) => {
+        token?.onCancellationRequested(() => {
+            return reject(new NestedError(`Cancelled downloading ${urlString}.`));
+        });
+
         const request = https.request(options, (response) => {
             if (response.statusCode === 301 || response.statusCode === 302) {
                 // Redirect - download from new location
@@ -82,7 +88,7 @@ async function downloadFile(
                     return reject(new NestedError('Missing location'));
                 }
                 return resolve(
-                    downloadFile(description, response.headers.location, eventStream, networkSettingsProvider)
+                    downloadFile(description, response.headers.location, eventStream, networkSettingsProvider, token)
                 );
             } else if (response.statusCode !== 200) {
                 // Download failed - print error message
@@ -130,10 +136,6 @@ async function downloadFile(
             });
         });
 
-        request.on('timeout', () => {
-            request.destroy(new Error(`Timed out trying to download ${urlString}.`));
-        });
-
         request.on('error', (err) => {
             reject(new NestedError(`Request error: ${err.message || 'NONE'}`, err));
         });

src/razor/razorTelemetryDownloader.ts

@@ -3,6 +3,7 @@
  *  Licensed under the MIT License. See License.txt in the project root for license information.
  *--------------------------------------------------------------------------------------------*/
 
+import * as vscode from 'vscode';
 import { PlatformInformation } from '../shared/platform';
 import { PackageInstallation, LogPlatformInfo, InstallationSuccess } from '../omnisharp/loggingEvents';
 import { EventStream } from '../eventStream';
@@ -42,17 +43,27 @@ export class RazorTelemetryDownloader {
         if (packagesToInstall.length > 0) {
             this.eventStream.post(new PackageInstallation(`Razor Telemetry Version = ${version}`));
             this.eventStream.post(new LogPlatformInfo(this.platformInfo));
-            if (
-                await downloadAndInstallPackages(
-                    packagesToInstall,
-                    this.networkSettingsProvider,
-                    this.eventStream,
-                    isValidDownload
-                )
-            ) {
-                this.eventStream.post(new InstallationSuccess());
-                return true;
-            }
+            await vscode.window.withProgress(
+                {
+                    location: vscode.ProgressLocation.Notification,
+                    title: vscode.l10n.t('Downloading Razor Telemetry Package'),
+                    cancellable: true,
+                },
+                async (_, token) => {
+                    if (
+                        await downloadAndInstallPackages(
+                            packagesToInstall,
+                            this.networkSettingsProvider,
+                            this.eventStream,
+                            isValidDownload,
+                            token
+                        )
+                    ) {
+                        this.eventStream.post(new InstallationSuccess());
+                        return true;
+                    }
+                }
+            );
         }
 
         return false;

tasks/offlinePackagingTasks.ts

@@ -29,6 +29,7 @@ import { getPackageJSON } from '../tasks/packageJson';
 import { createPackageAsync } from '../tasks/vsceTasks';
 import { isValidDownload } from '../src/packageManager/isValidDownload';
 import path = require('path');
+import { CancellationToken } from 'vscode';
 // There are no typings for this library.
 // eslint-disable-next-line @typescript-eslint/no-var-requires
 const argv = require('yargs').argv;
@@ -153,7 +154,8 @@ async function installDebugger(packageJSON: any, platformInfo: PlatformInformati
 async function installPackageJsonDependency(
     dependencyName: string,
     packageJSON: any,
-    platformInfo: PlatformInformation
+    platformInfo: PlatformInformation,
+    token?: CancellationToken
 ) {
     const eventStream = new EventStream();
     const logger = new Logger((message) => process.stdout.write(message));
@@ -168,7 +170,7 @@ async function installPackageJsonDependency(
         codeExtensionPath
     );
     const provider = () => new NetworkSettings('', true);
-    if (!(await downloadAndInstallPackages(packagesToInstall, provider, eventStream, isValidDownload))) {
+    if (!(await downloadAndInstallPackages(packagesToInstall, provider, eventStream, isValidDownload, token))) {
         throw Error('Failed to download package.');
     }
 }