From 42e647c9321913593c6718a56dce09136099ced0 Mon Sep 17 00:00:00 2001
From: fish <zhongpengapp@hotmail.com>
Date: Wed, 15 Jul 2026 22:49:26 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E5=90=8E=E5=8F=B0=E8=BF=94?=
 =?UTF-8?q?=E5=9B=9E=E9=A6=96=E9=A1=B5=E6=94=B6=E9=9B=86=E7=AE=B1=E6=95=B0?=
 =?UTF-8?q?=E9=87=8F=E4=B8=8D=E5=88=B7=E6=96=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 Note/ViewController.swift | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/Note/ViewController.swift b/Note/ViewController.swift
index 58a7151..d6469f2 100644
--- a/Note/ViewController.swift
+++ b/Note/ViewController.swift
@@ -35,6 +35,7 @@ class ViewController: UIViewController {
         super.viewDidLoad()
         menuItems = [loadMenuOrder()]
         setupUI()
+        setupAppStateObserver()
     }
 
     override func viewWillAppear(_ animated: Bool) {
@@ -42,6 +43,23 @@ class ViewController: UIViewController {
         updateInboxBadge()
     }
 
+    deinit {
+        NotificationCenter.default.removeObserver(self)
+    }
+
+    private func setupAppStateObserver() {
+        NotificationCenter.default.addObserver(
+            self,
+            selector: #selector(appDidBecomeActive),
+            name: UIApplication.didBecomeActiveNotification,
+            object: nil
+        )
+    }
+
+    @objc private func appDidBecomeActive() {
+        updateInboxBadge()
+    }
+
     private func setupUI() {
         title = "首页"
         navigationController?.navigationBar.prefersLargeTitles = true