/*
 * Copyright (C) 2022 Yubico.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *       http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

import 'dart:async';

import 'package:flutter/material.dart';
import 'package:flutter_gen/gen_l10n/app_localizations.dart';
import 'package:flutter_riverpod/flutter_riverpod.dart';

import '../../app/message.dart';
import '../../app/models.dart';
import '../../app/shortcuts.dart';
import '../../app/views/action_list.dart';
import '../../app/views/app_failure_page.dart';
import '../../app/views/app_list_item.dart';
import '../../app/views/app_page.dart';
import '../../app/views/message_page.dart';
import '../../core/state.dart';
import '../../management/models.dart';
import '../../widgets/list_title.dart';
import '../features.dart' as features;
import '../models.dart';
import '../state.dart';
import 'actions.dart';
import 'add_fingerprint_dialog.dart';
import 'fingerprint_dialog.dart';
import 'key_actions.dart';
import 'pin_dialog.dart';
import 'pin_entry_form.dart';

class FingerprintsScreen extends ConsumerWidget {
  final YubiKeyData deviceData;
  const FingerprintsScreen(this.deviceData, {super.key});

  @override
  Widget build(BuildContext context, WidgetRef ref) {
    final l10n = AppLocalizations.of(context)!;
    return ref.watch(fidoStateProvider(deviceData.node.path)).when(
        loading: () => AppPage(
              centered: true,
              delayedContent: true,
              builder: (context, _) => const CircularProgressIndicator(),
            ),
        error: (error, _) {
          final enabled = deviceData
                  .info.config.enabledCapabilities[deviceData.node.transport] ??
              0;
          if (Capability.fido2.value & enabled == 0) {
            return MessagePage(
              title: l10n.s_fingerprints,
              header: l10n.s_fido_disabled,
              message: l10n.l_webauthn_req_fido2,
            );
          }

          return AppFailurePage(
            cause: error,
          );
        },
        data: (fidoState) {
          return fidoState.unlocked
              ? _FidoUnlockedPage(deviceData.node, fidoState)
              : _FidoLockedPage(deviceData.node, fidoState);
        });
  }
}

class _FidoLockedPage extends ConsumerWidget {
  final DeviceNode node;
  final FidoState state;

  const _FidoLockedPage(this.node, this.state);

  @override
  Widget build(BuildContext context, WidgetRef ref) {
    final l10n = AppLocalizations.of(context)!;
    final hasFeature = ref.watch(featureProvider);
    final hasActions = hasFeature(features.actions);

    if (!state.hasPin) {
      return MessagePage(
        actions: [
          ActionChip(
            label: Text(l10n.s_set_pin),
            onPressed: () async {
              await showBlurDialog(
                  context: context,
                  builder: (context) => FidoPinDialog(node.path, state));
            },
            avatar: const Icon(Icons.pin_outlined),
          )
        ],
        title: l10n.s_fingerprints,
        header: '${l10n.s_fingerprints_get_started} (1/2)',
        message: l10n.p_set_fingerprints_desc,
        keyActionsBuilder: hasActions ? _buildActions : null,
        keyActionsBadge: fidoShowActionsNotifier(state),
      );
    }

    if (state.forcePinChange) {
      return MessagePage(
        title: l10n.s_fingerprints,
        header: l10n.s_pin_change_required,
        message: l10n.l_pin_change_required_desc,
        keyActionsBuilder: hasActions ? _buildActions : null,
        keyActionsBadge: fidoShowActionsNotifier(state),
      );
    }

    return AppPage(
      title: l10n.s_fingerprints,
      keyActionsBuilder: hasActions ? _buildActions : null,
      builder: (context, _) => Column(
        children: [
          PinEntryForm(state, node),
        ],
      ),
    );
  }

  Widget _buildActions(BuildContext context) =>
      fidoBuildActions(context, node, state, -1);
}

class _FidoUnlockedPage extends ConsumerStatefulWidget {
  final DeviceNode node;
  final FidoState state;

  _FidoUnlockedPage(this.node, this.state) : super(key: ObjectKey(node.path));

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

class _FidoUnlockedPageState extends ConsumerState<_FidoUnlockedPage> {
  Fingerprint? _selected;

  @override
  Widget build(BuildContext context) {
    final l10n = AppLocalizations.of(context)!;
    List<Widget Function(bool expanded)> children = [];

    int nFingerprints = 0;
    if (widget.state.bioEnroll != null) {
      final data = ref.watch(fingerprintProvider(widget.node.path)).asData;
      if (data == null) {
        return _buildLoadingPage(context);
      }
      final fingerprints = data.value;
      if (fingerprints.isNotEmpty) {
        nFingerprints = fingerprints.length;
        children.addAll(fingerprints.map(
          (fp) => (expanded) => _FingerprintListItem(
                fp,
                expanded: expanded,
                selected: fp == _selected,
              ),
        ));
      }
    }

    final hasFeature = ref.watch(featureProvider);
    final hasActions = hasFeature(features.actions);
    final fingerprint = _selected;

    if (children.isNotEmpty) {
      return FidoActions(
        devicePath: widget.node.path,
        actions: (context) => {
          EscapeIntent: CallbackAction<EscapeIntent>(onInvoke: (intent) {
            if (_selected != null) {
              setState(() {
                _selected = null;
              });
            } else {
              Actions.invoke(context, intent);
            }
            return false;
          }),
          OpenIntent<Fingerprint>:
              CallbackAction<OpenIntent<Fingerprint>>(onInvoke: (intent) {
            return showBlurDialog(
              context: context,
              barrierColor: Colors.transparent,
              builder: (context) => FingerprintDialog(intent.target),
            );
          }),
          if (hasFeature(features.fingerprintsEdit))
            EditIntent<Fingerprint>: CallbackAction<EditIntent<Fingerprint>>(
                onInvoke: (intent) async {
              final renamed =
                  await (Actions.invoke(context, intent) as Future<dynamic>?);
              if (_selected == intent.target && renamed is Fingerprint) {
                setState(() {
                  _selected = renamed;
                });
              }
              return renamed;
            }),
          if (hasFeature(features.fingerprintsDelete))
            DeleteIntent<Fingerprint>:
                CallbackAction<DeleteIntent<Fingerprint>>(
                    onInvoke: (intent) async {
              final deleted =
                  await (Actions.invoke(context, intent) as Future<dynamic>?);
              if (deleted == true && _selected == intent.target) {
                setState(() {
                  _selected = null;
                });
              }
              return deleted;
            }),
        },
        builder: (context) => AppPage(
          title: l10n.s_fingerprints,
          detailViewBuilder: fingerprint != null
              ? (context) => Column(
                    crossAxisAlignment: CrossAxisAlignment.stretch,
                    children: [
                      ListTitle(l10n.s_details),
                      Card(
                        child: Padding(
                          padding: const EdgeInsets.all(16),
                          // TODO: Reuse from fingerprint_dialog
                          child: Column(
                            children: [
                              Text(
                                fingerprint.label,
                                style:
                                    Theme.of(context).textTheme.headlineSmall,
                                softWrap: true,
                                textAlign: TextAlign.center,
                              ),
                              const SizedBox(height: 16),
                              const Icon(Icons.fingerprint, size: 72),
                            ],
                          ),
                        ),
                      ),
                      ActionListSection.fromMenuActions(
                        context,
                        l10n.s_actions,
                        actions: buildFingerprintActions(fingerprint, l10n),
                      ),
                    ],
                  )
              : null,
          keyActionsBuilder: hasActions
              ? (context) => fidoBuildActions(
                  context, widget.node, widget.state, nFingerprints)
              : null,
          keyActionsBadge: fidoShowActionsNotifier(widget.state),
          builder: (context, expanded) {
            // De-select if window is resized to be non-expanded.
            if (!expanded && _selected != null) {
              Timer.run(() {
                setState(() {
                  _selected = null;
                });
              });
            }
            return Actions(
              actions: {
                if (expanded) ...{
                  OpenIntent<Fingerprint>:
                      CallbackAction<OpenIntent<Fingerprint>>(
                          onInvoke: (intent) {
                    setState(() {
                      _selected = intent.target;
                    });
                    return null;
                  }),
                }
              },
              child: Column(
                  crossAxisAlignment: CrossAxisAlignment.start,
                  children: children.map((f) => f(expanded)).toList()),
            );
          },
        ),
      );
    }

    return MessagePage(
      actions: [
        ActionChip(
          label: Text(l10n.s_add_fingerprint),
          onPressed: () async {
            await showBlurDialog(
                context: context,
                builder: (context) => AddFingerprintDialog(widget.node.path));
          },
          avatar: const Icon(Icons.fingerprint_outlined),
        )
      ],
      title: l10n.s_webauthn,
      header: '${l10n.s_fingerprints_get_started} (2/2)',
      message: l10n.l_add_one_or_more_fps,
      keyActionsBuilder: hasActions
          ? (context) => fidoBuildActions(context, widget.node, widget.state, 0)
          : null,
      keyActionsBadge: fidoShowActionsNotifier(widget.state),
    );
  }

  Widget _buildLoadingPage(BuildContext context) => AppPage(
        centered: true,
        delayedContent: true,
        builder: (context, _) => const CircularProgressIndicator(),
      );
}

class _FingerprintListItem extends StatelessWidget {
  final Fingerprint fingerprint;
  final bool selected;
  final bool expanded;

  const _FingerprintListItem(this.fingerprint,
      {required this.expanded, required this.selected});

  @override
  Widget build(BuildContext context) {
    return AppListItem(
      fingerprint,
      selected: selected,
      leading: CircleAvatar(
        foregroundColor: Theme.of(context).colorScheme.onSecondary,
        backgroundColor: Theme.of(context).colorScheme.secondary,
        child: const Icon(Icons.fingerprint),
      ),
      title: fingerprint.label,
      trailing: expanded
          ? null
          : OutlinedButton(
              onPressed: Actions.handler(context, OpenIntent(fingerprint)),
              child: const Icon(Icons.more_horiz),
            ),
      tapIntent: isDesktop && !expanded ? null : OpenIntent(fingerprint),
      doubleTapIntent: isDesktop && !expanded ? OpenIntent(fingerprint) : null,
      buildPopupActions: (context) =>
          buildFingerprintActions(fingerprint, AppLocalizations.of(context)!),
    );
  }
}