fix: update index, when using random()

KuramaSyu · Jan 30, 2025 · 556f584 · 556f584
1 parent a706e24
commit 556f584
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/utils/impl/ImageIterator.swift b/utils/impl/ImageIterator.swift
@@ -74,7 +74,12 @@ class StrategyBasedImageIterator: IteratorProtocol {
     }
 
     func random() -> NamedImage? {
-        return strategy.selectImage(from: items)
+        let image = strategy.selectImage(from: items)
+        guard let image = image else {return nil}
+        if let index = items.firstIndex(of: image) {
+            self.currentIndex = index
+        }
+        return image
     }
 
     func setStrategy(_ newStrategy: ImageSelectionStrategy) {