mirror of https://github.com/oxen-io/session-ios
add a Notification Service Extension for PN with preview
parent
93c5fd4b6d
commit
ca20f3f979
@ -0,0 +1,31 @@
|
|||||||
|
<?xml version="1.0" encoding="UTF-8"?>
|
||||||
|
<!DOCTYPE plist PUBLIC "-//Apple//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">
|
||||||
|
<plist version="1.0">
|
||||||
|
<dict>
|
||||||
|
<key>CFBundleDevelopmentRegion</key>
|
||||||
|
<string>$(DEVELOPMENT_LANGUAGE)</string>
|
||||||
|
<key>CFBundleDisplayName</key>
|
||||||
|
<string>LokiPushNotificationService</string>
|
||||||
|
<key>CFBundleExecutable</key>
|
||||||
|
<string>$(EXECUTABLE_NAME)</string>
|
||||||
|
<key>CFBundleIdentifier</key>
|
||||||
|
<string>$(PRODUCT_BUNDLE_IDENTIFIER)</string>
|
||||||
|
<key>CFBundleInfoDictionaryVersion</key>
|
||||||
|
<string>6.0</string>
|
||||||
|
<key>CFBundleName</key>
|
||||||
|
<string>$(PRODUCT_NAME)</string>
|
||||||
|
<key>CFBundlePackageType</key>
|
||||||
|
<string>$(PRODUCT_BUNDLE_PACKAGE_TYPE)</string>
|
||||||
|
<key>CFBundleShortVersionString</key>
|
||||||
|
<string>1.0</string>
|
||||||
|
<key>CFBundleVersion</key>
|
||||||
|
<string>1</string>
|
||||||
|
<key>NSExtension</key>
|
||||||
|
<dict>
|
||||||
|
<key>NSExtensionPointIdentifier</key>
|
||||||
|
<string>com.apple.usernotifications.service</string>
|
||||||
|
<key>NSExtensionPrincipalClass</key>
|
||||||
|
<string>$(PRODUCT_MODULE_NAME).NotificationService</string>
|
||||||
|
</dict>
|
||||||
|
</dict>
|
||||||
|
</plist>
|
@ -0,0 +1,14 @@
|
|||||||
|
<?xml version="1.0" encoding="UTF-8"?>
|
||||||
|
<!DOCTYPE plist PUBLIC "-//Apple//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">
|
||||||
|
<plist version="1.0">
|
||||||
|
<dict>
|
||||||
|
<key>com.apple.security.application-groups</key>
|
||||||
|
<array>
|
||||||
|
<string>group.com.loki-project.loki-messenger</string>
|
||||||
|
</array>
|
||||||
|
<key>keychain-access-groups</key>
|
||||||
|
<array>
|
||||||
|
<string>$(AppIdentifierPrefix)com.loki-project.loki-messenger</string>
|
||||||
|
</array>
|
||||||
|
</dict>
|
||||||
|
</plist>
|
@ -0,0 +1,211 @@
|
|||||||
|
import UserNotifications
|
||||||
|
import SignalServiceKit
|
||||||
|
import SignalMessaging
|
||||||
|
|
||||||
|
class NotificationService: UNNotificationServiceExtension {
|
||||||
|
|
||||||
|
static let threadIdKey = "Signal.AppNotificationsUserInfoKey.threadId"
|
||||||
|
var areVersionMigrationsComplete = false
|
||||||
|
var contentHandler: ((UNNotificationContent) -> Void)?
|
||||||
|
var notificationContent: UNMutableNotificationContent?
|
||||||
|
|
||||||
|
override func didReceive(_ request: UNNotificationRequest, withContentHandler contentHandler: @escaping (UNNotificationContent) -> Void) {
|
||||||
|
self.contentHandler = contentHandler
|
||||||
|
notificationContent = (request.content.mutableCopy() as? UNMutableNotificationContent)
|
||||||
|
|
||||||
|
DispatchQueue.main.sync { self.setupIfNecessary() }
|
||||||
|
|
||||||
|
if let notificationContent = notificationContent {
|
||||||
|
// Modify the notification content here...
|
||||||
|
let base64EncodedData = notificationContent.userInfo["ENCRYPTED_DATA"] as! String
|
||||||
|
let data = Data(base64Encoded: base64EncodedData)!
|
||||||
|
let envelope = try? LokiMessageWrapper.unwrap(data: data)
|
||||||
|
let envelopeData = try? envelope?.serializedData()
|
||||||
|
let decrypter = SSKEnvironment.shared.messageDecrypter
|
||||||
|
if (envelope != nil && envelopeData != nil) {
|
||||||
|
decrypter.decryptEnvelope(envelope!, envelopeData: envelopeData!,
|
||||||
|
successBlock: { result,transaction in
|
||||||
|
if let envelope = try? SSKProtoEnvelope.parseData(result.envelopeData) {
|
||||||
|
self.removeDecryptionChain(envelope: envelope, transaction: transaction)
|
||||||
|
self.handelDecryptionResult(result: result, notificationContent: notificationContent, transaction: transaction)
|
||||||
|
} else {
|
||||||
|
self.completeWithFailure(content: notificationContent)
|
||||||
|
}
|
||||||
|
},
|
||||||
|
failureBlock: {
|
||||||
|
self.completeWithFailure(content: notificationContent)
|
||||||
|
})
|
||||||
|
} else {
|
||||||
|
self.completeWithFailure(content: notificationContent)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func removeDecryptionChain(envelope: SSKProtoEnvelope, transaction: YapDatabaseReadWriteTransaction) {
|
||||||
|
let sessionRecord = SSKEnvironment.shared.primaryStorage.loadSession(envelope.source!, deviceId: Int32(envelope.sourceDevice), protocolContext: transaction)
|
||||||
|
let sessionState = sessionRecord.sessionState()
|
||||||
|
}
|
||||||
|
|
||||||
|
func handelDecryptionResult(result: OWSMessageDecryptResult, notificationContent: UNMutableNotificationContent, transaction: YapDatabaseReadWriteTransaction) {
|
||||||
|
let contentProto = try? SSKProtoContent.parseData(result.plaintextData!)
|
||||||
|
var thread: TSThread
|
||||||
|
var newNotificationBody = ""
|
||||||
|
let masterHexEncodedPublicKey: String = LokiDatabaseUtilities.objc_getMasterHexEncodedPublicKey(for: result.source, in: transaction) ?? result.source
|
||||||
|
var displayName = masterHexEncodedPublicKey
|
||||||
|
if let groupId = contentProto?.dataMessage?.group?.id {
|
||||||
|
thread = TSGroupThread.getOrCreateThread(withGroupId: groupId, groupType: .closedGroup, transaction: transaction)
|
||||||
|
displayName = thread.name()
|
||||||
|
if displayName.count < 1 {
|
||||||
|
displayName = MessageStrings.newGroupDefaultTitle
|
||||||
|
}
|
||||||
|
let group: SSKProtoGroupContext = (contentProto?.dataMessage?.group!)!
|
||||||
|
let oldGroupModel = (thread as! TSGroupThread).groupModel
|
||||||
|
var removeMembers = Set(arrayLiteral: oldGroupModel.groupMemberIds)
|
||||||
|
let newGroupModel = TSGroupModel.init(title: group.name,
|
||||||
|
memberIds:group.members,
|
||||||
|
image: oldGroupModel.groupImage,
|
||||||
|
groupId: group.id,
|
||||||
|
groupType: oldGroupModel.groupType,
|
||||||
|
adminIds: group.admins)
|
||||||
|
removeMembers.subtract(Set(arrayLiteral: newGroupModel.groupMemberIds))
|
||||||
|
newGroupModel.removedMembers = removeMembers as! NSMutableSet
|
||||||
|
switch contentProto?.dataMessage?.group?.type {
|
||||||
|
case .update:
|
||||||
|
newNotificationBody = oldGroupModel.getInfoStringAboutUpdate(to: newGroupModel, contactsManager: SSKEnvironment.shared.contactsManager)
|
||||||
|
break
|
||||||
|
case .quit:
|
||||||
|
let nameString = SSKEnvironment.shared.contactsManager.displayName(forPhoneIdentifier: masterHexEncodedPublicKey, transaction: transaction)
|
||||||
|
newNotificationBody = NSLocalizedString("GROUP_MEMBER_LEFT", comment: nameString)
|
||||||
|
break
|
||||||
|
default:
|
||||||
|
break
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
thread = TSContactThread.getOrCreateThread(withContactId: result.source, transaction: transaction)
|
||||||
|
displayName = contentProto?.dataMessage?.profile?.displayName ?? displayName
|
||||||
|
}
|
||||||
|
let userInfo: [String: Any] = [NotificationService.threadIdKey: thread.uniqueId!]
|
||||||
|
notificationContent.title = displayName
|
||||||
|
notificationContent.userInfo = userInfo
|
||||||
|
if newNotificationBody.count < 1 {
|
||||||
|
newNotificationBody = contentProto?.dataMessage?.body ?? ""
|
||||||
|
}
|
||||||
|
notificationContent.body = newNotificationBody
|
||||||
|
if notificationContent.body.count < 1 {
|
||||||
|
self.completeWithFailure(content: notificationContent)
|
||||||
|
} else {
|
||||||
|
self.contentHandler!(notificationContent)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private var hasSetup = false
|
||||||
|
func setupIfNecessary() {
|
||||||
|
AssertIsOnMainThread()
|
||||||
|
|
||||||
|
// The NSE will often re-use the same process, so if we're
|
||||||
|
// already setup we want to do nothing. We're already ready
|
||||||
|
// to process new messages.
|
||||||
|
guard !hasSetup else { return }
|
||||||
|
|
||||||
|
hasSetup = true
|
||||||
|
|
||||||
|
// This should be the first thing we do.
|
||||||
|
SetCurrentAppContext(NotificationServiceExtensionContext())
|
||||||
|
|
||||||
|
DebugLogger.shared().enableTTYLogging()
|
||||||
|
if _isDebugAssertConfiguration() {
|
||||||
|
DebugLogger.shared().enableFileLogging()
|
||||||
|
}
|
||||||
|
|
||||||
|
Logger.info("")
|
||||||
|
|
||||||
|
_ = AppVersion.sharedInstance()
|
||||||
|
|
||||||
|
Cryptography.seedRandom()
|
||||||
|
|
||||||
|
// We should never receive a non-voip notification on an app that doesn't support
|
||||||
|
// app extensions since we have to inform the service we wanted these, so in theory
|
||||||
|
// this path should never occur. However, the service does have our push token
|
||||||
|
// so it is possible that could change in the future. If it does, do nothing
|
||||||
|
// and don't disturb the user. Messages will be processed when they open the app.
|
||||||
|
guard OWSPreferences.isReadyForAppExtensions() else { return completeSilenty() }
|
||||||
|
|
||||||
|
AppSetup.setupEnvironment(
|
||||||
|
appSpecificSingletonBlock: {
|
||||||
|
// TODO: calls..
|
||||||
|
SSKEnvironment.shared.callMessageHandler = NoopCallMessageHandler()
|
||||||
|
SSKEnvironment.shared.notificationsManager = NoopNotificationsManager()
|
||||||
|
},
|
||||||
|
migrationCompletion: { [weak self] in
|
||||||
|
self?.versionMigrationsDidComplete()
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
NotificationCenter.default.addObserver(self,
|
||||||
|
selector: #selector(storageIsReady),
|
||||||
|
name: .StorageIsReady,
|
||||||
|
object: nil)
|
||||||
|
|
||||||
|
Logger.info("completed.")
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
override func serviceExtensionTimeWillExpire() {
|
||||||
|
// Called just before the extension will be terminated by the system.
|
||||||
|
// Use this as an opportunity to deliver your "best attempt" at modified content, otherwise the original push payload will be used.
|
||||||
|
if let contentHandler = contentHandler, let notificationContent = notificationContent {
|
||||||
|
contentHandler(notificationContent)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func wasReceivedByUD(envelope: SSKProtoEnvelope) -> Bool {
|
||||||
|
return (envelope.type == .unidentifiedSender && (!envelope.hasSource || envelope.source!.count < 1))
|
||||||
|
}
|
||||||
|
|
||||||
|
@objc
|
||||||
|
func versionMigrationsDidComplete() {
|
||||||
|
AssertIsOnMainThread()
|
||||||
|
|
||||||
|
Logger.debug("")
|
||||||
|
|
||||||
|
areVersionMigrationsComplete = true
|
||||||
|
|
||||||
|
checkIsAppReady()
|
||||||
|
}
|
||||||
|
|
||||||
|
@objc
|
||||||
|
func storageIsReady() {
|
||||||
|
AssertIsOnMainThread()
|
||||||
|
|
||||||
|
Logger.debug("")
|
||||||
|
|
||||||
|
checkIsAppReady()
|
||||||
|
}
|
||||||
|
|
||||||
|
@objc
|
||||||
|
func checkIsAppReady() {
|
||||||
|
AssertIsOnMainThread()
|
||||||
|
|
||||||
|
// Only mark the app as ready once.
|
||||||
|
guard !AppReadiness.isAppReady() else { return }
|
||||||
|
|
||||||
|
// App isn't ready until storage is ready AND all version migrations are complete.
|
||||||
|
guard OWSStorage.isStorageReady() && areVersionMigrationsComplete else { return }
|
||||||
|
|
||||||
|
// Note that this does much more than set a flag; it will also run all deferred blocks.
|
||||||
|
AppReadiness.setAppIsReady()
|
||||||
|
|
||||||
|
// AppVersion.sharedInstance().nseLaunchDidComplete()
|
||||||
|
}
|
||||||
|
|
||||||
|
func completeSilenty() {
|
||||||
|
contentHandler?(.init())
|
||||||
|
}
|
||||||
|
|
||||||
|
func completeWithFailure(content: UNMutableNotificationContent) {
|
||||||
|
content.body = "You've got a new message."
|
||||||
|
content.title = "Session"
|
||||||
|
contentHandler?(content)
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
@ -0,0 +1,94 @@
|
|||||||
|
//
|
||||||
|
// Copyright (c) 2020 Open Whisper Systems. All rights reserved.
|
||||||
|
//
|
||||||
|
|
||||||
|
import Foundation
|
||||||
|
import SignalServiceKit
|
||||||
|
import SignalMessaging
|
||||||
|
|
||||||
|
class NotificationServiceExtensionContext: NSObject, AppContext {
|
||||||
|
var wasWokenUpBySilentPushNotification: Bool = true
|
||||||
|
|
||||||
|
var openSystemSettingsAction: UIAlertAction?
|
||||||
|
|
||||||
|
let isMainApp = false
|
||||||
|
let isMainAppAndActive = false
|
||||||
|
|
||||||
|
func isInBackground() -> Bool { true }
|
||||||
|
func isAppForegroundAndActive() -> Bool { false }
|
||||||
|
func mainApplicationStateOnLaunch() -> UIApplication.State { .inactive }
|
||||||
|
|
||||||
|
var shouldProcessIncomingMessages: Bool { true }
|
||||||
|
func canPresentNotifications() -> Bool { true }
|
||||||
|
|
||||||
|
let appLaunchTime = Date()
|
||||||
|
lazy var buildTime: Date = {
|
||||||
|
guard let buildTimestamp = Bundle.main.object(forInfoDictionaryKey: "BuildTimestamp") as? TimeInterval, buildTimestamp > 0 else {
|
||||||
|
Logger.debug("No build timestamp, assuming app never expires.")
|
||||||
|
return .distantFuture
|
||||||
|
}
|
||||||
|
|
||||||
|
return .init(timeIntervalSince1970: buildTimestamp)
|
||||||
|
}()
|
||||||
|
|
||||||
|
func keychainStorage() -> SSKKeychainStorage {
|
||||||
|
return SSKDefaultKeychainStorage.shared
|
||||||
|
}
|
||||||
|
|
||||||
|
func appDocumentDirectoryPath() -> String {
|
||||||
|
guard let documentDirectoryURL = FileManager.default.urls(for: .documentDirectory, in: .userDomainMask).last else {
|
||||||
|
owsFail("failed to query document directory")
|
||||||
|
}
|
||||||
|
return documentDirectoryURL.path
|
||||||
|
}
|
||||||
|
|
||||||
|
func appSharedDataDirectoryPath() -> String {
|
||||||
|
guard let groupContainerURL = FileManager.default.containerURL(forSecurityApplicationGroupIdentifier: SignalApplicationGroup) else {
|
||||||
|
owsFail("failed to query group container")
|
||||||
|
}
|
||||||
|
return groupContainerURL.path
|
||||||
|
}
|
||||||
|
|
||||||
|
func appDatabaseBaseDirectoryPath() -> String {
|
||||||
|
return appSharedDataDirectoryPath()
|
||||||
|
}
|
||||||
|
|
||||||
|
func appUserDefaults() -> UserDefaults {
|
||||||
|
guard let userDefaults = UserDefaults(suiteName: SignalApplicationGroup) else {
|
||||||
|
owsFail("failed to initialize user defaults")
|
||||||
|
}
|
||||||
|
return userDefaults
|
||||||
|
}
|
||||||
|
|
||||||
|
override init() { super.init() }
|
||||||
|
|
||||||
|
// MARK: - Unused in this extension
|
||||||
|
|
||||||
|
let isRTL = false
|
||||||
|
let isRunningTests = false
|
||||||
|
|
||||||
|
var mainWindow: UIWindow?
|
||||||
|
let frame: CGRect = .zero
|
||||||
|
let interfaceOrientation: UIInterfaceOrientation = .unknown
|
||||||
|
let reportedApplicationState: UIApplication.State = .background
|
||||||
|
let statusBarHeight: CGFloat = .zero
|
||||||
|
|
||||||
|
func beginBackgroundTask(expirationHandler: @escaping BackgroundTaskExpirationHandler) -> UInt { 0 }
|
||||||
|
func endBackgroundTask(_ backgroundTaskIdentifier: UInt) {}
|
||||||
|
|
||||||
|
func beginBackgroundTask(expirationHandler: @escaping BackgroundTaskExpirationHandler) -> UIBackgroundTaskIdentifier { .invalid }
|
||||||
|
func endBackgroundTask(_ backgroundTaskIdentifier: UIBackgroundTaskIdentifier) {}
|
||||||
|
|
||||||
|
func ensureSleepBlocking(_ shouldBeBlocking: Bool, blockingObjectsDescription: String) {}
|
||||||
|
|
||||||
|
func setMainAppBadgeNumber(_ value: Int) {}
|
||||||
|
func setStatusBarHidden(_ isHidden: Bool, animated isAnimated: Bool) {}
|
||||||
|
|
||||||
|
func frontmostViewController() -> UIViewController? { nil }
|
||||||
|
|
||||||
|
func setNetworkActivityIndicatorVisible(_ value: Bool) {}
|
||||||
|
|
||||||
|
func runNowOr(whenMainAppIsActive block: @escaping AppActiveBlock) {}
|
||||||
|
|
||||||
|
func ensureSleepBlocking(_ shouldBeBlocking: Bool, blockingObjects: [Any]) {}
|
||||||
|
}
|
Loading…
Reference in New Issue