objectstorage.js

/**
 * Copyright 2017 IBM Corp. 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
 *
 *  https://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.
 */

const pkgcloud = require('pkgcloud');
const async = require('async');

function ObjectStorage(osConfig, initializeStorageCallback/* err, objectStorage*/) {
  const self = this;

  const storageClient = pkgcloud.storage.createClient(osConfig);
  const storageContainerName = 'openwhisk-darkvision';
  let storageContainer;

  async.waterfall([
    // authenticate
    function(callback) {
      storageClient.auth((err) => {
        if (err) {
          callback(err);
        } else {
          console.log('[objectstorage]', 'Connected to storage with tenant', storageClient._identity.options.tenantId);
          callback(null);
        }
      });
    },
    // create container
    function(callback) {
      console.log('[objectstorage]', 'Creating storage container...');
      storageClient.createContainer({
        name: storageContainerName,
        metadata: {
          'Web-Index': 'index.html'
        }
      }, (err, container) => {
        if (err) {
          callback(err);
        } else {
          console.log('[objectstorage]', 'Got container at', container.client._serviceUrl);
          storageContainer = container;
          callback(null);
        }
      });
    },
    // make container public
    function(callback) {
      console.log('[objectstorage]', 'Making storage container public...');
      require('request')({
        url: self.storageUrl(),
        method: 'POST',
        headers: {
          'X-Container-Read': '.r:*',
          'X-Auth-Token': storageClient._identity.token.id
        }
      }, (error) => {
        callback(error);
      });
    }
  ], (err) => {
    if (err) {
      console.log('[objectstorage]', err);
    } else {
      console.log('[objectstorage]', 'Storage container is ready at', self.storageUrl());
    }
    if (initializeStorageCallback) {
      initializeStorageCallback(err, self);
    }
  });

  self.storageUrl = function() {
    return `${storageContainer.client._serviceUrl}/${storageContainer.name}`;
  };

  self.write = function(filename) {
    console.log('[objectstorage]', 'New upload to', filename);
    const upload = storageClient.upload({
      container: storageContainerName,
      remote: filename
    });
    // upload.on('error', function(err) { });
    // upload.on('success', function(file) { });
    return upload;
  };

  self.read = function(filename) {
    console.log('[objectstorage]', 'Read', filename);
    const download = storageClient.download({
      container: storageContainerName,
      remote: filename
    });
    return download;
  };

  self.delete = function(filename, callback) {
    console.log('[objectstorage]', 'Delete', filename);
    storageClient.removeFile(storageContainer, filename, (err) => {
      if (callback) {
        callback(err);
      }
    });
  };
}

module.exports = function(osConfig, initializeStorageCallback/* err, objectStorage*/) {
  return new ObjectStorage(osConfig, initializeStorageCallback);
};