package it.unitn.molerat.cmd.experiments;

import it.unitn.molerat.data.csv.InputDataPoint;
import it.unitn.molerat.data.csv.PerfOutputDataPoint;
import it.unitn.molerat.repos.trackers.vuln.VulnerabilityEvidenceTracker;
import it.unitn.molerat.repos.trackers.vuln.VulnerabilityEvidenceTrackerFactory;
import java.util.Set;

public class VulnEvdPerformanceExperiment extends AbstractExperiment {

    public static void main(String[] args) {
        if (args.length < 2) {
            System.out.println("USAGE: [input.csv] [output.csv]");
            return;
        }
        String inputCsvPath = args[0];
        String outputCsvPath = args[1];

        try {
            Set<InputDataPoint> inputs = readInputDataPoints(inputCsvPath);
            saveOutput(outputCsvPath, new PerfOutputDataPoint().getHeader());
            for (InputDataPoint idp : inputs) {

                System.out.println("Processing the " + idp.CVEID + "...");
                long timeNow = System.currentTimeMillis();

                VulnerabilityEvidenceTracker vulnTracker = VulnerabilityEvidenceTrackerFactory.getTracker(idp.REPO_ROOT, idp.FIX_REV, idp.REPO_TYPE, idp.TRACKER_TYPE);
                vulnTracker.trackEvidence();

                long timeAfter = System.currentTimeMillis();

                String[] output = new String[]{
                        idp.CVEID,
                        idp.TRACKER_TYPE,
                        String.valueOf(timeAfter - timeNow)
                };
                PerfOutputDataPoint odp = new PerfOutputDataPoint(output);
                saveOutput(outputCsvPath, odp.toString());

            }
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}