keychain/ckks/CKKSAPSReceiver.h

   1 /*
   2  * Copyright (c) 2016 Apple Inc. All Rights Reserved.
   3  *
   4  * @APPLE_LICENSE_HEADER_START@
   5  *
   6  * This file contains Original Code and/or Modifications of Original Code
   7  * as defined in and that are subject to the Apple Public Source License
   8  * Version 2.0 (the 'License'). You may not use this file except in
   9  * compliance with the License. Please obtain a copy of the License at
  10  * http://www.opensource.apple.com/apsl/ and read it before using this
  11  * file.
  12  *
  13  * The Original Code and all software distributed under the License are
  14  * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
  15  * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
  16  * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
  17  * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
  18  * Please see the License for the specific language governing rights and
  19  * limitations under the License.
  20  *
  21  * @APPLE_LICENSE_HEADER_END@
  22  */
  23
  24 #import <Foundation/Foundation.h>
  25
  26 #if OCTAGON
  27 #import <ApplePushService/ApplePushService.h>
  28 #import <CloudKit/CloudKit.h>
  29 #import "keychain/ckks/CKKSCondition.h"
  30 #import "keychain/ckks/CloudKitDependencies.h"
  31
  32
  33 NS_ASSUME_NONNULL_BEGIN
  34
  35 // APS is giving us a tracingUUID and a tracingEnabled bool, but our interfaces take a CKRecordZoneNotification. Add them to that class, then.
  36 @interface CKRecordZoneNotification (CKKSPushTracing)
  37 @property (nonatomic, assign) BOOL ckksPushTracingEnabled;
  38 @property (nonatomic, strong, nullable) NSString* ckksPushTracingUUID;
  39 @property (nonatomic, strong, nullable) NSDate* ckksPushReceivedDate;
  40 @end
  41
  42 @protocol CKKSZoneUpdateReceiver <NSObject>
  43 - (void)notifyZoneChange:(CKRecordZoneNotification* _Nullable)notification;
  44 @end
  45
  46 @interface CKKSAPSReceiver : NSObject <APSConnectionDelegate>
  47
  48 @property NSMapTable<CKRecordZoneID*, id<CKKSZoneUpdateReceiver>>* zoneMap;
  49
  50 // class dependencies (for injection)
  51 @property (readonly) Class<CKKSAPSConnection> apsConnectionClass;
  52 @property (nullable) id<CKKSAPSConnection> apsConnection;
  53
  54 + (instancetype)receiverForEnvironment:(NSString*)environmentName
  55                      namedDelegatePort:(NSString*)namedDelegatePort
  56                     apsConnectionClass:(Class<CKKSAPSConnection>)apsConnectionClass;
  57 - (CKKSCondition*)registerReceiver:(id<CKKSZoneUpdateReceiver>)receiver forZoneID:(CKRecordZoneID*)zoneID;
  58
  59 // Test support:
  60 - (instancetype)initWithEnvironmentName:(NSString*)environmentName
  61                       namedDelegatePort:(NSString*)namedDelegatePort
  62                      apsConnectionClass:(Class<CKKSAPSConnection>)apsConnectionClass;
  63 // This is the queue that APNS will use send the notifications to us
  64 + (dispatch_queue_t)apsDeliveryQueue;
  65
  66 @end
  67
  68 NS_ASSUME_NONNULL_END
  69 #endif  // OCTAGON