/* 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 org.mozilla.samples.browser.request

import android.content.Context
import mozilla.components.browser.errorpages.ErrorPages
import mozilla.components.browser.errorpages.ErrorType
import mozilla.components.concept.engine.EngineSession
import mozilla.components.concept.engine.request.RequestInterceptor
import mozilla.components.concept.engine.request.RequestInterceptor.InterceptionResponse
import mozilla.components.concept.engine.request.RequestInterceptor.ErrorResponse
import org.mozilla.samples.browser.ext.components

class SampleRequestInterceptor(val context: Context) : RequestInterceptor {

    override fun onLoadRequest(
        engineSession: EngineSession,
        uri: String,
        lastUri: String?,
        hasUserGesture: Boolean,
        isSameDomain: Boolean,
        isRedirect: Boolean,
        isDirectNavigation: Boolean,
        isSubframeRequest: Boolean
    ): InterceptionResponse? {
        return when (uri) {
            "sample:about" -> InterceptionResponse.Content("<h1>I am the sample browser</h1>")
            else -> {
                var response = context.components.appLinksInterceptor.onLoadRequest(
                        engineSession, uri, lastUri, hasUserGesture, isSameDomain, isRedirect,
                        isDirectNavigation, isSubframeRequest
                )

                if (response == null && !isDirectNavigation) {
                    response = context.components.webAppInterceptor.onLoadRequest(
                        engineSession, uri, lastUri, hasUserGesture, isSameDomain, isRedirect,
                        isDirectNavigation, isSubframeRequest
                    )
                }

                response
            }
        }
    }

    override fun onErrorRequest(
        session: EngineSession,
        errorType: ErrorType,
        uri: String?
    ): ErrorResponse {
        val errorPage = ErrorPages.createErrorPage(context, errorType, uri)
        return ErrorResponse.Content(errorPage)
    }
}
