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

package software.aws.toolkits.jetbrains.services.lambda.completion

import com.intellij.codeInsight.completion.PrefixMatcher
import com.intellij.codeInsight.completion.impl.CamelHumpMatcher
import com.intellij.codeInsight.lookup.LookupElement
import com.intellij.codeInsight.lookup.LookupElementBuilder
import com.intellij.openapi.project.Project
import com.jetbrains.rd.framework.impl.RpcTimeouts
import com.jetbrains.rd.ui.icons.toIdeaIcon
import com.jetbrains.rider.projectView.solution
import software.aws.toolkits.jetbrains.protocol.HandlerCompletionItem
import software.aws.toolkits.jetbrains.protocol.lambdaPsiModel

class DotNetHandlerCompletion : HandlerCompletion {
    override fun getPrefixMatcher(prefix: String): PrefixMatcher = CamelHumpMatcher(prefix)

    override fun getLookupElements(project: Project): Collection<LookupElement> {
        val completionItems = getHandlersFromBackend(project)
        return completionItems.map { completionItem ->
            LookupElementBuilder.create(completionItem.handler).let { element ->
                if (completionItem.iconId != null) {
                    element.withIcon(completionItem.iconId.toIdeaIcon(project))
                } else {
                    element
                }
            }.withInsertHandler { context, item -> context.document.setText(item.lookupString) }
        }
    }

    fun getHandlersFromBackend(project: Project): List<HandlerCompletionItem> =
        project.solution.lambdaPsiModel.determineHandlers.sync(Unit, RpcTimeouts.default)
}