feat: updated screenshot manager

Author: jhomlala

example/lib/screenshot_example.dart

@@ -57,13 +57,13 @@ class _MyAppState extends State<MyApp> {
   Widget build(BuildContext context) {
     return MaterialApp(
       navigatorKey: Catcher.navigatorKey,
-      home: CatcherScreenshot(
-        catcher: Catcher.getInstance(),
-        child: Scaffold(
-          appBar: AppBar(
-            title: const Text('Plugin example app'),
-          ),
-          body: const ChildWidget(),
+      home: Scaffold(
+        appBar: AppBar(
+          title: const Text('Plugin example app'),
+        ),
+        body: CatcherScreenshot(
+          catcher: Catcher.getInstance(),
+          child: const ChildWidget(),
         ),
       ),
     );

lib/core/catcher_screenshot_manager.dart

@@ -55,46 +55,60 @@ class CatcherScreenshotManager {
 
   Future<Uint8List?> _capture({
     double? pixelRatio,
-    Duration? delay = const Duration(milliseconds: 20),
+    Duration delay = const Duration(milliseconds: 20),
   }) {
-    return Future.delayed(delay ?? const Duration(milliseconds: 20), () async {
-      final image = await _captureAsUiImage(
-        delay: Duration.zero,
-        pixelRatio: pixelRatio,
-      );
-      final byteData = await image.toByteData(format: ui.ImageByteFormat.png);
-      final pngBytes = byteData?.buffer.asUint8List();
+    //Delay is required. See Issue https://github.com/flutter/flutter/issues/22308
+    return Future.delayed(delay, () async {
+      try {
+        final image = await captureAsUiImage(
+          delay: Duration.zero,
+          pixelRatio: pixelRatio,
+        );
+        final byteData =
+            await image?.toByteData(format: ui.ImageByteFormat.png);
+        image?.dispose();
+
+        final pngBytes = byteData?.buffer.asUint8List();
 
-      return pngBytes;
+        return pngBytes;
+      } catch (exception) {
+        _logger.severe('Failed to capture screenshot: $exception');
+      }
+      return null;
     });
   }
 
-  Future<ui.Image> _captureAsUiImage({
-    double? pixelRatio,
-    Duration delay = const Duration(milliseconds: 100),
+  Future<ui.Image?> captureAsUiImage({
+    double? pixelRatio = 1,
+    Duration delay = const Duration(milliseconds: 20),
   }) {
+    //Delay is required. See Issue https://github.com/flutter/flutter/issues/22308
     return Future.delayed(delay, () async {
-      // ignore: cast_nullable_to_non_nullable
-
-      final renderObject = _containerKey.currentContext?.findRenderObject();
+      try {
+        final findRenderObject =
+            _containerKey.currentContext?.findRenderObject();
 
-      // ignore: unnecessary_null_comparison
-      if (renderObject == null) {
-        throw StateError('No boundary found');
-      }
-
-      final boundary = renderObject as RenderRepaintBoundary;
+        print(containerKey.currentContext);
+        print( _containerKey.currentContext?.findRenderObject());
+        if (findRenderObject == null) {
+          return null;
+        }
 
-      final context = _containerKey.currentContext;
-      var pixelRatioValue = pixelRatio;
-      if (pixelRatioValue == null) {
-        if (context != null) {
-          pixelRatioValue =
-              pixelRatioValue ?? MediaQuery.of(context).devicePixelRatio;
+        final boundary = findRenderObject as RenderRepaintBoundary;
+        final context = _containerKey.currentContext;
+        var pixelRatioValue = pixelRatio;
+        if (pixelRatio == null) {
+          if (context != null) {
+            pixelRatioValue =
+                pixelRatio ?? MediaQuery.of(context).devicePixelRatio;
+          }
         }
+        final image = await boundary.toImage(pixelRatio: pixelRatioValue ?? 1);
+        return image;
+      } catch (exception) {
+        _logger.severe('Failed to capture screenshot: $exception');
       }
-      final image = await boundary.toImage(pixelRatio: pixelRatio ?? 1);
-      return image;
+      return null;
     });
   }