package cc.unitmesh.git.provider

import cc.unitmesh.devti.devins.ShireActionLocation
import cc.unitmesh.devti.devins.provider.ActionLocationEditor
import com.intellij.openapi.application.invokeAndWaitIfNeeded
import com.intellij.openapi.diagnostic.logger
import com.intellij.openapi.editor.Editor
import com.intellij.openapi.project.Project
import com.intellij.openapi.vcs.ui.CommitMessage

class GitActionLocationEditor : ActionLocationEditor {
    private var commitUi: CommitMessage? = null

    override fun isApplicable(hole: ShireActionLocation): Boolean  {
        val commitMessage = getCommitUi(hole)
        if (commitMessage != null) {
            commitUi = commitMessage
        }

        return hole == ShireActionLocation.COMMIT_MENU && commitMessage != null
    }

    override fun resolve(project: Project, hole: ShireActionLocation): Editor? {
        val commitMessageUi = commitUi ?: getCommitUi(hole) ?: return null
        val editorField = commitMessageUi.editorField

        @Suppress("UnstableApiUsage")
        invokeAndWaitIfNeeded { editorField.text = "" }

        return editorField.editor
    }

    private fun getCommitUi(hole: ShireActionLocation): CommitMessage? {
        if (hole != ShireActionLocation.COMMIT_MENU) return null

        val commitMessageUi = getCommitWorkflowUi()?.commitMessageUi as? CommitMessage

        if (commitMessageUi == null) {
            logger<GitActionLocationEditor>().error("Failed to get commit message UI")
            return null
        }

        return commitMessageUi
    }
}