Added Android code
[wl-app.git] / iOS / Pods / FirebaseMessaging / Firebase / Messaging / FIRMessagingRegistrar.m
1 /*
2  * Copyright 2017 Google
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *      http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16
17 #import "FIRMessagingRegistrar.h"
18
19 #import "FIRMessagingDefines.h"
20 #import "FIRMessagingLogger.h"
21 #import "FIRMessagingPubSubRegistrar.h"
22 #import "FIRMessagingUtilities.h"
23 #import "NSError+FIRMessaging.h"
24
25 @interface FIRMessagingRegistrar ()
26
27 @property(nonatomic, readwrite, assign) BOOL stopAllSubscriptions;
28
29 @property(nonatomic, readwrite, strong) FIRMessagingCheckinService *checkinService;
30 @property(nonatomic, readwrite, strong) FIRMessagingPubSubRegistrar *pubsubRegistrar;
31
32 @end
33
34 @implementation FIRMessagingRegistrar
35
36 - (NSString *)deviceAuthID {
37   return self.checkinService.deviceAuthID;
38 }
39
40 - (NSString *)secretToken {
41   return self.checkinService.secretToken;
42 }
43
44 - (instancetype)init {
45   self = [super init];
46   if (self) {
47     _checkinService = [[FIRMessagingCheckinService alloc] init];
48     // TODO(chliangGoogle): Merge pubsubRegistrar with Registrar as it is hard to track how many
49     // checkinService instances by separating classes too often.
50     _pubsubRegistrar = [[FIRMessagingPubSubRegistrar alloc] initWithCheckinService:_checkinService];
51   }
52   return self;
53 }
54
55 #pragma mark - Checkin
56
57 - (BOOL)tryToLoadValidCheckinInfo {
58   [self.checkinService tryToLoadPrefetchedCheckinPreferences];
59   return [self.checkinService hasValidCheckinInfo];
60 }
61
62 #pragma mark - Subscribe/Unsubscribe
63
64 - (void)updateSubscriptionToTopic:(NSString *)topic
65                         withToken:(NSString *)token
66                           options:(NSDictionary *)options
67                      shouldDelete:(BOOL)shouldDelete
68                           handler:(FIRMessagingTopicOperationCompletion)handler {
69   _FIRMessagingDevAssert(handler, @"Invalid nil handler");
70
71   if ([self tryToLoadValidCheckinInfo]) {
72     [self doUpdateSubscriptionForTopic:topic
73                                  token:token
74                                options:options
75                           shouldDelete:shouldDelete
76                             completion:handler];
77
78   } else {
79     FIRMessagingLoggerDebug(kFIRMessagingMessageCodeRegistrar000,
80                             @"Device check in error, no auth credentials found");
81     NSError *error = [NSError errorWithFCMErrorCode:kFIRMessagingErrorCodeMissingDeviceID];
82     handler(error);
83   }
84 }
85
86 - (void)cancelAllRequests {
87   self.stopAllSubscriptions = YES;
88   [self.pubsubRegistrar stopAllSubscriptionRequests];
89 }
90
91 #pragma mark - Private
92
93 - (void)doUpdateSubscriptionForTopic:(NSString *)topic
94                                token:(NSString *)token
95                              options:(NSDictionary *)options
96                         shouldDelete:(BOOL)shouldDelete
97                           completion:(FIRMessagingTopicOperationCompletion)completion {
98   _FIRMessagingDevAssert([self.checkinService hasValidCheckinInfo],
99                 @"No valid checkin info found before subscribe");
100
101   [self.pubsubRegistrar updateSubscriptionToTopic:topic
102                                         withToken:token
103                                           options:options
104                                      shouldDelete:shouldDelete
105                                           handler:completion];
106 }
107
108 @end