1 files changed, 65 insertions, 0 deletions
diff --git a/Firebase/Storage/FIRStorageTask.m b/Firebase/Storage/FIRStorageTask.m
new file mode 100644
index 0000000..3c1cf6f
--- /dev/null
+++ b/Firebase/Storage/FIRStorageTask.m
@@ -0,0 +1,65 @@
+// Copyright 2017 Google
+//
+// 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.
+
+#import "FIRStorageTask.h"
+
+#import "FIRStorage.h"
+#import "FIRStorage_Private.h"
+#import "FIRStorageReference.h"
+#import "FIRStorageReference_Private.h"
+#import "FIRStorageTaskSnapshot.h"
+#import "FIRStorageTaskSnapshot_Private.h"
+#import "FIRStorageTask_Private.h"
+
+#import "GTMSessionFetcherService.h"
+
+@implementation FIRStorageTask
+
+- (instancetype)init {
+  FIRStorage *storage = [FIRStorage storage];
+  FIRStorageReference *reference = [storage reference];
+  FIRStorageTask *task =
+      [self initWithReference:reference fetcherService:storage.fetcherServiceForApp];
+  return task;
+}
+
+- (instancetype)initWithReference:(FIRStorageReference *)reference
+                   fetcherService:(GTMSessionFetcherService *)service {
+  self = [super init];
+  if (self) {
+    _reference = reference;
+    _baseRequest = [FIRStorageUtils defaultRequestForPath:reference.path];
+    _fetcherService = service;
+    _fetcherService.maxRetryInterval = _reference.storage.maxOperationRetryTime;
+  }
+  return self;
+}
+
+- (FIRStorageTaskSnapshot *)snapshot {
+  @synchronized(self) {
+    NSProgress *progress =
+        [NSProgress progressWithTotalUnitCount:self.progress.totalUnitCount];
+    progress.completedUnitCount = self.progress.completedUnitCount;
+    FIRStorageTaskSnapshot *snapshot =
+        [[FIRStorageTaskSnapshot alloc] initWithTask:self
+                                               state:self.state
+                                            metadata:self.metadata
+                                           reference:self.reference
+                                            progress:progress
+                                               error:[self.error copy]];
+    return snapshot;
+  }
+}
+
+@end