me.snowdrop.licenses.sanitiser.AliasLicenseSanitiser Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2016-2017 Red Hat, Inc, and individual contributors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package me.snowdrop.licenses.sanitiser;

import me.snowdrop.licenses.xml.DependencyElement;
import me.snowdrop.licenses.xml.LicenseElement;

import java.util.Optional;
import java.util.Set;

import static me.snowdrop.licenses.utils.JsonUtils.loadJsonToSet;

/**
 * @author Gytis Trikleris
 */
public class AliasLicenseSanitiser implements LicenseSanitiser {

    private final Set redHatLicenses;

    private final LicenseSanitiser next;

    public AliasLicenseSanitiser(String namesFilePath, LicenseSanitiser next) {
        this.redHatLicenses = loadJsonToSet(namesFilePath, RedHatLicense::new);
        this.next = next;
    }

    @Override
    public DependencyElement fix(DependencyElement originalDependencyElement) {
        boolean shouldCallNext = false;
        DependencyElement dependencyElement = new DependencyElement(originalDependencyElement);

        for (LicenseElement licenseElement : dependencyElement.getLicenses()) {
            Optional redHatLicenseOptional = getRedHatLicenseByUrl(licenseElement.getUrl());
            if (!redHatLicenseOptional.isPresent()) {
                redHatLicenseOptional = getRedHatLicenseByName(licenseElement.getName());
            }
            if (redHatLicenseOptional.isPresent()) {
                RedHatLicense redHatLicense = redHatLicenseOptional.get();
                licenseElement.setName(redHatLicense.getName());
                licenseElement.setUrl(redHatLicense.getUrl());
                licenseElement.setTextUrl(redHatLicense.getTextUrl());
            } else {
                shouldCallNext = true;
            }
        }

        if (shouldCallNext) {
            return next.fix(dependencyElement);
        }

        return dependencyElement;
    }

    private Optional getRedHatLicenseByUrl(String url) {
        if (url == null) {
            return Optional.empty();
        }
        return redHatLicenses.stream()
                .filter(l -> l.getUrlAliases()
                        .contains(url.toLowerCase()))
                .findFirst();
    }

    private Optional getRedHatLicenseByName(String name) {
        if (name == null) {
            return Optional.empty();
        }
        return redHatLicenses.stream()
                .filter(l -> l.getAliases()
                        .contains(name.toLowerCase()))
                .findFirst();
    }

}