Analytics improvement (#9)

Co-authored-by: Quentin Bandera <q.bandera@openium.fr>
Reviewed-on: #9
Co-authored-by: Loris Perret <l.perret@openium.fr>
Co-committed-by: Loris Perret <l.perret@openium.fr>

Sources/ResgenSwift/Analytics/Generator/AnalyticsGenerator.swift

@@ -89,36 +89,76 @@ enum AnalyticsGenerator {
     ) -> String {
         """
         // Generated by ResgenSwift.\(Analytics.toolName) \(ResgenSwiftVersion)
-
-        \(getImport(targets: targets))
-
-        \(getAnalyticsProtocol(targets: targets))
+        
+        \(Self.getImport(targets: targets))
+        
+        \(Self.getAnalyticsProtocol(targets: targets))
+        
+        \(Self.getTrackerTypeEnum())
+        
         // MARK: - Manager
 
         class AnalyticsManager {
-
+        
             static var shared = AnalyticsManager()
+        
+            private init() {}
 
             // MARK: - Properties
-
-            var managers: [AnalyticsManagerProtocol] = []
-
-            \(getEnabledContent())
-
-            \(getAnalyticsProperties(targets: targets))
-
-            \(getPrivateLogFunction())
+        
+            var managers: [TrackerType: AnalyticsManagerProtocol] = [:]
+        
+            \(Self.getEnabledContent())
+        
+            \(Self.getAnalyticsProperties(targets: targets))
+                
+            \(Self.getPrivateLogFunction())
         """
     }
-
+    
+    private static func getTrackerTypeEnum() -> String {
+        var result: [String] = []
+        TrackerType.allCases.forEach { type in
+            result.append("    case \(type)")
+        }
+        
+        return """
+        // MARK: - Traker Type
+        
+        enum TrackerType: CaseIterable {
+        \(result.joined(separator: "\n"))
+        }
+        """
+    }
+    
     private static func getEnabledContent() -> String {
         """
-        private var isEnabled: Bool = true
-
+        private var isEnabled: Bool {
+                if ProcessInfo.processInfo.environment["XCODE_RUNNING_FOR_PREVIEWS"] == "1" {
+                    false
+                } else {
+                    true
+                }
+            }
+        
             // MARK: - Methods
-
-            func setAnalyticsEnabled(_ enable: Bool) {
-                isEnabled = enable
+        
+            private func setAnalytics(enable: Bool, _ analytics: [TrackerType]) {
+                managers.forEach { (key, value) in
+                    if analytics.contains(where: { type in
+                        type == key
+                    }) {
+                        value.setEnable(enable)
+                    }
+                }
+            }
+        
+            func enableAnalytics(_ analytics: [TrackerType] = TrackerType.allCases) {
+                setAnalytics(enable: true, analytics)
+            }
+        
+            func disableAnalytics(_ analytics: [TrackerType] = TrackerType.allCases) {
+                setAnalytics(enable: false, analytics)
             }
         """
     }
@@ -126,6 +166,8 @@ enum AnalyticsGenerator {
     private static func getImport(targets: [TrackerType]) -> String {
         var result: [String] = []
 
+        result.append("import Foundation")
+
         if targets.contains(TrackerType.matomo) {
             result.append("import MatomoTracker")
         }
@@ -138,11 +180,19 @@ enum AnalyticsGenerator {
 
     private static func getPrivateLogFunction() -> String {
         """
-        private func logScreen(name: String, path: String) {
+        private func logScreen(
+                name: String,
+                path: String,
+                params: [String: Any]?
+            ) {
                 guard isEnabled else { return }
-
-                managers.forEach { manager in
-                    manager.logScreen(name: name, path: path)
+        
+                managers.values.forEach { manager in
+                    manager.logScreen(
+                        name: name,
+                        path: path,
+                        params: params
+                    )
                 }
             }
 
@@ -153,8 +203,8 @@ enum AnalyticsGenerator {
                 params: [String: Any]?
             ) {
                 guard isEnabled else { return }
-
-                managers.forEach { manager in
+        
+                managers.values.forEach { manager in
                     manager.logEvent(
                         name: name,
                         action: action,
@@ -179,16 +229,14 @@ enum AnalyticsGenerator {
 
         if targets.contains(TrackerType.matomo) {
             content.append("""
-                    managers.append(
-                        MatomoAnalyticsManager(
-                            siteId: siteId,
-                            url: url
-                        )
+                    managers[TrackerType.matomo] = MatomoAnalyticsManager(
+                        siteId: siteId,
+                        url: url
                     )
             """)
         }
         if targets.contains(TrackerType.firebase) {
-            content.append("        managers.append(FirebaseAnalyticsManager())")
+            content.append("        managers[TrackerType.firebase] = FirebaseAnalyticsManager()")
         }
 
         return [
@@ -204,14 +252,20 @@ enum AnalyticsGenerator {
         // MARK: - Protocol
 
         protocol AnalyticsManagerProtocol {
-
-            func logScreen(name: String, path: String)
+            func logScreen(
+                name: String,
+                path: String,
+                params: [String: Any]?
+            )
+        
             func logEvent(
                 name: String,
                 action: String,
                 category: String,
                 params: [String: Any]?
             )
+        
+            func setEnable(_ enable: Bool)
         }
 
         """