Add flow to storage.js

Author: BlooJeans

lib/modules/storage.js

@@ -1,3 +1,4 @@
+/* @flow */
 
 import {NativeModules, NativeEventEmitter} from 'react-native';
 const FirestackStorage = NativeModules.FirestackStorage;
@@ -13,7 +14,7 @@ class StorageRef extends ReferenceBase {
     this.storage = storage;
   }
 
-  downloadUrl() {
+  downloadUrl(): Promise<Object> {
     const path = this.pathToString();
     return promisify('downloadUrl', FirestackStorage)(this.storage.storageUrl, path);
   }
@@ -23,7 +24,7 @@ class StorageRef extends ReferenceBase {
    * @param {String} downloadPath Where to store the file
    * @return {Promise}
    */
-  download (downloadPath, cb) {
+  download (downloadPath: string, cb: (evt: Object) => Object): Promise<Object> {
     let callback = cb;
     if (!callback || typeof callback !== 'function') {
       callback = (evt) => {};
@@ -38,7 +39,7 @@ class StorageRef extends ReferenceBase {
     return promisify('downloadFile', FirestackStorage)(this.storage.storageUrl, path, downloadPath)
       .then((res) => {
         console.log('res --->', res);
-        listeners.forEach(this.storage._removeListener);
+        listeners.forEach(listener => listener.remove());
         return res;
       })
       .catch(err => {
@@ -47,8 +48,11 @@ class StorageRef extends ReferenceBase {
   }
 }
 
+type StorageOptionsType = {
+  storageBucket?: ?string,
+};
 export default class Storage extends Base {
-  constructor(firestack, options={}) {
+  constructor(firestack: Object, options:StorageOptionsType={}) {
     super(firestack, options);
 
     if (this.options.storageBucket) {
@@ -58,8 +62,8 @@ export default class Storage extends Base {
     this.refs = {};
   }
 
-  ref(...path) {
-    const key = this._pathKey(path);
+  ref(...path: Array<string>): StorageRef {
+    const key = this._pathKey(...path);
     if (!this.refs[key]) {
       const ref = new StorageRef(this, path);
       this.refs[key] = ref;
@@ -74,10 +78,10 @@ export default class Storage extends Base {
    * @param  {object} metadata An object containing metadata
    * @return {Promise}
    */
-  uploadFile(name, filepath, metadata={}, cb) {
+  uploadFile(name: string, filepath: string, metadata: Object={}, cb: (evt: Object) => Object): Promise<Object> {
     let callback = cb;
     if (!callback || typeof callback !== 'function') {
-      callback = (evt) => {}
+      callback = (evt: Object) => ({});
     }
 
     filepath = filepath.replace("file://", "");
@@ -88,29 +92,26 @@ export default class Storage extends Base {
     listeners.push(this._addListener('upload_resumed', callback));
     return promisify('uploadFile', FirestackStorage)(this.storageUrl, name, filepath, metadata)
       .then((res) => {
-        listeners.forEach(this._removeListener);
+        listeners.forEach(listener => listener.remove());
         return res;
       });
   }
 
-  getRealPathFromURI(uri) {
+  getRealPathFromURI(uri: string): Promise<string> {
     return promisify('getRealPathFromURI', FirestackStorage)(uri);
   }
 
-  _addListener(evt, cb) {
-    return FirestackStorageEvt.addListener(evt, cb);
+  _addListener(evt: string, cb: (evt: Object) => Object): {remove: () => void} {
+    let listener = FirestackStorageEvt.addListener(evt, cb);
+    return listener;
   }
 
-  _removeListener(evt) {
-    return FirestackStorageEvt.removeListener(evt);
-  }
-
-  setStorageUrl(url) {
+  setStorageUrl(url: string): void {
     // return promisify('setStorageUrl', FirestackStorage)(url);
     this.storageUrl = `gs://${url}`;
   }
 
-  _pathKey(...path) {
+  _pathKey(...path: Array<string>): string {
     return path.join('-');
   }
 
@@ -126,7 +127,7 @@ export default class Storage extends Base {
       'FILETYPE_DIRECTORY': FirestackStorage.FILETYPE_DIRECTORY
   };
 
-  get namespace() {
+  get namespace(): string {
     return 'firestack:storage'
   }
 }