From 88b00f3fc74446a1727c93722c7b64179b45a9db Mon Sep 17 00:00:00 2001
From: zhujie <leon.zhu@cloudroam.com.cn>
Date: 星期五, 18 四月 2025 17:40:45 +0800
Subject: [PATCH] Merge branch 'master' of http://47.96.225.205:8888/r/FirstApp2

---
 app/src/main/java/com/example/firstapp/ui/home/HomeFragment.kt |  458 +++++++++++++++++++++++++++++++++++++++++++++++----------
 1 files changed, 376 insertions(+), 82 deletions(-)

diff --git a/app/src/main/java/com/example/firstapp/ui/home/HomeFragment.kt b/app/src/main/java/com/example/firstapp/ui/home/HomeFragment.kt
index 481756d..743a5ab 100644
--- a/app/src/main/java/com/example/firstapp/ui/home/HomeFragment.kt
+++ b/app/src/main/java/com/example/firstapp/ui/home/HomeFragment.kt
@@ -13,32 +13,42 @@
 import androidx.core.content.ContextCompat
 import androidx.fragment.app.Fragment
 import androidx.lifecycle.ViewModelProvider
+import androidx.lifecycle.lifecycleScope
 import androidx.recyclerview.widget.LinearLayoutManager
-import com.bumptech.glide.Glide
 import com.example.firstapp.R
 import com.example.firstapp.activity.PickupActivity
+import com.example.firstapp.activity.VipActivity
 import com.example.firstapp.adapter.ExpressAdapter
 import com.example.firstapp.adapter.FinanceAdapter
 import com.example.firstapp.adapter.CategorySelectorAdapter
+import com.example.firstapp.adapter.IncomeAdapter
+import com.example.firstapp.adapter.TrainAdapter
+import com.example.firstapp.adapter.FlightAdapter
+import com.example.firstapp.database.service.RetrofitClient
 import com.example.firstapp.databinding.FragmentHomeBinding
 import com.example.firstapp.databinding.DialogCategorySelectorBinding
+import com.example.firstapp.model.CategoryConfig
+import com.example.firstapp.utils.PreferencesManager
 import com.google.android.material.bottomsheet.BottomSheetDialog
+import kotlinx.coroutines.launch
+import com.example.firstapp.view.UnderlineTextView
+import com.example.firstapp.activity.ReminderListActivity
 
 class HomeFragment : Fragment() {
 
     private var _binding: FragmentHomeBinding? = null
 
-    // This property is only valid between onCreateView and
-    // onDestroyView.
     private val binding get() = _binding!!
 
     private lateinit var homeViewModel: HomeViewModel
     private lateinit var expressAdapter: ExpressAdapter
     private lateinit var financeAdapter: FinanceAdapter
-    private lateinit var incomeAdapter: FinanceAdapter
-    private lateinit var flightAdapter: FinanceAdapter
-    private lateinit var trainAdapter: FinanceAdapter
+    private lateinit var incomeAdapter: IncomeAdapter
+    private lateinit var flightAdapter: FlightAdapter
+    private lateinit var trainAdapter: TrainAdapter
     private lateinit var dataUpdateReceiver: BroadcastReceiver
+    private lateinit var reminderUpdateReceiver: BroadcastReceiver
+    private var reminderBadge: TextView? = null
 
     //onCreateView这个方法创建后被调用,通常是初始化视图组件和观察者
     override fun onCreateView(
@@ -54,17 +64,37 @@
         super.onViewCreated(view, savedInstanceState)
 
         homeViewModel = ViewModelProvider(this).get(HomeViewModel::class.java)
-        // 假设从某处获取用户ID
-//        val userId = getUserId() // 需要实现这个方法
-        val userId ="123456"
+        val userId = "123456"
         homeViewModel.initialize(requireContext(), userId)
-        
-        //调用这个方法来设置 RecyclerView用于设置 RecyclerView 的布局和适配器。
-        setupRecyclerViews()
+
+        // 检查是否是首次安装
+        val isFirstInstall = PreferencesManager.isFirstInstall()
+        if (isFirstInstall) {
+            // 首次安装,设置默认显示快递和还款
+            val defaultCategories = listOf(
+                CategoryConfig(
+                    id = 1,
+                    name = "快递",
+                    order = 1,
+                    isEnabled = true
+                ),
+                CategoryConfig(
+                    id = 2,
+                    name = "还款",
+                    order = 2,
+                    isEnabled = true
+                )
+            )
+            homeViewModel.saveCategories(defaultCategories)
+            // 标记为非首次安装
+            PreferencesManager.setFirstInstall(false)
+        }
+
+        setupAdapters()
         setupTabSwitching()
-        //调用这个方法来观察 ViewModel 中的数据变化
-        observeViewModelData()
+        setupObservers()
         setupCategorySelector()
+        setupUnreadBadge()
     }
 
     override fun onCreate(savedInstanceState: Bundle?) {
@@ -79,9 +109,20 @@
                 }
             }
         }
