iOS 通知ダイアログを許可したタイミングでviewDidLoad()を再度読ませたい

Question

表題の通りです。

webviewでデフォルトのURLを指定しているのですが、初回起動時の通知ダイアログがある場合にルートパス（/）に飛んでしまいます。

想定している動きとしてはルートパスのあとにtokenを埋め込む形です（/tokenの文字列〜）。

【ViewController.swift】

    override func viewDidLoad() {
        super.viewDidLoad()
        // Do any additional setup after loading the view, typically from a nib.
        
        let appDelegate:AppDelegate = UIApplication.shared.delegate as! AppDelegate
        appDelegate.viewController = self
        
        let fcmToken = UserDefaults.standard.string(forKey: "fcmToken") ?? ""
        print("トークン取得情報: (fcmToken)")

        let urlString = "localhost:8080/(fcmToken)"
        let urlRequest = URLRequest(url: URL(string: urlString)!)
        self.browserWebView.load(urlRequest)
    }

    @IBAction func goBack(_ sender: Any) {
        self.browserWebView.goBack()
    }
    @IBAction func goForward(_ sender: Any) {
        self.browserWebView.goForward()
    }
    @IBAction func reload(_ sender: Any) {
        self.browserWebView.reload()
    }
}

これを回避するためにAppDelegateのapplicationDidBecomeActiveでviewDidLoad()を再読込するように処理を入れました。

【AppDelegate.swift】

import UIKit
import Firebase
import FirebaseMessaging
import UserNotifications

@UIApplicationMain
class AppDelegate: UIResponder, UIApplicationDelegate {

    var window: UIWindow?
    var viewController: ViewController!


    func application(_ application: UIApplication, didFinishLaunchingWithOptions launchOptions: [UIApplication.LaunchOptionsKey: Any]?) -> Bool {
        // Override point for customization after application launch.
        FirebaseApp.configure()
        
        Messaging.messaging().delegate = self
        
        if #available(iOS 10.0, *) {
            // For iOS 10 display notification (sent via APNS)
            UNUserNotificationCenter.current().delegate = self as? UNUserNotificationCenterDelegate
            
            let authOptions: UNAuthorizationOptions = [.alert, .badge, .sound]
            UNUserNotificationCenter.current().requestAuthorization(
                options: authOptions,
                completionHandler: {_, _ in })
        } else {
            let settings: UIUserNotificationSettings =
                UIUserNotificationSettings(types: [.alert, .badge, .sound], categories: nil)
            application.registerUserNotificationSettings(settings)
        }
        
        application.registerForRemoteNotifications()
        
        let defaults = UserDefaults.standard
        let dic = ["firstLaunch": true]
        defaults.register(defaults: dic)
    
        return true
    }

    func applicationWillResignActive(_ application: UIApplication) {
        // Sent when the application is about to move from active to inactive state. This can occur for certain types of temporary interruptions (such as an incoming phone call or SMS message) or when the user quits the application and it begins the transition to the background state.
        // Use this method to pause ongoing tasks, disable timers, and invalidate graphics rendering callbacks. Games should use this method to pause the game.
    }

    func applicationDidEnterBackground(_ application: UIApplication) {
        // Use this method to release shared resources, save user data, invalidate timers, and store enough application state information to restore your application to its current state in case it is terminated later.
        // If your application supports background execution, this method is called instead of applicationWillTerminate: when the user quits.
    }

    func applicationWillEnterForeground(_ application: UIApplication) {
        // Called as part of the transition from the background to the active state; here you can undo many of the changes made on entering the background.
    }

    func applicationDidBecomeActive(_ application: UIApplication) {
        // Restart any tasks that were paused (or not yet started) while the application was inactive. If the application was previously in the background, optionally refresh the user interface.
        
        let defaults = UserDefaults.standard
        if defaults.bool(forKey: "firstLaunch") {
            // Some Process will be here
            let appDelegate = UIApplication.shared.delegate as! AppDelegate
            appDelegate.viewController?.viewDidLoad()

            // off the flag to know if it is first time to launch
            defaults.set(false, forKey: "firstLaunch")
        }
    }

    func applicationWillTerminate(_ application: UIApplication) {
        // Called when the application is about to terminate. Save data if appropriate. See also applicationDidEnterBackground:.
    }


}

extension AppDelegate : MessagingDelegate {
    // [START refresh_token]
    func messaging(_ messaging: Messaging, didReceiveRegistrationToken fcmToken: String) {
        print("ファイアベースに登録するやつ: (fcmToken)")
        
        let dataDict:[String: String] = ["token": fcmToken]
        NotificationCenter.default.post(name: Notification.Name("FCMToken"), object: nil, userInfo: dataDict)
        
        //save token
        UserDefaults.standard.set(fcmToken, forKey: "fcmToken")
    }
    // [END refresh_token]
    // [START ios_10_data_message]
    // Receive data messages on iOS 10+ directly from FCM (bypassing APNs) when the app is in the foreground.
    // To enable direct data messages, you can set Messaging.messaging().shouldEstablishDirectChannel to true.
    func messaging(_ messaging: Messaging, didReceive remoteMessage: MessagingRemoteMessage) {
        print("Received data message: (remoteMessage.appData)")
    }
    // [END ios_10_data_message]
}

シミュレーター、ipad実機でうまく動作（ダイアログを閉じるとviewDidLoad()が再読込された）したので問題ないなと思っていたのですが、どうやらapplicationDidBecomeActiveが呼ばれるときと呼ばれないときがあるようでうまく動作しなくなってしまいました。

手元のシミュレーター、ipadでも確認しましたが、うまくいくときもあればうまくいかないときもあって頭を抱えています。
※アプリをアンインストール→ビルドの繰り返しで確認

こういった問題があるので、通知ダイアログで許可（Allow）をしたタイミングでviewDidLoad()を読み直したいのですが、どういったコードをどこに書けばいいのか分かりませんでした。
（https://qiita.com/tokorom/items/6c6864fa8e841e50e37bの記事を参考にしました）

ダイアログの処理について詳しい方にご教示をいただきたいです、よろしくお願いいたします。
ダイアログは関係ありませんでした。

最初のアプリ立ち上げ時にfcmTokenの取得前にwebViewを読み込んでいるのが原因でした。
AppDelegateでfcmTokenを取得している箇所をviewDidLoadの前に読み込むことができれば解決すると思うのですが、可能でしょうか？

追記：起動ログのスクリーンショット

Answer 1

生成されたときに呼ばれる viewDidLoad ではなく、
画面に表示される直前直後に呼ばれる viewWillAppear, viewDidAppear を利用すれば良いと考えます。

特定の処理の後に何かをしたい場合。
【Swift】NotificationCenterの使い方

Comments

[oooeee]

dsuzuki様
ご回答ありがとうございます。
viewDidLoadの処理をviewWillAppear, viewDidAppearでそれぞれ試してみましたがうまくいきませんでした。

そこで、調査したところAppDelegate.swiftで定義しているfcmTokenの取得が、viewDidLoad()のあとで読み込まれていました。
※ログのスクショを質問に貼ります

delegateのfcmTokenの取得をviewDidLoadまたはiewWillAppear, viewDidAppearの前に読み込ませることはできないでしょうか？

[dsuzuki]

fcmTokenの取得をviewDidLoad等の前に呼ばせることを考えるよりも、NotificationCenterを利用して、fcmTokenが呼ばれたことをViewControllerに通知して、webViewを更新する方法はどうでしょうか。

[oooeee]

dsuzuki様

そのような方法があるのですね！調べて試してみたいと思います。