org.grails.cli.gradle.FetchAllTaskSelectorsBuildAction Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of grace-shell Show documentation
Grace Framework : Grace Shell
There is a newer version: 2023.1.0-RC1
/*
 * Copyright 2014-2022 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
 *
 *      https://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.grails.cli.gradle;

import java.io.File;
import java.io.Serializable;
import java.util.HashMap;
import java.util.LinkedHashMap;
import java.util.LinkedHashSet;
import java.util.Map;
import java.util.Set;

import org.gradle.tooling.BuildAction;
import org.gradle.tooling.BuildController;
import org.gradle.tooling.model.Task;
import org.gradle.tooling.model.TaskSelector;
import org.gradle.tooling.model.gradle.BasicGradleProject;
import org.gradle.tooling.model.gradle.BuildInvocations;

import org.grails.cli.gradle.FetchAllTaskSelectorsBuildAction.AllTasksModel;

/**
 * A {@link org.gradle.tooling.BuildAction} that calculates all the tasks from the Gradle build
 *
 * @author Lari Hotari
 * @since 3.0
 *
 */
public class FetchAllTaskSelectorsBuildAction implements BuildAction {

    private static final long serialVersionUID = 1L;

    private final String currentProjectPath;

    public FetchAllTaskSelectorsBuildAction(File currentProjectDir) {
        this.currentProjectPath = currentProjectDir.getAbsolutePath();
    }

    public AllTasksModel execute(BuildController controller) {
        AllTasksModel model = new AllTasksModel();
        Map> allTaskSelectors = new LinkedHashMap<>();
        model.allTaskSelectors = allTaskSelectors;
        Map> allTasks = new LinkedHashMap<>();
        model.allTasks = allTasks;
        Map projectPaths = new HashMap<>();
        model.projectPaths = projectPaths;

        for (BasicGradleProject project : controller.getBuildModel().getProjects()) {
            BuildInvocations entryPointsForProject = controller.getModel(project, BuildInvocations.class);
            Set selectorNames = new LinkedHashSet<>();
            for (TaskSelector selector : entryPointsForProject.getTaskSelectors()) {
                selectorNames.add(selector.getName());
            }
            allTaskSelectors.put(project.getName(), selectorNames);

            Set taskNames = new LinkedHashSet<>();
            for (Task task : entryPointsForProject.getTasks()) {
                taskNames.add(task.getName());
            }
            allTasks.put(project.getName(), taskNames);

            projectPaths.put(project.getName(), project.getPath());
            if (project.getProjectDirectory().getAbsolutePath().equals(this.currentProjectPath)) {
                model.currentProject = project.getName();
            }
        }
        return model;
    }

    public static class AllTasksModel implements Serializable {

        private static final long serialVersionUID = 1L;

        public Map> allTasks;

        public Map> allTaskSelectors;

        public Map projectPaths;

        public String currentProject;

    }

}