skyquake/plugins/composer/api/packageFileHandler.js

   1 var request = require('request');
   2 var Promise = require('bluebird');
   3 var rp = require('request-promise');
   4 var utils = require('../../../framework/core/api_utils/utils.js');
   5 var constants = require('../../../framework/core/api_utils/constants.js');
   6 var fs = require('fs');
   7 var _ = require('lodash');
   8
   9 PackageFileHandler = {};
  10
  11 function deleteFile(filename) {
  12         setTimeout(function() {
  13                 fs.unlinkSync(constants.BASE_PACKAGE_UPLOAD_DESTINATION + filename);
  14         }, constants.PACKAGE_FILE_DELETE_DELAY_MILLISECONDS);
  15 };
  16
  17 function checkStatus(req, transactionId, isUpdate) {
  18         var upload_server = req.query['upload_server'];
  19         var headers = _.extend({},
  20         {
  21             'Authorization': req.get('Authorization')
  22         }
  23     );
  24     var type = isUpdate ? 'update' : 'upload';
  25         request({
  26                 url: upload_server + ':' + constants.PACKAGE_MANAGER_SERVER_PORT + '/api/' + type + '/' + transactionId + '/state',
  27                 type: 'GET',
  28                 headers: headers,
  29                 forever: constants.FOREVER_ON,
  30         rejectUnauthorized: false
  31         }, function(error, response, body) {
  32                 if (error) {
  33                         console.log('Error checking status for transaction', transactionId, '. Will delete file', req.file.filename, ' in ', constants.PACKAGE_FILE_DELETE_DELAY_MILLISECONDS ,' milliseconds');
  34                         deleteFile(req.file.filename);
  35                 } else {
  36                         var jsonStatus = null;
  37                         if (typeof body == 'string' || body instanceof String) {
  38                                 jsonStatus = JSON.parse(body);
  39                         } else {
  40                                 jsonStatus = body;
  41                         }
  42
  43                         if (jsonStatus.status && (jsonStatus.status == 'success' || jsonStatus.status == 'failure')) {
  44                                 console.log('Transaction ', transactionId, ' completed with status ', jsonStatus.status ,'. Will delete file', req.file.filename, ' in ', constants.PACKAGE_FILE_DELETE_DELAY_MILLISECONDS ,' milliseconds');
  45                                 deleteFile(req.file.filename);
  46                         } else {
  47                                 setTimeout(function() {
  48                                         checkStatus(req, transactionId, isUpdate);
  49                                 }, constants.PACKAGE_FILE_ONBOARD_TRANSACTION_STATUS_CHECK_DELAY_MILLISECONDS);
  50                         }
  51                 }
  52         });
  53 };
  54
  55 PackageFileHandler.checkCreatePackageStatusAndHandleFile = function(req, transactionId, isUpdate) {
  56         checkStatus(req, transactionId, isUpdate);
  57 };
  58
  59
  60 module.exports = PackageFileHandler;