From b8af00bd9414da77f3bcb6f46233ec6be518870c Mon Sep 17 00:00:00 2001
From: Brody Roberts <brodyroberts2002@gmail.com>
Date: Mon, 14 Jul 2025 11:09:39 -0700
Subject: [PATCH 1/4] Fixed the snapshot bug fix implementation

---
 Sources/FlowStack/FlowLink.swift | 27 ++++++++++++---------------
 1 file changed, 12 insertions(+), 15 deletions(-)

diff --git a/Sources/FlowStack/FlowLink.swift b/Sources/FlowStack/FlowLink.swift
index e2458d2..d1bec72 100644
--- a/Sources/FlowStack/FlowLink.swift
+++ b/Sources/FlowStack/FlowLink.swift
@@ -302,7 +302,7 @@ public struct FlowLink<Label>: View where Label: View {
         return path?.wrappedValue.elements.map(\.context?.linkDepth).contains(flowDepth) ?? false
     }
 
-    private func createSnapshot(colorScheme: ColorScheme) -> UIImage? {
+    private func createSnapshot(colorScheme: ColorScheme) async -> UIImage? {
         guard let size = size else { return nil }
 
         let frame = CGRect(origin: .zero, size: size)
@@ -365,13 +365,13 @@ public struct FlowLink<Label>: View where Label: View {
             .onButtonGesture {
                 buttonPressed = true
                 // check for sibling elements and return early if we already have a presented element at this depth
-                guard !hasSiblingElement else {
-                    return
-                }
+                guard !hasSiblingElement else { return }
                 Task {
                     if configuration.transitionFromSnapshot {
-                        context?.snapshot = snapshots[colorScheme]
+                        await initSnapshots()
+                        self.context?.snapshot = snapshots[colorScheme]
                     }
+
                     if let value = value {
                         withTransaction(transaction) {
                             path?.wrappedValue.append(value, context: context)
@@ -406,7 +406,7 @@ public struct FlowLink<Label>: View where Label: View {
             snapshots[newScheme]
             path?.wrappedValue.updateSnapshots(from: newScheme)
         }
-        .onAppear { initSnapshots() }
+//        .onAppear { initSnapshots() }
         .background(
             GeometryReader { proxy in
                 Color.clear
@@ -459,15 +459,12 @@ public struct FlowLink<Label>: View where Label: View {
         }
     }
 
-    private func initSnapshots() {
-        Task {
-            // Prevent Snapshot from being taken too early before Fetchable content loads
-            try? await Task.sleep(10000)
-                let lightImage = createSnapshot(colorScheme: .light)
-                let darkImage = createSnapshot(colorScheme: .dark)
-                snapshots[.light] = lightImage
-                snapshots[.dark] = darkImage
-        }
+    private func initSnapshots() async {
+        guard snapshots.isEmpty else { return }
+        let lightImage = await createSnapshot(colorScheme: .light)
+        let darkImage = await createSnapshot(colorScheme: .dark)
+        self.snapshots[.light] = lightImage
+        self.snapshots[.dark] = darkImage
     }
 }
 

From 89c9e3a31df5c9ccec74bfe804620544db9861e4 Mon Sep 17 00:00:00 2001
From: Brody Roberts <brodyroberts2002@gmail.com>
Date: Mon, 14 Jul 2025 13:19:14 -0700
Subject: [PATCH 2/4] fixed feature

---
 Sources/FlowStack/FlowLink.swift | 5 ++---
 Sources/FlowStack/FlowPath.swift | 8 ++++++++
 2 files changed, 10 insertions(+), 3 deletions(-)

diff --git a/Sources/FlowStack/FlowLink.swift b/Sources/FlowStack/FlowLink.swift
index d1bec72..db63828 100644
--- a/Sources/FlowStack/FlowLink.swift
+++ b/Sources/FlowStack/FlowLink.swift
@@ -369,6 +369,7 @@ public struct FlowLink<Label>: View where Label: View {
                 Task {
                     if configuration.transitionFromSnapshot {
                         await initSnapshots()
+                        self.context?.snapshotDict = snapshots
                         self.context?.snapshot = snapshots[colorScheme]
                     }
 
@@ -402,11 +403,9 @@ public struct FlowLink<Label>: View where Label: View {
             }
         }
         .onChange(of: colorScheme) { newScheme in
-            refreshButton = UUID()
-            snapshots[newScheme]
             path?.wrappedValue.updateSnapshots(from: newScheme)
+            refreshButton = UUID()
         }
-//        .onAppear { initSnapshots() }
         .background(
             GeometryReader { proxy in
                 Color.clear
diff --git a/Sources/FlowStack/FlowPath.swift b/Sources/FlowStack/FlowPath.swift
index 06f8b8a..8a74d06 100644
--- a/Sources/FlowStack/FlowPath.swift
+++ b/Sources/FlowStack/FlowPath.swift
@@ -104,7 +104,15 @@ public struct FlowPath: Equatable, Hashable {
     ///    - colorScheme: The new color scheme to be used for snapshots
     public mutating func updateSnapshots(from colorScheme: ColorScheme) {
         for i in elements.indices {
+            // This isn't running on inital go !?!?!??!
+            print("🦦 item \(elements.count)")
+            print("🦦 elements context? \(elements[i].context)")
+            print("🦦 elements dict? \(elements[i].context?.snapshotDict)")
             guard var context = elements[i].context else { continue }
+//            print("🦦 item \(elements.count) entered here? ")
+//            context.snapshot = context.snapshotDict[colorScheme]
+//            elements[i].context?.snapshot = context.snapshot
+            print("🦦 \(context.snapshotDict[colorScheme])\nDictionary? \(context.snapshotDict)")
             if let newSnapshot = context.snapshotDict[colorScheme] {
                 context.snapshot = newSnapshot
                 elements[i].context?.snapshot = context.snapshot

From 9c01447c0a0fbdbb678d3bcfde541d3e0bafd813 Mon Sep 17 00:00:00 2001
From: Brody Roberts <brodyroberts2002@gmail.com>
Date: Mon, 14 Jul 2025 13:23:23 -0700
Subject: [PATCH 3/4] removed prints

---
 Sources/FlowStack/FlowPath.swift | 8 --------
 1 file changed, 8 deletions(-)

diff --git a/Sources/FlowStack/FlowPath.swift b/Sources/FlowStack/FlowPath.swift
index 8a74d06..06f8b8a 100644
--- a/Sources/FlowStack/FlowPath.swift
+++ b/Sources/FlowStack/FlowPath.swift
@@ -104,15 +104,7 @@ public struct FlowPath: Equatable, Hashable {
     ///    - colorScheme: The new color scheme to be used for snapshots
     public mutating func updateSnapshots(from colorScheme: ColorScheme) {
         for i in elements.indices {
-            // This isn't running on inital go !?!?!??!
-            print("🦦 item \(elements.count)")
-            print("🦦 elements context? \(elements[i].context)")
-            print("🦦 elements dict? \(elements[i].context?.snapshotDict)")
             guard var context = elements[i].context else { continue }
-//            print("🦦 item \(elements.count) entered here? ")
-//            context.snapshot = context.snapshotDict[colorScheme]
-//            elements[i].context?.snapshot = context.snapshot
-            print("🦦 \(context.snapshotDict[colorScheme])\nDictionary? \(context.snapshotDict)")
             if let newSnapshot = context.snapshotDict[colorScheme] {
                 context.snapshot = newSnapshot
                 elements[i].context?.snapshot = context.snapshot

From 3f660afd56f4f7cf1ae2b8083842a8167efe943d Mon Sep 17 00:00:00 2001
From: Brody Roberts <brodyroberts2002@gmail.com>
Date: Mon, 14 Jul 2025 13:43:08 -0700
Subject: [PATCH 4/4] Removed async calls

---
 Sources/FlowStack/FlowLink.swift | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/Sources/FlowStack/FlowLink.swift b/Sources/FlowStack/FlowLink.swift
index db63828..58da7b2 100644
--- a/Sources/FlowStack/FlowLink.swift
+++ b/Sources/FlowStack/FlowLink.swift
@@ -302,7 +302,7 @@ public struct FlowLink<Label>: View where Label: View {
         return path?.wrappedValue.elements.map(\.context?.linkDepth).contains(flowDepth) ?? false
     }
 
-    private func createSnapshot(colorScheme: ColorScheme) async -> UIImage? {
+    private func createSnapshot(colorScheme: ColorScheme) -> UIImage? {
         guard let size = size else { return nil }
 
         let frame = CGRect(origin: .zero, size: size)
@@ -368,7 +368,7 @@ public struct FlowLink<Label>: View where Label: View {
                 guard !hasSiblingElement else { return }
                 Task {
                     if configuration.transitionFromSnapshot {
-                        await initSnapshots()
+                        initSnapshots()
                         self.context?.snapshotDict = snapshots
                         self.context?.snapshot = snapshots[colorScheme]
                     }
@@ -458,10 +458,10 @@ public struct FlowLink<Label>: View where Label: View {
         }
     }
 
-    private func initSnapshots() async {
+    private func initSnapshots() {
         guard snapshots.isEmpty else { return }
-        let lightImage = await createSnapshot(colorScheme: .light)
-        let darkImage = await createSnapshot(colorScheme: .dark)
+        let lightImage = createSnapshot(colorScheme: .light)
+        let darkImage = createSnapshot(colorScheme: .dark)
         self.snapshots[.light] = lightImage
         self.snapshots[.dark] = darkImage
     }