fluffychat/lib/pages/bootstrap_dialog.dart

import 'package:adaptive_dialog/adaptive_dialog.dart';
import 'package:fluffychat/widgets/layouts/one_page_card.dart';
import 'package:matrix/encryption.dart';
import 'package:matrix/encryption/utils/bootstrap.dart';
import 'package:matrix/matrix.dart';
import 'package:fluffychat/widgets/adaptive_flat_button.dart';
import 'package:flutter/services.dart';
import 'package:fluffychat/utils/platform_infos.dart';
import 'package:flutter/cupertino.dart';
import 'package:flutter/material.dart';
import 'package:flutter_gen/gen_l10n/l10n.dart';
import 'package:future_loading_dialog/future_loading_dialog.dart';

import 'key_verification_dialog.dart';

class BootstrapDialog extends StatefulWidget {
  final bool wipe;
  final Client client;
  const BootstrapDialog({
    Key key,
    this.wipe = false,
    @required this.client,
  }) : super(key: key);

  Future<bool> show(BuildContext context) => PlatformInfos.isCupertinoStyle
      ? showCupertinoDialog(
          context: context,
          builder: (context) => this,
          barrierDismissible: true,
          useRootNavigator: false,
        )
      : showDialog(
          context: context,
          builder: (context) => this,
          barrierDismissible: true,
          useRootNavigator: false,
        );

  @override
  _BootstrapDialogState createState() => _BootstrapDialogState();
}

class _BootstrapDialogState extends State<BootstrapDialog> {
  final TextEditingController _recoveryKeyTextEditingController =
      TextEditingController();

  Bootstrap bootstrap;

  String _recoveryKeyInputError;

  bool _recoveryKeyInputLoading = false;

  String titleText;

  bool _recoveryKeyStored = false;

  bool _wipe;

  void _createBootstrap(bool wipe) {
    setState(() {
      _wipe = wipe;
      titleText = null;
      _recoveryKeyStored = false;
      bootstrap = widget.client.encryption
          .bootstrap(onUpdate: () => setState(() => null));
    });
  }

