From ffa9328dff4e1830477328e729b6f3cb120e9671 Mon Sep 17 00:00:00 2001
From: Maximilian Fischer <github@maaeps.de>
Date: Wed, 21 Feb 2024 22:28:09 +0100
Subject: [PATCH] add childWrapper

---
 lib/photo_view_gallery.dart | 24 ++++++++++++++++++++++--
 1 file changed, 22 insertions(+), 2 deletions(-)

diff --git a/lib/photo_view_gallery.dart b/lib/photo_view_gallery.dart
index ad3bda64..e6bb7a5b 100644
--- a/lib/photo_view_gallery.dart
+++ b/lib/photo_view_gallery.dart
@@ -15,6 +15,7 @@ import 'package:photo_view/src/controller/photo_view_controller.dart';
 import 'package:photo_view/src/controller/photo_view_scalestate_controller.dart';
 import 'package:photo_view/src/core/photo_view_gesture_detector.dart';
 import 'package:photo_view/src/photo_view_scale_state.dart';
+import 'package:photo_view/src/photo_view_wrappers.dart';
 import 'package:photo_view/src/utils/photo_view_hero_attributes.dart';
 
 /// A type definition for a [Function] that receives a index after a page change in [PhotoViewGallery]
@@ -24,6 +25,11 @@ typedef PhotoViewGalleryPageChangedCallback = void Function(int index);
 typedef PhotoViewGalleryBuilder = PhotoViewGalleryPageOptions Function(
     BuildContext context, int index);
 
+/// A type definition for a [Function] that wraps the [child] in an other
+/// [Widget]. Used to apply an overlay to the items.
+typedef PhotoViewGalleryChildWrapper = Widget Function(
+    BuildContext context, int index, Widget child);
+
 /// A [StatefulWidget] that shows multiple [PhotoView] widgets in a [PageView]
 ///
 /// Some of [PhotoView] constructor options are passed direct to [PhotoViewGallery] constructor. Those options will affect the gallery in a whole.
@@ -117,6 +123,7 @@ class PhotoViewGallery extends StatefulWidget {
     this.scrollDirection = Axis.horizontal,
     this.customSize,
     this.allowImplicitScrolling = false,
+    this.childWrapper = defaultChildWrapper,
   })  : itemCount = null,
         builder = null,
         super(key: key);
@@ -141,11 +148,17 @@ class PhotoViewGallery extends StatefulWidget {
     this.scrollDirection = Axis.horizontal,
     this.customSize,
     this.allowImplicitScrolling = false,
+    this.childWrapper = defaultChildWrapper,
   })  : pageOptions = null,
         assert(itemCount != null),
         assert(builder != null),
         super(key: key);
 
+  static Widget defaultChildWrapper(
+      BuildContext context, int index, Widget child) {
+    return child;
+  }
+
   /// A list of options to describe the items in the gallery
   final List<PhotoViewGalleryPageOptions>? pageOptions;
 
@@ -155,6 +168,9 @@ class PhotoViewGallery extends StatefulWidget {
   /// Called to build items for the gallery when using [PhotoViewGallery.builder]
   final PhotoViewGalleryBuilder? builder;
 
+  /// Wraps each item. Useful for applying an overlay to all pictures.
+  final PhotoViewGalleryChildWrapper childWrapper;
+
   /// [ScrollPhysics] for the internal [PageView]
   final ScrollPhysics? scrollPhysics;
 
@@ -300,8 +316,12 @@ class _PhotoViewGalleryState extends State<PhotoViewGallery> {
             errorBuilder: pageOption.errorBuilder,
           );
 
-    return ClipRect(
-      child: photoView,
+    return widget.childWrapper(
+      context,
+      index,
+      ClipRect(
+        child: photoView,
+      ),
     );
   }