uploadVMImage.js

/**
* Copyright (c) Microsoft.  All rights reserved.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*   http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/

/**
 * UploadVMImage 
 */

var util = require('util');
var qs = require('querystring');
var pageBlob = require('./pageBlob');
var blobUtils = require('../../blobUtils');
var callbackAggregator = require('../../callbackAggregator');

// get account info: account key and blob endpoint
// callback: function(error, primaryKey, destUri)
function getAccountInfo(iaasClient, destinationUri, logger, callback) {
  var splitDestination = blobUtils.splitDestinationUri(destinationUri);
  
  var caagg = callbackAggregator.callbackAndAggregator(function(results, key, args) {
    if (key) { // This means error somewhere - defined by key
      var error = args[0];
      var resp = args[2];
      if (error && typeof error === 'object') {
        error.whichFunction = key;
        if (!error.code && !error.statusCode && resp) {
          // Sometimes error object does not have useful information. Add it to make it more debuggable.
          error.statusCode = resp.statusCode;
          error.response = resp;
        }
      }
      callback(error);
      return;
    }
    var responseKeys = results.storageAccountKeys[1]; // function(error, response)
    if (results.storageAccountProperties) {
      var responseProps = results.storageAccountProperties[1]; // function(error, response)
      var blobEndpoint = '';
      if (responseProps && responseProps.body && responseProps.body.StorageServiceProperties && 
          responseProps.body.StorageServiceProperties.Endpoints) {
        blobEndpoint = responseProps.body.StorageServiceProperties.Endpoints[0];
      }
      if (blobEndpoint) {
        var destOld = destinationUri;
        destinationUri = blobEndpoint + splitDestination.resourceName;
        if (destOld !== destinationUri && 'http://' + destOld !== destinationUri && splitDestination.host) {
          logger.warn('Incorrect blob destination: ' + destOld);
          logger.warn('Using instead             : ' + destinationUri);
        }
      } else {
        logger.warn('Couldn\'t get blob service endpoint'); // ignorable, will use default or specified endpoint
      }
    }
    callback(null, responseKeys.body.StorageServiceKeys.Primary, destinationUri);
  });
  
  
  // Run two calls in parallel 
  iaasClient.getStorageAccountKeys(splitDestination.accountName, caagg.getCallback('storageAccountKeys'));
  iaasClient.getStorageAccountProperties(splitDestination.accountName, caagg.getCallback('storageAccountProperties'));
    
}

var uploadPageBlobFromIaasClient = 
  exports.uploadPageBlobFromIaasClient = 
    function uploadPageBlobFromIaasClient(destinationUri, iaasClient, fileName, options, callback) {
  var logger = (options || {}).logger || console;
  var errorFunc = logger.error || console.error;
  if (typeof callback !== 'function') {
    throw 'Callback is expected in uploadPageBlobFromIaasClient(). Found: ' + utils.inspect(callback);
  }
  
  getAccountInfo(iaasClient, destinationUri, logger, function(error, primaryKey, newDestUri) {
    if (!error) {
      pageBlob.uploadPageBlob(newDestUri, primaryKey, fileName, options, function(error, alreadyExisted) {
        callback(error, newDestUri, alreadyExisted);
      });
    } else {
      errorFunc('There was an error in getStorageAccountKeys() or getStorageAccountProperties() for ' + destinationUri);
      errorFunc(util.inspect(error));
      callback(error);
    }
  });
};

var deleteBlobFromIaasClient = 
  exports.deleteBlobFromIaasClient =
    function deleteBlobFromIaasClient(destinationUri, iaasClient, options, callback) {
  var logger = (options || {}).logger || console;
  var errorFunc = logger.error || console.error;
  if (typeof callback !== 'function') {
    throw 'Callback is expected in deleteBlobFromIaasClient(). Found: ' + utils.inspect(callback);
  }

  destinationUri = destinationUri.replace(/^https\:\/\//i, ''); // avoid https/http url mismatch warning for delete when we get https:// MediaLink
  getAccountInfo(iaasClient, destinationUri, logger, function(error, primaryKey, newDestUri) {
    if (!error) {
      if (logger.verbose) logger.verbose('Uri : ' + newDestUri);
      pageBlob.deleteBlob(newDestUri, primaryKey, callback);
    } else {
      errorFunc('There was an error in getStorageAccountKeys() or getStorageAccountProperties() for ' + destinationUri);
      errorFunc(util.inspect(error));
      callback(error);
    }
  });
};

var copyBlobFromIaasClient = 
  exports.copyBlobFromIaasClient = 
    function copyBlobFromIaasClient(iaasClient, sourceUri, destinationUri, options, callback) {
  var logger = (options || {}).logger || console;
  var errorFunc = logger.error || console.error;
  if (typeof callback !== 'function') {
    throw 'Callback is expected in copyBlobFromIaasClient(). Found: ' + utils.inspect(callback);
  }
  
  getAccountInfo(iaasClient, destinationUri, logger, function(error, primaryKey, newDestUri) {
    if (!error) {
      pageBlob.copyBlob(sourceUri, newDestUri, primaryKey, function(error, blob, response) {
        callback(error, blob, response, newDestUri);
      });
    } else {
      errorFunc('There was an error in getStorageAccountKeys() or getStorageAccountProperties() for ' + destinationUri);
      errorFunc(util.inspect(error));
      callback(error);
    }
  });
};