Added 'modifyEntries' helper

keemobile · Jun 19, 2022 · 2d4494d · 2d4494d
1 parent 53a1253
commit 2d4494d
Show file tree

Hide file tree

Showing 2 changed files with 43 additions and 0 deletions.
diff --git a/kotpass/src/main/kotlin/io/github/anvell/kotpass/database/modifiers/Entry.kt b/kotpass/src/main/kotlin/io/github/anvell/kotpass/database/modifiers/Entry.kt
@@ -36,6 +36,12 @@ fun KeePassDatabase.modifyEntry(
     copy(group = group.modifyEntry(uuid, block))
 }
 
+fun KeePassDatabase.modifyEntries(
+    block: Entry.() -> Entry
+) = modifyContent {
+    copy(group = group.modifyEntries(block))
+}
+
 fun KeePassDatabase.removeEntry(
     uuid: UUID
 ) = modifyContent {
@@ -74,6 +80,27 @@ private fun Group.modifyEntry(
     }
 }
 
+private fun Group.modifyEntries(
+    block: Entry.() -> Entry
+): Group = copy(
+    entries = entries.map { entry ->
+        val newEntry = block(entry)
+
+        if (newEntry != entry) {
+            val now = Instant.now()
+            newEntry.copy(
+                times = entry.times?.copy(
+                    lastAccessTime = now,
+                    lastModificationTime = now
+                ) ?: TimeData.create()
+            )
+        } else {
+            newEntry
+        }
+    },
+    groups = groups.map { it.modifyEntries(block) }
+)
+
 private fun Group.removeChildEntry(
     uuid: UUID
 ): Group {

diff --git a/kotpass/src/test/kotlin/io/github/anvell/kotpass/database/KeePassDatabaseSpec.kt b/kotpass/src/test/kotlin/io/github/anvell/kotpass/database/KeePassDatabaseSpec.kt
@@ -237,6 +237,22 @@ class KeePassDatabaseSpec : DescribeSpec({
             group.name shouldBe "Hello"
         }
 
+        it("Entries mass modification") {
+            val label = "Hello"
+            val database = loadDatabase(
+                rawData = DatabaseRes.GroupsAndEntries.DbGroupsAndEntries,
+                passphrase = "1"
+            ).modifyEntries {
+                copy(overrideUrl = label)
+            }
+
+            database.traverse { element ->
+                if (element is Entry) {
+                    element.overrideUrl shouldBe label
+                }
+            }
+        }
+
         it("Entry modification with history") {
             val (_, entry) = loadDatabase(
                 rawData = DatabaseRes.GroupsAndEntries.DbGroupsAndEntries,