shaka-player/lib/offline/db_engine.js

/**
 * @license
 * Copyright 2016 Google Inc.
 *
 * 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.
 */

goog.provide('shaka.offline.DBEngine');

goog.require('goog.asserts');
goog.require('shaka.offline.IStorageEngine');
goog.require('shaka.util.Error');
goog.require('shaka.util.Functional');
goog.require('shaka.util.PublicPromise');



/**
 * This manages all operations on an IndexedDB.  This wraps all operations
 * in Promises.  All Promises will resolve once the transaction has completed.
 * Depending on the browser, this may or may not be after the data is flushed
 * to disk.  https://goo.gl/zMOeJc
 *
 * @struct
 * @constructor
 * @param {string} name
 * @param {number=} opt_updateRetries
 * @implements {shaka.offline.IStorageEngine}
 */
shaka.offline.DBEngine = function(name, opt_updateRetries) {
  goog.asserts.assert(
      shaka.offline.DBEngine.isSupported(),
      'DBEngine should not be called when DBEngine is not supported');

  /** @private {string} */
  this.name_ = name;

  /** @private {number} */
  this.retries_ = opt_updateRetries || 0;

  /** @private {IDBDatabase} */
  this.db_ = null;

  /** @private {!Array.<shaka.offline.DBEngine.Operation>} */
  this.operations_ = [];

  /** @private {number} */
  this.nextManifestId_ = 0;

  /** @private {number} */
  this.nextSegmentId_ = 0;
};


/**
 * @typedef {{
 *   transaction: !IDBTransaction,
 *   promise: !shaka.util.PublicPromise
 * }}
 *
 * @property {!IDBTransaction} transaction
 *   The transaction that this operation is using.
 * @property {!shaka.util.PublicPromise} promise
 *   The promise associated with the operation.
 */
shaka.offline.DBEngine.Operation;


/** @private @const {number} */
shaka.offline.DBEngine.DB_VERSION_ = 1;


/**
 * @enum {string}
 * @private
 */
shaka.offline.DBEngine.Mode_ = {
  READ_ONLY: 'readonly',
  READ_WRITE: 'readwrite'
};


/**
 * @enum {string}
 * @private
 */
shaka.offline.DBEngine.Store_ = {
  MANIFEST: 'manifest',
  SEGMENT: 'segment'
};


/**
 * Determines if the browsers supports IndexedDB.
 * @return {boolean}
 */
shaka.offline.DBEngine.isSupported = function() {
  return window.indexedDB != null;
};


/**
 * Delete the database.  There must be no open connections to the database.
 * @param {string} name
 * @return {!Promise}
 */
shaka.offline.DBEngine.deleteDatabase = function(name) {
  if (!window.indexedDB)
    return Promise.resolve();
  var request = window.indexedDB.deleteDatabase(name);

  var p = new shaka.util.PublicPromise();
  request.onsuccess = function(event) {
    goog.asserts.assert(event.newVersion == null, 'Unexpected database update');
    p.resolve();
  };
  request.onerror = shaka.offline.DBEngine.onError_.bind(null, request, p);
  return p;
};


/**
 * @return {!Promise}
 */
shaka.offline.DBEngine.prototype.init = function() {
  /** @const */
  var DBEngine = shaka.offline.DBEngine;
  /** @const */
  var READ_ONLY = DBEngine.Mode_.READ_ONLY;
  /** @const */
  var MANIFEST = DBEngine.Store_.MANIFEST;
  /** @const */
  var SEGMENT = DBEngine.Store_.SEGMENT;

  /** @type {!shaka.offline.DBEngine} */
  var self = this;

  return self.createConnection_()
      .then(function() {
        return self.createTransaction_(MANIFEST, READ_ONLY, function(store) {
          DBEngine.getLastId_(store, function(id) {
            self.nextManifestId_ = id;
          });
        });
      })
      .then(function() {
        return self.createTransaction_(SEGMENT, READ_ONLY, function(store) {
          DBEngine.getLastId_(store, function(id) {
            self.nextSegmentId_ = id;
          });
        });
      });
};


/** @override */
shaka.offline.DBEngine.prototype.destroy = function() {
  return Promise.all(this.operations_.map(function(op) {
    try {
      // If the transaction is considered finished but has not called the
      // callbacks yet, it will still be in the list and this call will fail.
      // Simply ignore errors.
      op.transaction.abort();
    } catch (e) {}

    var Functional = shaka.util.Functional;
    return op.promise.catch(Functional.noop);
  })).then(function() {
    goog.asserts.assert(this.operations_.length == 0,
                        'All operations should have been closed');
    if (this.db_) {
      this.db_.close();
      this.db_ = null;
    }
  }.bind(this));
};


/** @override */
shaka.offline.DBEngine.prototype.getManifest = function(key) {
  return this.get_(
      shaka.offline.DBEngine.Store_.MANIFEST,
      key);
};


/** @override */
shaka.offline.DBEngine.prototype.forEachManifest = function(each) {
  return this.forEach_(
      shaka.offline.DBEngine.Store_.MANIFEST,
      each);
};


