Main.java

package org.example;

import org.example.coverage.CoverageCollector;
import org.example.coverage.CoverageReport;
import org.example.coverage.CoverageTransformer;
import org.example.metrics.Metrics;
import org.example.visitor.ClassAnalyzer;
import org.objectweb.asm.ClassReader;

import java.io.FileWriter;
import java.io.IOException;
import java.lang.instrument.Instrumentation;
import java.util.Enumeration;
import java.util.Locale;
import java.util.jar.JarEntry;
import java.util.jar.JarFile;

public class Main {
    private static final CoverageCollector collector = CoverageCollector.getInstance();

    public static void premain(String agentArgs, Instrumentation inst) {
        inst.addTransformer(new CoverageTransformer(), true);
        Runtime.getRuntime().addShutdownHook(new Thread(() -> {
            System.out.println("[LineCoverageAgent] Saving coverage report...");
            CoverageReport report = collector.generateReport();
            try {
                report.saveToFile("coverage-report.json");
            } catch (IOException e) {
                throw new RuntimeException(e);
            }
            report.printSummary();
        }));
    }

    public static void main(String[] args) throws IOException {
        String inputJar = "src/main/resources/fescar.zip";
        String outputFile = "src/main/resources/output.json";

        Metrics metrics = new Metrics();

        try (JarFile inputJarFile = new JarFile(inputJar)) {
            Enumeration<JarEntry> enumeration = inputJarFile.entries();

            while (enumeration.hasMoreElements()) {
                JarEntry entry = enumeration.nextElement();
                if (entry.getName().endsWith(".class")) {
                    ClassReader reader = new ClassReader(inputJarFile.getInputStream(entry));
                    ClassAnalyzer analyzer = new ClassAnalyzer(metrics);
                    reader.accept(analyzer, 0);
                }
            }
        }
        metrics.calculateClassesDepth();
        printMetricsToConsole(metrics);
        saveMetricsToJson(metrics, outputFile);
    }

    private static void printMetricsToConsole(Metrics metrics) {
        System.out.println("Max Depth: " + metrics.getMaxDepth());
        System.out.println("Average Depth: " + metrics.getAverageDepth());
        System.out.println("Average ABC: " + metrics.getAverageAbc());
        System.out.println("Average Overridden Methods: " + metrics.getAverageFieldsCount());
        System.out.println("Average Fields Count: " + metrics.getAverageOverriddenMethods());

    }

    private static void saveMetricsToJson(Metrics metrics, String outputFile) {
        String json = String.format(Locale.US,
                "{\n" +
                        "  \"maxDepth\": %d,\n" +
                        "  \"averageDepth\": %f,\n" +
                        "  \"averageAbc\": %f,\n" +
                        "  \"averageOverriddenMethods\": %f,\n" +
                        "  \"averageFieldsCount\": %f\n" +
                        "}",
                metrics.getMaxDepth(),
                metrics.getAverageDepth(),
                metrics.getAverageAbc(),
                metrics.getAverageOverriddenMethods(),
                metrics.getAverageFieldsCount()
        );

        try (FileWriter writer = new FileWriter(outputFile)) {
            writer.write(json);
            System.out.println("Metrics saved to file: " + outputFile);
        } catch (IOException e) {
            System.err.println("Error write file: " + e.getMessage());
        }
    }
}