From 6403ddc78af1190995fa4f6335b57b3820fa2fc6 Mon Sep 17 00:00:00 2001
From: cloudroam <cloudroam>
Date: 星期二, 22 四月 2025 14:20:10 +0800
Subject: [PATCH] add： 消息提醒图片，code工具类
---
app/src/main/java/com/example/firstapp/ui/home/HomeViewModel.kt | 267 +++++++++++++++++++++++++++++++++++++++++++----------
1 files changed, 215 insertions(+), 52 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 9699b1e..1507d6e 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
@@ -17,21 +17,28 @@
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 _expressItems = MutableLiveData<List<ExpressGroup>>()
private val _financeItems = MutableLiveData<List<FinanceGroup>>()
private val _incomeItems = MutableLiveData<List<IncomeGroup>>()
- private val _flightItems = MutableLiveData<List<FinanceGroup>>()
- private val _trainItems = MutableLiveData<List<FinanceGroup>>()
+ 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<FinanceGroup>> = _flightItems
- val trainItems: LiveData<List<FinanceGroup>> = _trainItems
+ val flightItems: LiveData<List<FlightGroup>> = _flightItems
+ val trainItems: LiveData<List<TrainGroup>> = _trainItems
private val _categories = MutableLiveData<List<CategoryConfig>>()
val categories: LiveData<List<CategoryConfig>> = _categories
@@ -39,8 +46,13 @@
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 lateinit var currentUserId: String
+ private var currentUserId: String = ""
+ private lateinit var reminderRecordRepository: ReminderRecordRepository
+ private var categoriesLoaded = false
init {
// 初始化时加载包裹列表数据
@@ -52,9 +64,18 @@
fun initialize(context: Context, userId: String) {
secureStorage = SecureStorage(context)
currentUserId = userId
- loadCategories()
+ reminderRecordRepository = ReminderRecordRepository(context)
+
+ // 只在首次加载或者用户修改分类后加载分类数据
+ if (!categoriesLoaded) {
+ loadCategories()
+ categoriesLoaded = true
+ }
+
// 初始化时更新可见分类
_categories.value?.let { updateVisibleCategories(it) }
+ // 加载未读提醒数量
+ checkUnreadReminders()
}
private fun loadDataByType(type: String) {
@@ -95,8 +116,38 @@
}
_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(
@@ -112,8 +163,6 @@
// 根据类型更新对应的 LiveData
when (type) {
"还款" -> _financeItems.postValue(groups)
- "航班" -> _flightItems.postValue(groups)
- "火车票" -> _trainItems.postValue(groups)
}
}
}
@@ -143,66 +192,159 @@
loadDataByType("火车票")
}
- fun loadCategories() {
+
+ 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 serverCategories = RetrofitClient.apiService.getUserCategories(currentUserId)
- if (serverCategories.isNotEmpty()) {
- _categories.value = serverCategories
- secureStorage.saveCategories(currentUserId, serverCategories)
- } else {
- // 如果服务器没有配置，尝试获取本地配置
+ // 默认完整分类列表
+// 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.isEmpty()) {
- // 如果本地也没有配置，使用默认配置
- val defaultCategories = listOf(
- CategoryConfig(1, "快递", 0),
- CategoryConfig(2, "还款", 1),
- CategoryConfig(3, "收入", 2),
- CategoryConfig(4, "航班", 3),
- CategoryConfig(5, "火车票", 4)
- )
- _categories.value = defaultCategories
- syncCategoriesToServer(defaultCategories)
- } else {
+
+ if (localCategories.isNotEmpty()) {
+ // 使用本地缓存的配置
_categories.value = localCategories
- syncCategoriesToServer(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) {
- // 如果网络请求失败，使用本地数据
- val localCategories = secureStorage.getCategories(currentUserId)
- _categories.value = localCategories.ifEmpty {
- listOf(
- CategoryConfig(1, "快递", 0),
- CategoryConfig(2, "还款", 1),
- CategoryConfig(3, "收入", 2),
- CategoryConfig(4, "航班", 3),
- CategoryConfig(5, "火车票", 4)
- )
- }
+ 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 fun syncCategoriesToServer(categories: List<CategoryConfig>) {
- viewModelScope.launch {
- try {
- RetrofitClient.apiService.saveUserCategories(
- CategoryConfigSync(currentUserId, categories)
- )
- } catch (e: Exception) {
- // 同步失败，可以稍后重试或者显示提示
- Log.e("CategorySync", "Failed to sync categories: ${e.message}")
- }
+ 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}")
+ }
+ }
// 更新可见分类
- _visibleCategories.value = categories.filter { it.isEnabled }
+ updateVisibleCategories(categories)
+ // 标记分类已被修改
+ categoriesLoaded = true
}
private fun updateVisibleCategories(categories: List<CategoryConfig>) {
@@ -214,10 +356,31 @@
_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
+ // 删除token
+ PreferencesManager.clearUserData()
}
}
\ No newline at end of file
--
Gitblit v1.9.3