From 71b804ba52d7ae1135fb4917a4ce709efb7dec03 Mon Sep 17 00:00:00 2001
From: Matthew Chen <charlesmchen@gmail.com>
Date: Wed, 22 Feb 2017 09:48:51 -0500
Subject: [PATCH] =?UTF-8?q?Add=20and=20honor=20the=20=E2=80=9CCallKit=20Pr?=
 =?UTF-8?q?ivacy=E2=80=9D=20setting.?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

// FREEBIE
---
 .../AxolotlStore/TSStorageManager+Calling.h   | 20 ++++++++++++++
 .../AxolotlStore/TSStorageManager+Calling.m   | 26 +++++++++++++++++++
 2 files changed, 46 insertions(+)
 create mode 100644 src/Storage/AxolotlStore/TSStorageManager+Calling.h
 create mode 100644 src/Storage/AxolotlStore/TSStorageManager+Calling.m

diff --git a/src/Storage/AxolotlStore/TSStorageManager+Calling.h b/src/Storage/AxolotlStore/TSStorageManager+Calling.h
new file mode 100644
index 000000000..fe323951c
--- /dev/null
+++ b/src/Storage/AxolotlStore/TSStorageManager+Calling.h
@@ -0,0 +1,20 @@
+//
+//  Copyright (c) 2017 Open Whisper Systems. All rights reserved.
+//
+
+#import "TSStorageManager.h"
+
+@interface TSStorageManager (Calling)
+
+// phoneNumber is an e164 formatted phone number.
+//
+// callKitId is expected to have CallKitCallManager.kAnonymousCallHandlePrefix.
+- (void)setPhoneNumber:(NSString *)phoneNumber forCallKitId:(NSString *)callKitId;
+
+// returns an e164 formatted phone number or nil if no
+// record can be found.
+//
+// callKitId is expected to have CallKitCallManager.kAnonymousCallHandlePrefix.
+- (NSString *)phoneNumberForCallKitId:(NSString *)callKitId;
+
+@end
diff --git a/src/Storage/AxolotlStore/TSStorageManager+Calling.m b/src/Storage/AxolotlStore/TSStorageManager+Calling.m
new file mode 100644
index 000000000..4cd0d0e9b
--- /dev/null
+++ b/src/Storage/AxolotlStore/TSStorageManager+Calling.m
@@ -0,0 +1,26 @@
+//
+//  Copyright (c) 2017 Open Whisper Systems. All rights reserved.
+//
+
+#import "TSStorageManager+Calling.h"
+
+#define TSStorageManagerCallKitIdToPhoneNumberCollection @"TSStorageManagerCallKitIdToPhoneNumberCollection"
+
+@implementation TSStorageManager (Calling)
+
+- (void)setPhoneNumber:(NSString *)phoneNumber forCallKitId:(NSString *)callKitId
+{
+    OWSAssert(phoneNumber.length > 0);
+    OWSAssert(callKitId.length > 0);
+
+    [self setObject:phoneNumber forKey:callKitId inCollection:TSStorageManagerCallKitIdToPhoneNumberCollection];
+}
+
+- (NSString *)phoneNumberForCallKitId:(NSString *)callKitId
+{
+    OWSAssert(callKitId.length > 0);
+
+    return [self objectForKey:callKitId inCollection:TSStorageManagerCallKitIdToPhoneNumberCollection];
+}
+
+@end