From 9886c71fcff6e228b7048e5a3aad28ec033c7424 Mon Sep 17 00:00:00 2001
From: Bugfix Jacques <jacques@iconica.nl>
Date: Mon, 16 Oct 2023 15:46:23 +0200
Subject: [PATCH] feat: ability to wait for the onfinished

---
 lib/src/auth_screen.dart                 | 17 ++++++++++-------
 lib/src/config/registration_options.dart |  4 +++-
 2 files changed, 13 insertions(+), 8 deletions(-)

diff --git a/lib/src/auth_screen.dart b/lib/src/auth_screen.dart
index 984e558..d007268 100644
--- a/lib/src/auth_screen.dart
+++ b/lib/src/auth_screen.dart
@@ -2,6 +2,7 @@
 //
 // SPDX-License-Identifier: BSD-3-Clause
 
+import 'dart:async';
 import 'dart:collection';
 import 'package:flutter/material.dart';
 import 'package:flutter_registration/flutter_registration.dart';
@@ -22,7 +23,7 @@ class AuthScreen extends StatefulWidget {
   }) : assert(steps.length > 0, 'At least one step is required');
 
   final String title;
-  final Function({
+  final Future<void> Function({
     required HashMap<String, String> values,
     required void Function(int? pageToReturn) onError,
   }) onFinish;
@@ -32,7 +33,7 @@ class AuthScreen extends StatefulWidget {
   final String previousBtnTitle;
   final AppBar? customAppBar;
   final Color? customBackgroundColor;
-  final Widget Function(VoidCallback, String)? nextButtonBuilder;
+  final Widget Function(Future<void> Function(), String)? nextButtonBuilder;
   final Widget Function(VoidCallback, String)? previousButtonBuilder;
 
   @override
@@ -58,7 +59,7 @@ class _AuthScreenState extends State<AuthScreen> {
     );
   }
 
-  void onNext(AuthStep step) {
+  Future<void> onNext(AuthStep step) async {
     if (!_formKey.currentState!.validate()) {
       return;
     }
@@ -77,7 +78,7 @@ class _AuthScreenState extends State<AuthScreen> {
         }
       }
 
-      widget.onFinish(
+      await widget.onFinish(
         values: values,
         onError: (int? pageToReturn) => _pageController.animateToPage(
           pageToReturn ?? 0,
@@ -169,14 +170,16 @@ class _AuthScreenState extends State<AuthScreen> {
                                 ),
                               ),
                         widget.nextButtonBuilder?.call(
-                              () => onNext(step),
+                              () async {
+                                await onNext(step);
+                              },
                               widget.steps.last == step
                                   ? widget.submitBtnTitle
                                   : widget.nextBtnTitle,
                             ) ??
                             ElevatedButton(
-                              onPressed: () {
-                                onNext(step);
+                              onPressed: () async {
+                                await onNext(step);
                               },
                               child: Row(
                                 children: [
diff --git a/lib/src/config/registration_options.dart b/lib/src/config/registration_options.dart
index a1862c5..8ed741b 100644
--- a/lib/src/config/registration_options.dart
+++ b/lib/src/config/registration_options.dart
@@ -2,6 +2,8 @@
 //
 // SPDX-License-Identifier: BSD-3-Clause
 
+import 'dart:async';
+
 import 'package:flutter/material.dart';
 import 'package:flutter_registration/flutter_registration.dart';
 
@@ -24,7 +26,7 @@ class RegistrationOptions {
   final VoidCallback afterRegistration;
   final RegistrationRepository registrationRepository;
   final AppBar Function(String title)? customAppbarBuilder;
-  final Widget Function(VoidCallback onPressed, String label)?
+  final Widget Function(Future<void> Function() onPressed, String label)?
       nextButtonBuilder;
   final Widget Function(VoidCallback onPressed, String label)?
       previousButtonBuilder;