/** @override */
shaka.offline.DBEngine.prototype.insertManifest = function(value) {
  return this.insert_(
      shaka.offline.DBEngine.Store_.MANIFEST,
      value);
};


/** @override */
shaka.offline.DBEngine.prototype.removeManifests =
    function(keys, onKeyRemoved) {
  return this.remove_(
      shaka.offline.DBEngine.Store_.MANIFEST,
      keys,
      onKeyRemoved);
};


/** @override */
shaka.offline.DBEngine.prototype.reserveManifestId = function() {
  return this.nextManifestId_++;
};


/** @override */
shaka.offline.DBEngine.prototype.getSegment = function(key) {
  return this.get_(
      shaka.offline.DBEngine.Store_.SEGMENT,
      key);
};


/** @override */
shaka.offline.DBEngine.prototype.forEachSegment = function(each) {
  return this.forEach_(
      shaka.offline.DBEngine.Store_.SEGMENT,
      each);
};


/** @override */
shaka.offline.DBEngine.prototype.insertSegment = function(value) {
  return this.insert_(
      shaka.offline.DBEngine.Store_.SEGMENT,
      value);
};


/** @override */
shaka.offline.DBEngine.prototype.removeSegments =
    function(keys, onKeyRemoved) {
  return this.remove_(
      shaka.offline.DBEngine.Store_.SEGMENT,
      keys,
      onKeyRemoved);
};


/** @override */
shaka.offline.DBEngine.prototype.reserveSegmentId = function() {
  return this.nextSegmentId_++;
};


/**
 * @param {shaka.offline.DBEngine.Store_} store
 * @param {number} key
 * @return {!Promise<T>}
 * @template T
 * @private
 */
shaka.offline.DBEngine.prototype.get_ = function(store, key) {
  /** @const */
  var DBEngine = shaka.offline.DBEngine;
  /** @const */
  var READ_ONLY = DBEngine.Mode_.READ_ONLY;

  /** @type {IDBRequest} */
  var request;
  return this.createTransaction_(store, READ_ONLY, function(store) {
    request = store.get(key);
  }).then(function() { return request.result; });
};


/**
 * @param {shaka.offline.DBEngine.Store_} store
 * @param {function(T)} each
 * @return {!Promise}
 * @template T
 * @private
 */
shaka.offline.DBEngine.prototype.forEach_ = function(store, each) {
  /** @const */
  var DBEngine = shaka.offline.DBEngine;
  /** @const */
  var READ_ONLY = DBEngine.Mode_.READ_ONLY;

  return this.createTransaction_(store, READ_ONLY, function(store) {
    /** @type {IDBRequest} */
    var request = store.openCursor();

    request.onsuccess = function(event) {
      var cursor = event.target.result;
      if (cursor) {
        each(cursor.value);
        cursor.continue();
      }
    };
  });
};


/**
 * @param {shaka.offline.DBEngine.Store_} store
 * @param {T} value
 * @return {!Promise}
 * @template T
 * @private
 */
shaka.offline.DBEngine.prototype.insert_ = function(store, value) {
  /** @const */
  var DBEngine = shaka.offline.DBEngine;
  /** @const */
  var READ_WRITE = DBEngine.Mode_.READ_WRITE;

  return this.createTransaction_(store, READ_WRITE, function(store) {
    store.put(value);
  });
};


/**
 * @param {shaka.offline.DBEngine.Store_} store
 * @param {!Array<number>} keys
 * @param {?function(number)} onKeyRemoved
 * @return {!Promise}
 * @template T
 * @private
 */
shaka.offline.DBEngine.prototype.remove_ = function(store, keys, onKeyRemoved) {
  /** @const */
  var DBEngine = shaka.offline.DBEngine;
  /** @const */
  var READ_WRITE = DBEngine.Mode_.READ_WRITE;

  return this.createTransaction_(store, READ_WRITE, function(store) {
    keys.forEach(function(key) {
      /** @type {IDBRequest} */
      var request = store.delete(key);
      request.onsuccess = function() {
        if (onKeyRemoved) {
          onKeyRemoved(key);
        }
      };
    });
  });
};


/**
 * Creates a new transaction for the given store name and calls |action| to
 * modify the store. The transaction will resolve or reject the promise
 * returned by this function.
 *
 * @param {string} storeName
 * @param {string} type
 * @param {!function(IDBObjectStore)} action
 *
 * @return {!Promise}
 * @private
 */
