From f67cf3b81a00f732ca743431258ae6b78f5f40ab Mon Sep 17 00:00:00 2001
From: tj <1378534974@qq.com>
Date: 星期四, 17 四月 2025 15:05:28 +0800
Subject: [PATCH] 11、我的	切换头像  点击切换头像没有显示允许存储权限的窗口,华为的手机目前有 49、首页	实时刷新  点击全部取件或其他分类后,回到上一层,内容没有刷新(5个分类) 52、数据统计	数据统计  1.周月年的柱状图统计逻辑需要修改为只统计快递类的数据  2.按年的图形统计,右下方加上图示说明 53、首页	首页登录  点击用户协议、隐私政策无反应

---
 app/src/main/java/com/example/firstapp/ui/home/HomeViewModel.kt |  380 ++++++++++++++++++++++++++++++++++++++++++++++++++++--
 1 files changed, 364 insertions(+), 16 deletions(-)

diff --git a/app/src/main/java/com/example/firstapp/ui/home/HomeViewModel.kt b/app/src/main/java/com/example/firstapp/ui/home/HomeViewModel.kt
index 630b6af..e1357b3 100644
--- a/app/src/main/java/com/example/firstapp/ui/home/HomeViewModel.kt
+++ b/app/src/main/java/com/example/firstapp/ui/home/HomeViewModel.kt
@@ -1,36 +1,384 @@
 package com.example.firstapp.ui.home
 
+import android.content.Context
+import android.util.Log
 import androidx.lifecycle.LiveData
 import androidx.lifecycle.MutableLiveData
 import androidx.lifecycle.ViewModel
+import androidx.lifecycle.viewModelScope
 import com.example.firstapp.core.Core
