Fix telemetry opt-out through attribution dialog (better fix)

CHANGELOG.md

@@ -7,6 +7,7 @@ Mapbox welcomes participation and contributions from everyone.
 * Make `PointAnnotation.Image`'s fields public. ([#753](https://github.com/mapbox/mapbox-maps-ios/pull/753))
 * Passing an unsupported locale into `Style.localizeLabels(into:forLayerIds:)` throws an error instead of crashing. ([#752](https://github.com/mapbox/mapbox-maps-ios/pull/752))
 * Set `MapboxMap` flags during gestures and animations. ([#754](https://github.com/mapbox/mapbox-maps-ios/pull/754))
+* Fix telemetry opt-out through attribution dialog (better fix). ([#758](https://github.com/mapbox/mapbox-maps-ios/pull/758))
 
 ## 10.0.0 - October 6, 2021
 

Sources/MapboxMaps/Foundation/Events/EventsManager.swift

@@ -1,12 +1,26 @@
 import UIKit
 import MapboxMobileEvents
 
+extension UserDefaults {
+    // dynamic var name have to be the same as corresponding key in UserDefaults
+    // to make KVO observing work properly
+    @objc dynamic var MGLMapboxMetricsEnabled: Bool {
+        get {
+            return bool(forKey: #keyPath(MGLMapboxMetricsEnabled))
+        }
+        set {
+            set(newValue, forKey: #keyPath(MGLMapboxMetricsEnabled))
+        }
+    }
+}
+
 internal class EventsManager: EventsListener {
     private enum Constants {
         static let MGLAPIClientUserAgentBase = "mapbox-maps-ios"
     }
 
     var telemetry: TelemetryProtocol!
+    private var metricsEnabledObservation: NSKeyValueObservation?
 
     init(accessToken: String) {
         let sdkVersion = "10.0.0"
@@ -16,10 +30,12 @@ internal class EventsManager: EventsListener {
                                     userAgentBase: Constants.MGLAPIClientUserAgentBase,
                                     hostSDKVersion: sdkVersion)
         mmeEventsManager.skuId = "00"
-    }
 
-    init(with telemetry: TelemetryProtocol?) {
-        self.telemetry = telemetry
+        metricsEnabledObservation = UserDefaults.standard.observe(\.MGLMapboxMetricsEnabled, options: [.new, .initial]) { _, change in
+            guard let newValue = change.newValue else { return }
+            UserDefaults.mme_configuration().mme_isCollectionEnabled = newValue
+            MMEEventsManager.shared().pauseOrResumeMetricsCollectionIfRequired()
+        }
     }
 
     func push(event: EventType) {

Sources/MapboxMaps/Ornaments/MapboxInfoButtonOrnament.swift

@@ -18,7 +18,7 @@ internal class InfoButtonOrnament: UIView {
     }
 
     internal var isMetricsEnabled: Bool {
-        return UserDefaults.standard.bool(forKey: Ornaments.metricsEnabledKey)
+        return UserDefaults.standard.MGLMapboxMetricsEnabled
     }
 
     internal weak var delegate: InfoButtonOrnamentDelegate?

Sources/MapboxMaps/Ornaments/OrnamentsManager.swift

@@ -16,7 +16,6 @@ public enum OrnamentVisibility: String, Equatable {
 
 internal struct Ornaments {
     static let localizableTableName = "OrnamentsLocalizable"
-    static let metricsEnabledKey = "MGLMapboxMetricsEnabled"
     static let telemetryURL = "https://www.mapbox.com/telemetry/"
 }
 

Sources/MapboxMaps/Style/AttributionDialogManager.swift

@@ -21,10 +21,10 @@ internal class AttributionDialogManager {
 
     internal var isMetricsEnabled: Bool {
         get {
-            UserDefaults.standard.bool(forKey: Ornaments.metricsEnabledKey)
+            UserDefaults.standard.MGLMapboxMetricsEnabled
         }
         set {
-            UserDefaults.standard.set(newValue, forKey: Ornaments.metricsEnabledKey)
+            UserDefaults.standard.MGLMapboxMetricsEnabled = newValue
         }
     }
 

Tests/MapboxMapsTests/Foundation/Events/EventsManagerTests.swift

@@ -0,0 +1,27 @@
+import XCTest
+@testable import MapboxMaps
+
+final class EventsManagerTests: XCTestCase {
+
+    let eventsManager = EventsManager(accessToken: "empty")
+
+    override class func setUp() {
+        UserDefaults.mme_configuration().set(true, forKey: "MMECollectionEnabledInSimulator")
+    }
+
+    func testUserDefaultsDynamicProperty() {
+        UserDefaults.standard.MGLMapboxMetricsEnabled = true
+        XCTAssertEqual(UserDefaults.standard.bool(forKey: "MGLMapboxMetricsEnabled"), true)
+
+        UserDefaults.standard.MGLMapboxMetricsEnabled = false
+        XCTAssertEqual(UserDefaults.standard.bool(forKey: "MGLMapboxMetricsEnabled"), false)
+    }
+
+    func testMGLMapboxMetricsEqualsMMECollectionEnabled() {
+        UserDefaults.standard.MGLMapboxMetricsEnabled = true
+        XCTAssertEqual(UserDefaults.mme_configuration().mme_isCollectionEnabled, true)
+
+        UserDefaults.standard.MGLMapboxMetricsEnabled = false
+        XCTAssertEqual(UserDefaults.mme_configuration().mme_isCollectionEnabled, false)
+    }
+}

Tests/MapboxMapsTests/Ornaments/InfoButton/MapboxInfoButtonOrnamentTests.swift

@@ -36,7 +36,7 @@ class InfoButtonOrnamentTests: XCTestCase {
         infoButton.delegate = attributionDialogManager
 
         parentViewController.view.addSubview(infoButton)
-        UserDefaults.standard.set(true, forKey: Ornaments.metricsEnabledKey)
+        UserDefaults.standard.MGLMapboxMetricsEnabled = true
         infoButton.infoTapped()
 
         var infoAlert = try XCTUnwrap(parentViewController.currentAlert, "The info alert controller could not be found.")
@@ -70,7 +70,7 @@ class InfoButtonOrnamentTests: XCTestCase {
     }
 
     func testTelemetryOptIn() throws {
-        UserDefaults.standard.set(false, forKey: Ornaments.metricsEnabledKey)
+        UserDefaults.standard.MGLMapboxMetricsEnabled = false
         let infoButton = InfoButtonOrnament()
         infoButton.delegate = attributionDialogManager