/*
 * Copyright 2014 the original author or authors.
 *
 * 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 org.gradle.nativeplatform.test.plugins;

import org.gradle.api.Incubating;
import org.gradle.api.Plugin;
import org.gradle.api.Project;
import org.gradle.api.tasks.TaskContainer;
import org.gradle.internal.reflect.Instantiator;
import org.gradle.internal.service.ServiceRegistry;
import org.gradle.language.base.plugins.LifecycleBasePlugin;
import org.gradle.language.nativeplatform.DependentSourceSet;
import org.gradle.model.Finalize;
import org.gradle.model.Model;
import org.gradle.model.RuleSource;
import org.gradle.nativeplatform.NativeBinarySpec;
import org.gradle.nativeplatform.internal.NativeBinarySpecInternal;
import org.gradle.nativeplatform.plugins.NativeComponentPlugin;
import org.gradle.nativeplatform.tasks.InstallExecutable;
import org.gradle.nativeplatform.test.NativeTestSuiteBinarySpec;
import org.gradle.platform.base.test.TestSuiteContainer;
import org.gradle.platform.base.internal.test.DefaultTestSuiteContainer;
import org.gradle.nativeplatform.test.tasks.RunTestExecutable;
import org.gradle.platform.base.BinaryContainer;
import org.gradle.platform.base.internal.BinaryNamingScheme;

import java.io.File;

/**
 * A plugin that sets up the infrastructure for testing native binaries with CUnit.
 */
@Incubating
public class NativeBinariesTestPlugin implements Plugin<Project> {
    public void apply(final Project project) {
        project.getPluginManager().apply(NativeComponentPlugin.class);
    }

    /**
     * Model rules.
     */
    @SuppressWarnings("UnusedDeclaration")
    @RuleSource
    public static class Rules {
        @Model
        TestSuiteContainer testSuites(ServiceRegistry serviceRegistry) {
            Instantiator instantiator = serviceRegistry.get(Instantiator.class);
            return instantiator.newInstance(DefaultTestSuiteContainer.class, instantiator);
        }

        @Finalize
            // Must run after test binaries have been created (currently in CUnit plugin)
        void attachTestedBinarySourcesToTestBinaries(BinaryContainer binaries) {
            for (NativeTestSuiteBinarySpec testSuiteBinary : binaries.withType(NativeTestSuiteBinarySpec.class)) {
                NativeBinarySpec testedBinary = testSuiteBinary.getTestedBinary();
                testSuiteBinary.source(testedBinary.getSource());

                for (DependentSourceSet testSource : testSuiteBinary.getSource().withType(DependentSourceSet.class)) {
                    testSource.lib(testedBinary.getSource());
                }
            }
        }

        @Finalize
        public void createTestTasks(final TaskContainer tasks, BinaryContainer binaries) {
            for (NativeTestSuiteBinarySpec testBinary : binaries.withType(NativeTestSuiteBinarySpec.class)) {
                NativeBinarySpecInternal binary = (NativeBinarySpecInternal) testBinary;
                final BinaryNamingScheme namingScheme = binary.getNamingScheme();

                RunTestExecutable runTask = tasks.create(namingScheme.getTaskName("run"), RunTestExecutable.class);
                final Project project = runTask.getProject();
                runTask.setDescription(String.format("Runs the %s", binary));

                final InstallExecutable installTask = binary.getTasks().withType(InstallExecutable.class).iterator().next();
                runTask.getInputs().files(installTask.getOutputs().getFiles());
                runTask.setExecutable(installTask.getRunScript().getPath());
                runTask.setOutputDir(new File(project.getBuildDir(), "/test-results/" + namingScheme.getOutputDirectoryBase()));

                testBinary.getTasks().add(runTask);

                tasks.getByName(LifecycleBasePlugin.CHECK_TASK_NAME).dependsOn(runTask);
            }
        }
    }
}

@@ -685,79 +685,9 @@
 api.
-Incubating;%0Aimport org.gradle.api.Plugin;%0Aimport org.gradle.api.Project
+*
 ;%0Aim
@@ -1016,16 +1016,48 @@
 .Model;%0A
+import org.gradle.model.Mutate;%0A
 import o
@@ -1084,16 +1084,70 @@
 Source;%0A
+import org.gradle.model.collection.CollectionBuilder;%0A
 import o
@@ -1448,138 +1448,8 @@
 ec;%0A
-import org.gradle.platform.base.test.TestSuiteContainer;%0Aimport org.gradle.platform.base.internal.test.DefaultTestSuiteContainer;%0A
 impo
@@ -1616,16 +1616,146 @@
 gScheme;
+%0Aimport org.gradle.platform.base.internal.test.DefaultTestSuiteContainer;%0Aimport org.gradle.platform.base.test.TestSuiteContainer;
 %0A%0Aimport
@@ -4255,16 +4255,29 @@
 nTask);%0A
+            %7D
 %0A       
@@ -4277,24 +4277,27 @@
         
+%7D%0A%0A
         
 tasks.ge
@@ -4292,23 +4292,240 @@
     
-tasks.getByN
+@Mutate%0A        void attachBinariesToCheckLifecycle(CollectionBuilder%3CTask%3E tasks, final BinaryContainer binaries) %7B%0A            // TODO - binaries aren't an input to this rule, they're an input to the action%0A            tasks.n
 ame
+d
 (Lif
@@ -4560,43 +4560,361 @@
 NAME
-).dependsOn(runTask);%0A            %7D
+, new Action%3CTask%3E() %7B%0A                @Override%0A                public void execute(Task checkTask) %7B%0A                    for (NativeTestSuiteBinarySpec testBinary : binaries.withType(NativeTestSuiteBinarySpec.class)) %7B%0A                        checkTask.dependsOn(testBinary.getTasks().getRun());%0A                    %7D%0A                %7D%0A            %7D);
 %0A   

Split out the rule that defines the tasks for a native test suite from the rule that attaches each test suite to the 'check' task.