update

Author: tchowski

example/ios/Podfile.lock

@@ -25,4 +25,4 @@ SPEC CHECKSUMS:
 
 PODFILE CHECKSUM: 819463e6a0290f5a72f145ba7cde16e8b6ef0796
 
-COCOAPODS: 1.13.0
+COCOAPODS: 1.15.2

example/lib/main.dart

@@ -28,6 +28,20 @@ class _Home extends StatefulWidget {
 }
 
 class _HomeState extends State<_Home> {
+  bool hasScreenDetected = false;
+  @override
+  void initState() {
+    // TODO: implement initState
+    super.initState();
+    ScreenshotDetector.instance.startListening(() {});
+  }
+
+  @override
+  void dispose() {
+    ScreenshotDetector.instance.dispose();
+    super.dispose();
+  }
+
   @override
   Widget build(BuildContext context) {
     return Scaffold(
@@ -39,15 +53,37 @@ class _HomeState extends State<_Home> {
           crossAxisAlignment: CrossAxisAlignment.center,
           mainAxisAlignment: MainAxisAlignment.center,
           children: [
-            ScreenShotDetectorWrapper(
-              onScreenshot: const Text('Screenshot Detected'),
-              child: TextButton(
-                onPressed: () => {},
-                child: const Text(
-                  'Share',
-                ),
-              ),
-            )
+            if (hasScreenDetected) const Text('Screenshot Detected'),
+            TextButton(
+                onPressed: () => Navigator.of(context).push(MaterialPageRoute(builder: (context) => const Page1())),
+                child: const Text('push'))
+          ],
+        ),
+      ),
+    );
+  }
+}
+
+class Page1 extends StatefulWidget {
+  const Page1({super.key});
+
+  @override
+  State<Page1> createState() => _Page1State();
+}
+
+class _Page1State extends State<Page1> {
+  @override
+  Widget build(BuildContext context) {
+    return Scaffold(
+      appBar: AppBar(
+        title: const Text('Plugin example app'),
+      ),
+      body: const Center(
+        child: Column(
+          crossAxisAlignment: CrossAxisAlignment.center,
+          mainAxisAlignment: MainAxisAlignment.center,
+          children: [
+            Text('Page 1'),
           ],
         ),
       ),

example/pubspec.lock

@@ -192,7 +192,7 @@ packages:
       path: ".."
       relative: true
     source: path
-    version: "0.0.1"
+    version: "1.0.0"
   sky_engine:
     dependency: transitive
     description: flutter
@@ -280,4 +280,4 @@ packages:
     version: "3.0.3"
 sdks:
   dart: ">=3.3.1 <4.0.0"
-  flutter: ">=3.3.0"
+  flutter: ">=3.19.3"

ios/Classes/ScreenshotDetectorPlugin.swift

@@ -8,6 +8,18 @@ public class ScreenshotDetectorPlugin: NSObject, FlutterPlugin, FlutterStreamHan
         let instance = ScreenshotDetectorPlugin()
         let eventChannel = FlutterEventChannel(name: "screenshot_detector", binaryMessenger: registrar.messenger())
         eventChannel.setStreamHandler(instance)
+
+        let methodChannel = FlutterMethodChannel(name: "screenshot_detector_method", binaryMessenger: registrar.messenger())
+        registrar.addMethodCallDelegate(instance, channel: methodChannel)
+    }
+
+    public func handle(_ call: FlutterMethodCall, result: @escaping FlutterResult) {
+        if call.method == "cancelStream" {
+            onCancel(withArguments: nil)
+            result(nil)
+        } else {
+            result(FlutterMethodNotImplemented)
+        }
     }
 
     public func onListen(withArguments arguments: Any?, eventSink events: @escaping FlutterEventSink) -> FlutterError? {

lib/screenshot_detector.dart

@@ -16,7 +16,7 @@ class ScreenshotDetector {
         onScreenshotController.add(true);
       });
 
-  void disposeStream() {
+  void dispose() {
     onScreenshotController.close();
     ScreenshotDetectorPlatform.instance.dispose();
   }
@@ -41,6 +41,12 @@ class _ScreenShotDetectorWrapperState extends State<ScreenShotDetectorWrapper> {
     screenshotDetector._startListeningStream();
   }
 
+  @override
+  void dispose() {
+    screenshotDetector.dispose();
+    super.dispose();
+  }
+
   @override
   Widget build(BuildContext context) {
     return StreamBuilder<bool>(

lib/screenshot_detector_method_channel.dart

@@ -10,6 +10,8 @@ class MethodChannelScreenshotDetector extends ScreenshotDetectorPlatform {
   /// The method channel used to interact with the native platform.
   @visibleForTesting
   static const EventChannel eventChannel = EventChannel('screenshot_detector');
+  static const MethodChannel _methodChannel = MethodChannel('screenshot_detector_method');
+
   StreamSubscription<dynamic>? _eventSubscription;
 
   @override
@@ -22,7 +24,8 @@ class MethodChannelScreenshotDetector extends ScreenshotDetectorPlatform {
   }
 
   @override
-  void dispose() {
+  void dispose() async {
+    await _methodChannel.invokeMethod('dispose');
     _eventSubscription?.cancel();
     _eventSubscription = null;
   }

lib/screenshot_detector_platform_interface.dart

@@ -27,5 +27,5 @@ abstract class ScreenshotDetectorPlatform extends PlatformInterface {
     return instance.startListening(onScreenshot);
   }
 
-  void dispose();
+  void dispose() => instance.dispose();
 }