From 556f584183b4deec17d82439320ccb292a98255d Mon Sep 17 00:00:00 2001
From: kurama <paul.zenker.pz@gmail.com>
Date: Thu, 30 Jan 2025 23:11:17 +0100
Subject: [PATCH] fix: update index, when using random()

---
 utils/impl/ImageIterator.swift | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/utils/impl/ImageIterator.swift b/utils/impl/ImageIterator.swift
index 0855b53..5670169 100644
--- 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) {