Hold on to dart-side refs for shader libraries

Author: bdero

lib/gpu/lib/src/shader_library.dart

@@ -18,14 +18,26 @@ base class ShaderLibrary extends NativeFieldWrapperClass1 {
 
   ShaderLibrary._();
 
+  // Hold a Dart-side reference to shaders in the library as they're wrapped for
+  // the first time. This prevents the wrapper from getting prematurely
+  // destroyed.
+  Map<String, Shader> shaders_ = {};
+
   Shader? operator [](String shaderName) {
     // This `flutter_gpu` library isn't always registered as part of the builtin
     // DartClassLibrary, and so we can't instantiate the Dart classes on the
     // engine side.
     // Providing a new wrapper to [_getShader] for wrapping the native
     // counterpart (if it hasn't been wrapped already) is a hack to work around
     // this.
-    return _getShader(shaderName, Shader._());
+    Shader? result = shaders_[shaderName];
+    if (result == null) {
+      result = _getShader(shaderName, Shader._());
+      if (result != null) {
+        shaders_[shaderName] = result;
+      }
+    }
+    return result;
   }
 
   @Native<Handle Function(Handle, Handle)>(