appcenter-cli/src/util/misc/download.ts

Command line tool for Visual Studio App Center

import { AppCommand, ErrorCodes, failure } from "../../util/commandline";
import { StreamingArrayOutput } from "../../util/interaction";
import * as fsHelper from "../../util/misc/fs-helper";
import * as Fs from "fs";
import * as os from "os";
import * as path from "path";
import * as Request from "request";
import { inspect } from "util";

const debug = require("debug")("appcenter-cli:util:misc:download");

export function downloadFileAndSave(downloadUrl: string, filePath: string): Promise<void> {
  debug(`Downloading file from ${downloadUrl} to the path ${filePath}`);
  return new Promise<void>((resolve, reject) => {
    Request.get(downloadUrl)
    .on("error", (error) => {
      debug(`Failed to download the file from ${downloadUrl} - ${inspect(error)}`);
      reject(failure(ErrorCodes.Exception, `failed to download the file from ${downloadUrl}`));
    })
    .pipe(
      Fs.createWriteStream(filePath)
      .on("error", (error: Error) => {
        debug(`Failed to save the file to ${filePath} - ${inspect(error)}`);
        reject(failure(ErrorCodes.Exception, `failed to save the file to ${filePath}`));
      })
      .on("finish", () => resolve()));
  });
}

export async function downloadArtifacts(command: AppCommand, streamingOutput: StreamingArrayOutput, outputDir: string, testRunId: string, artifacts: { [propertyName: string]: string }): Promise<void> {
  for (const key in artifacts) {

    const reportPath: string = fsHelper.generateAbsolutePath(outputDir);
    const pathToArchive: string = path.join(reportPath, `${key.toString()}.zip`);
    fsHelper.createLongPath(reportPath);
    await downloadFileAndSave(artifacts[key], pathToArchive);

    // Print only in VSTS environment
    // https://docs.microsoft.com/en-us/vsts/build-release/concepts/definitions/build/variables?view=vsts&tabs=batch#tfbuild
    if (process.env["TF_BUILD"]) {
      streamingOutput.text((command: AppCommand): string => {
        return `##vso[task.setvariable variable=${key}]${pathToArchive}${os.EOL}`;
      }, command);
    }

    streamingOutput.text((command: AppCommand): string => {
      return `Downloaded artifacts to ${pathToArchive}`;
    }, command);
  }
}