yubioath-flutter/lib/desktop/management/state.dart

import 'dart:async';
import 'dart:convert';

import 'package:flutter_riverpod/flutter_riverpod.dart';
import 'package:logging/logging.dart';
import 'package:yubico_authenticator/management/models.dart';

import '../../app/models.dart';
import '../../app/state.dart';
import '../../management/state.dart';
import '../rpc.dart';
import '../state.dart';

final _log = Logger('desktop.management.state');

final _sessionProvider =
    Provider.autoDispose.family<RpcNodeSession, DevicePath>(
  (ref, devicePath) {
    final currentDevice = ref.watch(currentDeviceProvider);
    final protocol = currentDevice?.when(
          usbYubiKey: (path, name, pid, info) {
            final interfaces = UsbInterfaces.forCapabilites(
                info.config.enabledCapabilities[Transport.usb] ?? 0);
            return [UsbInterface.ccid, UsbInterface.otp, UsbInterface.fido]
                .firstWhere((iface) => iface.value & interfaces != 0);
          },
          nfcReader: (_, __) => UsbInterface.ccid,
        ) ??
        UsbInterface.ccid;
    return RpcNodeSession(
        ref.watch(rpcProvider), devicePath, [protocol.name, 'management']);
  },
);

final desktopManagementState = StateNotifierProvider.autoDispose
    .family<ManagementStateNotifier, DeviceInfo?, DevicePath>(
  (ref, devicePath) {
    final session = ref.watch(_sessionProvider(devicePath));
    final notifier = _DesktopManagementStateNotifier(session);
    session.setErrorHandler('state-reset', (_) async {
      ref.refresh(_sessionProvider(devicePath));
    });
    ref.onDispose(() {
      session.unserErrorHandler('state-reset');
    });
    return notifier..refresh();
  },
);

class _DesktopManagementStateNotifier extends ManagementStateNotifier {
  final RpcNodeSession _session;
  _DesktopManagementStateNotifier(this._session) : super();

  void refresh() async {
    var result = await _session.command('get');
    _log.config('application status', jsonEncode(result));
    if (mounted) {
      state = DeviceInfo.fromJson(result['data']);
    }
  }

  @override
  Future<void> setMode(int mode,
      {int challengeResponseTimeout = 0, int autoEjectTimeout = 0}) async {
    await _session.command('set_mode', params: {
      'mode': mode,
      'challenge_response_timeout': challengeResponseTimeout,
      'auto_eject_timeout': autoEjectTimeout,
    });
  }

  @override
  Future<void> writeConfig(DeviceConfig config,
      {String currentLockCode = '',
      String newLockCode = '',
      bool reboot = false}) async {
    if (reboot) {
      state = null;
    }
    await _session.command('configure', params: {
      ...config.toJson(),
      'cur_lock_code': currentLockCode,
      'new_lock_code': newLockCode,
      'reboot': reboot,
    });
    if (mounted && !reboot) {
      // Make sure we get a deep copy
      state = state?.copyWith(
        config: DeviceConfig(
          {...config.enabledCapabilities},
          config.autoEjectTimeout ?? state?.config.autoEjectTimeout,
          config.challengeResponseTimeout ??
              state?.config.challengeResponseTimeout,
          config.deviceFlags ?? state?.config.deviceFlags,
        ),
      );
    }
  }
}