aptabase-swift/Sources/Aptabase/Aptabase.swift

import Foundation

#if os(iOS) || os(visionOS)
import UIKit
#elseif os(macOS)
import AppKit
#elseif os(watchOS)
import WatchKit
#elseif os(tvOS)
import TVUIKit
#endif

/// The Aptabase client used to track events.
public class Aptabase: NSObject {
    private var env = EnvironmentInfo.current()
    private var client: AptabaseClient?

    /// The shared client instance.
    @objc public static let shared = Aptabase()

    /// Initializes the client with given App Key.
    /// - Parameters:
    ///   - appKey: The App Key to use.
    ///   - options: Optional initialization options.
    public func initialize(appKey: String, with options: InitOptions? = nil) {
        let parts = appKey.components(separatedBy: "-")
        if parts.count != 3 || hosts[parts[1]] == nil {
            debugPrint("The Aptabase App Key \(appKey) is invalid. Tracking will be disabled.")
            return
        }

        guard let baseUrl = getBaseUrl(parts[1], options?.host) else {
            return
        }

        if let trackingMode = options?.trackingMode, trackingMode != .readFromEnvironment {
            env.isDebug = trackingMode.isDebug
        }

        client = AptabaseClient(appKey: appKey, baseUrl: baseUrl, env: env, options: options)

        let notifications = NotificationCenter.default
        #if os(tvOS) || os(iOS) || os(visionOS)
        notifications.addObserver(self, selector: #selector(startPolling), name: UIApplication.willEnterForegroundNotification, object: nil)
        notifications.addObserver(self, selector: #selector(stopPolling), name: UIApplication.didEnterBackgroundNotification, object: nil)
        #elseif os(macOS)
        notifications.addObserver(self, selector: #selector(startPolling), name: NSApplication.didBecomeActiveNotification, object: nil)
        notifications.addObserver(self, selector: #selector(stopPolling), name: NSApplication.willTerminateNotification, object: nil)
        #elseif os(watchOS)
        if #available(watchOS 7.0, *) {
            notifications.addObserver(self, selector: #selector(startPolling), name: WKExtension.applicationWillEnterForegroundNotification, object: nil)
            notifications.addObserver(self, selector: #selector(stopPolling), name: WKExtension.applicationDidEnterBackgroundNotification, object: nil)
        }
        #endif
    }

    /// Track an event using given properties.
    /// - Parameters:
    ///   - eventName: The name of the event to track.
    ///   - props: Additional given properties.
    public func trackEvent(_ eventName: String, with props: [String: Value] = [:]) {
        guard let codable = toCodableProps(from: props) else {
            return
        }

        enqueueEvent(eventName, with: codable)
    }

    /// Initializes the client with given App Key.
    /// - Parameter appKey: The App Key to use.
    @objc public func initialize(appKey: String) {
        initialize(appKey: appKey, with: nil)
    }

    /// Initializes the client with given App Key.
    /// - Parameters:
    ///   - appKey: The App Key to use.
    ///   - options: Optional initialization options.
    @objc public func initialize(appKey: String, options: InitOptions?) {
        initialize(appKey: appKey, with: options)
    }

    /// Track an event using given properties.
    /// - Parameters:
    ///   - eventName: The name of the event to track.
    ///   - props: Additional given properties.
    @objc public func trackEvent(_ eventName: String, with props: [String: Any] = [:]) {
        guard let codable = toCodableProps(from: props) else {
            return
        }

        enqueueEvent(eventName, with: codable)
    }

    /// Forces all queued events to be sent to the server
    @objc public func flush() {
        Task {
            await self.client?.flush()
        }
    }

    private func enqueueEvent(_ eventName: String, with props: [String: AnyCodableValue] = [:]) {
        guard let client else {
            return
        }

        client.trackEvent(eventName, with: props)
    }

    @objc private func startPolling() {
        client?.startPolling()
    }

    @objc private func stopPolling() {
        client?.stopPolling()
    }

    private var hosts = [
        "US": "https://us.aptabase.com",
        "EU": "https://eu.aptabase.com",
        "DEV": "http://localhost:3000",
        "SH": ""
    ]

    private func getBaseUrl(_ region: String, _ host: String?) -> String? {
        guard var baseURL = hosts[region] else { return nil }
        if region == "SH" {
            guard let host else {
                debugPrint("Aptabase: Host parameter must be defined when using Self-Hosted App Key. Tracking will be disabled.")
                return nil
            }
            baseURL = host
        }

        return baseURL
    }

    private func toCodableProps(from props: [String: Any]) -> [String: AnyCodableValue]? {
        var codableProps: [String: AnyCodableValue] = [:]
        for (key, value) in props {
            if let intValue = value as? Int {
                codableProps[key] = .integer(intValue)
            } else if let doubleValue = value as? Double {
                codableProps[key] = .double(doubleValue)
            } else if let stringValue = value as? String {
                codableProps[key] = .string(stringValue)
            } else if let floatValue = value as? Float {
                codableProps[key] = .float(floatValue)
            } else if let boolValue = value as? Bool {
                codableProps[key] = .boolean(boolValue)
            } else {
                debugPrint("Aptabase: Props with key \(key) has an unsupported value type. Supported types are: String, Int, Double, Float and Boolean. Event will be discarded")
                return nil
            }
        }
        return codableProps
    }
}