83 lines
2.6 KiB
Swift
83 lines
2.6 KiB
Swift
//
|
||
// AllFunctionsViewModel.swift
|
||
// suixinkan
|
||
//
|
||
|
||
import Foundation
|
||
|
||
/// 全部功能页 ViewModel，管理常用应用与更多功能的增删与持久化。
|
||
final class AllFunctionsViewModel {
|
||
|
||
private(set) var commonMenus: [HomeMenuItem] = []
|
||
private(set) var moreMenus: [HomeMenuItem] = []
|
||
|
||
var onStateChange: (() -> Void)?
|
||
private(set) var didCustomize = false
|
||
|
||
private let appStore: AppStore
|
||
private let commonMenuStore: HomeCommonMenuStore
|
||
|
||
init(
|
||
appStore: AppStore = .shared,
|
||
commonMenuStore: HomeCommonMenuStore = HomeCommonMenuStore()
|
||
) {
|
||
self.appStore = appStore
|
||
self.commonMenuStore = commonMenuStore
|
||
}
|
||
|
||
/// 加载全部功能并按常用配置拆分。
|
||
func loadFunctions() {
|
||
let permissions = appStore.permissionItems()
|
||
let allMenus = HomeMenuCatalog.visibleMenus(from: permissions)
|
||
let accountScope = appStore.accountCachePrefix
|
||
let roleCode = appStore.roleCode
|
||
let savedURIs = commonMenuStore.savedCommonURIs(accountScope: accountScope, roleCode: roleCode)
|
||
let defaultURIs = HomeCommonMenuStore.defaultCommonURIs(from: permissions)
|
||
let commonURIs = savedURIs.isEmpty ? defaultURIs : savedURIs
|
||
|
||
let split = HomeCommonMenuStore.splitMenus(from: allMenus, commonURIs: commonURIs)
|
||
commonMenus = split.common
|
||
moreMenus = split.more
|
||
|
||
if savedURIs.isEmpty, !split.common.isEmpty {
|
||
commonMenuStore.saveCommonURIs(split.common.map(\.uri), accountScope: accountScope, roleCode: roleCode)
|
||
}
|
||
|
||
notifyStateChange()
|
||
}
|
||
|
||
/// 将菜单加入常用应用。
|
||
func addToCommon(_ menu: HomeMenuItem) {
|
||
guard !commonMenus.contains(where: { $0.uri == menu.uri }) else { return }
|
||
commonMenus.append(menu)
|
||
moreMenus.removeAll { $0.uri == menu.uri }
|
||
persistCommonMenus()
|
||
didCustomize = true
|
||
notifyStateChange()
|
||
}
|
||
|
||
/// 将菜单移出常用应用。
|
||
func removeFromCommon(_ menu: HomeMenuItem) {
|
||
guard commonMenus.contains(where: { $0.uri == menu.uri }) else { return }
|
||
commonMenus.removeAll { $0.uri == menu.uri }
|
||
if !moreMenus.contains(where: { $0.uri == menu.uri }) {
|
||
moreMenus.append(menu)
|
||
}
|
||
persistCommonMenus()
|
||
didCustomize = true
|
||
notifyStateChange()
|
||
}
|
||
|
||
private func persistCommonMenus() {
|
||
commonMenuStore.saveCommonURIs(
|
||
commonMenus.map(\.uri),
|
||
accountScope: appStore.accountCachePrefix,
|
||
roleCode: appStore.roleCode
|
||
)
|
||
}
|
||
|
||
private func notifyStateChange() {
|
||
onStateChange?()
|
||
}
|
||
}
|