package org.jetbrains

import com.jfrog.bintray.gradle.BintrayExtension
import org.gradle.api.DefaultTask
import org.gradle.api.GradleException
import org.gradle.api.Project
import org.gradle.api.artifacts.ProjectDependency
import org.gradle.api.publish.PublishingExtension
import org.gradle.api.publish.maven.MavenPublication
import org.gradle.api.tasks.TaskAction
import org.gradle.kotlin.dsl.findByType
import org.gradle.kotlin.dsl.provideDelegate

open class ValidatePublications : DefaultTask() {
    class MissingBintrayPublicationException(project: Project, publication: MavenPublication) : GradleException(
        "Project ${project.path} has publication ${publication.name} that is not configured for bintray publication"
    )

    class UnpublishedProjectDependencyException(
        project: Project, dependencyProject: Project
    ) : GradleException(
        "Published project ${project.path} cannot depend on unpublished projed ${dependencyProject.path}"
    )


    @TaskAction
    fun validatePublicationConfiguration() {
        @Suppress("LocalVariableName")
        project.subprojects.forEach { subProject ->
            val publishing = subProject.extensions.findByType<PublishingExtension>() ?: return@forEach
            publishing.publications
                .filterIsInstance<MavenPublication>()
                .filter { it.version == project.dokkaVersion }
                .forEach { publication ->
                    if (project.publicationChannels.any { it.isBintrayRepository }) {
                        checkPublicationIsConfiguredForBintray(subProject, publication)
                    }
                    checkProjectDependenciesArePublished(subProject)
                }
        }
    }

    private fun checkPublicationIsConfiguredForBintray(project: Project, publication: MavenPublication) {
        val bintrayExtension = project.extensions.findByType<BintrayExtension>()
            ?: throw MissingBintrayPublicationException(project, publication)

        val isPublicationConfiguredForBintray = bintrayExtension.publications.orEmpty()
            .any { publicationName -> publicationName == publication.name }

        if (!isPublicationConfiguredForBintray) {
            throw MissingBintrayPublicationException(project, publication)
        }
    }

    private fun checkProjectDependenciesArePublished(project: Project) {
        (project.configurations.findByName("implementation")?.allDependencies.orEmpty() +
                project.configurations.findByName("api")?.allDependencies.orEmpty())
            .filterIsInstance<ProjectDependency>()
            .forEach { projectDependency ->
                val publishing = projectDependency.dependencyProject.extensions.findByType<PublishingExtension>()
                    ?: throw UnpublishedProjectDependencyException(
                        project = project, dependencyProject = projectDependency.dependencyProject
                    )

                val isPublished = publishing.publications.filterIsInstance<MavenPublication>()
                    .filter { it.version == project.dokkaVersion }
                    .any()

                if (!isPublished) {
                    throw UnpublishedProjectDependencyException(project, projectDependency.dependencyProject)
                }
            }
    }

    init {
        group = "verification"
        project.tasks.named("check") {
            dependsOn(this@ValidatePublications)
        }
    }
}