// Copyright 2020 Amazon.com, Inc. or its affiliates. All Rights Reserved.
// SPDX-License-Identifier: Apache-2.0

package software.aws.toolkits.jetbrains.uitests.fixtures

import com.intellij.remoterobot.RemoteRobot
import com.intellij.remoterobot.data.RemoteComponent
import com.intellij.remoterobot.fixtures.ComponentFixture
import com.intellij.remoterobot.fixtures.ContainerFixture
import com.intellij.remoterobot.fixtures.FixtureName
import com.intellij.remoterobot.fixtures.JTextFieldFixture
import com.intellij.remoterobot.search.locators.byXpath
import com.intellij.remoterobot.stepsProcessing.step
import com.intellij.remoterobot.utils.keyboard
import com.intellij.remoterobot.utils.waitForIgnoringError
import org.assertj.swing.timing.Pause
import java.io.File
import java.nio.file.Path
import java.time.Duration

fun ContainerFixture.fileBrowser(
    partialTitle: String,
    timeout: Duration = Duration.ofSeconds(20),
    function: FileBrowserFixture.() -> Unit = {}
) {
    step("Search for file explorer with title matching '$partialTitle'") {
        val dialog = find<FileBrowserFixture>(DialogFixture.byTitleContains(partialTitle), timeout)

        dialog.apply(function)

        if (dialog.isShowing) {
            dialog.close()
        }

        dialog
    }
}

@FixtureName("FileBrowser")
class FileBrowserFixture(
    remoteRobot: RemoteRobot,
    remoteComponent: RemoteComponent
) : DialogFixture(remoteRobot, remoteComponent) {
    private val treeXpath = byXpath("//div[@class='Tree']")
    private val tree by lazy {
        find<JTreeFixture>(treeXpath, Duration.ofSeconds(10))
    }

    fun selectFile(path: Path) {
        val absolutePath = path.toAbsolutePath()
        step("Select $absolutePath") {
            step("Refresh file explorer to make sure the file ${path.fileName} is loaded") {
                waitForIgnoringError(duration = Duration.ofSeconds(30), interval = Duration.ofSeconds(10)) {
                    setFilePath(absolutePath)

                    if (findAll<ComponentFixture>(treeXpath).isNotEmpty()) {
                        findAndClick("//div[@accessiblename='Refresh']")
                        tree.collectSelectedPaths().any { it == absolutePath.toParts() }
                    } else {
                        keyboard {
                            enter()
                        }
                        true
                    }
                }
            }

            pressOk()
        }
    }

    private fun setFilePath(path: Path) {
        step("Set file path to $path") {
            val pathBox: JTextFieldFixture = if (remoteRobot.ideMajorVersion() <= 202) {
                find(byXpath("//div[@class='JTextField']"), Duration.ofSeconds(5))
            } else {
                find(byXpath("//div[@class='BorderlessTextField']"), Duration.ofSeconds(5))
            }
            // clear the path box then type in the path. needs to be set slowly and not typed due to the tree can steal focus when loading.
            // so break the path up and set it in segments
            pathBox.text = ""
            step("Type path '$path'") {
                val pathSoFar = mutableListOf<String>()
                path.toParts().forEach {
                    pathSoFar += it
                    pathBox.text = pathSoFar.joinToString(separator = "/").replace("//", "/")
                    Pause.pause(100)
                }
            }
        }
    }

    private fun Path.toParts(): List<String> {
        val parts = this.toString().split(File.separatorChar).toMutableList()
        // Need to re-add "/" on linux
        if (this.toString().startsWith(File.separatorChar)) {
            parts[0] = File.separator
        }
        return parts
    }
}