-import com.example.firstapp.database.entity.Code
+import com.example.firstapp.database.service.RetrofitClient
+import com.example.firstapp.model.CategoryConfig
+import com.example.firstapp.model.CategoryConfigSync
+import com.example.firstapp.model.ExpressGroup
+import com.example.firstapp.model.ExpressPackage
+import com.example.firstapp.model.FinanceGroup
+import com.example.firstapp.model.FinancePackage
+import com.example.firstapp.model.IncomeGroup
+import com.example.firstapp.model.IncomePackage
+import com.example.firstapp.util.SecureStorage
+import com.example.firstapp.utils.PreferencesManager
+import kotlinx.coroutines.launch
+import com.example.firstapp.database.repository.ReminderRecordRepository
+import com.example.firstapp.database.entity.ReminderRecord
+import com.example.firstapp.model.TrainGroup
+import com.example.firstapp.model.TrainPackage
+import com.example.firstapp.model.FlightGroup
+import com.example.firstapp.model.FlightPackage
 
 class HomeViewModel : ViewModel() {
 
-    private val _text = MutableLiveData<String>().apply {
-        value = "短信主页面"
-    }
-    val text: LiveData<String> = _text
+    private val _expressItems = MutableLiveData<List<ExpressGroup>>()
+    private val _financeItems = MutableLiveData<List<FinanceGroup>>()
+    private val _incomeItems = MutableLiveData<List<IncomeGroup>>()
+    private val _flightItems = MutableLiveData<List<FlightGroup>>()
+    private val _trainItems = MutableLiveData<List<TrainGroup>>()
+    
+    val expressItems: LiveData<List<ExpressGroup>> = _expressItems
+    val financeItems: LiveData<List<FinanceGroup>> = _financeItems
+    val incomeItems: LiveData<List<IncomeGroup>> = _incomeItems
+    val flightItems: LiveData<List<FlightGroup>> = _flightItems
+    val trainItems: LiveData<List<TrainGroup>> = _trainItems
 
-    private val _codeList = MutableLiveData<List<Code>>()
+    private val _categories = MutableLiveData<List<CategoryConfig>>()
+    val categories: LiveData<List<CategoryConfig>> = _categories
 
-    val codeList: LiveData<List<Code>> get() = _codeList
+    private val _visibleCategories = MutableLiveData<List<CategoryConfig>>()
+    val visibleCategories: LiveData<List<CategoryConfig>> = _visibleCategories
+
+    private val _unreadReminderCount = MutableLiveData<Int>()
+    val unreadReminderCount: LiveData<Int> = _unreadReminderCount
+
+    private lateinit var secureStorage: SecureStorage
+    private var currentUserId: String = ""
+    private lateinit var reminderRecordRepository: ReminderRecordRepository
+    private var categoriesLoaded = false
 
     init {
-        // 初始化时加载数据
-        loadData()
+        // 初始化时加载包裹列表数据
+        loadExpressData()
+        // 初始化时不加载财务列表数据 0317
+        //  loadFinanceData()
     }
 
-    // 加载数据的方法
-    fun loadData() {
-        // 获取数据,并更新 LiveData
-        _codeList.value = Core.code.getAllDesc() // 假设这是获取最新的 data 的方法
+    fun initialize(context: Context, userId: String) {
+        secureStorage = SecureStorage(context)
+        currentUserId = userId
+        reminderRecordRepository = ReminderRecordRepository(context)
+        
+        // 只在首次加载或者用户修改分类后加载分类数据
+        if (!categoriesLoaded) {
+            loadCategories()
+            categoriesLoaded = true
+        }
+        
+        // 初始化时更新可见分类
+        _categories.value?.let { updateVisibleCategories(it) }
+        // 加载未读提醒数量
+        checkUnreadReminders()
     }
 
-    // 如果需要手动更新数据,可以调用这个方法
-    fun updateData() {
-        _codeList.value = Core.code.getAllDesc() // 重新获取并更新数据
+    private fun loadDataByType(type: String) {
+        viewModelScope.launch {
+            try {
+                // 获取该类型下的所有站点分组
+                val stations = Core.code.getStationsByType(type)
+                
+                when (type) {
+                    "快递" -> {
+                        // 处理快递类型
+                        val groups = stations.map { station ->
+                            val packages = Core.code.getPackagesByTypeAndStation(type, station.stationName).map { code ->
+                                ExpressPackage(
+                                    id = code.id,
+                                    company = code.secondLevel,
+                                    trackingNumber = code.code,
+                                    createTime = code.createTime
+                                )
+                            }
+                            ExpressGroup(stationName = station.stationName, packages = packages)
+                        }
+                        _expressItems.postValue(groups)
+                    }
+                    "收入" -> {
+                        // 特殊处理收入类型
+                        val groups = stations.map { station ->
+                            val packages = Core.code.getPackagesByTypeAndStation(type, station.stationName).map { code ->
+                                IncomePackage(
+                                    id = code.id,
+                                    company = code.secondLevel,  // 交易对方
+                                    trackingNumber = code.code,  // 金额
+                                    createTime = code.createTime,
+                                    balance = code.remarks.replace("余额", "") // 去掉"余额"前缀
+                                )
+                            }
+                            IncomeGroup(stationName = station.stationName, packages = packages)
+                        }
+                        _incomeItems.postValue(groups)
+                    }
+                    "火车票" -> {
+                        // 处理火车票类型
+                        val groups = stations.map { station ->
+                            val packages = Core.code.getPackagesByTypeAndStation(type, station.stationName).map { code ->
+                                TrainPackage(
+                                    id = code.id,
+                                    company = code.secondLevel,
+                                    trackingNumber = code.code,
+                                    createTime = code.createTime
+                                )
+                            }
+                            TrainGroup(stationName = station.stationName, packages = packages)
+                        }
+                        _trainItems.postValue(groups)
+                    }
+                    "航班" -> {
+                        // 处理航班类型
+                        val groups = stations.map { station ->
+                            val packages = Core.code.getPackagesByTypeAndStation(type, station.stationName).map { code ->
+                                FlightPackage(
+                                    id = code.id,
+                                    company = code.secondLevel,
+                                    trackingNumber = code.code,
+                                    createTime = code.createTime
+                                )
+                            }
+                            FlightGroup(stationName = station.stationName, packages = packages)
+                        }
+                        _flightItems.postValue(groups)
+                    }
+                    else -> {
+                        // 处理其他类型(还款)
+                        val groups = stations.map { station ->
+                            val packages = Core.code.getPackagesByTypeAndStation(type, station.stationName).map { code ->
+                                FinancePackage(
+                                    id = code.id,
+                                    company = code.secondLevel,
+                                    trackingNumber = code.code,
+                                    createTime = code.createTime
+                                )
+                            }
+                            FinanceGroup(stationName = station.stationName, packages = packages)
+                        }
+                        
+                        // 根据类型更新对应的 LiveData
+                        when (type) {
+                            "还款" -> _financeItems.postValue(groups)
+                        }
+                    }
+                }
+            } catch (e: Exception) {
+                Log.e("HomeViewModel", "Failed to load $type data: ${e.message}")
+            }
+        }
+    }
+
+    fun loadExpressData() {
+        loadDataByType("快递")
+    }
+
+    fun loadFinanceData() {
+        loadDataByType("还款")
+    }
+
+    fun loadIncomeData() {
+        loadDataByType("收入")
+    }
+
+    fun loadFlightData() {
+        loadDataByType("航班")
+    }
+
+    fun loadTrainData() {
+        loadDataByType("火车票")
+    }
+
+
+    fun loadAllCategoryData() {
+        getFullCategories().forEach { category ->
+            loadDataByType(category.name)
+        }
+    }
+
+
+    fun getFullCategories(): List<CategoryConfig> {
+        return listOf(
+            CategoryConfig(1, "快递", 0, true),
+            CategoryConfig(2, "还款", 1, true),
+            CategoryConfig(3, "收入", 2, true),
+            CategoryConfig(4, "航班", 3, true),
+            CategoryConfig(5, "火车票", 4, true)
+        )
+    }
+
+    private fun loadCategories() {
+        viewModelScope.launch {
+            try {
+                // 默认完整分类列表
+//                val fullCategories = listOf(
+//                    CategoryConfig(1, "快递", 0, true),
+//                    CategoryConfig(2, "还款", 1, true),
+//                    CategoryConfig(3, "收入", 2, true),
+//                    CategoryConfig(4, "航班", 3, true),
+//                    CategoryConfig(5, "火车票", 4, true)
+//                )
+
+                val fullCategories=getFullCategories()
+                
+                // 基础分类(非会员可见)
+                val basicCategories = listOf(
+                    CategoryConfig(1, "快递", 0, true),
+                    CategoryConfig(2, "还款", 1, true)
+                )
+
+                try {
+                    // 获取会员状态
+                    val savedPhone = PreferencesManager.getPhone()
+                    val userResponse = RetrofitClient.apiService.getUserInfo(savedPhone ?: "")
+                    val isMember = userResponse.code == "0" && userResponse.data?.isMember == true
+                    
+                    // 从用户信息中获取正确的userId
+                    if (userResponse.code == "0" && userResponse.data != null) {
+                        currentUserId = userResponse.data?.id.toString()
+                    }
+
+                    // 首先检查本地是否有缓存的分类配置
+                    val localCategories = secureStorage.getCategories(currentUserId)
+                    
+                    if (localCategories.isNotEmpty()) {
+                        // 使用本地缓存的配置
+                        _categories.value = localCategories
+                    } else {
+                        // 本地无缓存,尝试从服务器获取
+                        try {
+                            val serverCategories = RetrofitClient.apiService.getUserCategories(currentUserId)
+                            
+                            if (serverCategories.isNotEmpty()) {
+                                // 服务器有配置,使用服务器配置
+                                // 如果不是会员,需要过滤掉会员专属分类
+                                val filteredCategories = if (isMember) {
+                                    serverCategories
+                                } else {
+                                    serverCategories.filter { it.name == "快递" || it.name == "还款" }
+                                }
+                                
+                                _categories.value = filteredCategories
+                                // 同时更新本地缓存
+                                secureStorage.saveCategories(currentUserId, filteredCategories)
+                                // 同步回服务器(如果有变化)
+                                if (filteredCategories.size != serverCategories.size) {
+                                    syncCategoriesToServer(filteredCategories)
+                                }
+                            } else {
+                                // 服务器返回空,根据会员状态设置默认分类
+                                val defaultCategories = if (isMember) fullCategories else basicCategories
+                                _categories.value = defaultCategories
+                                // 更新本地缓存
+                                secureStorage.saveCategories(currentUserId, defaultCategories)
+                                // 同步到服务器
+                                syncCategoriesToServer(defaultCategories)
+                            }
+                        } catch (e: Exception) {
+                            // 服务器获取失败,使用默认分类
+                            Log.e("HomeViewModel", "Failed to get categories from server: ${e.message}")
+                            val defaultCategories = if (isMember) fullCategories else basicCategories
+                            _categories.value = defaultCategories
+                            secureStorage.saveCategories(currentUserId, defaultCategories)
+                        }
+                    }
+                } catch (e: Exception) {
+                    // 网络连接失败,尝试从本地获取配置
+                    Log.e("HomeViewModel", "Failed to get user info: ${e.message}")
+                    val localCategories = secureStorage.getCategories(currentUserId)
+                    
+                    if (localCategories.isNotEmpty()) {
+                        // 使用本地缓存的配置
+                        _categories.value = localCategories
+                    } else {
+                        // 本地也没有配置,使用基础分类
+                        _categories.value = basicCategories
+                        // 更新本地缓存
+                        secureStorage.saveCategories(currentUserId, basicCategories)
+                    }
+                }
+                
+                // 更新可见分类
+                _categories.value?.let { updateVisibleCategories(it) }
+                
+            } catch (e: Exception) {
+                Log.e("HomeViewModel", "Failed to load categories: ${e.message}")
+                // 出现异常时,使用基础分类
+                val basicCategories = listOf(
+                    CategoryConfig(1, "快递", 0, true),
+                    CategoryConfig(2, "还款", 1, true)
+                )
+                _categories.value = basicCategories
+                // 更新可见分类
+                updateVisibleCategories(basicCategories)
+            }
+        }
+    }
+
+    private suspend fun syncCategoriesToServer(categories: List<CategoryConfig>) {
+        try {
+            RetrofitClient.apiService.saveUserCategories(
+                CategoryConfigSync(currentUserId, categories)
+            )
+        } catch (e: Exception) {
+            // 同步失败,可以稍后重试或者显示提示
+            Log.e("CategorySync", "Failed to sync categories: ${e.message}")
+        }
+    }
+
+    fun saveCategories(categories: List<CategoryConfig>) {
+        _categories.value = categories
+        // 保存到本地存储
+        secureStorage.saveCategories(currentUserId, categories)
+        // 同步到服务器
+        viewModelScope.launch {
+            try {
+                syncCategoriesToServer(categories)
+            } catch (e: Exception) {
+                Log.e("HomeViewModel", "Failed to sync categories: ${e.message}")
+            }
+        }
+        // 更新可见分类
+        updateVisibleCategories(categories)
+        // 标记分类已被修改
+        categoriesLoaded = true
+    }
+
+    private fun updateVisibleCategories(categories: List<CategoryConfig>) {
+        val visibleNames = categories
+            .filter { it.isEnabled }
+            .sortedBy { it.order }
+            .map { it.name }
+        
+        _visibleCategories.value = categories.filter { it.isEnabled }
+    }
+
+    // 添加检查未读提醒数量的方法
+    fun checkUnreadReminders() {
+        viewModelScope.launch {
+            try {
+                val unreadCount = reminderRecordRepository.getUnreadCount(ReminderRecord.STATUS_UNREAD)
+                _unreadReminderCount.postValue(unreadCount)
+            } catch (e: Exception) {
+                Log.e("HomeViewModel", "Failed to get unread reminder count: ${e.message}")
+            }
+        }
+    }
+
+    // 添加方法以强制刷新分类
+    fun refreshCategories() {
+        categoriesLoaded = false
+        loadCategories()
+    }
+
+    // 登出时不再清除本地数据
+    fun logout() {
+        // 只清除内存中的数据
+        _categories.value = emptyList()
+        categoriesLoaded = false
     }
 
 }
\ No newline at end of file

--
Gitblit v1.9.3