yubioath-flutter/lib/management/views/management_screen.dart

import 'package:flutter/material.dart';
import 'package:flutter_riverpod/flutter_riverpod.dart';
import 'package:collection/collection.dart';

import '../../app/models.dart';
import '../../app/state.dart';
import '../../app/views/app_failure_screen.dart';
import '../../app/views/app_loading_screen.dart';
import '../../app/views/responsive_dialog.dart';
import '../../core/models.dart';
import '../models.dart';
import '../state.dart';

final _mapEquals = const DeepCollectionEquality().equals;

class _CapabilityForm extends StatelessWidget {
  final int capabilities;
  final int enabled;
  final Function(int) onChanged;
  const _CapabilityForm(
      {required this.capabilities,
      required this.enabled,
      required this.onChanged,
      Key? key})
      : super(key: key);

  @override
  Widget build(BuildContext context) {
    return Wrap(
      spacing: 6.0,
      runSpacing: 6.0,
      children: Capability.values
          .where((c) => capabilities & c.value != 0)
          .map((c) => FilterChip(
                showCheckmark: true,
                selected: enabled & c.value != 0,
                label: Text(c.name),
                onSelected: (_) {
                  onChanged(enabled ^ c.value);
                },
              ))
          .toList(),
    );
  }
}

class _ModeForm extends StatefulWidget {
  final int initialInterfaces;
  final Function(int) onSubmit;
  const _ModeForm(this.initialInterfaces, {required this.onSubmit, Key? key})
      : super(key: key);

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

class _ModeFormState extends State<_ModeForm> {
  int _enabledInterfaces = 0;

  @override
  void initState() {
    super.initState();
    _enabledInterfaces = widget.initialInterfaces;
  }

  @override
  Widget build(BuildContext context) {
    final valid = _enabledInterfaces != 0 &&
        _enabledInterfaces != widget.initialInterfaces;
    return Column(children: [
      ...UsbInterface.values.map(
        (iface) => CheckboxListTile(
          title: Text(iface.name.toUpperCase()),
          value: iface.value & _enabledInterfaces != 0,
          onChanged: (_) {
            setState(() {
              _enabledInterfaces ^= iface.value;
            });
          },
        ),
      ),
      Container(
        padding: const EdgeInsets.all(16.0),
        alignment: Alignment.centerRight,
        child: ElevatedButton(
          onPressed: valid
              ? () {
                  widget.onSubmit(_enabledInterfaces);
                }
              : null,
          child: const Text('Apply changes'),
        ),
      )
    ]);
  }
}

class _CapabilitiesForm extends StatelessWidget {
  final Map<Transport, int> supported;
  final Map<Transport, int> enabled;
  final Function(Map<Transport, int> enabled) onChanged;

  const _CapabilitiesForm({
    required this.onChanged,
    required this.supported,
    required this.enabled,
    Key? key,
  }) : super(key: key);

  @override
  Widget build(BuildContext context) {
    final usbCapabilities = supported[Transport.usb] ?? 0;
    final nfcCapabilities = supported[Transport.nfc] ?? 0;

    return Column(
      crossAxisAlignment: CrossAxisAlignment.start,
      children: [
        if (usbCapabilities != 0)
          const ListTile(
            leading: Icon(Icons.usb),
            title: Text('USB applications'),
          ),
        _CapabilityForm(
          capabilities: usbCapabilities,
          enabled: enabled[Transport.usb] ?? 0,
          onChanged: (value) {
            onChanged({...enabled, Transport.usb: value});
          },
        ),
        if (nfcCapabilities != 0)
          const ListTile(
            leading: Icon(Icons.wifi),
            title: Text('NFC applications'),
          ),
        _CapabilityForm(
          capabilities: nfcCapabilities,
          enabled: enabled[Transport.nfc] ?? 0,
          onChanged: (value) {
            onChanged({...enabled, Transport.nfc: value});
          },
        ),
      ],
    );
  }
}

class ManagementScreen extends ConsumerStatefulWidget {
  final YubiKeyData deviceData;
  const ManagementScreen(this.deviceData, {Key? key}) : super(key: key);

  @override
  ConsumerState<ConsumerStatefulWidget> createState() =>
      _ManagementScreenState();
}

class _ManagementScreenState extends ConsumerState<ManagementScreen> {
  late Map<Transport, int> _enabled;

  @override
  void initState() {
    super.initState();
    _enabled = widget.deviceData.info.config.enabledCapabilities;
  }

  Widget _buildCapabilitiesForm(
      BuildContext context, WidgetRef ref, DeviceInfo info) {
    return _CapabilitiesForm(
      supported: widget.deviceData.info.supportedCapabilities,
      enabled: _enabled,
      onChanged: (enabled) {
        setState(() {
          _enabled = enabled;
        });
      },
    );
  }

  void _submitCapabilitiesForm() async {
    final bool reboot;
    if (widget.deviceData.node is UsbYubiKeyNode) {
      // Reboot if USB device descriptor is changed.
      final oldInterfaces = UsbInterfaces.forCapabilites(
          widget.deviceData.info.config.enabledCapabilities[Transport.usb] ??
              0);
      final newInterfaces =
          UsbInterfaces.forCapabilites(_enabled[Transport.usb] ?? 0);
      reboot = oldInterfaces != newInterfaces;
    } else {
      reboot = false;
    }

    Function()? close;
    try {
      if (reboot) {
        // This will take longer, show a message
        close = ScaffoldMessenger.of(context)
            .showSnackBar(const SnackBar(
              content: Text('Reconfiguring YubiKey...'),
              duration: Duration(seconds: 8),
            ))
            .close;
      }
      await ref
          .read(managementStateProvider(widget.deviceData.node.path).notifier)
          .writeConfig(
            widget.deviceData.info.config
                .copyWith(enabledCapabilities: _enabled),
            reboot: reboot,
          );
      ScaffoldMessenger.of(context).showSnackBar(const SnackBar(
        content: Text('Configuration updated'),
        duration: Duration(seconds: 2),
      ));
    } finally {
      close?.call();
    }
  }

  Widget _buildModeForm(BuildContext context, WidgetRef ref, DeviceInfo info) =>
      _ModeForm(
          UsbInterfaces.forCapabilites(
              info.config.enabledCapabilities[Transport.usb] ?? 0),
          onSubmit: (enabledInterfaces) {
        ScaffoldMessenger.of(context).showSnackBar(const SnackBar(
          content: Text('Not yet implemented!'),
          duration: Duration(seconds: 1),
        ));
      });

  @override
  Widget build(BuildContext context) {
    ref.listen<DeviceNode?>(currentDeviceProvider, (_, __) {
      //TODO: This can probably be checked better to make sure it's the main page.
      Navigator.of(context).popUntil((route) => route.isFirst);
    });

    bool changed = false;

    return ResponsiveDialog(
      title: const Text('Toggle applications'),
      child:
          ref.watch(managementStateProvider(widget.deviceData.node.path)).when(
                none: () => const AppLoadingScreen(),
                failure: (reason) => AppFailureScreen(reason),
                success: (info) {
                  // TODO: Check mode for < YK5 intead
                  changed = !_mapEquals(
                    _enabled,
                    info.config.enabledCapabilities,
                  );
                  return Column(
                    children: [
                      info.version.major > 4
                          ? _buildCapabilitiesForm(context, ref, info)
                          : _buildModeForm(context, ref, info),
                    ],
                  );
                },
              ),
      actions: [
        TextButton(
          onPressed: changed ? _submitCapabilitiesForm : null,
          child: const Text('Save'),
        ),
      ],
    );
  }
}