3 * Copyright (C) 2009-2024 SonarSource SA
4 * mailto:info AT sonarsource DOT com
6 * This program is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 3 of the License, or (at your option) any later version.
11 * This program is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
16 * You should have received a copy of the GNU Lesser General Public License
17 * along with this program; if not, write to the Free Software Foundation,
18 * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
20 package org.sonar.ce.task.projectanalysis.qualitymodel;
22 import java.util.Optional;
23 import org.sonar.ce.task.projectanalysis.component.Component;
24 import org.sonar.ce.task.projectanalysis.component.PathAwareVisitor;
25 import org.sonar.ce.task.projectanalysis.component.PathAwareVisitorAdapter;
26 import org.sonar.ce.task.projectanalysis.issue.ComponentIssuesRepository;
27 import org.sonar.ce.task.projectanalysis.measure.MeasureRepository;
28 import org.sonar.ce.task.projectanalysis.measure.RatingMeasures;
29 import org.sonar.ce.task.projectanalysis.metric.Metric;
30 import org.sonar.ce.task.projectanalysis.metric.MetricRepository;
32 import static org.sonar.api.measures.CoreMetrics.SECURITY_HOTSPOTS_REVIEWED_KEY;
33 import static org.sonar.api.measures.CoreMetrics.SECURITY_HOTSPOTS_REVIEWED_STATUS_KEY;
34 import static org.sonar.api.measures.CoreMetrics.SECURITY_HOTSPOTS_TO_REVIEW_STATUS_KEY;
35 import static org.sonar.api.measures.CoreMetrics.SECURITY_REVIEW_RATING_KEY;
36 import static org.sonar.api.rules.RuleType.SECURITY_HOTSPOT;
37 import static org.sonar.ce.task.projectanalysis.component.ComponentVisitor.Order.POST_ORDER;
38 import static org.sonar.ce.task.projectanalysis.component.CrawlerDepthLimit.FILE;
39 import static org.sonar.ce.task.projectanalysis.measure.Measure.newMeasureBuilder;
40 import static org.sonar.server.security.SecurityReviewRating.computePercent;
41 import static org.sonar.server.security.SecurityReviewRating.computeRating;
43 public class SecurityReviewMeasuresVisitor extends PathAwareVisitorAdapter<SecurityReviewCounter> {
45 private final ComponentIssuesRepository componentIssuesRepository;
46 private final MeasureRepository measureRepository;
47 private final Metric securityReviewRatingMetric;
48 private final Metric securityHotspotsReviewedMetric;
49 private final Metric securityHotspotsReviewedStatusMetric;
50 private final Metric securityHotspotsToReviewStatusMetric;
52 public SecurityReviewMeasuresVisitor(ComponentIssuesRepository componentIssuesRepository, MeasureRepository measureRepository, MetricRepository metricRepository) {
53 super(FILE, POST_ORDER, SecurityReviewMeasuresVisitor.CounterFactory.INSTANCE);
54 this.componentIssuesRepository = componentIssuesRepository;
55 this.measureRepository = measureRepository;
56 this.securityReviewRatingMetric = metricRepository.getByKey(SECURITY_REVIEW_RATING_KEY);
57 this.securityHotspotsReviewedMetric = metricRepository.getByKey(SECURITY_HOTSPOTS_REVIEWED_KEY);
58 this.securityHotspotsReviewedStatusMetric = metricRepository.getByKey(SECURITY_HOTSPOTS_REVIEWED_STATUS_KEY);
59 this.securityHotspotsToReviewStatusMetric = metricRepository.getByKey(SECURITY_HOTSPOTS_TO_REVIEW_STATUS_KEY);
63 public void visitProject(Component project, Path<SecurityReviewCounter> path) {
64 computeMeasure(project, path);
68 public void visitDirectory(Component directory, PathAwareVisitor.Path<SecurityReviewCounter> path) {
69 computeMeasure(directory, path);
73 public void visitFile(Component file, PathAwareVisitor.Path<SecurityReviewCounter> path) {
74 computeMeasure(file, path);
77 private void computeMeasure(Component component, PathAwareVisitor.Path<SecurityReviewCounter> path) {
78 componentIssuesRepository.getIssues(component)
80 .filter(issue -> issue.type().equals(SECURITY_HOTSPOT))
81 .forEach(issue -> path.current().processHotspot(issue));
83 measureRepository.add(component, securityHotspotsReviewedStatusMetric, newMeasureBuilder().create(path.current().getHotspotsReviewed()));
84 measureRepository.add(component, securityHotspotsToReviewStatusMetric, newMeasureBuilder().create(path.current().getHotspotsToReview()));
85 Optional<Double> percent = computePercent(path.current().getHotspotsToReview(), path.current().getHotspotsReviewed());
86 measureRepository.add(component, securityReviewRatingMetric, RatingMeasures.get(computeRating(percent.orElse(null))));
87 percent.ifPresent(p -> measureRepository.add(component, securityHotspotsReviewedMetric, newMeasureBuilder().create(p, securityHotspotsReviewedMetric.getDecimalScale())));
90 path.parent().add(path.current());
94 private static final class CounterFactory extends PathAwareVisitorAdapter.SimpleStackElementFactory<SecurityReviewCounter> {
95 public static final SecurityReviewMeasuresVisitor.CounterFactory INSTANCE = new SecurityReviewMeasuresVisitor.CounterFactory();
97 private CounterFactory() {
98 // prevents instantiation
102 public SecurityReviewCounter createForAny(Component component) {
103 return new SecurityReviewCounter();