/*
 * Copyright 2023 Amazon.com, Inc. or its affiliates. All Rights Reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License").
 * You may not use this file except in compliance with the License.
 * A copy of the License is located at
 *
 *  http://aws.amazon.com/apache2.0
 *
 * or in the "license" file accompanying this file. This file is distributed
 * on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
 * express or implied. See the License for the specific language governing
 * permissions and limitations under the License.
 */

package com.amplifyframework.pushnotifications.pinpoint

import com.amplifyframework.annotations.InternalAmplifyApi
import com.amplifyframework.notifications.pushnotifications.NotificationContentProvider
import com.amplifyframework.notifications.pushnotifications.NotificationPayload

@InternalAmplifyApi
class PinpointNotificationPayload internal constructor(
    val title: String? = null,
    val body: String? = null,
    val imageUrl: String? = null,
    val action: Map<String, String> = mapOf(),
    val silentPush: Boolean = false,
    channelId: String? = null,
    targetClass: Class<*>? = null,
    contentProvider: NotificationContentProvider,
) : NotificationPayload(contentProvider, channelId, targetClass) {
    companion object {
        fun isPinpointNotificationPayload(payload: NotificationPayload) = payload.rawData.keys.any {
            it.contains(PushNotificationsConstants.PINPOINT_PREFIX)
        }

        fun fromNotificationPayload(payload: NotificationPayload): PinpointNotificationPayload? {
            if (!isPinpointNotificationPayload(payload)) return null

            val data = payload.rawData
            val title = data[PushNotificationsConstants.TITLE]
                ?: data[PushNotificationsConstants.PINPOINT_NOTIFICATION_TITLE]
            val body = data[PushNotificationsConstants.MESSAGE]
                ?: data[PushNotificationsConstants.PINPOINT_NOTIFICATION_BODY]
            val imageUrl = data[PushNotificationsConstants.IMAGEURL]
                ?: data[PushNotificationsConstants.PINPOINT_NOTIFICATION_IMAGEURL]
            val channelId = payload.channelId ?: PushNotificationsConstants.DEFAULT_NOTIFICATION_CHANNEL_ID
            val silentPush = data[PushNotificationsConstants.PINPOINT_NOTIFICATION_SILENTPUSH].equals("1")
            val action: MutableMap<String, String> = mutableMapOf()

            data[PushNotificationsConstants.PINPOINT_OPENAPP]?.let {
                action.put(PushNotificationsConstants.OPENAPP, it)
            }
            data[PushNotificationsConstants.PINPOINT_URL]?.let {
                // force HTTPS URL scheme
                val urlHttps = it.replaceFirst("http://", "https://")
                action.put(PushNotificationsConstants.URL, urlHttps)
            }
            data[PushNotificationsConstants.PINPOINT_DEEPLINK]?.let {
                action.put(PushNotificationsConstants.DEEPLINK, it)
            }

            return PinpointNotificationPayload(
                title, body, imageUrl,
                action, silentPush, channelId,
                payload.targetClass, payload.contentProvider
            )
        }
    }
}