[auth] Add bitwarden folder as tag

ente-io · Jun 3, 2024 · 7cd998b · 7cd998b
1 parent ad00690
commit 7cd998b
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/auth/lib/ui/settings/data/import/bitwarden_import.dart b/auth/lib/ui/settings/data/import/bitwarden_import.dart
@@ -4,6 +4,7 @@ import 'dart:io';
 
 import 'package:ente_auth/l10n/l10n.dart';
 import 'package:ente_auth/models/code.dart';
+import 'package:ente_auth/models/code_display.dart';
 import 'package:ente_auth/services/authenticator_service.dart';
 import 'package:ente_auth/store/code_store.dart';
 import 'package:ente_auth/ui/components/buttons/button_widget.dart';
@@ -80,10 +81,19 @@ Future<int?> _processBitwardenExportFile(
   final jsonString = await file.readAsString();
   final data = jsonDecode(jsonString);
   List<dynamic> jsonArray = data['items'];
+  final Map<String, String> folderIdToName = {};
+  try {
+    for (var item in data['folders']) {
+      folderIdToName[item['id']] = item['name'];
+    }
+  } catch (e) {
+    debugPrint("Failed to get folder details $e");
+  }
   final parsedCodes = [];
   for (var item in jsonArray) {
     if (item['login'] != null && item['login']['totp'] != null) {
       var totp = item['login']['totp'];
+      String? folderID = item['folderId'];
 
       Code code;
       if (totp.contains("otpauth://")) {
@@ -110,6 +120,11 @@ Future<int?> _processBitwardenExportFile(
           Code.defaultDigits,
         );
       }
+      if (folderID != null && folderIdToName.containsKey(folderID)) {
+        code = code.copyWith(
+          display: CodeDisplay(tags: [folderIdToName[folderID]!]),
+        );
+      }
 
       parsedCodes.add(code);
     }