1 files changed, 84 insertions, 0 deletions
diff --git a/Firebase/Storage/FIRStorageGetMetadataTask.m b/Firebase/Storage/FIRStorageGetMetadataTask.m
new file mode 100644
index 0000000..d0e8981
--- /dev/null
+++ b/Firebase/Storage/FIRStorageGetMetadataTask.m
@@ -0,0 +1,84 @@
+// 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 "FIRStorageGetMetadataTask.h"
+
+#import "FIRStorageConstants.h"
+#import "FIRStorageMetadata_Private.h"
+#import "FIRStorageTask_Private.h"
+#import "FIRStorageUtils.h"
+
+#import "FirebaseStorage.h"
+
+@implementation FIRStorageGetMetadataTask {
+ @private
+  FIRStorageVoidMetadataError _completion;
+}
+
+- (instancetype)initWithReference:(FIRStorageReference *)reference
+                   fetcherService:(GTMSessionFetcherService *)service
+                       completion:(FIRStorageVoidMetadataError)completion {
+  self = [super initWithReference:reference fetcherService:service];
+  if (self) {
+    _completion = [completion copy];
+  }
+  return self;
+}
+
+- (void)enqueue {
+  NSMutableURLRequest *request = [self.baseRequest mutableCopy];
+  request.HTTPMethod = @"GET";
+  request.timeoutInterval = self.reference.storage.maxDownloadRetryTime;
+
+  FIRStorageVoidMetadataError callback = _completion;
+  _completion = nil;
+
+  GTMSessionFetcher *fetcher = [self.fetcherService fetcherWithRequest:request];
+  fetcher.comment = @"GetMetadataTask";
+  [fetcher beginFetchWithCompletionHandler:^(NSData *data, NSError *error) {
+    if (error) {
+      if (!self.error) {
+        self.error = [FIRStorageErrors errorWithServerError:error reference:self.reference];
+      }
+      if (callback) {
+        callback(nil, self.error);
+      }
+      return;
+    }
+
+    NSDictionary *responseDictionary = [NSDictionary frs_dictionaryFromJSONData:data];
+    if (responseDictionary != nil) {
+      FIRStorageMetadata *metadata =
+          [[FIRStorageMetadata alloc] initWithDictionary:responseDictionary];
+      [metadata setType:FIRStorageMetadataTypeFile];
+      if (callback) {
+        callback(metadata, nil);
+      }
+    } else {
+      NSString *returnedData = [[NSString alloc] initWithData:data encoding:NSUTF8StringEncoding];
+      NSString *invalidDataString =
+          [NSString stringWithFormat:kFIRStorageInvalidDataFormat, returnedData];
+      NSDictionary *dict;
+      if (invalidDataString.length > 0) {
+        dict = @{NSLocalizedFailureReasonErrorKey : invalidDataString};
+      }
+      self.error = [FIRStorageErrors errorWithCode:FIRStorageErrorCodeUnknown infoDictionary:dict];
+      if (callback) {
+        callback(nil, self.error);
+      }
+    }
+  }];
+}
+
+@end