+
+        // 创建提醒更新广播接收器
+        reminderUpdateReceiver = object : BroadcastReceiver() {
+            override fun onReceive(context: Context, intent: Intent) {
+                if (intent.action == "com.example.firstapp.REMINDER_UPDATED") {
+                    // 收到提醒更新广播时重新检查未读提醒数量
+                    homeViewModel.checkUnreadReminders()
+                }
+            }
+        }
     }
 
-    private fun setupRecyclerViews() {
+
+    private fun setupAdapters() {
         binding.expressRecycler.apply {
             layoutManager = LinearLayoutManager(context)
             expressAdapter = ExpressAdapter()
@@ -89,10 +130,24 @@
 
             // 设置点击监听
             expressAdapter.setOnPackageClickListener { group, pack ->
+                // 根据驿站名称设置相应的图标
+                val stationName = group.stationName.lowercase()
+                val stationIconResId = when {
+                    stationName.isEmpty() || stationName == "未知" -> R.drawable.express_normal
+                    stationName.contains("菜鸟") -> R.drawable.express_cainiao
+                    stationName.contains("多多") -> R.drawable.express_duoduo
+                    stationName.contains("兔喜") -> R.drawable.express_tuxi
+                    stationName.contains("妈妈") -> R.drawable.express_mama
+                    stationName.contains("小兵") -> R.drawable.express_xiaobing
+                    stationName.contains("蓝店") -> R.drawable.express_landian
+                    else -> R.drawable.express_normal
+                }
                 // 跳转到取件页面
                 val intent = Intent(requireContext(), PickupActivity::class.java).apply {
                     putExtra("station_name", group.stationName)
                     putExtra("company", pack.company)
+                    putExtra("page_type", PickupActivity.TYPE_EXPRESS)
+                    putExtra("station_icon_res_id", stationIconResId) // 添加图标资源ID
                 }
                 startActivity(intent)
             }
@@ -109,10 +164,29 @@
 
             // 设置点击监听
             financeAdapter.setOnPackageClickListener { group, pack ->
+                val stationName = group.stationName.lowercase()
+                val stationIconResId = when {
+                    stationName.isEmpty() || stationName == "未知" -> R.drawable.finance_huankuan_normal
+                    stationName.contains("安逸花") -> R.drawable.finance_anyihua
+                    stationName.contains("房贷") -> R.drawable.finance_fangdai
+                    stationName.contains("还款") -> R.drawable.finance_huankuan
+                    stationName.contains("花呗") -> R.drawable.finance_huabei
+                    stationName.contains("借呗") -> R.drawable.finance_jiebei
+                    stationName.contains("京东白条") -> R.drawable.finance_jingdongbaitiao
+                    stationName.contains("京东金条") -> R.drawable.finance_jingdongjintiao
+                    stationName.contains("汽车分期") -> R.drawable.finance_qichefenqi
+                    stationName.contains("省呗") -> R.drawable.finance_shenbei
+                    stationName.contains("信用卡") -> R.drawable.finance_xingyongka
+                    stationName.contains("支付宝") -> R.drawable.finance_zhifubao
+                    else -> R.drawable.finance_huankuan_normal
+                }
                 // 跳转到取件页面
                 val intent = Intent(requireContext(), PickupActivity::class.java).apply {
                     putExtra("station_name", group.stationName)
                     putExtra("company", pack.company)
+                    putExtra("page_type", PickupActivity.TYPE_REPAYMENT)
+                    putExtra("station_icon_res_id", stationIconResId) // 添加图标资源ID
+
                 }
                 startActivity(intent)
             }
@@ -121,23 +195,85 @@
         // 添加新的 RecyclerView
         binding.incomeRecycler.apply {
             layoutManager = LinearLayoutManager(context)
-            incomeAdapter = FinanceAdapter()
+            incomeAdapter = IncomeAdapter()
             adapter = incomeAdapter
-            visibility = View.GONE
+
+            // 设置初始状态 - 添加这行
+            binding.incomeRecycler.visibility = View.GONE
+
+            // 设置点击监听
+            incomeAdapter.setOnPackageClickListener { group, pack ->
+
+                val stationName = group.stationName.lowercase()
+                val stationIconResId = when {
+                    stationName.isEmpty() || stationName == "未知" -> R.drawable.finance_income_bank_normal
+                    stationName.contains("工商银行") -> R.drawable.bank_gongshang
+                    stationName.contains("光大银行") -> R.drawable.bank_guangda
+                    stationName.contains("国家开发银行") -> R.drawable.bank_guojiakaifa
+                    stationName.contains("恒丰银行") -> R.drawable.bank_hengfeng
+                    stationName.contains("华夏银行") -> R.drawable.bank_huaxia
+                    stationName.contains("花呗分期") -> R.drawable.bank_huabeifenqi
+                    stationName.contains("建设银行") -> R.drawable.bank_jianshe
+                    stationName.contains("交通银行") -> R.drawable.bank_jiaotong
+                    stationName.contains("民生银行") -> R.drawable.bank_mingsheng
+                    stationName.contains("南京银行") -> R.drawable.bank_nanjing
+                    stationName.contains("农商银行") -> R.drawable.bank_nongshang
+                    stationName.contains("农业银行") -> R.drawable.bank_nongye
+                    stationName.contains("平安银行") -> R.drawable.bank_pingan
+                    stationName.contains("浦发银行") -> R.drawable.bank_pufa
+                    stationName.contains("苏州银行") -> R.drawable.bank_suzhou
+                    stationName.contains("邮储银行") -> R.drawable.bank_youzheng
+                    stationName.contains("招商银行") -> R.drawable.bank_zhaoshang
+                    stationName.contains("中国银行") -> R.drawable.bank_zhongguo
+                    stationName.contains("中信银行") -> R.drawable.bank_zhongxin
+                    stationName.contains("中原银行") -> R.drawable.bank_zhongyuan
+                    else -> R.drawable.finance_income_bank_normal
+                }
+                // 跳转到取件页面
+                val intent = Intent(requireContext(), PickupActivity::class.java).apply {
+                    putExtra("station_name", group.stationName)
+                    putExtra("company", pack.company)
+                    putExtra("page_type", PickupActivity.TYPE_INCOME)
+                    putExtra("station_icon_res_id", stationIconResId) // 添加图标资源ID
+                }
+                startActivity(intent)
+            }
         }
 
         binding.flightRecycler.apply {
             layoutManager = LinearLayoutManager(context)
-            flightAdapter = FinanceAdapter()
+            flightAdapter = FlightAdapter()
             adapter = flightAdapter
             visibility = View.GONE
+
+            // 设置点击监听
+            flightAdapter.setOnPackageClickListener { group, pack ->
+                // 跳转到航班处理页面
+                val intent = Intent(requireContext(), PickupActivity::class.java).apply {
+                    putExtra("station_name", group.stationName)
+                    putExtra("company", pack.company)
+                    putExtra("page_type", PickupActivity.TYPE_FLIGHT)
+                }
+                startActivity(intent)
+            }
         }
 
         binding.trainRecycler.apply {
             layoutManager = LinearLayoutManager(context)
-            trainAdapter = FinanceAdapter()
+            trainAdapter = TrainAdapter()
             adapter = trainAdapter
             visibility = View.GONE
+
+            // 设置点击监听
+            trainAdapter.setOnPackageClickListener { group, pack ->
+                // 跳转到火车票页面
+                val intent = Intent(requireContext(), PickupActivity::class.java).apply {
+                    putExtra("station_name", group.stationName)
+                    putExtra("company", pack.company)
+                    putExtra("page_type", PickupActivity.TYPE_TRAIN)
+                }
+                startActivity(intent)
+            }
         }
     }
 
@@ -154,8 +290,8 @@
                 updateTabStyles(tabExpress)
                 homeViewModel.loadExpressData()
             }
-            
-            // 财务标签点击事件
+
+            // 还款标签点击事件 - 非会员也可以使用
             tabFinance.setOnClickListener {
                 hideAllRecyclers()
                 financeRecycler.visibility = View.VISIBLE
@@ -163,25 +299,58 @@
                 homeViewModel.loadFinanceData()
             }
 
-            tabIncome.setOnClickListener {
-                hideAllRecyclers()
-                incomeRecycler.visibility = View.VISIBLE
-                updateTabStyles(tabIncome)
-                homeViewModel.loadIncomeData()
-            }
+            // 其他标签点击事件需要检查会员状态
+            val memberOnlyTabs = mapOf(
+                tabIncome to { homeViewModel.loadIncomeData() },
+                tabFlight to { homeViewModel.loadFlightData() },
+                tabTrain to { homeViewModel.loadTrainData() }
+            )
 
-            tabFlight.setOnClickListener {
-                hideAllRecyclers()
-                flightRecycler.visibility = View.VISIBLE
-                updateTabStyles(tabFlight)
-                homeViewModel.loadFlightData()
+            memberOnlyTabs.forEach { (tab, loadAction) ->
+                tab.setOnClickListener {
+                    checkMembershipAndExecute(tab) {
+                        hideAllRecyclers()
+                        when (tab) {
+                            tabIncome -> incomeRecycler.visibility = View.VISIBLE
+                            tabFlight -> flightRecycler.visibility = View.VISIBLE
+                            tabTrain -> trainRecycler.visibility = View.VISIBLE
+                        }
+                        updateTabStyles(tab)
+                        loadAction()
+                    }
+                }
             }
+        }
+    }
 
