Meshtastic-Apple/Meshtastic/Views/Nodes/Helpers/Actions/FavoriteNodeButton.swift
2026-04-16 12:10:00 -07:00

96 lines
2.6 KiB
Swift

import SwiftData
import OSLog
import SwiftUI
struct FavoriteNodeButton: View {
@EnvironmentObject var accessoryManager: AccessoryManager
@Environment(\.modelContext) private var context
@Bindable var node: NodeInfoEntity
@State var isShowingClientBaseConfirmation = false
var body: some View {
let connectedRoleIsClientBase = accessoryManager.connectedDeviceRole == DeviceRoles.clientBase
Button {
// Special case for CLIENT_BASE: show confirmation when attempting to favorite a node
if connectedRoleIsClientBase && !node.favorite {
isShowingClientBaseConfirmation = true
return
}
// Normal case: perform action immediately
guard let connectedNodeNum = accessoryManager.activeDeviceNum else { return }
Task {
await assignFavorite(node: node, setToFavorite: !node.favorite, connectedNodeNum: Int64(connectedNodeNum))
}
} label: {
Label {
Text(node.favorite ? "Remove from favorites" : "Add to favorites")
} icon: {
Image(systemName: node.favorite ? "star.fill" : "star")
.symbolRenderingMode(.multicolor)
}
}
.confirmationDialog(
"Are you sure?",
isPresented: $isShowingClientBaseConfirmation,
titleVisibility: .visible
) {
Button("Yes, I control this node") {
guard let connectedNodeNum = accessoryManager.activeDeviceNum else { return }
Task {
await assignFavorite(node: node, setToFavorite: true, connectedNodeNum: Int64(connectedNodeNum))
}
}
Button("Cancel", role: .cancel) { }
} message: {
Text("Client Base should only favorite other nodes you control. Improper use will hurt your local mesh.")
}
}
private func assignFavorite (node: NodeInfoEntity, setToFavorite: Bool, connectedNodeNum: Int64) async {
do {
if setToFavorite {
try await accessoryManager.setFavoriteNode(
node: node,
connectedNodeNum: Int64(connectedNodeNum)
)
} else {
try await accessoryManager.removeFavoriteNode(
node: node,
connectedNodeNum: Int64(connectedNodeNum)
)
}
Task { @MainActor in
// Update CoreData
node.favorite = setToFavorite
do {
try context.save()
} catch {
context.rollback()
Logger.data.error("Save Node Favorite Error")
}
Logger.data.debug("Favorited a node")
}
} catch {
}
}
}
// TODO: Fix preview for SwiftData
/*
#Preview {
let node = NodeInfoEntity()
node.num = 123456789
let user = UserEntity()
user.longName = "Test Node"
user.shortName = "TN"
node.user = user
FavoriteNodeButton(node: node)
.environmentObject(AccessoryManager.shared)
.modelContainer(PersistenceController.preview.container)
}
*/