From 0cf485be5b682f49638924bfdb8a2e9f9938bb12 Mon Sep 17 00:00:00 2001
From: Frank Merkel <138444693+frankmer@users.noreply.github.com>
Date: Mon, 10 Feb 2025 10:46:05 +0100
Subject: [PATCH] add container from link

---
 android/app/src/main/AndroidManifest.xml      |  1 +
 ios/Runner/Info.plist                         |  2 +
 lib/l10n/app_en.arb                           |  2 +-
 lib/model/processor_result.dart               |  4 +-
 .../home_widget_navigate_processor.dart       |  4 +-
 .../token_container_notifier.dart             |  8 ++--
 .../generated_providers/token_notifier.dart   |  8 ++--
 .../token_container_deep_link_listener.dart   | 42 +++++++++++++++++++
 lib/utils/utils.dart                          |  2 +-
 .../dialogs/select_tokens_dialog.dart         |  2 +-
 lib/widgets/app_wrapper.dart                  |  2 +
 .../token_container_notifier_test.dart        |  2 +-
 12 files changed, 63 insertions(+), 16 deletions(-)
 create mode 100644 lib/utils/riverpod/state_listeners/token_container_deep_link_listener.dart

diff --git a/android/app/src/main/AndroidManifest.xml b/android/app/src/main/AndroidManifest.xml
index 1a0a3ab90..4b59c807e 100644
--- a/android/app/src/main/AndroidManifest.xml
+++ b/android/app/src/main/AndroidManifest.xml
@@ -46,6 +46,7 @@
                 <!-- Accepts URIs that begin with YOUR_SCHEME://-->
                 <data android:scheme="otpauth" />
                 <data android:scheme="otpauth-migration" />
+                <data android:scheme="pia" />
             </intent-filter>
             <!-- Specifies an Android theme to apply to this Activity as soon as
                  the Android process has started. This theme is visible to the user
diff --git a/ios/Runner/Info.plist b/ios/Runner/Info.plist
index 48db00e09..f7cec3a22 100644
--- a/ios/Runner/Info.plist
+++ b/ios/Runner/Info.plist
@@ -37,6 +37,8 @@
 			<key>CFBundleURLSchemes</key>
 			<array>
 				<string>otpauth</string>
+				<string>otpauth-migration</string>
+				<string>pia</string>
 			</array>
 		</dict>
 	</array>
diff --git a/lib/l10n/app_en.arb b/lib/l10n/app_en.arb
index 46f079fed..d6a6e966a 100644
--- a/lib/l10n/app_en.arb
+++ b/lib/l10n/app_en.arb
@@ -1920,7 +1920,7 @@
   "selectFile": "Select file",
   "selectImportSource": "Select import source",
   "selectImportType": "How do you want to import the tokens?",
-  "selectTokensToExport": "{count, plural, one{select the token to be exported} other{select the tokens to be exported}}",
+  "selectTokensToExport": "{count, plural, one{Select the token to be exported} other{Select the tokens to be exported}}",
   "selectTokensToExportHelpContent1": "If a token is not listed, it is not guaranteed that it is not a privacyIDEA token.",
   "selectTokensToExportHelpContent2": "Currently only manually added and imported tokens are exportable.",
   "selectTokensToExportHelpContent3": "We are working on a solution to differentiate between privacyIDEA tokens and private tokens.",
