DevTools: Promisify WebInspector.TempFile

Source/devtools/front_end/bindings/TempFile.js

 /*
  * Copyright (C) 2013 Google Inc. All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions are
  * met:
  *
  *     * Redistributions of source code must retain the above copyright
  * notice, this list of conditions and the following disclaimer.
  *     * Redistributions in binary form must reproduce the above
@@ skipping 14 matching lines @@
  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  */
 
 window.requestFileSystem = window.requestFileSystem || window.webkitRequestFileSystem;
 
 /**
  * @constructor
- * @param {!string} dirPath
- * @param {!string} name
- * @param {!function(?WebInspector.TempFile)} callback
  */
-WebInspector.TempFile = function(dirPath, name, callback)
+WebInspector.TempFile = function()
 {
     this._fileEntry = null;
     this._writer = null;
+}
+
+/**
+ * @param {string} dirPath
+ * @param {string} name
+ * @return {!Promise.<!WebInspector.TempFile>}
+ */
+WebInspector.TempFile.create = function(dirPath, name)
+{
+    var file = new WebInspector.TempFile();
+
+    function requestTempFileSystem()
+    {
+        return new Promise(window.requestFileSystem.bind(window, window.TEMPORARY, 10));
+    }
 
     /**
      * @param {!FileSystem} fs
-     * @this {WebInspector.TempFile}
      */
-    function didInitFs(fs)
+    function getDirectoryEntry(fs)
     {
-        fs.root.getDirectory(dirPath, { create: true }, didGetDir.bind(this), errorHandler);
+        return new Promise(fs.root.getDirectory.bind(fs.root, dirPath, { create: true }));
     }
 
     /**
      * @param {!DirectoryEntry} dir
-     * @this {WebInspector.TempFile}
      */
-    function didGetDir(dir)
+    function getFileEntry(dir)
     {
-        dir.getFile(name, { create: true }, didCreateFile.bind(this), errorHandler);
+        return new Promise(dir.getFile.bind(dir, name, { create: true }));
     }
 
     /**
      * @param {!FileEntry} fileEntry
-     * @this {WebInspector.TempFile}
      */
-    function didCreateFile(fileEntry)
+    function createFileWriter(fileEntry)
     {
-        this._fileEntry = fileEntry;
-        fileEntry.createWriter(didCreateWriter.bind(this), errorHandler);
+        file._fileEntry = fileEntry;
+        return new Promise(fileEntry.createWriter.bind(fileEntry));
     }
 
     /**
      * @param {!FileWriter} writer
-     * @this {WebInspector.TempFile}
      */
-    function didCreateWriter(writer)
+    function truncateFile(writer)
     {
+        if (!writer.length) {
+            file._writer = writer;
+            return Promise.resolve(file);
+        }
+
         /**
-         * @this {WebInspector.TempFile}
+         * @param {function(?)} fulfill
+         * @param {function(*)} reject
          */
-        function didTruncate(e)
+        function truncate(fulfill, reject)
         {
-            this._writer = writer;
+            writer.onwriteend = fulfill;
+            writer.onerror = reject;
+            writer.truncate(0);
+        }
+
+        function didTruncate()
+        {
+            file._writer = writer;
             writer.onwriteend = null;
             writer.onerror = null;
-            callback(this);
+            return Promise.resolve(file);
         }
 
         function onTruncateError(e)
         {
-            WebInspector.console.error("Failed to truncate temp file " + e.code + " : " + e.message);
-            callback(null);
+            writer.onwriteend = null;
+            writer.onerror = null;
+            throw e;
         }
 
-        if (writer.length) {
-            writer.onwriteend = didTruncate.bind(this);
-            writer.onerror = onTruncateError;
-            writer.truncate(0);
-        } else {
-            this._writer = writer;
-            callback(this);
-        }
+        return new Promise(truncate).then(didTruncate, onTruncateError);
     }
 
-    function errorHandler(e)
-    {
-        WebInspector.console.error("Failed to create temp file " + e.code + " : " + e.message);
-        callback(null);
-    }
-
-    /**
-     * @this {WebInspector.TempFile}
-     */
-    function didClearTempStorage()
-    {
-        window.requestFileSystem(window.TEMPORARY, 10, didInitFs.bind(this), errorHandler);
-    }
-    WebInspector.TempFile._ensureTempStorageCleared(didClearTempStorage.bind(this));
+    return WebInspector.TempFile.ensureTempStorageCleared()
+        .then(requestTempFileSystem)
+        .then(getDirectoryEntry)
+        .then(getFileEntry)
+        .then(createFileWriter)
+        .then(truncateFile);
 }
 
 WebInspector.TempFile.prototype = {
     /**
      * @param {!Array.<string>} strings
      * @param {!function(boolean)} callback
      */
     write: function(strings, callback)
     {
         var blob = new Blob(strings, {type: 'text/plain'});
@@ skipping 90 matching lines @@
     }
 }
 
 /**
  * @constructor
  * @param {!string} dirPath
  * @param {!string} name
  */
 WebInspector.DeferredTempFile = function(dirPath, name)
 {
+    /** @type {?Array.<string>} */
     this._chunks = [];
     this._tempFile = null;
     this._isWriting = false;
     this._finishCallback = null;
     this._finishedWriting = false;
     this._callsPendingOpen = [];
     this._pendingReads = [];
-    new WebInspector.TempFile(dirPath, name, this._didCreateTempFile.bind(this));
+    WebInspector.TempFile.create(dirPath, name)
+        .then(this._didCreateTempFile.bind(this), this._failedToCreateTempFile.bind(this));
 }
 
 WebInspector.DeferredTempFile.prototype = {
     /**
      * @param {!Array.<string>} strings
      */
     write: function(strings)
     {
         if (!this._chunks)
             return;
         if (this._finishCallback)
             throw new Error("No writes are allowed after close.");
         this._chunks.push.apply(this._chunks, strings);
         if (this._tempFile && !this._isWriting)
             this._writeNextChunk();
     },
 
     /**
      * @param {!function(?WebInspector.TempFile)} callback
      */
     finishWriting: function(callback)
     {
         this._finishCallback = callback;
         if (this._finishedWriting)
             callback(this._tempFile);
         else if (!this._isWriting && !this._chunks.length)
             this._notifyFinished();
     },
 
+    /**
+     * @param {*} e
+     */
+    _failedToCreateTempFile: function(e)
+    {
+        WebInspector.console.error("Failed to create temp file " + e.code + " : " + e.message);
+        this._chunks = null;
+        this._notifyFinished();
+    },
+
+    /**
+     * @param {!WebInspector.TempFile} tempFile
+     */
     _didCreateTempFile: function(tempFile)
     {
         this._tempFile = tempFile;
         var callsPendingOpen = this._callsPendingOpen;
         this._callsPendingOpen = null;
         for (var i = 0; i < callsPendingOpen.length; ++i)
             callsPendingOpen[i]();
-        if (!tempFile) {
-            this._chunks = null;
-            this._notifyFinished();
-            return;
-        }
         if (this._chunks.length)
             this._writeNextChunk();
     },
 
     _writeNextChunk: function()
     {
         var chunks = this._chunks;
         this._chunks = [];
         this._isWriting = true;
-        this._tempFile.write(chunks, this._didWriteChunk.bind(this));
+        this._tempFile.write(/** @type {!Array.<string>} */(chunks), this._didWriteChunk.bind(this));
     },
 
     _didWriteChunk: function(success)
     {
         this._isWriting = false;
         if (!success) {
             this._tempFile = null;
             this._chunks = null;
             this._notifyFinished();
             return;
@@ skipping 54 matching lines @@
         if (this._callsPendingOpen) {
             this._callsPendingOpen.push(this.remove.bind(this));
             return;
         }
         if (this._tempFile)
             this._tempFile.remove();
     }
 }
 
 /**
- * @constructor
+ * @param {function(?)} fulfill
+ * @param {function(*)} reject
  */
-WebInspector.TempStorageCleaner = function()
+WebInspector.TempFile._clearTempStorage = function(fulfill, reject)
 {
+    /**
+     * @param {!Event} event
+     */
+    function handleError(event)
+    {
+        WebInspector.console.error(WebInspector.UIString("Failed to clear temp storage: %s", event.data));
+        reject(event.data);
+    }
+
+    /**
+     * @param {!Event} event
+     */
+    function handleMessage(event)
+    {
+        if (event.data.type === "tempStorageCleared") {
+            if (event.data.error)
+                WebInspector.console.error(event.data.error);
+            else
+                fulfill(undefined);
+            return;
+        }
+        reject(event.data);
+    }
+
     try {
-        this._worker = new WorkerRuntime.Worker("temp_storage_shared_worker", "TempStorageCleaner");
-        this._worker.onerror = this._handleError.bind(this);
-        this._callbacks = [];
-        this._worker.port.onmessage = this._handleMessage.bind(this);
-        this._worker.port.onerror = this._handleError.bind(this);
+        var worker = new WorkerRuntime.Worker("temp_storage_shared_worker", "TempStorageCleaner");
+        worker.onerror = handleError;
+        worker.port.onmessage = handleMessage;
+        worker.port.onerror = handleError;
     } catch (e) {
         if (e.name === "URLMismatchError")
             console.log("Shared worker wasn't started due to url difference. " + e);
         else
             throw e;
     }
 }
 
-WebInspector.TempStorageCleaner.prototype = {
-    /**
-     * @param {!function()} callback
-     */
-    ensureStorageCleared: function(callback)
-    {
-        if (this._callbacks)
-            this._callbacks.push(callback);
-        else
-            callback();
-    },
-
-    _handleMessage: function(event)
-    {
-        if (event.data.type === "tempStorageCleared") {
-            if (event.data.error)
-                WebInspector.console.error(event.data.error);
-            this._notifyCallbacks();
-        }
-    },
-
-    _handleError: function(event)
-    {
-        WebInspector.console.error(WebInspector.UIString("Failed to clear temp storage: %s", event.data));
-        this._notifyCallbacks();
-    },
-
-    _notifyCallbacks: function()
-    {
-        var callbacks = this._callbacks;
-        this._callbacks = null;
-        for (var i = 0; i < callbacks.length; i++)
-            callbacks[i]();
-    }
+/**
+ * @return {!Promise.<undefined>}
+ */
+WebInspector.TempFile.ensureTempStorageCleared = function()
+{
+    if (!WebInspector.TempFile._storageCleanerPromise)
+        WebInspector.TempFile._storageCleanerPromise = new Promise(WebInspector.TempFile._clearTempStorage);
+    return WebInspector.TempFile._storageCleanerPromise;
 }
-
-/**
- * @param {!function()} callback
- */
-WebInspector.TempFile._ensureTempStorageCleared = function(callback)
-{
-    if (!WebInspector.TempFile._storageCleaner)
-        WebInspector.TempFile._storageCleaner = new WebInspector.TempStorageCleaner();
-    WebInspector.TempFile._storageCleaner.ensureStorageCleared(callback);
-}