/src/main/java/com/atlassian/bamboo/plugin/dotnet/msbuild/MsBuildConfigurator.java

package com.atlassian.bamboo.plugin.dotnet.msbuild;



import com.atlassian.bamboo.build.Job;

import com.atlassian.bamboo.collections.ActionParametersMap;

import com.atlassian.bamboo.deployments.environments.DeploymentTaskRequirementSupport;

import com.atlassian.bamboo.deployments.environments.Environment;

import com.atlassian.bamboo.plugin.dotnet.support.AbstractDotNetTaskConfigurator;

import com.atlassian.bamboo.task.BuildTaskRequirementSupport;

import com.atlassian.bamboo.task.TaskConfigConstants;

import com.atlassian.bamboo.task.TaskDefinition;

import com.atlassian.bamboo.utils.error.ErrorCollection;

import com.atlassian.bamboo.v2.build.agent.capability.Requirement;

import com.atlassian.bamboo.v2.build.agent.capability.RequirementImpl;

import com.google.common.collect.Sets;

import org.apache.commons.lang3.StringUtils;

import org.apache.log4j.Logger;

import org.jetbrains.annotations.NotNull;

import org.jetbrains.annotations.Nullable;



import java.util.Map;

import java.util.Set;



public class MsBuildConfigurator extends AbstractDotNetTaskConfigurator implements BuildTaskRequirementSupport, DeploymentTaskRequirementSupport

{

    @SuppressWarnings("UnusedDeclaration")

    private static final Logger log = Logger.getLogger(MsBuildConfigurator.class);

    // ------------------------------------------------------------------------------------------------------- Constants



    private static final Set<String> FIELDS_TO_COPY = Sets.newHashSet(

            MsBuildConfig.LABEL,

            MsBuildConfig.SOLUTION,

            MsBuildConfig.OPTIONS,

            MsBuildConfig.ENVIRONMENT,

            TaskConfigConstants.CFG_WORKING_SUB_DIRECTORY

    );

    private static final String DEFAULT_SOLUTION = "YourSolution.sln";



    // ------------------------------------------------------------------------------------------------- Type Properties

    // ---------------------------------------------------------------------------------------------------- Dependencies

    // ---------------------------------------------------------------------------------------------------- Constructors

    // ----------------------------------------------------------------------------------------------- Interface Methods



    @NotNull

    @Override

    public Set<Requirement> calculateRequirements(@NotNull TaskDefinition taskDefinition, @NotNull Environment environment)

    {

        final String label = taskDefinition.getConfiguration().get(MsBuildConfig.LABEL);

        return Sets.<Requirement>newHashSet(new RequirementImpl(MsBuildTaskType.MSBUILD_CAPABILITY_PREFIX + "." + label, true, ".*"));

    }



    @Override

    @NotNull

    public Set<Requirement> calculateRequirements(@NotNull TaskDefinition taskDefinition, @NotNull Job buildable)

    {

        final String label = taskDefinition.getConfiguration().get(MsBuildConfig.LABEL);

        return Sets.<Requirement>newHashSet(new RequirementImpl(MsBuildTaskType.MSBUILD_CAPABILITY_PREFIX + "." + label, true, ".*"));

    }



    @Override

    public void validate(@NotNull final ActionParametersMap actionParametersMap, @NotNull final ErrorCollection errorCollection)

    {

        final String solution = actionParametersMap.getString(MsBuildConfig.SOLUTION);

        if (StringUtils.isEmpty(solution))

        {

            errorCollection.addError(MsBuildConfig.SOLUTION, textProvider.getText("msbuild.solution.error"));

        }



        final String label = actionParametersMap.getString(MsBuildConfig.LABEL);

        if (StringUtils.isEmpty(label))

        {

            errorCollection.addError(MsBuildConfig.LABEL, textProvider.getText("msbuild.label.error"));

        }

    }



    @Override

    public void populateContextForCreate(@NotNull final Map<String, Object> context)

    {

        super.populateContextForCreate(context);

        context.put(MsBuildConfig.SOLUTION, DEFAULT_SOLUTION);

    }



    @Override

    public void populateContextForEdit(@NotNull final Map<String, Object> context, @NotNull final TaskDefinition taskDefinition)

    {

        super.populateContextForEdit(context, taskDefinition);

        taskConfiguratorHelper.populateContextWithConfiguration(context, taskDefinition, FIELDS_TO_COPY);

    }



    @Override

    public void populateContextForView(@NotNull final Map<String, Object> context, @NotNull final TaskDefinition taskDefinition)

    {

        super.populateContextForView(context, taskDefinition);

        taskConfiguratorHelper.populateContextWithConfiguration(context, taskDefinition, FIELDS_TO_COPY);

    }



    @NotNull

    @Override

    public Map<String, String> generateTaskConfigMap(@NotNull final ActionParametersMap params, @Nullable final TaskDefinition previousTaskDefinition)

    {

        final Map<String, String> map = super.generateTaskConfigMap(params, previousTaskDefinition);

        taskConfiguratorHelper.populateTaskConfigMapWithActionParameters(map, params, FIELDS_TO_COPY);

        return map;

    }



    // -------------------------------------------------------------------------------------------------- Action Methods

    // -------------------------------------------------------------------------------------------------- Public Methods

    // -------------------------------------------------------------------------------------- Basic Accessors / Mutators

}