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.squid.bridges;
021
022import org.apache.commons.lang.StringUtils;
023import org.sonar.api.batch.SensorContext;
024import org.sonar.api.batch.SquidUtils;
025import org.sonar.api.resources.JavaFile;
026import org.sonar.api.resources.JavaPackage;
027import org.sonar.api.resources.Project;
028import org.sonar.api.resources.Resource;
029import org.sonar.java.api.JavaClass;
030import org.sonar.java.api.JavaMethod;
031import org.sonar.squid.Squid;
032import org.sonar.squid.api.*;
033import org.sonar.squid.indexer.QueryByMeasure;
034import org.sonar.squid.indexer.QueryByType;
035import org.sonar.squid.measures.Metric;
036
037import java.util.Collection;
038import java.util.HashMap;
039
040public final class ResourceIndex extends HashMap<SourceCode, Resource> {
041
042  private static final long serialVersionUID = -918346378374943773L;
043
044  public ResourceIndex loadSquidResources(Squid squid, SensorContext context, Project project) {
045    loadSquidProject(squid, project);
046    loadSquidPackages(squid, context);
047    loadSquidFiles(squid, context);
048    loadSquidClasses(squid, context);
049    loadSquidMethods(squid, context);
050    return this;
051  }
052
053  private void loadSquidProject(Squid squid, Project project) {
054    put(squid.getProject(), project);
055  }
056
057  private void loadSquidPackages(Squid squid, SensorContext context) {
058    Collection<SourceCode> packages = squid.search(new QueryByType(SourcePackage.class));
059    for (SourceCode squidPackage : packages) {
060      JavaPackage sonarPackage = SquidUtils.convertJavaPackageKeyFromSquidFormat(squidPackage.getKey());
061      context.index(sonarPackage);
062      put(squidPackage, context.getResource(sonarPackage)); // resource is reloaded to get the id
063    }
064  }
065
066  private void loadSquidFiles(Squid squid, SensorContext context) {
067    Collection<SourceCode> files = squid.search(new QueryByType(SourceFile.class));
068    for (SourceCode squidFile : files) {
069      JavaFile sonarFile = SquidUtils.convertJavaFileKeyFromSquidFormat(squidFile.getKey());
070      JavaPackage sonarPackage = (JavaPackage) get(squidFile.getParent(SourcePackage.class));
071      context.index(sonarFile, sonarPackage);
072      put(squidFile, context.getResource(sonarFile)); // resource is reloaded to get the id
073    }
074  }
075
076  private void loadSquidClasses(Squid squid, SensorContext context) {
077    Collection<SourceCode> classes = squid.search(new QueryByType(SourceClass.class), new QueryByMeasure(Metric.CLASSES, QueryByMeasure.Operator.GREATER_THAN_EQUALS, 1));
078    for (SourceCode squidClass : classes) {
079      JavaFile sonarFile = (JavaFile) get(squidClass.getParent(SourceFile.class));
080      JavaClass sonarClass = new JavaClass.Builder()
081          .setName(convertClassKey(squidClass.getKey()))
082          .setFromLine(squidClass.getStartAtLine())
083          .setToLine(squidClass.getEndAtLine())
084          .create();
085      context.index(sonarClass, sonarFile);
086      put(squidClass, sonarClass);
087    }
088  }
089
090  private void loadSquidMethods(Squid squid, SensorContext context) {
091    Collection<SourceCode> methods = squid.search(new QueryByType(SourceMethod.class));
092    for (SourceCode squidMethod : methods) {
093      SourceClass squidClass = squidMethod.getParent(SourceClass.class);
094      JavaClass sonarClass = (JavaClass) get(squidClass);
095      if (sonarClass != null) {
096        JavaMethod sonarMethod = new JavaMethod.Builder()
097            .setClass(sonarClass)
098            .setSignature(squidMethod.getName())
099            .setFromLine(squidMethod.getStartAtLine())
100            .setToLine(squidMethod.getEndAtLine())
101            .setAccessor(squidMethod.getInt(Metric.ACCESSORS) > 0)
102            .create();
103
104        context.index(sonarMethod, sonarClass);
105        put(squidMethod, sonarMethod);
106      }
107    }
108  }
109
110  static String convertClassKey(String squidClassKey) {
111    return StringUtils.replace(squidClassKey, "/", ".");
112  }
113
114}