001    /*
002     * SonarQube, open source software quality management tool.
003     * Copyright (C) 2008-2014 SonarSource
004     * mailto:contact AT sonarsource DOT com
005     *
006     * SonarQube is free software; you can redistribute it and/or
007     * modify it under the terms of the GNU Lesser General Public
008     * License as published by the Free Software Foundation; either
009     * version 3 of the License, or (at your option) any later version.
010     *
011     * SonarQube is distributed in the hope that it will be useful,
012     * but WITHOUT ANY WARRANTY; without even the implied warranty of
013     * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
014     * Lesser General Public License for more details.
015     *
016     * You should have received a copy of the GNU Lesser General Public License
017     * along with this program; if not, write to the Free Software Foundation,
018     * Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
019     */
020    package org.sonar.api.batch.bootstrap;
021    
022    import org.sonar.api.BatchComponent;
023    
024    import java.util.ArrayList;
025    import java.util.List;
026    
027    /**
028     * @since 2.9
029     */
030    public class ProjectReactor implements BatchComponent {
031    
032      private ProjectDefinition root;
033    
034      public ProjectReactor(ProjectDefinition root) {
035        if (root.getParent() != null) {
036          throw new IllegalArgumentException("Not a root project: " + root);
037        }
038        this.root = root;
039      }
040    
041      public List<ProjectDefinition> getProjects() {
042        return collectProjects(root, new ArrayList<ProjectDefinition>());
043      }
044    
045      /**
046       * Populates list of projects from hierarchy.
047       */
048      private static List<ProjectDefinition> collectProjects(ProjectDefinition def, List<ProjectDefinition> collected) {
049        collected.add(def);
050        for (ProjectDefinition child : def.getSubProjects()) {
051          collectProjects(child, collected);
052        }
053        return collected;
054      }
055    
056      public ProjectDefinition getRoot() {
057        return root;
058      }
059    
060      public ProjectDefinition getProject(String key) {
061        for (ProjectDefinition p : getProjects()) {
062          if (key.equals(p.getKey())) {
063            return p;
064          }
065        }
066        return null;
067      }
068    }