extraxtExpressionUi.kt

/*
 * Use of this source code is governed by the MIT license that can be
 * found in the LICENSE file.
 */

package org.rust.ide.refactoring

import com.intellij.openapi.editor.Editor
import com.intellij.openapi.project.Project
import com.intellij.openapi.util.Pass
import com.intellij.openapiext.isUnitTestMode
import com.intellij.refactoring.IntroduceTargetChooser
import com.intellij.refactoring.RefactoringBundle
import com.intellij.refactoring.introduce.inplace.OccurrencesChooser
import com.intellij.refactoring.util.CommonRefactoringUtil
import org.jetbrains.annotations.TestOnly
import org.rust.lang.core.psi.RsExpr
import org.rust.lang.core.psi.RsFunction


fun showExpressionChooser(
    editor: Editor,
    exprs: List<RsExpr>,
    callback: (RsExpr) -> Unit
) {
    if (isUnitTestMode) {
        callback(MOCK!!.chooseTarget(exprs))
    } else {
        IntroduceTargetChooser.showChooser(editor, exprs, callback.asPass) { it.text }
    }
}

fun showOccurrencesChooser(
    editor: Editor,
    expr: RsExpr,
    occurrences: List<RsExpr>,
    callback: (List<RsExpr>) -> Unit
) {
    if (isUnitTestMode && occurrences.size > 1) {
        callback(MOCK!!.chooseOccurrences(expr, occurrences))
    } else {
        OccurrencesChooser.simpleChooser<RsExpr>(editor)
            .showChooser(
                expr,
                occurrences,
                { choice: OccurrencesChooser.ReplaceChoice ->
                    val toReplace = if (choice == OccurrencesChooser.ReplaceChoice.ALL) occurrences else listOf(expr)
                    callback(toReplace)
                }.asPass
            )
    }
}

fun showErrorMessageForExtractParameter(project: Project, editor: Editor, message: String) {
    val title = RefactoringBundle.message("introduce.parameter.title")
    val helpId = "refactoring.extractParameter"
    CommonRefactoringUtil.showErrorHint(project, editor, message, title, helpId)
}

private val <T> ((T) -> Unit).asPass: Pass<T>
    get() = object : Pass<T>() {
        override fun pass(t: T) = this@asPass(t)
    }

interface ExtractExpressionUi {
    fun chooseTarget(exprs: List<RsExpr>): RsExpr
    fun chooseOccurrences(expr: RsExpr, occurrences: List<RsExpr>): List<RsExpr>
    fun chooseMethod(methods: List<RsFunction>): RsFunction {
        return methods.first()
    }
}

var MOCK: ExtractExpressionUi? = null
@TestOnly
fun withMockTargetExpressionChooser(mock: ExtractExpressionUi, f: () -> Unit) {
    MOCK = mock
    try {
        f()
    } finally {
        MOCK = null
    }
}