/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

package mozilla.components.feature.sitepermissions

import mozilla.components.concept.engine.permission.Permission
import mozilla.components.concept.engine.permission.PermissionRequest
import mozilla.components.concept.engine.permission.SitePermissions
import mozilla.components.concept.engine.permission.SitePermissions.AutoplayStatus
import mozilla.components.feature.sitepermissions.SitePermissionsRules.Action.ASK_TO_ALLOW
import mozilla.components.feature.sitepermissions.SitePermissionsRules.Action.BLOCKED

/**
 * Indicate how site permissions must behave by permission category.
 */
data class SitePermissionsRules constructor(
    val camera: Action,
    val location: Action,
    val notification: Action,
    val microphone: Action,
    val autoplayAudible: AutoplayAction,
    val autoplayInaudible: AutoplayAction,
    val persistentStorage: Action,
    val mediaKeySystemAccess: Action,
    val crossOriginStorageAccess: Action,
) {

    enum class Action {
        ALLOWED, BLOCKED, ASK_TO_ALLOW;

        fun toStatus(): SitePermissions.Status = when (this) {
            ALLOWED -> SitePermissions.Status.ALLOWED
            BLOCKED -> SitePermissions.Status.BLOCKED
            ASK_TO_ALLOW -> SitePermissions.Status.NO_DECISION
        }
    }

    /**
     * Autoplay requests will never prompt the user
     */
    enum class AutoplayAction {
        ALLOWED, BLOCKED;

        internal fun toAction(): Action = when (this) {
            ALLOWED -> Action.ALLOWED
            BLOCKED -> Action.BLOCKED
        }
        /**
         * Convert from an AutoplayAction to an AutoplayStatus.
         */
        fun toAutoplayStatus(): AutoplayStatus = when (this) {
            ALLOWED -> AutoplayStatus.ALLOWED
            BLOCKED -> AutoplayStatus.BLOCKED
        }
    }

    internal fun getActionFrom(request: PermissionRequest): Action {
        return if (request.containsVideoAndAudioSources()) {
            getActionForCombinedPermission()
        } else {
            getActionForSinglePermission(request.permissions.first())
        }
    }

    private fun getActionForSinglePermission(permission: Permission): Action {
        return when (permission) {
            is Permission.ContentGeoLocation -> {
                location
            }
            is Permission.ContentNotification -> {
                notification
            }
            is Permission.ContentPersistentStorage -> {
                persistentStorage
            }
            is Permission.ContentAudioCapture, is Permission.ContentAudioMicrophone -> {
                microphone
            }
            is Permission.ContentVideoCamera, is Permission.ContentVideoCapture -> {
                camera
            }
            is Permission.ContentAutoPlayAudible -> {
                autoplayAudible.toAction()
            }
            is Permission.ContentAutoPlayInaudible -> {
                autoplayInaudible.toAction()
            }
            is Permission.ContentMediaKeySystemAccess -> {
                mediaKeySystemAccess
            }
            is Permission.ContentCrossOriginStorageAccess -> {
                crossOriginStorageAccess
            }
            else -> ASK_TO_ALLOW
        }
    }

    private fun getActionForCombinedPermission(): Action {
        return if (camera == BLOCKED || microphone == BLOCKED) {
            BLOCKED
        } else {
            ASK_TO_ALLOW
        }
    }

    /**
     * Converts a [SitePermissionsRules] object into a [SitePermissions] .
     */
    fun toSitePermissions(origin: String, savedAt: Long = System.currentTimeMillis()): SitePermissions {
        return SitePermissions(
            origin = origin,
            location = location.toStatus(),
            notification = notification.toStatus(),
            microphone = microphone.toStatus(),
            camera = camera.toStatus(),
            autoplayAudible = autoplayAudible.toAutoplayStatus(),
            autoplayInaudible = autoplayInaudible.toAutoplayStatus(),
            localStorage = persistentStorage.toStatus(),
            mediaKeySystemAccess = mediaKeySystemAccess.toStatus(),
            crossOriginStorageAccess = crossOriginStorageAccess.toStatus(),
            savedAt = savedAt
        )
    }
}
