From 9df53d7e775671cc5becd192b362c3958ddb1c24 Mon Sep 17 00:00:00 2001
From: Chris Leonavicius <chris.leonavicius@amplitude.com>
Date: Wed, 2 Oct 2024 13:46:56 -0700
Subject: [PATCH] fix: stop session replay when plugin/middleware is shut down

---
 Package.resolved                                     | 12 ++++++------
 Package.swift                                        |  2 +-
 .../AmplitudeSegmentSessionReplayPlugin.swift        |  4 ++++
 .../AmplitudeiOSSessionReplayMiddleware.swift        |  4 ++++
 4 files changed, 15 insertions(+), 7 deletions(-)

diff --git a/Package.resolved b/Package.resolved
index b4d36f6..bd31ccb 100644
--- a/Package.resolved
+++ b/Package.resolved
@@ -5,8 +5,8 @@
       "kind" : "remoteSourceControl",
       "location" : "https://github.com/amplitude/Amplitude-iOS.git",
       "state" : {
-        "revision" : "681c49824a2b011256f3e66ab910f2e514629bd3",
-        "version" : "8.21.0"
+        "revision" : "4471ff73fffdff9a9b256422a0a0b9233582f33d",
+        "version" : "8.22.0"
       }
     },
     {
@@ -14,8 +14,8 @@
       "kind" : "remoteSourceControl",
       "location" : "https://github.com/amplitude/Amplitude-Swift.git",
       "state" : {
-        "revision" : "ce41c105bf9f0a1518f8bd6e594177c69512f054",
-        "version" : "1.9.2"
+        "revision" : "3ac8de1f8fed85fb9907b34579c1500c218dfedb",
+        "version" : "1.9.3"
       }
     },
     {
@@ -32,8 +32,8 @@
       "kind" : "remoteSourceControl",
       "location" : "https://github.com/segmentio/analytics-swift",
       "state" : {
-        "revision" : "f6f3a7fafcace7bcdbda329f9b309f31bf9a42a1",
-        "version" : "1.5.11"
+        "revision" : "41df3103293bbed6f17489fcc28e16a6e495168d",
+        "version" : "1.6.1"
       }
     },
     {
diff --git a/Package.swift b/Package.swift
index a8476db..c5d3a47 100644
--- a/Package.swift
+++ b/Package.swift
@@ -24,7 +24,7 @@ let package = Package(
     ],
     dependencies: [
         .package(url: "https://github.com/amplitude/Amplitude-Swift.git", from: "1.9.2"),
-        .package(url: "https://github.com/amplitude/Amplitude-iOS.git", from: "8.21.0"),
+        .package(url: "https://github.com/amplitude/Amplitude-iOS.git", from: "8.22.0"),
         .package(url: "https://github.com/segmentio/analytics-swift", "1.5.0"..<"2.0.0"),
     ],
     targets: [
diff --git a/Sources/AmplitudeSegmentSessionReplayPlugin/AmplitudeSegmentSessionReplayPlugin.swift b/Sources/AmplitudeSegmentSessionReplayPlugin/AmplitudeSegmentSessionReplayPlugin.swift
index 95eea58..feb49c1 100644
--- a/Sources/AmplitudeSegmentSessionReplayPlugin/AmplitudeSegmentSessionReplayPlugin.swift
+++ b/Sources/AmplitudeSegmentSessionReplayPlugin/AmplitudeSegmentSessionReplayPlugin.swift
@@ -81,4 +81,8 @@ public class AmplitudeSegmentSessionReplayPlugin: Plugin {
 
         return event
     }
+
+    public func shutdown() {
+        sessionReplay.stop()
+    }
 }
diff --git a/Sources/AmplitudeiOSSessionReplayMiddleware/AmplitudeiOSSessionReplayMiddleware.swift b/Sources/AmplitudeiOSSessionReplayMiddleware/AmplitudeiOSSessionReplayMiddleware.swift
index adf51e9..3eee5f1 100644
--- a/Sources/AmplitudeiOSSessionReplayMiddleware/AmplitudeiOSSessionReplayMiddleware.swift
+++ b/Sources/AmplitudeiOSSessionReplayMiddleware/AmplitudeiOSSessionReplayMiddleware.swift
@@ -85,4 +85,8 @@ import AmplitudeSessionReplay
             sessionReplay?.flush()
         }
     }
+
+    public func amplitudeDidRemove(_ amplitude: Amplitude) {
+        sessionReplay?.stop()
+    }
 }