gradle-master/subprojects/tooling-api/src/crossVersionTest/groovy/org/gradle/integtests/tooling/r18/UseOtherTypesToFetchProjectModel.java
/*
 * Copyright 2013 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.integtests.tooling.r18;

import org.gradle.integtests.tooling.r16.CustomModel;
import org.gradle.tooling.BuildAction;
import org.gradle.tooling.BuildController;
import org.gradle.tooling.model.GradleProject;
import org.gradle.tooling.model.HierarchicalElement;
import org.gradle.tooling.model.eclipse.EclipseProject;
import org.gradle.tooling.model.idea.IdeaModule;
import org.gradle.tooling.model.idea.IdeaProject;

import java.util.HashMap;
import java.util.Map;

public class UseOtherTypesToFetchProjectModel implements BuildAction<Map<String, CustomModel>> {
    public Map<String, CustomModel> execute(BuildController controller) {
        // Use an IdeaModule to reference a project
        IdeaProject ideaProject = controller.getModel(IdeaProject.class);
        for (IdeaModule ideaModule : ideaProject.getModules()) {
            visit(ideaModule, controller, new HashMap<String, CustomModel>());
        }

        // Use an EclipseProject to reference a project
        EclipseProject eclipseProject = controller.getModel(EclipseProject.class);
        visit(eclipseProject, controller, new HashMap<String, CustomModel>());

        // Use a GradleProject to reference a project
        GradleProject rootProject = controller.getModel(GradleProject.class);
        Map<String, CustomModel> projects = new HashMap<String, CustomModel>();
        visit(rootProject, controller, projects);
        return projects;
    }

    void visit(HierarchicalElement element, BuildController buildController, Map<String, CustomModel> results) {
        results.put(element.getName(), buildController.getModel(element, CustomModel.class));
        for (HierarchicalElement child : element.getChildren()) {
            visit(child, buildController, results);
        }
    }
}
