resgen.swift/Sources/ResgenSwift/Analytics/Generator/AnalyticsGenerator.swift

//
//  AnalyticsGenerator.swift
//
//
//  Created by Loris Perret on 08/12/2023.
//

import CoreVideo
import Foundation
import ToolCore

// Disabled cause it's a pain to handle in generated string

enum AnalyticsGenerator {

    // MARK: - Write content

    static func writeExtensionFiles(
        sections: [AnalyticsCategory],
        target: String,
        tags: [String],
        staticVar: Bool,
        extensionName: String,
        extensionFilePath: String
    ) {
        // Get target type from enum
        let targetsString: [String] = target.components(separatedBy: " ")
        let targets = {
            var targets = [TrackerType]()
            TrackerType.allCases.forEach { enumTarget in
                if targetsString.contains(enumTarget.value) {
                    targets.append(enumTarget)
                }
            }
            return targets
        }()

        // Get extension content
        let extensionFileContent = getExtensionContent(
            targets: targets,
            sections: sections,
            tags: tags,
            staticVar: staticVar,
            extensionName: extensionName
        )

        // Write content
        let extensionFilePathURL = URL(fileURLWithPath: extensionFilePath)
        do {
            try extensionFileContent.write(to: extensionFilePathURL, atomically: false, encoding: .utf8)
        } catch {
            let error = AnalyticsError.writeFile(extensionFilePath, error.localizedDescription)
            print(error.description)
            Analytics.exit(withError: error)
        }
    }

    // MARK: - Extension content

    static func getExtensionContent(
        targets: [TrackerType],
        sections: [AnalyticsCategory],
        tags: [String],
        staticVar: Bool,
        extensionName: String
    ) -> String {
        [
            getHeader(
                targets: targets,
                extensionClassname: extensionName,
                staticVar: staticVar
            ),
            getProperties(
                sections: sections,
                tags: tags,
                staticVar: staticVar
            ),
            getFooter()
        ]
            .joined(separator: "\n")
    }

    // MARK: - Extension part

    private static func getHeader(
        targets: [TrackerType],
        extensionClassname: String,
        staticVar: Bool
    ) -> String {
        """
        // Generated by ResgenSwift.\(Analytics.toolName) \(ResgenSwiftVersion)
        
        \(Self.getImport(targets: targets))
        
        \(Self.getAnalyticsProtocol(targets: targets))
        
        \(Self.getTrackerTypeEnum())
        
        // MARK: - Manager

        class AnalyticsManager {
        
            static var shared = AnalyticsManager()
        
            private init() {}

            // MARK: - Properties
        
            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 {
                if ProcessInfo.processInfo.environment["XCODE_RUNNING_FOR_PREVIEWS"] == "1" {
                    false
                } else {
                    true
                }
            }
        
            // MARK: - Methods
        
            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)
            }
        """
    }

    private static func getImport(targets: [TrackerType]) -> String {
        var result: [String] = []

        result.append("import Foundation")

        if targets.contains(TrackerType.matomo) {
            result.append("import MatomoTracker")
        }
        if targets.contains(TrackerType.firebase) {
            result.append("import FirebaseAnalytics")
        }

        return result.joined(separator: "\n")
    }

    private static func getPrivateLogFunction() -> String {
        """
        private func logScreen(
                name: String,
                path: String,
                params: [String: Any]?
            ) {
                guard isEnabled else { return }
        
                managers.values.forEach { manager in
                    manager.logScreen(
                        name: name,
                        path: path,
                        params: params
                    )
                }
            }

            private func logEvent(
                name: String,
                action: String,
                category: String,
                params: [String: Any]?
            ) {
                guard isEnabled else { return }
        
                managers.values.forEach { manager in
                    manager.logEvent(
                        name: name,
                        action: action,
                        category: category,
                        params: params
                    )
                }
            }
        """
    }

    private static func getAnalyticsProperties(targets: [TrackerType]) -> String {
        var header = ""
        var content: [String] = []
        let footer = "    }"

        if targets.contains(TrackerType.matomo) {
            header = "func configure(siteId: String, url: String) {"
        } else if targets.contains(TrackerType.firebase) {
            header = "func configure() {"
        }

        if targets.contains(TrackerType.matomo) {
            content.append("""
                    managers[TrackerType.matomo] = MatomoAnalyticsManager(
                        siteId: siteId,
                        url: url
                    )
            """)
        }
        if targets.contains(TrackerType.firebase) {
            content.append("        managers[TrackerType.firebase] = FirebaseAnalyticsManager()")
        }

        return [
            header,
            content.joined(separator: "\n"),
            footer
        ]
            .joined(separator: "\n")
    }

    private static func getAnalyticsProtocol(targets: [TrackerType]) -> String {
        let proto = """
        // MARK: - Protocol

        protocol AnalyticsManagerProtocol {
            func logScreen(
                name: String,
                path: String,
                params: [String: Any]?
            )
        
            func logEvent(
                name: String,
                action: String,
                category: String,
                params: [String: Any]?
            )
        
            func setEnable(_ enable: Bool)
        }

        """

        var result: [String] = [proto]

        if targets.contains(TrackerType.matomo) {
            result.append(MatomoGenerator.service)
        }

        if targets.contains(TrackerType.firebase) {
            result.append(FirebaseGenerator.service)
        }

        return result.joined(separator: "\n")
    }

    private static func getProperties(
        sections: [AnalyticsCategory],
        tags: [String],
        staticVar: Bool
    ) -> String {
        sections
            .compactMap { section in
                // Check that at least one string will be generated
                guard section.hasOneOrMoreMatchingTags(tags: tags) else {
                    return nil// Go to next section
                }

                var res = "\n    // MARK: - \(section.id)"
                section.definitions.forEach { definition in
                    guard definition.hasOneOrMoreMatchingTags(inputTags: tags) == true else {
                        return // Go to next definition
                    }

                    if staticVar {
                        res += "\n\n\(definition.getStaticProperty())"
                    } else {
                        res += "\n\n\(definition.getProperty())"
                    }
                }
                return res
            }
            .joined(separator: "\n")
    }

    private static func getFooter() -> String {
        """
        }

        """
    }
}