Added StorageRef to track download urls on iOS

Author: auser

ios/Firestack/FirestackStorage.m

@@ -15,6 +15,29 @@ @implementation FirestackStorage
 
 RCT_EXPORT_MODULE(FirestackStorage);
 
+RCT_EXPORT_METHOD(downloadUrl: (NSString *) path
+    callback:(RCTResponseSenderBlock) callback)
+{
+    FIRStorageReference *storageRef = [[FIRStorage storage] referenceWithPath:path];
+    [storageRef downloadURLWithCompletion:^(NSURL * _Nullable URL, NSError * _Nullable error) {
+        if (error != nil) {
+            NSDictionary *evt = @{
+                                  @"status": @"error",
+                                  @"path": path,
+                                  @"msg": [error debugDescription]
+                                  };
+            callback(@[evt]);
+        } else {
+            NSDictionary *resp = @{
+                                   @"status": @"success",
+                                   @"path": path,
+                                   @"url": URL
+                                   };
+            callback(@[[NSNull null], resp]);
+        }
+    }];
+}
+
 RCT_EXPORT_METHOD(uploadFile: (NSString *) urlStr
                   name: (NSString *) name
                   path:(NSString *)path

lib/modules/base.js

@@ -72,4 +72,23 @@ export class Base {
       }
     });
   }
+}
+
+export class ReferenceBase extends Base {
+  constructor(firestack, path) {
+    super(firestack);
+    this.path = Array.isArray(path) ?
+                      path :
+                      (typeof path == 'string' ?
+                        [path] : []);
+  }
+
+  pathToString() {
+    let path = this.path;
+    let pathStr = (path.length > 0 ? path.join('/') : '/');
+    if (pathStr[0] != '/') {
+      pathStr = `/${pathStr}`
+    }
+    return pathStr;
+  }
 }

lib/modules/database.js

@@ -6,7 +6,7 @@ const FirestackDatabase = NativeModules.FirestackDatabase;
 const FirestackDatabaseEvt = new NativeEventEmitter(FirestackDatabase);
 
 import promisify from '../promisify'
-import { Base } from './base'
+import { Base, ReferenceBase } from './base'
 
 let dbSubscriptions = {};
 
@@ -114,16 +114,11 @@ class DatabaseQuery {
 
 // https://firebase.google.com/docs/reference/js/firebase.database.Reference
 const separator = '/';
-class DatabaseRef extends Base {
+class DatabaseRef extends ReferenceBase {
   constructor(db, path) {
-    super(db.firestack);
+    super(db.firestack, path);
 
     this.db = db;
-    this.path = Array.isArray(path) ?
-                      path :
-                      (typeof path == 'string' ?
-                        [path] : []);
-
     this.query = new DatabaseQuery(this);
 
     // Aliases
@@ -255,24 +250,6 @@ class DatabaseRef extends Base {
 
   dbPath() {
     let path = this.path;
-    // let key, value;
-    // if (arr && arr.length > 0) {
-    //   key = arr[0];
-    //   value = arr[1];
-    //   // A key and a value were passed, to parse, if necessary
-    //   if (arr.length == 1) {
-    //     if (key && typeof(key) == 'string') {
-    //       path.push(key);
-    //     }
-    //   } else {
-    //     if (key && typeof(key) == 'string' && value) {
-    //       path.push(key);
-    //     } else {
-    //       value = key;
-    //     }
-    //   }
-    // }
-
     let pathStr = (path.length > 0 ? path.join('/') : '/');
     if (pathStr[0] != '/') {
       pathStr = `/${pathStr}`

lib/modules/storage.js

@@ -4,7 +4,19 @@ const FirestackStorage = NativeModules.FirestackStorage;
 const FirestackStorageEvt = new NativeEventEmitter(FirestackStorage);
 
 import promisify from '../promisify'
-import { Base } from './base'
+import { Base, ReferenceBase } from './base'
+
+class StorageRef extends ReferenceBase {
+  constructor(storage, path) {
+    super(storage.firestack, path);
+  }
+
+  downloadUrl() {
+    console.log('downloadUrl called', this.path);
+    const path = this.pathToString();
+    return promisify('downloadUrl', FirestackStorage)(path);
+  }
+}
 
 export class Storage extends Base {
   constructor(firestack, options={}) {
@@ -14,10 +26,21 @@ export class Storage extends Base {
       this.setStorageUrl(this.options.storageBucket);
     }
 
+    this.refs = {};
+
     this._addToFirestackInstance(
       'uploadFile'
     )
   }
+
+  ref(...path) {
+    const key = this._pathKey(path);
+    if (!this.refs[key]) {
+      const ref = new StorageRef(this, path);
+      this.refs[key] = ref;
+    }
+    return this.refs[key];
+  }
 
   /**
    * Upload a filepath
@@ -58,6 +81,10 @@ export class Storage extends Base {
     this.storageUrl = `gs://${url}`;
   }
 
+  _pathKey(...path) {
+    return path.join('-');
+  }
+
   get namespace() {
     return 'firestack:storage'
   }