/*
 * Copyright 2010-2021 JetBrains s.r.o. and Kotlin Programming Language contributors.
 * Use of this source code is governed by the Apache 2.0 license that can be found in the license/LICENSE.txt file.
 */

package org.jetbrains.kotlin.fir.pipeline

import org.jetbrains.kotlin.diagnostics.KtDiagnostic
import org.jetbrains.kotlin.diagnostics.impl.BaseDiagnosticsCollector
import org.jetbrains.kotlin.fir.FirSession
import org.jetbrains.kotlin.fir.analysis.checkers.MppCheckerKind
import org.jetbrains.kotlin.fir.analysis.collectors.DiagnosticCollectorComponents
import org.jetbrains.kotlin.fir.analysis.collectors.CliDiagnosticsCollector
import org.jetbrains.kotlin.fir.analysis.collectors.components.DiagnosticComponentsFactory
import org.jetbrains.kotlin.fir.analysis.collectors.components.LossDiagnosticCollectorComponent
import org.jetbrains.kotlin.fir.analysis.collectors.components.ReportCommitterDiagnosticComponent
import org.jetbrains.kotlin.fir.declarations.FirFile
import org.jetbrains.kotlin.fir.resolve.ScopeSession
import org.jetbrains.kotlin.fir.resolve.transformers.FirTotalResolveProcessor
import org.jetbrains.kotlin.fir.withFileAnalysisExceptionWrapping

fun FirSession.runResolution(firFiles: List<FirFile>): Pair<ScopeSession, List<FirFile>> {
    val resolveProcessor = FirTotalResolveProcessor(this)
    resolveProcessor.process(firFiles)
    return resolveProcessor.scopeSession to firFiles
}

fun FirSession.runCheckers(
    scopeSession: ScopeSession,
    firFiles: Collection<FirFile>,
    reporter: BaseDiagnosticsCollector,
    mppCheckerKind: MppCheckerKind
): Map<FirFile, List<KtDiagnostic>> {
    val collector = DiagnosticComponentsFactory.create(this, scopeSession, mppCheckerKind)
    for (file in firFiles) {
        withFileAnalysisExceptionWrapping(file) {
            collector.collectDiagnostics(file, reporter)
        }
    }
    collector.collectDiagnosticsInSettings(reporter)
    return firFiles.associateWith {
        val path = it.sourceFile?.path ?: return@associateWith emptyList()
        reporter.diagnosticsByFilePath[path] ?: emptyList()
    }
}

fun FirSession.collectLostDiagnosticsOnFile(
    scopeSession: ScopeSession,
    file: FirFile,
    reporter: BaseDiagnosticsCollector,
): List<KtDiagnostic> {
    val collector = CliDiagnosticsCollector(this, scopeSession) { reporter ->
        DiagnosticCollectorComponents(
            arrayOf(LossDiagnosticCollectorComponent(this, reporter)),
            ReportCommitterDiagnosticComponent(this, reporter)
        )
    }
    withFileAnalysisExceptionWrapping(file) {
        collector.collectDiagnostics(file, reporter)
    }
    val path = file.sourceFile?.path ?: return emptyList()
    return reporter.diagnosticsByFilePath[path] ?: emptyList()
}
