[darwin] Use weak pointer to mbgl::Source

1 files changed, 8 insertions, 2 deletions

diff --git a/platform/darwin/src/MGLSource.mm b/platform/darwin/src/MGLSource.mm
index fd54627471..2c673fbb30 100644
--- a/platform/darwin/src/MGLSource.mm
+++ b/platform/darwin/src/MGLSource.mm
@@ -19,6 +19,7 @@
 
 @implementation MGLSource {
     std::unique_ptr<mbgl::style::Source> _pendingSource;
+    mapbox::base::WeakPtr<mbgl::style::Source> _weakSource;
 }
 
 
@@ -33,13 +34,18 @@
 - (instancetype)initWithRawSource:(mbgl::style::Source *)rawSource mapView:(MGLMapView *)mapView {
     NSString *identifier = @(rawSource->getID().c_str());
     if (self = [self initWithIdentifier:identifier]) {
-        _rawSource = rawSource;
-        _rawSource->peer = SourceWrapper { self };
+        _weakSource = rawSource->makeWeakPtr();
+        rawSource->peer = SourceWrapper { self };
         _mapView = mapView;
     }
     return self;
 }
 
+- (mbgl::style::Source *)rawSource
+{
+    return _weakSource.get();
+}
+
 - (instancetype)initWithPendingSource:(std::unique_ptr<mbgl::style::Source>)pendingSource {
     if (self = [self initWithRawSource:pendingSource.get() mapView:nil]) {
         _pendingSource = std::move(pendingSource);