/*
 * Copyright 2020 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package androidx.compose.material.icons.generator.tasks

import androidx.compose.material.icons.generator.IconTestingManifestGenerator
import java.io.File
import org.gradle.api.Project
import org.gradle.api.tasks.CacheableTask
import org.gradle.api.tasks.OutputDirectory

/**
 * Task responsible for generating files related to testing.
 *
 * - Generates a list of all icons mapped to the drawable ID used in testing, so we can bitmap
 * compare the programmatic icon with the original source drawable.
 *
 * - Flattens all the source drawables into a drawable folder that will be used in comparison tests.
 */
@CacheableTask
open class IconTestingGenerationTask : IconGenerationTask() {
    /**
     * Directory to generate the flattened drawables used for testing to.
     */
    @get:OutputDirectory
    val drawableDirectory: File
        get() = generatedResourceDirectory.resolve("drawable")

    override fun run() {
        // Copy all drawables to the drawable directory
        loadIcons().forEach { icon ->
            drawableDirectory.resolve("${icon.xmlFileName}.xml").apply {
                createNewFile()
                writeText(icon.fileContent)
            }
        }

        // Generate the testing manifest to the androidTest directory
        IconTestingManifestGenerator(loadIcons()).generateTo(generatedSrcAndroidTestDirectory)
    }

    companion object {
        /**
         * Registers [IconTestingGenerationTask] in [project] for [variant].
         */
        @Suppress("DEPRECATION") // BaseVariant
        fun register(project: Project, variant: com.android.build.gradle.api.BaseVariant) {
            val (task, buildDirectory) = project.registerGenerationTask(
                "generateTestFiles",
                IconTestingGenerationTask::class.java,
                variant
            )

            val generatedResourceDirectory = buildDirectory.resolve(GeneratedResource)

            variant.registerGeneratedResFolders(
                project.files(generatedResourceDirectory).builtBy(task)
            )

            val generatedSrcAndroidTestDirectory = buildDirectory.resolve(GeneratedSrcAndroidTest)
            variant.registerJavaGeneratingTask(task, generatedSrcAndroidTestDirectory)
        }
    }
}