-            tabTrain.setOnClickListener {
-                hideAllRecyclers()
-                trainRecycler.visibility = View.VISIBLE
-                updateTabStyles(tabTrain)
-                homeViewModel.loadTrainData()
+    private fun checkMembershipAndExecute(tab: TextView, action: () -> Unit) {
+        // 从本地获取保存的手机号
+        val savedPhone = PreferencesManager.getPhone()
+        if (savedPhone.isNullOrEmpty()) {
+            Toast.makeText(requireContext(), "请先登录", Toast.LENGTH_SHORT).show()
+            return
+        }
+
+        // 使用协程检查会员状态
+        lifecycleScope.launch {
+            try {
+                val response = RetrofitClient.apiService.getUserInfo(savedPhone)
+                if (response.code == "0" && response.data != null) {
+                    if (response.data.isMember) {
+                        action()
+                    } else {
+                        Toast.makeText(requireContext(), "该功能仅对会员开放", Toast.LENGTH_SHORT).show()
+                        // 切回快递标签
+                        binding.tabExpress.performClick()
+                    }
+                } else {
+                    Toast.makeText(requireContext(), "获取用户信息失败", Toast.LENGTH_SHORT).show()
+                    binding.tabExpress.performClick()
+                }
+            } catch (e: Exception) {
+                e.printStackTrace()
+                Toast.makeText(requireContext(), "网络错误,请稍后重试", Toast.LENGTH_SHORT).show()
+                binding.tabExpress.performClick()
             }
         }
     }
@@ -196,19 +365,22 @@
         }
     }
 
