import RiviumPush

func application(_ application: UIApplication, didFinishLaunchingWithOptions launchOptions: [UIApplication.LaunchOptionsKey: Any]?) -> Bool {

    let config = RiviumPushConfig(apiKey: "your_api_key_here")
    RiviumPush.shared.initialize(config: config)
    RiviumPush.shared.delegate = self

    // Request notification permission
    UNUserNotificationCenter.current().requestAuthorization(options: [.alert, .sound, .badge]) { granted, _ in
        if granted {
            DispatchQueue.main.async {
                UIApplication.shared.registerForRemoteNotifications()
            }
        }
    }

    // Register device
    RiviumPush.shared.register(userId: "user_123") // userId is optional

    return true
}

func application(_ application: UIApplication, didRegisterForRemoteNotificationsWithDeviceToken deviceToken: Data) {
    RiviumPush.shared.setAPNsToken(deviceToken)
}

extension AppDelegate: RiviumPushDelegate {

    func riviumPush(_ riviumPush: RiviumPush, didRegisterWithDeviceId deviceId: String) {
        print("Registered: \(deviceId)")
    }

    func riviumPush(_ riviumPush: RiviumPush, didReceiveMessage message: RiviumPushMessage) {
        print("Message: \(message.title ?? "")")
    }

    func riviumPush(_ riviumPush: RiviumPush, didTapNotification message: RiviumPushMessage) {
        // Handle notification tap
    }

    func riviumPush(_ riviumPush: RiviumPush, didChangeConnectionState connected: Bool) {
        print("Connected: \(connected)")
    }

    func riviumPush(_ riviumPush: RiviumPush, didFailWithError error: Error) {
        print("Error: \(error)")
    }
}

// Set user ID (after login)
RiviumPush.shared.setUserId("user_123")

// Clear user ID (after logout)
RiviumPush.shared.clearUserId()

// Subscribe
RiviumPush.shared.subscribeTopic("news")

// Unsubscribe
RiviumPush.shared.unsubscribeTopic("news")

// Trigger messages
RiviumPush.shared.triggerInAppOnAppOpen()
RiviumPush.shared.triggerInAppEvent("viewed_product")

// Set up callback
RiviumPush.shared.setInAppMessageCallback(handler)

// Fetch messages
RiviumPush.shared.getInboxMessages(
    filter: InboxFilter(limit: 50),
    onSuccess: { response in
        let messages = response.messages
        let unread = response.unreadCount
    },
    onError: { error in print(error) }
)

// Real-time updates
RiviumPush.shared.setInboxCallback(handler)

// Mark as read
RiviumPush.shared.markInboxMessageAsRead(messageId: "msg_123")

// Get unread count
let count = RiviumPush.shared.getInboxManager().getUnreadCount()

let manager = RiviumPush.shared.getABTestingManager()

// Get active tests
manager.getActiveTests { result in
    // handle tests
}

// Get variant assignment
manager.getVariant(testId: "test_123") { result in
    if case .success(let variant) = result {
        print("Assigned to: \(variant.variantName)")
    }
}

// Track events
manager.trackEvent(testId: "test_123", variantId: "variant_456", event: .clicked) { _ in }

let config = RiviumPushConfig.builder(apiKey: "your_key")
    .usePushKit(false)           // VoIP push (only for calling apps)
    .useAPNs(true)               // Standard APNs (recommended)
    .showNotificationInForeground(true)
    .autoConnect(true)
    .autoReconnect(true)
    .build()