org.sonar.core.technicaldebt.DefaultTechnicalDebtModel.java Source code

Introduction

Here is the source code for org.sonar.core.technicaldebt.DefaultTechnicalDebtModel.java

Source

/*
 * SonarQube, open source software quality management tool.
 * Copyright (C) 2008-2014 SonarSource
 * mailto:contact AT sonarsource DOT com
 *
 * SonarQube is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 3 of the License, or (at your option) any later version.
 *
 * SonarQube is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this program; if not, write to the Free Software Foundation,
 * Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
 */

package org.sonar.core.technicaldebt;

import com.google.common.base.Predicate;
import com.google.common.collect.Iterables;
import org.sonar.api.batch.debt.DebtCharacteristic;
import org.sonar.api.batch.debt.DebtModel;
import org.sonar.api.batch.debt.internal.DefaultDebtCharacteristic;
import org.sonar.api.rule.RuleKey;
import org.sonar.api.technicaldebt.batch.TechnicalDebtModel;
import org.sonar.api.technicaldebt.batch.internal.DefaultCharacteristic;
import org.sonar.api.technicaldebt.batch.internal.DefaultRequirement;

import javax.annotation.CheckForNull;
import javax.annotation.Nullable;

import java.util.Collections;
import java.util.List;

import static com.google.common.collect.Lists.newArrayList;

public class DefaultTechnicalDebtModel implements TechnicalDebtModel {

    private final DebtModel model;

    public DefaultTechnicalDebtModel(DebtModel model) {
        this.model = model;
    }

    public List<DefaultCharacteristic> rootCharacteristics() {
        return newArrayList(Iterables.filter(characteristics(), new Predicate<DefaultCharacteristic>() {
            @Override
            public boolean apply(DefaultCharacteristic input) {
                return input.isRoot();
            }
        }));
    }

    @Override
    @CheckForNull
    public DefaultCharacteristic characteristicByKey(final String key) {
        return Iterables.find(characteristics(), new Predicate<DefaultCharacteristic>() {
            @Override
            public boolean apply(DefaultCharacteristic input) {
                return input.key().equals(key);
            }
        }, null);
    }

    @Override
    @CheckForNull
    public DefaultCharacteristic characteristicById(final Integer id) {
        return Iterables.find(characteristics(), new Predicate<DefaultCharacteristic>() {
            @Override
            public boolean apply(DefaultCharacteristic input) {
                return input.id().equals(id);
            }
        }, null);
    }

    @Override
    @CheckForNull
    public DefaultRequirement requirementsByRule(final RuleKey ruleKey) {
        return null;
    }

    @Override
    @CheckForNull
    public DefaultRequirement requirementsById(final Integer id) {
        return null;
    }

    @Override
    public List<DefaultCharacteristic> characteristics() {
        List<DefaultCharacteristic> flatCharacteristics = newArrayList();
        for (DebtCharacteristic characteristic : model.characteristics()) {
            DefaultCharacteristic root = toDefaultCharacteristic((DefaultDebtCharacteristic) characteristic, null);
            flatCharacteristics.add(root);
            for (DebtCharacteristic subCharacteristic : model.subCharacteristics(characteristic.key())) {
                flatCharacteristics
                        .add(toDefaultCharacteristic((DefaultDebtCharacteristic) subCharacteristic, root));
            }
        }
        return flatCharacteristics;
    }

    @Override
    public List<DefaultRequirement> requirements() {
        return Collections.emptyList();
    }

    public boolean isEmpty() {
        return model.allCharacteristics().isEmpty();
    }

    private static DefaultCharacteristic toDefaultCharacteristic(DefaultDebtCharacteristic debtCharacteristic,
            @Nullable DefaultCharacteristic parentCharacteristic) {
        return new DefaultCharacteristic().setId(debtCharacteristic.id()).setKey(debtCharacteristic.key())
                .setName(debtCharacteristic.name()).setOrder(debtCharacteristic.order())
                .setParent(parentCharacteristic).setRoot(parentCharacteristic)
                .setCreatedAt(debtCharacteristic.createdAt()).setUpdatedAt(debtCharacteristic.updatedAt());
    }

}