-    private fun updateTabStyles(selectedTab: TextView) {
+    private fun updateTabStyles(selectedTab: UnderlineTextView) {
         binding.apply {
             val tabs = listOf(tabExpress, tabFinance, tabIncome, tabFlight, tabTrain)
             tabs.forEach { tab ->
-                tab.setTextColor(ContextCompat.getColor(requireContext(), 
-                    if (tab == selectedTab) R.color.tab_selected else R.color.gray))
+                // 设置文字颜色为黑色或灰色
+                tab.setTextColor(ContextCompat.getColor(requireContext(),
+                    if (tab == selectedTab) android.R.color.black else R.color.gray))
+                // 设置文字大小
                 tab.textSize = if (tab == selectedTab) 16f else 14f
+                // 设置下划线
+                tab.setUnderlineVisible(tab == selectedTab)
             }
         }
     }
 
-    //这个方法用于观察 homeViewModel 中的 expressItems 数据。
-    private fun observeViewModelData() {
+    private fun setupObservers() {
         //当 expressItems 数据发生变化时,更新 RecyclerView 的数据。
         homeViewModel.expressItems.observe(viewLifecycleOwner) { items ->
             //将新的数据列表提交给适配器,以更新 RecyclerView 的显示内容。
@@ -219,6 +391,7 @@
             financeAdapter.submitList(items)
         }
 
+        // 观察收入数据变化
         homeViewModel.incomeItems.observe(viewLifecycleOwner) { items ->
             incomeAdapter.submitList(items)
         }
@@ -231,8 +404,13 @@
             trainAdapter.submitList(items)
         }
 
+        // 观察未读提醒数量变化
+        homeViewModel.unreadReminderCount.observe(viewLifecycleOwner) { unreadCount ->
+            updateReminderBadge(unreadCount)
+        }
+
         // 观察可见分类的变化
-        homeViewModel.visibleCategories.observe(viewLifecycleOwner) { categories: List<String> ->
+        homeViewModel.visibleCategories.observe(viewLifecycleOwner) { categories: List<CategoryConfig> ->
             binding.apply {
                 // 隐藏所有标签
                 tabExpress.visibility = View.GONE
@@ -241,29 +419,55 @@
                 tabFlight.visibility = View.GONE
                 tabTrain.visibility = View.GONE
 
-                // 根据选中的分类显示对应的标签
-                categories.forEachIndexed { index: Int, categoryName: String ->
-                    when (categoryName) {
-                        "快递" -> {
+                // 获取用户信息判断是否是会员
+                val savedPhone = PreferencesManager.getPhone()
+                lifecycleScope.launch {
+                    try {
+                        val response = RetrofitClient.apiService.getUserInfo(savedPhone ?: "")
+                        val isMember = response.code == "0" && response.data?.isMember == true
+
+                        if (!isMember) {
+                            // 非会员只显示快递和还款
                             tabExpress.visibility = View.VISIBLE
-                            if (index == 0) tabExpress.performClick()
-                        }
-                        "还款" -> {
                             tabFinance.visibility = View.VISIBLE
-                            if (index == 0) tabFinance.performClick()
+                            if (categories.firstOrNull()?.name == "快递") {
+                                tabExpress.performClick()
+                            } else {
+                                tabFinance.performClick()
+                            }
+                        } else {
+                            // 会员显示所有选中的分类
+                            categories.forEach { category ->
+                                when (category.name) {
+                                    "快递" -> {
+                                        tabExpress.visibility = if (category.isEnabled) View.VISIBLE else View.GONE
+                                        if (categories.indexOf(category) == 0 && category.isEnabled) tabExpress.performClick()
+                                    }
+                                    "还款" -> {
+                                        tabFinance.visibility = if (category.isEnabled) View.VISIBLE else View.GONE
+                                        if (categories.indexOf(category) == 0 && category.isEnabled) tabFinance.performClick()
+                                    }
+                                    "收入" -> {
+                                        tabIncome.visibility = if (category.isEnabled) View.VISIBLE else View.GONE
+                                        if (categories.indexOf(category) == 0 && category.isEnabled) tabIncome.performClick()
+                                    }
+                                    "航班" -> {
+                                        tabFlight.visibility = if (category.isEnabled) View.VISIBLE else View.GONE
+                                        if (categories.indexOf(category) == 0 && category.isEnabled) tabFlight.performClick()
+                                    }
+                                    "火车票" -> {
+                                        tabTrain.visibility = if (category.isEnabled) View.VISIBLE else View.GONE
+                                        if (categories.indexOf(category) == 0 && category.isEnabled) tabTrain.performClick()
+                                    }
+                                }
+                            }
                         }
-                        "收入" -> {
-                            tabIncome.visibility = View.VISIBLE
-                            if (index == 0) tabIncome.performClick()
-                        }
-                        "航班" -> {
-                            tabFlight.visibility = View.VISIBLE
-                            if (index == 0) tabFlight.performClick()
-                        }
-                        "火车票" -> {
-                            tabTrain.visibility = View.VISIBLE
-                            if (index == 0) tabTrain.performClick()
-                        }
+                    } catch (e: Exception) {
+                        e.printStackTrace()
+                        // 发生错误时默认显示快递和还款
+                        tabExpress.visibility = View.VISIBLE
+                        tabFinance.visibility = View.VISIBLE
+                        tabExpress.performClick()
                     }
                 }
             }
@@ -279,8 +483,26 @@
             IntentFilter("com.example.firstapp.DATA_UPDATED"),
             ContextCompat.RECEIVER_NOT_EXPORTED
         )
+
+        // 注册提醒更新广播接收器
+        ContextCompat.registerReceiver(
+            requireContext(),
+            reminderUpdateReceiver,
+            IntentFilter("com.example.firstapp.REMINDER_UPDATED"),
+            ContextCompat.RECEIVER_NOT_EXPORTED
+        )
+
         // 加载数据
-        homeViewModel.loadExpressData()
+        homeViewModel.loadAllCategoryData()
+//        homeViewModel.loadExpressData()
+//        homeViewModel.loadFinanceData()
+//        homeViewModel.loadIncomeData()
+//        homeViewModel.loadFlightData()
+//        homeViewModel.loadTrainData()
+
+        // 检查未读提醒数量
+        homeViewModel.checkUnreadReminders()
+
     }
 
     override fun onPause() {
@@ -288,6 +510,7 @@
         try {
             // 取消注册广播接收器
             requireContext().unregisterReceiver(dataUpdateReceiver)
+            requireContext().unregisterReceiver(reminderUpdateReceiver)
         } catch (e: Exception) {
             // 处理可能的异常
             e.printStackTrace()
@@ -299,26 +522,55 @@
         _binding = null
     }
 
-    private fun loadAdvertisements() {
-        // 使用 Glide 加载网络图片
-        Glide.with(this)
-            .load("http://192.168.1.235:9999/advertisement/up.png")
-            .into(binding.adBanner)
+//    private fun loadAdvertisements() {
+//        // 使用 Glide 加载网络图片
+//        Glide.with(this)
+//            .load("http://192.168.1.235:9999/advertisement/up.png")
+//            .into(binding.adBanner)
+//
+//        Glide.with(this)
+//            .load("http://192.168.1.235:9999/advertisement/down.png")
+//            .into(binding.bottomAdBanner)
+//    }
 
-        Glide.with(this)
-            .load("http://192.168.1.235:9999/advertisement/down.png")
-            .into(binding.bottomAdBanner)
-    }
-
+    // 设置分类选择器 检查会员状态
     private fun setupCategorySelector() {
         binding.categoryButton.setOnClickListener {
-            // TODO: 检查会员状态
-            if (true) { // 临时设置为true,实际应该检查会员状态
-                showCategorySelectorDialog()
-            } else {
-                // 显示会员提示
-                Toast.makeText(requireContext(), "该功能仅对会员开放", Toast.LENGTH_SHORT).show()
+            // 从本地获取保存的手机号
+            val savedPhone = PreferencesManager.getPhone()
+            if (savedPhone.isNullOrEmpty()) {
+                Toast.makeText(requireContext(), "请先登录", Toast.LENGTH_SHORT).show()
+                return@setOnClickListener
             }
+
+            // 使用协程检查会员状态
+            lifecycleScope.launch {
+                try {
+                    val response = RetrofitClient.apiService.getUserInfo(savedPhone)
+                    if (response.code == "0" && response.data != null) {
+                        if (response.data.isMember) {
+                            showCategorySelectorDialog()
+                        } else {
+                            // 非会员跳转到VIP开通页面
+                            val intent = Intent(requireContext(), VipActivity::class.java)
+                            startActivity(intent)
+                        }
+                    } else {
+                        Toast.makeText(requireContext(), "获取用户信息失败", Toast.LENGTH_SHORT).show()
+                    }
+                } catch (e: Exception) {
+                    e.printStackTrace()
+                    Toast.makeText(requireContext(), "网络错误,请稍后重试", Toast.LENGTH_SHORT).show()
+                }
+            }
+        }
+
+        // 添加提醒按钮点击事件
+        binding.reminderButton.setOnClickListener {
+            // 跳转到提醒列表页面
+            startActivity(Intent(requireContext(), ReminderListActivity::class.java))
+            // 重新检查未读提醒数量
+            homeViewModel.checkUnreadReminders()
         }
     }
 
@@ -333,9 +585,33 @@
             this.adapter = adapter
         }
 
-        // 加载现有分类
+        // 加载所有分类
         homeViewModel.categories.observe(viewLifecycleOwner) { categories ->
-            adapter.setCategories(categories)
+            // 如果是会员,显示所有分类供选择
+            lifecycleScope.launch {
+                try {
+                    val savedPhone = PreferencesManager.getPhone()
+                    val response = RetrofitClient.apiService.getUserInfo(savedPhone ?: "")
+                    val isMember = response.code == "0" && response.data?.isMember == true
+
+                    if (isMember) {
+                        // 会员可以看到所有分类
+                        adapter.setCategories(categories)
+                    } else {
+                        // 非会员只能看到快递和还款
+                        val limitedCategories = categories.filter {
+                            it.name == "快递" || it.name == "还款"
+                        }
+                        adapter.setCategories(limitedCategories)
+                    }
+                } catch (e: Exception) {
+                    // 发生错误时只显示基础分类
+                    val limitedCategories = categories.filter {
+                        it.name == "快递" || it.name == "还款"
+                    }
+                    adapter.setCategories(limitedCategories)
+                }
+            }
         }
 
         dialogBinding.saveButton.setOnClickListener {
@@ -345,4 +621,22 @@
 
         dialog.show()
     }
+
+    // 添加设置未读提醒徽章的方法
+    private fun setupUnreadBadge() {
+        // 直接使用布局中定义的小红点
+        reminderBadge = binding.reminderBadge
+    }
+
+    // 更新未读提醒徽章
+    private fun updateReminderBadge(unreadCount: Int) {
+        reminderBadge?.apply {
+            if (unreadCount > 0) {
+                text = if (unreadCount > 99) "99+" else unreadCount.toString()
+                visibility = View.VISIBLE
+            } else {
+                visibility = View.GONE
+            }
+        }
+    }
 }
\ No newline at end of file

--
Gitblit v1.9.3