diff --git a/lib/model/processor_result.dart b/lib/model/processor_result.dart
index f19a25df9..1af80e5dd 100644
--- a/lib/model/processor_result.dart
+++ b/lib/model/processor_result.dart
@@ -78,6 +78,6 @@ extension ListProcessorResult<T> on List<ProcessorResult<T>> {
 
 mixin ResultHandler {
   static const argTokenOriginSourceType = "TokenOriginSourceType";
-  Future handleProcessorResult(ProcessorResult result, Map<String, dynamic> args);
-  Future handleProcessorResults(List<ProcessorResult> results, Map<String, dynamic> args);
+  Future handleProcessorResult(ProcessorResult result, {Map<String, dynamic> args = const {}});
+  Future handleProcessorResults(List<ProcessorResult> results, {Map<String, dynamic> args = const {}});
 }
diff --git a/lib/processors/scheme_processors/navigation_scheme_processors/home_widget_navigate_processor.dart b/lib/processors/scheme_processors/navigation_scheme_processors/home_widget_navigate_processor.dart
index 3c4982bd0..c39ab9795 100644
--- a/lib/processors/scheme_processors/navigation_scheme_processors/home_widget_navigate_processor.dart
+++ b/lib/processors/scheme_processors/navigation_scheme_processors/home_widget_navigate_processor.dart
@@ -148,7 +148,7 @@ class HomeWidgetNavigateProcessor implements NavigationSchemeProcessor {
 
 class NavigationHandler<R> with ResultHandler {
   @override
-  Future<R?> handleProcessorResult(ProcessorResult result, Map<String, dynamic> args) async {
+  Future<R?> handleProcessorResult(ProcessorResult result, {Map<String, dynamic> args = const {}}) async {
     if (result is! ProcessorResult<Navigation>) return null;
     if (result.isFailed) return null;
     validate(
@@ -162,7 +162,7 @@ class NavigationHandler<R> with ResultHandler {
   }
 
   @override
-  Future<List<R>?> handleProcessorResults(List<ProcessorResult> results, Map<String, dynamic> args) async {
+  Future<List<R>?> handleProcessorResults(List<ProcessorResult> results, {Map<String, dynamic> args = const {}}) async {
     final successResults = results.whereType<ProcessorResult<Navigation>>().toList().successResults;
     if (successResults.isEmpty) return null;
     final BuildContext context;
diff --git a/lib/utils/riverpod/riverpod_providers/generated_providers/token_container_notifier.dart b/lib/utils/riverpod/riverpod_providers/generated_providers/token_container_notifier.dart
index a89c5b4a0..c1c70389e 100644
--- a/lib/utils/riverpod/riverpod_providers/generated_providers/token_container_notifier.dart
+++ b/lib/utils/riverpod/riverpod_providers/generated_providers/token_container_notifier.dart
@@ -243,7 +243,7 @@ class TokenContainerNotifier extends _$TokenContainerNotifier with ResultHandler
     if (uri == null) throw ArgumentError('Invalid rollover uri');
     final result = (await TokenContainerProcessor().processUri(uri, fromInit: false))?.firstOrNull;
     if (result == null) throw StateError('Failed to process rollover uri');
-    final success = await handleProcessorResult(result, {TokenContainerProcessor.ARG_DO_REPLACE: true});
+    final success = await handleProcessorResult(result, args: {TokenContainerProcessor.ARG_DO_REPLACE: true});
     return success;
   }
 
@@ -416,14 +416,14 @@ class TokenContainerNotifier extends _$TokenContainerNotifier with ResultHandler
 
   /// Returns true if the processor result was handled successfully
   @override
-  Future<bool> handleProcessorResult(ProcessorResult result, Map<String, dynamic> args) async {
-    final failedContainer = await handleProcessorResults([result], args);
+  Future<bool> handleProcessorResult(ProcessorResult result, {Map<String, dynamic> args = const {}}) async {
+    final failedContainer = await handleProcessorResults([result], args: args);
     return failedContainer?.isEmpty ?? false;
   }
 
   /// Returns a list of containers that failed to add
   @override
-  Future<List<TokenContainerUnfinalized>?> handleProcessorResults(List<ProcessorResult> results, Map<String, dynamic> args) async {
+  Future<List<TokenContainerUnfinalized>?> handleProcessorResults(List<ProcessorResult> results, {Map<String, dynamic> args = const {}}) async {
     Logger.info('Handling processor results');
     final newContainers = results.getData().whereType<TokenContainerUnfinalized>().toList();
     final validatedArgs = TokenContainerProcessor.validateArgs(args);
diff --git a/lib/utils/riverpod/riverpod_providers/generated_providers/token_notifier.dart b/lib/utils/riverpod/riverpod_providers/generated_providers/token_notifier.dart
index 045cba40a..374bdf02a 100644
--- a/lib/utils/riverpod/riverpod_providers/generated_providers/token_notifier.dart
+++ b/lib/utils/riverpod/riverpod_providers/generated_providers/token_notifier.dart
@@ -775,19 +775,19 @@ class TokenNotifier extends _$TokenNotifier with ResultHandler {
   Future<void> handleLink(Uri uri) async {
     final tokenResults = await TokenImportSchemeProcessor.processUriByAny(uri);
     if (tokenResults == null) return;
-    await handleProcessorResults(tokenResults, {'TokenOriginSourceType': TokenOriginSourceType.link});
+    await handleProcessorResults(tokenResults, args: {'TokenOriginSourceType': TokenOriginSourceType.link});
   }
 
   @override
-  Future<void> handleProcessorResult(ProcessorResult result, Map<String, dynamic> args) {
+  Future<void> handleProcessorResult(ProcessorResult result, {Map<String, dynamic> args = const {}}) {
     if (result is ProcessorResult<Token>) {
-      return handleProcessorResults([result], args);
+      return handleProcessorResults([result], args: args);
     }
     return Future.value();
   }
 
   @override
-  Future handleProcessorResults(List<ProcessorResult> results, Map<String, dynamic> args) async {
+  Future handleProcessorResults(List<ProcessorResult> results, {Map<String, dynamic> args = const {}}) async {
     final List<ProcessorResult<Token>> tokenResults = results.whereType<ProcessorResult<Token>>().toList();
     if (tokenResults.isEmpty) return;
     final List<Token> resultTokens = tokenResults.getData();
diff --git a/lib/utils/riverpod/state_listeners/token_container_deep_link_listener.dart b/lib/utils/riverpod/state_listeners/token_container_deep_link_listener.dart
new file mode 100644
index 000000000..d4e6624a3
--- /dev/null
+++ b/lib/utils/riverpod/state_listeners/token_container_deep_link_listener.dart
@@ -0,0 +1,42 @@
+/*
+ * privacyIDEA Authenticator
+ *
+ * Author: Frank Merkel <frank.merkel@netknights.it>
+ *
+ * Copyright (c) 2025 NetKnights GmbH
+ *
+ * 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 'package:flutter_riverpod/flutter_riverpod.dart';
+import 'package:privacyidea_authenticator/processors/scheme_processors/token_container_processor.dart';
+import 'package:privacyidea_authenticator/utils/riverpod/riverpod_providers/generated_providers/token_container_notifier.dart';
+
+import '../../../interfaces/riverpod/state_listeners/state_notifier_provider_listeners/deep_link_listener.dart';
+import '../../../model/deeplink.dart';
+
+class TokenContainerDeepLinkListener extends DeepLinkListener {
+  const TokenContainerDeepLinkListener({
+    required super.deeplinkProvider,
+  }) : super(
+          onNewState: _onNewState,
+          listenerName: 'TokenContainerDeepLinkListener().processUri',
+        );
+
+  static void _onNewState(WidgetRef ref, AsyncValue<DeepLink>? previous, AsyncValue<DeepLink> next) {
+    next.whenData((next) async {
+      final processorResults = await TokenContainerProcessor().processUri(next.uri);
+      if (processorResults == null || processorResults.isEmpty) return;
+      ref.read(tokenContainerProvider.notifier).handleProcessorResults(processorResults);
+    });
+  }
+}
diff --git a/lib/utils/utils.dart b/lib/utils/utils.dart
index 2a1dd95e1..d43ec188c 100644
--- a/lib/utils/utils.dart
+++ b/lib/utils/utils.dart
@@ -241,7 +241,7 @@ Future<void> scanQrCode({BuildContext? context, required List<ResultHandler> res
       final results = resultHandlerTypeMap[resultHandlerType]!;
       final resultHandler = resultHandlerList.firstWhereOrNull((resultHandler) => resultHandlerType.isTypeOf(resultHandler));
       if (resultHandler != null) {
-        await resultHandler.handleProcessorResults(results, {ResultHandler.argTokenOriginSourceType: TokenOriginSourceType.qrScan});
+        await resultHandler.handleProcessorResults(results, args: {ResultHandler.argTokenOriginSourceType: TokenOriginSourceType.qrScan});
       }
     }
   }
diff --git a/lib/views/settings_view/settings_groups/import_export_tokens_widgets/dialogs/select_tokens_dialog.dart b/lib/views/settings_view/settings_groups/import_export_tokens_widgets/dialogs/select_tokens_dialog.dart
index b96766c31..0d6cd32c9 100644
--- a/lib/views/settings_view/settings_groups/import_export_tokens_widgets/dialogs/select_tokens_dialog.dart
+++ b/lib/views/settings_view/settings_groups/import_export_tokens_widgets/dialogs/select_tokens_dialog.dart
@@ -68,7 +68,7 @@ class _SelectTokensDialogState extends ConsumerState<SelectExportTokensDialog> {
                 });
               }
             : (selected, _) {
-                _showExportDialog(_selectedTokens);
+                _showExportDialog(selected);
               },
       ),
       actions: [
diff --git a/lib/widgets/app_wrapper.dart b/lib/widgets/app_wrapper.dart
index 9f6a2c454..80755e643 100644
--- a/lib/widgets/app_wrapper.dart
+++ b/lib/widgets/app_wrapper.dart
@@ -14,6 +14,7 @@ import '../utils/riverpod/riverpod_providers/generated_providers/token_notifier.
 import '../utils/riverpod/state_listeners/home_widget_deep_link_listener.dart';
 import '../utils/riverpod/state_listeners/home_widget_token_state_listener.dart';
 import '../utils/riverpod/state_listeners/navigation_deep_link_listener.dart';
+import '../utils/riverpod/state_listeners/token_container_deep_link_listener.dart';
 import '../utils/riverpod/state_listeners/token_deep_link_listener.dart';
 import 'app_wrappers/single_touch_recognizer.dart';
 import 'app_wrappers/state_observer.dart';
@@ -87,6 +88,7 @@ class _AppWrapperState extends ConsumerState<_AppWrapper> {
           NavigationDeepLinkListener(deeplinkProvider: deeplinkNotifierProvider),
           HomeWidgetDeepLinkListener(deeplinkProvider: deeplinkNotifierProvider),
           TokenImportDeepLinkListener(deeplinkProvider: deeplinkNotifierProvider),
+          TokenContainerDeepLinkListener(deeplinkProvider: deeplinkNotifierProvider),
         ],
         asyncNotifierProviderListeners: [],
         child: EasyDynamicThemeWidget(
diff --git a/test/unit_test/state_notifiers/token_container_notifier_test.dart b/test/unit_test/state_notifiers/token_container_notifier_test.dart
index 30ee58a3f..94e19cc26 100644
--- a/test/unit_test/state_notifiers/token_container_notifier_test.dart
+++ b/test/unit_test/state_notifiers/token_container_notifier_test.dart
@@ -524,7 +524,7 @@ void _testTokenContainerNotifier() {
       expect(processorResults, isNotNull);
       expect(processorResults!.length, 1);
       final result = processorResults.first;
-      await providerContainer.read(tokenContainerProvider.notifier).handleProcessorResult(result, {
+      await providerContainer.read(tokenContainerProvider.notifier).handleProcessorResult(result, args: {
         TokenContainerProcessor.ARG_DO_REPLACE: true,
         TokenContainerProcessor.ARG_ADD_DEVICE_INFOS: true,
         TokenContainerProcessor.ARG_INIT_SYNC: false,