com.metaeffekt.artifact.extractors.configuration.DirectoryScanExtractorConfiguration Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of ae-artifact-analysis Show documentation
The newest version!
/*
 * Copyright 2021-2024 the original author or authors.
 *
 * 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 com.metaeffekt.artifact.extractors.configuration;

import com.metaeffekt.artifact.analysis.utils.FileUtils;
import com.metaeffekt.artifact.analysis.utils.InventoryUtils;
import org.apache.commons.lang3.Validate;
import org.metaeffekt.core.inventory.processor.configuration.DirectoryScanAggregatorConfiguration;
import org.metaeffekt.core.inventory.processor.model.Inventory;
import org.metaeffekt.core.inventory.processor.reader.InventoryReader;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.File;
import java.io.IOException;

public class DirectoryScanExtractorConfiguration extends AbstractExtractorConfiguration {

    private static final Logger LOG = LoggerFactory.getLogger(DirectoryScanExtractorConfiguration.class);

    final private File scanBaseDir;

    final private File referenceInventoryFile;
    final private Inventory referenceInventory;


    public DirectoryScanExtractorConfiguration(String id, File referenceInventoryFile, File scanResultInventoryFile, File scanBaseDir) {
        super(id, scanResultInventoryFile);

        this.scanBaseDir = scanBaseDir;
        this.referenceInventoryFile = referenceInventoryFile;
        this.referenceInventory = null;
    }

    public DirectoryScanExtractorConfiguration(String id, Inventory referenceInventory, File scanResultInventoryFile, File scanBaseDir) {
        super(id, scanResultInventoryFile);

        this.scanBaseDir = scanBaseDir;
        this.referenceInventory = referenceInventory;
        this.referenceInventoryFile = null;
    }

    @Override
    public void contribute(File targetDir, Inventory aggregatedInventory) throws IOException {

        // load reference inventory
        final Inventory referenceInventory = loadReferenceInventory();

        // load result inventory
        final Inventory resultInventory = loadResultInventory();

        final DirectoryScanAggregatorConfiguration directoryScanAggregatorConfiguration =
                new DirectoryScanAggregatorConfiguration(referenceInventory, resultInventory, scanBaseDir);

        directoryScanAggregatorConfiguration.contribute(targetDir, aggregatedInventory);
    }

    private Inventory loadResultInventory() throws IOException {
        final File scanResultInventoryFile = getResultInventoryFile();
        FileUtils.validateExists(scanResultInventoryFile);
        return new InventoryReader().readInventory(scanResultInventoryFile);
    }

    private Inventory loadReferenceInventory() throws IOException {
        final Inventory referenceInventory;
        if (referenceInventoryFile != null) {
            FileUtils.validateExists(referenceInventoryFile);
            if (referenceInventoryFile.isDirectory()) {
                referenceInventory = InventoryUtils.readInventory(referenceInventoryFile, "*.xls, *.xlsx");
            } else {
                referenceInventory = new InventoryReader().readInventory(referenceInventoryFile);
            }
        } else {
            Validate.notNull(this.referenceInventory);
            referenceInventory = this.referenceInventory;
        }
        return referenceInventory;
    }

}