What doesn't work? If you add the @UIApplicationDelegateAdaptor private var appDelegate: AppDelegate entry to your App, and then create an AppDelegate.swift file that implements NSObject and UIApplicationDelegate then the push notification delegate methods will get called.
Inside the application(_:didFinishLaunchingWithOptions:) you still have to then register for notifications like normal.