001/*
002 * Sonar, open source software quality management tool.
003 * Copyright (C) 2008-2012 SonarSource
004 * mailto:contact AT sonarsource DOT com
005 *
006 * Sonar 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 * Sonar 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
017 * License along with Sonar; if not, write to the Free Software
018 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02
019 */
020package org.sonar.plugins.core.sensors;
021
022import org.sonar.api.batch.Decorator;
023import org.sonar.api.batch.DecoratorContext;
024import org.sonar.api.batch.DependedUpon;
025import org.sonar.api.measures.CoreMetrics;
026import org.sonar.api.measures.Measure;
027import org.sonar.api.measures.MeasureUtils;
028import org.sonar.api.measures.Metric;
029import org.sonar.api.resources.Project;
030import org.sonar.api.resources.Resource;
031
032import java.util.Collection;
033
034/**
035 * @since 2.2
036 */
037public final class FilesDecorator implements Decorator {
038
039  public boolean shouldExecuteOnProject(Project project) {
040    return true;
041  }
042
043  @DependedUpon
044  public Metric generateDirectoriesMetric() {
045    return CoreMetrics.FILES;
046  }
047
048  public void decorate(Resource resource, DecoratorContext context) {
049    if (MeasureUtils.hasValue(context.getMeasure(CoreMetrics.FILES))) {
050      return;
051    }
052
053    if (Resource.QUALIFIER_CLASS.equals(resource.getQualifier()) || Resource.QUALIFIER_FILE.equals(resource.getQualifier())) {
054      context.saveMeasure(CoreMetrics.FILES, 1.0);
055
056    } else {
057      Collection<Measure> childrenMeasures = context.getChildrenMeasures(CoreMetrics.FILES);
058      Double sum = MeasureUtils.sum(false, childrenMeasures);
059      if (sum != null) {
060        context.saveMeasure(CoreMetrics.FILES, sum);
061      }
062    }
063  }
064}