build-system/gradle/src/main/groovy/com/android/build/gradle/internal/ProductFlavorData.groovy

/*
 * Copyright (C) 2012 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 com.android.build.gradle.internal

import com.android.annotations.NonNull
import com.android.build.gradle.internal.api.DefaultAndroidSourceSet
import com.android.builder.core.BuilderConstants
import com.android.builder.core.DefaultProductFlavor
import org.gradle.api.Project
import org.gradle.api.Task
import org.gradle.api.artifacts.Configuration

/**
 * Class containing a ProductFlavor and associated data (sourcesets)
 */
public class ProductFlavorData<T extends DefaultProductFlavor> {

    private static class ConfigurationProviderImpl implements ConfigurationProvider {

        private final Project project
        private final DefaultAndroidSourceSet sourceSet

        ConfigurationProviderImpl(Project project, DefaultAndroidSourceSet sourceSet) {
            this.project = project
            this.sourceSet = sourceSet
        }

        @Override
        @NonNull
        public Configuration getCompileConfiguration() {
            return project.configurations.getByName(sourceSet.compileConfigurationName)
        }

        @Override
        @NonNull
        public Configuration getPackageConfiguration() {
            return project.configurations.getByName(sourceSet.packageConfigurationName)
        }

        @Override
        @NonNull
        Configuration getProvidedConfiguration() {
            return project.configurations.getByName(sourceSet.providedConfigurationName)
        }
    }

    final T productFlavor

    final DefaultAndroidSourceSet sourceSet
    final DefaultAndroidSourceSet testSourceSet
    final ConfigurationProvider mainProvider
    final ConfigurationProvider testProvider

    final Task assembleTask

    ProductFlavorData(
            T productFlavor,
            DefaultAndroidSourceSet sourceSet,
            DefaultAndroidSourceSet testSourceSet,
            Project project) {
        this.productFlavor = productFlavor
        this.sourceSet = sourceSet
        this.testSourceSet = testSourceSet
        mainProvider = new ConfigurationProviderImpl(project, sourceSet)
        testProvider = new ConfigurationProviderImpl(project, testSourceSet)

        if (!BuilderConstants.MAIN.equals(sourceSet.name)) {
            assembleTask = project.tasks.create("assemble${sourceSet.name.capitalize()}")
            assembleTask.description = "Assembles all ${sourceSet.name.capitalize()} builds"
            assembleTask.setGroup("Build")
        } else {
            assembleTask = null
        }
    }

    public static String getFlavoredName(ProductFlavorData[] flavorDataArray, boolean capitalized) {
        StringBuilder builder = new StringBuilder()
        for (ProductFlavorData data : flavorDataArray) {
            builder.append(capitalized ?
                data.productFlavor.name.capitalize() :
                data.productFlavor.name)
        }

        return builder.toString()
    }
}