yubioath-flutter/lib/piv/views/import_file_dialog.dart

/*
 * Copyright (C) 2023 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:io';

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

import '../../app/message.dart';
import '../../app/models.dart';
import '../../app/state.dart';
import '../../widgets/app_input_decoration.dart';
import '../../widgets/app_text_field.dart';
import '../../widgets/responsive_dialog.dart';
import '../keys.dart' as keys;
import '../models.dart';
import '../state.dart';
import 'cert_info_view.dart';
import 'overwrite_confirm_dialog.dart';
import 'utils.dart';

class ImportFileDialog extends ConsumerStatefulWidget {
  final DevicePath devicePath;
  final PivState pivState;
  final PivSlot pivSlot;
  final File file;
  const ImportFileDialog(
      this.devicePath, this.pivState, this.pivSlot, this.file,
      {super.key});

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

class _ImportFileDialogState extends ConsumerState<ImportFileDialog> {
  late String _data;
  PivExamineResult? _state;
  String _password = '';
  bool _passwordIsWrong = false;
  bool _importing = false;
  bool _isObscure = true;

  @override
  void initState() {
    super.initState();
    _init();
  }

  void _init() async {
    final bytes = await widget.file.readAsBytes();
    _data = bytes.map((e) => e.toRadixString(16).padLeft(2, '0')).join();
    _examine();
  }

  void _examine() async {
    setState(() {
      _state = null;
    });
    final result = await ref
        .read(pivSlotsProvider(widget.devicePath).notifier)
        .examine(_data, password: _password.isNotEmpty ? _password : null);
    setState(() {
      _state = result;
      _passwordIsWrong = result.maybeWhen(
        invalidPassword: () => _password.isNotEmpty,
        orElse: () => true,
      );
    });
  }

  @override
  Widget build(BuildContext context) {
    final l10n = AppLocalizations.of(context)!;
    final textTheme = Theme.of(context).textTheme;
    final colorScheme = Theme.of(context).colorScheme;
    // This is what ListTile uses for subtitle
    final subtitleStyle = textTheme.bodyMedium!.copyWith(
      color: colorScheme.onSurfaceVariant,
    );
    // This is what TextInput errors look like
    final errorStyle = textTheme.labelLarge!.copyWith(color: colorScheme.error);
    final state = _state;
    if (state == null) {
      return ResponsiveDialog(
        title: Text(l10n.l_import_file),
        actions: [
          TextButton(
            key: keys.unlockButton,
            onPressed: null,
            child: Text(l10n.s_unlock),
          ),
        ],
        child: const Padding(
            padding: EdgeInsets.symmetric(horizontal: 18.0),
            child: Center(
              child: CircularProgressIndicator(),
            )),
      );
    }

    return state.when(
      invalidPassword: () => ResponsiveDialog(
        title: Text(l10n.l_import_file),
        actions: [
          TextButton(
            key: keys.unlockButton,
            onPressed: () => _examine(),
            child: Text(l10n.s_unlock),
          ),
        ],
        child: Padding(
          padding: const EdgeInsets.symmetric(horizontal: 18.0),
          child: Column(
            crossAxisAlignment: CrossAxisAlignment.start,
            children: [
              Text(l10n.p_password_protected_file),
              AppTextField(
                autofocus: true,
                obscureText: _isObscure,
                autofillHints: const [AutofillHints.password],
                key: keys.managementKeyField,
                decoration: AppInputDecoration(
                  border: const OutlineInputBorder(),
                  labelText: l10n.s_password,
                  errorText: _passwordIsWrong ? l10n.s_wrong_password : null,
                  errorMaxLines: 3,
                  prefixIcon: const Icon(Symbols.password),
                  suffixIcon: IconButton(
                      icon: Icon(_isObscure
                          ? Symbols.visibility
                          : Symbols.visibility_off),
                      onPressed: () {
                        setState(() {
                          _isObscure = !_isObscure;
                        });
                      },
                      tooltip: _isObscure
                          ? l10n.s_show_password
                          : l10n.s_hide_password),
                ),
                textInputAction: TextInputAction.next,
                onChanged: (value) {
                  setState(() {
                    _passwordIsWrong = false;
                    _password = value;
                  });
                },
                onSubmitted: (_) => _examine(),
              ),
            ]
                .map((e) => Padding(
                      padding: const EdgeInsets.symmetric(vertical: 8.0),
                      child: e,
                    ))
                .toList(),
          ),
        ),
      ),
      result: (_, keyType, certInfo) {
        final isFips =
            ref.watch(currentDeviceDataProvider).valueOrNull?.info.isFips ??
                false;
        final unsupportedKey = keyType != null &&
            !getSupportedKeyTypes(widget.pivState.version, isFips)
                .contains(keyType);
        return ResponsiveDialog(
          title: Text(l10n.l_import_file),
          actions: [
            TextButton(
              key: keys.unlockButton,
              onPressed: (keyType == null && certInfo == null) ||
                      _importing ||
                      unsupportedKey
                  ? null
                  : () async {
                      final withContext = ref.read(withContextProvider);

                      if (!await confirmOverwrite(
                        context,
                        widget.pivSlot,
                        writeKey: keyType != null,
                        writeCert: certInfo != null,
                      )) {
                        return;
                      }

                      setState(() {
                        _importing = true;
                      });

                      void Function()? close;
                      try {
                        close = await withContext<void Function()>(
                            (context) async => showMessage(
                                  context,
                                  l10n.l_importing_file,
                                  duration: const Duration(seconds: 30),
                                ));
                        await ref
                            .read(pivSlotsProvider(widget.devicePath).notifier)
                            .import(widget.pivSlot.slot, _data,
                                password:
                                    _password.isNotEmpty ? _password : null);
                        await withContext(
                          (context) async {
                            Navigator.of(context).pop(true);
                            showMessage(context, l10n.s_file_imported);
                          },
                        );
                      } catch (err) {
                        // TODO: More error cases
                        setState(() {
                          _passwordIsWrong = true;
                          _importing = false;
                        });
                      } finally {
                        close?.call();
                      }
                    },
              child: Text(l10n.s_import),
            ),
          ],
          child: Padding(
            padding: const EdgeInsets.symmetric(horizontal: 18.0),
            child: Column(
              crossAxisAlignment: CrossAxisAlignment.start,
              children: [
                Text(l10n.p_import_items_desc(
                    widget.pivSlot.slot.getDisplayName(l10n))),
                if (keyType == null && certInfo == null) ...[
                  Row(
                    children: [
                      Icon(Symbols.error, color: colorScheme.error),
                      const SizedBox(width: 8),
                      Text(
                        l10n.l_import_nothing,
                        style: errorStyle,
                      ),
                    ],
                  ),
                ],
                if (keyType != null) ...[
                  Text(
                    l10n.s_private_key,
                    style: textTheme.bodyLarge,
                    softWrap: true,
                    textAlign: TextAlign.center,
                  ),
                  Row(
                    mainAxisSize: MainAxisSize.min,
                    children: [
                      Text(l10n.s_algorithm),
                      const SizedBox(width: 8),
                      Text(
                        keyType.name.toUpperCase(),
                        style: subtitleStyle,
                      ),
                    ],
                  ),
                  if (unsupportedKey)
                    Row(
                      children: [
                        Icon(Symbols.error, color: colorScheme.error),
                        const SizedBox(width: 8),
                        Text(
                          l10n.l_unsupported_key_type,
                          style: errorStyle,
                        ),
                      ],
                    ),
                ],
                if (certInfo != null) ...[
                  Text(
                    l10n.s_certificate,
                    style: textTheme.bodyLarge,
                    softWrap: true,
                    textAlign: TextAlign.center,
                  ),
                  SizedBox(
                    height:
                        140, // Needed for layout, adapt if text sizes changes
                    child: CertInfoTable(certInfo, null),
                  ),
                ]
              ]
                  .map((e) => Padding(
                        padding: const EdgeInsets.symmetric(vertical: 8.0),
                        child: e,
                      ))
                  .toList(),
            ),
          ),
        );
      },
    );
  }
}