1class NotificationViewModel(
2    private val context: Context,
3    private val repository: BookingRepository
4) : ViewModel() {
5
6    private val receiver = object : BroadcastReceiver() {
7        override fun onReceive(ctx: Context, intent: Intent) {
8            viewModelScope.launch {
9                val booking = repository.getActiveBooking()
10                if (booking != null) {
11                    showLocalNotification(ctx, booking)
12                }
13            }
14        }
15    }
16
17    init {
18        context.registerReceiver(
19            receiver,
20            IntentFilter("BOOKING_UPDATE"),
21            Context.RECEIVER_NOT_EXPORTED
22        )
23    }
24
25    val notifications = MutableStateFlow<List<Notification>>(emptyList())
26
27    fun markRead(id: String) {
28        viewModelScope.launch {
29            repository.markNotificationRead(id)
30            notifications.update { list ->
31                list.map { if (it.id == id) it.copy(read = true) else it }
32            }
33        }
34    }
35}