  @override
  Widget build(BuildContext context) {
    _wipe ??= widget.wipe;
    final buttons = <AdaptiveFlatButton>[];
    Widget body = PlatformInfos.isCupertinoStyle
        ? CupertinoActivityIndicator()
        : LinearProgressIndicator();
    titleText = L10n.of(context).loadingPleaseWait;

    if (bootstrap == null) {
      titleText = L10n.of(context).setupChatBackup;
      body = Column(
        mainAxisSize: MainAxisSize.min,
        children: [
          Image.asset('assets/backup.png', fit: BoxFit.contain),
          Text(L10n.of(context).setupChatBackupDescription),
        ],
      );
      buttons.add(AdaptiveFlatButton(
        label: L10n.of(context).next,
        onPressed: () => _createBootstrap(false),
      ));
    } else if (bootstrap.newSsssKey?.recoveryKey != null &&
        _recoveryKeyStored == false) {
      final key = bootstrap.newSsssKey.recoveryKey;
      titleText = L10n.of(context).securityKey;
      return OnePageCard(
        child: Scaffold(
          appBar: AppBar(
            leading: IconButton(
              icon: Icon(Icons.close),
              onPressed: Navigator.of(context).pop,
            ),
            title: Text(L10n.of(context).securityKey),
          ),
          body: ListView(
            padding: const EdgeInsets.all(16.0),
            children: [
              TextField(
                minLines: 4,
                maxLines: 4,
                readOnly: true,
                controller: TextEditingController(text: key),
              ),
              const SizedBox(height: 16),
              ElevatedButton.icon(
                icon: Icon(Icons.copy_outlined),
                label: Text(L10n.of(context).copyToClipboard),
                onPressed: () => Clipboard.setData(ClipboardData(text: key)),
              ),
              const SizedBox(height: 16),
              ElevatedButton.icon(
                style: ElevatedButton.styleFrom(
                  primary: Theme.of(context).secondaryHeaderColor,
                  onPrimary: Theme.of(context).primaryColor,
                ),
                icon: Icon(Icons.check_outlined),
                label: Text(L10n.of(context).iWroteDownTheKey),
                onPressed: () => setState(() => _recoveryKeyStored = true),
              ),
            ],
          ),
        ),
      );
    } else {
      switch (bootstrap.state) {
        case BootstrapState.loading:
          break;
        case BootstrapState.askWipeSsss:
          WidgetsBinding.instance.addPostFrameCallback(
            (_) => bootstrap.wipeSsss(_wipe),
          );
          break;
        case BootstrapState.askBadSsss:
          WidgetsBinding.instance.addPostFrameCallback(
            (_) => bootstrap.ignoreBadSecrets(true),
          );
          break;
        case BootstrapState.askUseExistingSsss:
          WidgetsBinding.instance.addPostFrameCallback(
            (_) => bootstrap.useExistingSsss(!_wipe),
          );
          break;
        case BootstrapState.askUnlockSsss:
          WidgetsBinding.instance.addPostFrameCallback(
            (_) => bootstrap.unlockedSsss(),
          );
          break;
        case BootstrapState.askNewSsss:
          WidgetsBinding.instance.addPostFrameCallback(
            (_) => bootstrap.newSsss(),
          );
          break;
        case BootstrapState.openExistingSsss:
          _recoveryKeyStored = true;
          return OnePageCard(
            child: Scaffold(
              appBar: AppBar(
                leading: IconButton(
                  icon: Icon(Icons.close),
                  onPressed: Navigator.of(context).pop,
                ),
                title: Text(L10n.of(context).pleaseEnterSecurityKey),
              ),
              body: ListView(
                padding: const EdgeInsets.all(16.0),
                children: [
                  TextField(
                    minLines: 4,
                    maxLines: 4,
                    autofocus: true,
                    autocorrect: false,
                    autofillHints: _recoveryKeyInputLoading
                        ? null
                        : [AutofillHints.password],
                    controller: _recoveryKeyTextEditingController,
                    decoration: InputDecoration(
                      hintText: 'Abc123 Def456',
                      labelText: L10n.of(context).securityKey,
                      errorText: _recoveryKeyInputError,
                    ),
                  ),
                  const SizedBox(height: 16),
                  ElevatedButton.icon(
                      icon: Icon(Icons.lock_open_outlined),
                      label: Text(L10n.of(context).unlockChatBackup),
                      onPressed: () async {
                        setState(() {
                          _recoveryKeyInputError = null;
                          _recoveryKeyInputLoading = true;
                        });
                        try {
                          await bootstrap.newSsssKey.unlock(
                            keyOrPassphrase:
                                _recoveryKeyTextEditingController.text,
                          );
                          await bootstrap.openExistingSsss();
                        } catch (e, s) {
                          Logs().w('Unable to unlock SSSS', e, s);
                          setState(() => _recoveryKeyInputError =
                              L10n.of(context).oopsSomethingWentWrong);
                        } finally {
                          setState(() => _recoveryKeyInputLoading = false);
                        }
                      }),
                  const SizedBox(height: 16),
                  Row(children: [
                    Expanded(child: Divider()),
                    Padding(
                      padding: const EdgeInsets.all(12.0),
                      child: Text(L10n.of(context).or),
                    ),
                    Expanded(child: Divider()),
                  ]),
                  const SizedBox(height: 16),
                  ElevatedButton.icon(
                    style: ElevatedButton.styleFrom(
                      primary: Theme.of(context).secondaryHeaderColor,
                      onPrimary: Theme.of(context).primaryColor,
                    ),
                    icon: Icon(Icons.transfer_within_a_station_outlined),
                    label: Text(L10n.of(context).transferFromAnotherDevice),
                    onPressed: () async {
                      final req = await showFutureLoadingDialog(
                        context: context,
                        future: () => widget
                            .client.userDeviceKeys[widget.client.userID]
                            .startVerification(),
                      );
                      if (req.error != null) return;
                      await KeyVerificationDialog(request: req.result)
                          .show(context);
                      Navigator.of(context, rootNavigator: false).pop();
                    },
                  ),
                  const SizedBox(height: 16),
                  ElevatedButton.icon(
                    style: ElevatedButton.styleFrom(
                      primary: Theme.of(context).secondaryHeaderColor,
                      onPrimary: Colors.red,
                    ),
                    icon: Icon(Icons.delete_outlined),
                    label: Text(L10n.of(context).securityKeyLost),
                    onPressed: () async {
                      if (OkCancelResult.ok ==
                          await showOkCancelAlertDialog(
                            useRootNavigator: false,
                            context: context,
                            title: L10n.of(context).securityKeyLost,
                            message: L10n.of(context).wipeChatBackup,
                            okLabel: L10n.of(context).ok,
                            cancelLabel: L10n.of(context).cancel,
                            isDestructiveAction: true,
                          )) {
                        _createBootstrap(true);
                      }
                    },
                  )
                ],
              ),
            ),
          );
        case BootstrapState.askWipeCrossSigning:
          WidgetsBinding.instance.addPostFrameCallback(
            (_) => bootstrap.wipeCrossSigning(_wipe),
          );
          break;
        case BootstrapState.askSetupCrossSigning:
          WidgetsBinding.instance.addPostFrameCallback(
            (_) => bootstrap.askSetupCrossSigning(
              setupMasterKey: true,
              setupSelfSigningKey: true,
              setupUserSigningKey: true,
            ),
          );
          break;
        case BootstrapState.askWipeOnlineKeyBackup:
          WidgetsBinding.instance.addPostFrameCallback(
            (_) => bootstrap.wipeOnlineKeyBackup(_wipe),
          );

          break;
        case BootstrapState.askSetupOnlineKeyBackup:
          WidgetsBinding.instance.addPostFrameCallback(
            (_) => bootstrap.askSetupOnlineKeyBackup(true),
          );
          break;
        case BootstrapState.error:
          titleText = L10n.of(context).oopsSomethingWentWrong;
          body = Icon(Icons.error_outline, color: Colors.red, size: 40);
          buttons.add(AdaptiveFlatButton(
            label: L10n.of(context).close,
            onPressed: () =>
                Navigator.of(context, rootNavigator: false).pop<bool>(false),
          ));
          break;
        case BootstrapState.done:
          titleText = L10n.of(context).everythingReady;
          body = Column(
            mainAxisSize: MainAxisSize.min,
            children: [
              Image.asset('assets/backup.png', fit: BoxFit.contain),
              Text(L10n.of(context).yourChatBackupHasBeenSetUp),
            ],
          );
          buttons.add(AdaptiveFlatButton(
            label: L10n.of(context).close,
            onPressed: () =>
                Navigator.of(context, rootNavigator: false).pop<bool>(false),
          ));
          break;
      }
    }

    final title = Text(titleText);
    if (PlatformInfos.isCupertinoStyle) {
      return CupertinoAlertDialog(
        title: title,
        content: body,
        actions: buttons,
      );
    }
    return AlertDialog(
      title: title,
      content: body,
      actions: buttons,
    );
  }
}