/*
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * See the NOTICE file distributed with this work for additional
 * information regarding copyright ownership.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License 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 io.appium.espressoserver.lib.handlers

import androidx.test.espresso.web.model.Atom
import androidx.test.espresso.web.sugar.Web.WebInteraction
import androidx.test.espresso.web.sugar.Web.onWebView
import androidx.test.espresso.web.webdriver.DriverAtoms
import io.appium.espressoserver.lib.handlers.exceptions.AppiumException
import io.appium.espressoserver.lib.handlers.exceptions.InvalidArgumentException
import io.appium.espressoserver.lib.helpers.AndroidLogger
import io.appium.espressoserver.lib.helpers.ReflectionUtils.invokeInstanceMethod
import io.appium.espressoserver.lib.helpers.ReflectionUtils.invokeStaticMethod
import io.appium.espressoserver.lib.model.EspressoElement
import io.appium.espressoserver.lib.model.web.WebAtomsParams
import io.appium.espressoserver.lib.viewmatcher.withView

class WebAtoms : RequestHandler<WebAtomsParams, Any?> {

    @Throws(AppiumException::class)
    override fun handleInternal(params: WebAtomsParams): Any? {
        var webViewInteraction: WebInteraction<*>

        // TODO: Add a 'waitForDocument' feature to wait for HTML DOCUMENT to be ready

        // Initialize onWebView with web view matcher (if webviewEl provided)
        params.webviewEl.let { webviewEl ->
            AndroidLogger.info("Initializing webView interaction on webview with the element")
            val viewState = EspressoElement.getCachedViewStateById(webviewEl)
            val matcher = withView(viewState.view)
            webViewInteraction = onWebView(matcher)
        }

        // Set forceJavascript enabled if provided
        if (params.forceJavascriptEnabled) {
            webViewInteraction.forceJavascriptEnabled()
        }

        // Iterate through methodsChain and call the atoms
        params.methodChain.forEach { method ->
            val atom = invokeStaticMethod(DriverAtoms::class.java, method.atom.name, *method.atom.args) as? Atom<*>
                ?: throw InvalidArgumentException(
                    "'${method.atom.name}' did not return a valid '${Atom::class.qualifiedName}' object",
                )

            AndroidLogger.info("Calling interaction '${method.name}' with the atom '${method.atom}'")

            val res = invokeInstanceMethod(webViewInteraction, method.name, atom) as? WebInteraction<*>
                ?: throw InvalidArgumentException(
                    "'${method.name}' does not return a valid '${WebInteraction::class.qualifiedName}' object",
                )

            webViewInteraction = res
        }
        // if you don't pass any action to perform `webViewInteraction.get()` will throw error
        // java.lang.IllegalStateException: Perform or Check never called on this WebInteraction!
        // when you just want to just check element existence call with `selectActiveElement` as action
        return webViewInteraction.get()
    }
}