zap-stream-flutter/lib/widgets/notifications_button.dart

import 'package:flutter/material.dart';
import 'package:zap_stream_flutter/notifications.dart';
import 'package:zap_stream_flutter/theme.dart';
import 'package:zap_stream_flutter/utils.dart';

class NotificationsButtonWidget extends StatefulWidget {
  final StreamEvent stream;

  const NotificationsButtonWidget({super.key, required this.stream});

  @override
  State<StatefulWidget> createState() => _NotificationsButtonWidget();
}

class _NotificationsButtonWidget extends State<NotificationsButtonWidget> {
  @override
  Widget build(BuildContext context) {
    return ValueListenableBuilder(
      valueListenable: notifications,
      builder: (context, state, _) {
        final isNotified = (state?.notifyKeys ?? []).contains(
          widget.stream.info.host,
        );
        return IconButton(
          iconSize: 20,
          onPressed: () async {
            final n = getNotificationService();
            if (n == null) return;

            if (isNotified) {
              await n.removeWatchPubkey(widget.stream.info.host);
            } else {
              await n.watchPubkey(widget.stream.info.host, [30311]);
            }
            await notifications.reload();
          },
          style: ButtonStyle(
            padding: WidgetStatePropertyAll(EdgeInsets.all(0)),
            backgroundColor: WidgetStateColor.resolveWith((_) => LAYER_2),
          ),
          icon: Icon(
            isNotified ? Icons.notifications_off : Icons.notification_add,
          ),
        );
      },
    );
  }
}