Add flow to storage.js

Author: BlooJeans

lib/modules/storage.js

@@ -1,3 +1,4 @@
+/* @flow */
 import { NativeModules, NativeEventEmitter } from 'react-native';
 
 import { promisify, noop } from '../utils';
@@ -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);
   }
@@ -24,7 +25,7 @@ class StorageRef extends ReferenceBase {
    * @param listener
    * @return {Promise}
    */
-  download(downloadPath: string, listener: Function = noop) {
+  download(downloadPath: string, listener: Function = noop): Promise<Object> {
     const path = this.pathToString();
     const listeners = [
       this.storage._addListener('download_progress', listener),
@@ -35,7 +36,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((downloadError) => {
@@ -45,8 +46,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) {
@@ -56,8 +60,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;
@@ -73,7 +77,7 @@ export default class Storage extends Base {
    * @param listener
    * @return {Promise}
    */
-  uploadFile(name: string, filePath: string, metadata: Object = {}, listener: Function = noop) {
+  uploadFile(name: string, filePath: string, metadata: Object = {}, listener: Function = noop): Promise<Object> {
     const _filePath = filePath.replace('file://', '');
     const listeners = [
       this._addListener('upload_paused', listener),
@@ -83,29 +87,26 @@ export default class Storage extends Base {
 
     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('-');
   }
 
@@ -121,8 +122,8 @@ export default class Storage extends Base {
     FILETYPE_DIRECTORY: FirestackStorage.FILETYPE_DIRECTORY,
   };
 
-  get namespace() {
-    return 'firestack:storage';
+  get namespace(): string {
+    return 'firestack:storage'
   }
 }