shaka.offline.DBEngine.prototype.createTransaction_ = function(storeName,
                                                               type,
                                                               action) {
  /** @const */
  var DBEngine = shaka.offline.DBEngine;
  /** @const {string} */
  var READ_ONLY = DBEngine.Mode_.READ_ONLY;
  /** @const {string} */
  var READ_WRITE = DBEngine.Mode_.READ_WRITE;

  /** @type {!shaka.offline.DBEngine} */
  var self = this;

  goog.asserts.assert(self.db_, 'DBEngine must not be destroyed');
  goog.asserts.assert(type == READ_ONLY || type == READ_WRITE,
                      'Unexpected transaction type.');

  var op = {
    transaction: self.db_.transaction([storeName], type),
    promise: new shaka.util.PublicPromise()
  };

  op.transaction.oncomplete = function(event) {
    self.closeOperation_(op);
    op.promise.resolve();
  };

  // We will see an onabort call via:
  //   1. request error -> transaction error -> transaction abort
  //   2. transaction commit fail -> transaction abort
  // As any transaction error will result in an abort, it is better to listen
  // for an abort so that we will catch all failed transaction operations.
  op.transaction.onabort = function(event) {
    self.closeOperation_(op);
    shaka.offline.DBEngine.onError_(op.transaction, op.promise, event);
  };

  // We need to prevent default on the onerror event or else Firefox will
  // raise an error which will cause a karma failure. This will not stop the
  // onabort callback from firing.
  op.transaction.onerror = function(event) {
    event.preventDefault();
  };

  var store = op.transaction.objectStore(storeName);
  action(store);

  self.operations_.push(op);

  return op.promise;
};


/**
 * Close an open operation.
 *
 * @param {!shaka.offline.DBEngine.Operation} op
 * @private
 */
shaka.offline.DBEngine.prototype.closeOperation_ = function(op) {
  var i = this.operations_.indexOf(op);
  goog.asserts.assert(i >= 0, 'Operation must be in the list.');
  this.operations_.splice(i, 1);
};


/**
 * Creates a new connection to the database.
 * @return {!Promise}
 * @private
 */
shaka.offline.DBEngine.prototype.createConnection_ = function() {
  /** @type {!shaka.util.PublicPromise} */
  var promise = new shaka.util.PublicPromise();
  shaka.offline.DBEngine.createConnection_(promise, this.name_, this.retries_);

  return promise.then(function(db) {
    this.db_ = db;
  }.bind(this));
};


/**
 * Creates a new connection to the database.
 *
 * On IE/Edge, it is possible for the database to not be deleted when the
 * success callback is fired.  This means that when we delete the database and
 * immediately create a new connection, we will connect to the old database.
 * If this happens, we need to close the connection and retry.
 *
 * @see https://goo.gl/hOYJvN
 *
 * @param {!shaka.util.PublicPromise} promise
 * @param {string} name
 * @param {number} retries The number of attempts opening the database when
 *                         expecting an upgrade. If there are no more retries
 *                         then the connection will be accepted.
 * @private
 */
shaka.offline.DBEngine.createConnection_ = function(promise, name, retries) {
  /** @const */
  var DBEngine = shaka.offline.DBEngine;

  var indexedDB = window.indexedDB;
  var request = indexedDB.open(name, DBEngine.DB_VERSION_);

  /** @type {boolean} */
  var upgraded = false;

  request.onupgradeneeded = function(event) {
    var db = event.target.result;
    goog.asserts.assert(event.oldVersion == 0,
                        'Must be upgrading from version 0');
    goog.asserts.assert(db.objectStoreNames.length == 0,
                        'Version 0 database should be empty');

    // TODO(vaage) : Change to use auto increment keys.
    db.createObjectStore(DBEngine.Store_.MANIFEST, {keyPath: 'key'});
    db.createObjectStore(DBEngine.Store_.SEGMENT, {keyPath: 'key'});

    upgraded = true;
  };

  request.onsuccess = function(event) {
    var db = event.target.result;

    if (retries == 0 || upgraded) {
      promise.resolve(db);
    } else {
      /** @type {function()} */
      var retry = function() {
        DBEngine.createConnection_(promise, name, retries - 1);
      };

      // Try creating the connection again in one second.
      db.close();
      setTimeout(retry, 1000 /* 1 second */);
    }
  };

  request.onerror = DBEngine.onError_.bind(null, request, promise);
};


/**
 * Rejects the given Promise using the error fromt the transaction.
 *
 * @param {!IDBTransaction|!IDBRequest} errorSource
 * @param {!shaka.util.PublicPromise} promise
 * @param {!Event} event
 * @private
 */
shaka.offline.DBEngine.onError_ = function(errorSource, promise, event) {

  var error;

  if (errorSource.error) {
    error = new shaka.util.Error(
        shaka.util.Error.Severity.CRITICAL,
        shaka.util.Error.Category.STORAGE,
        shaka.util.Error.Code.INDEXED_DB_ERROR,
        errorSource.error);
  } else {
    error = new shaka.util.Error(
        shaka.util.Error.Severity.CRITICAL,
        shaka.util.Error.Category.STORAGE,
        shaka.util.Error.Code.OPERATION_ABORTED);
  }

  promise.reject(error);

  // Firefox will raise an error which will cause a karma failure.
  event.preventDefault();
};


/**
 * @param {IDBObjectStore} store
 * @param {function(number)} onId
 * @private
 */
shaka.offline.DBEngine.getLastId_ = function(store, onId) {
  /** @type {IDBRequest} */
  var request = store.openCursor(null, 'prev');
  request.onsuccess = function(event) {
    var cursor = event.target.result;
    if (cursor) {
      onId(cursor.key);
    }
  };
};