package com.github.jcgay.maven.color.agent;

import org.objectweb.asm.ClassReader;
import org.objectweb.asm.ClassWriter;

import java.lang.instrument.ClassFileTransformer;
import java.lang.instrument.IllegalClassFormatException;
import java.lang.instrument.Instrumentation;
import java.security.ProtectionDomain;

import static com.github.jcgay.maven.color.agent.MavenSurefireVisitor.Version;

/**
 * Java agent to rewrite some part of Maven logging initialization.
 */
public class ChangeMavenLogger {

    public static void premain(String agentArgs, Instrumentation inst) {
        inst.addTransformer(new ReplaceMavenLoggerWithAnsiLogger(), true);
    }

    /**
     * Rewrite classes:
     * <ul>
     *     <li>Maven : MavenCli#setupLogger</li>
     *     <li>Surefire 2.13 : DefaultReporterFactory#createConsoleLogger</li>
     *     <li>Surefire 2.9 : FileReporterFactory#createConsoleLogger</li>
     * </ul>
     */
    private static class ReplaceMavenLoggerWithAnsiLogger implements ClassFileTransformer {

        public byte[] transform(ClassLoader classLoader, String s, Class<?> aClass, ProtectionDomain protectionDomain, byte[] bytes) throws IllegalClassFormatException {

            if (s.equals("org/apache/maven/cli/MavenCli")) {
                ClassReader reader = new ClassReader(bytes);
                ClassWriter writer = new ClassWriter(reader, 0);
                MavenCliVisitor visitor = new MavenCliVisitor(writer);
                reader.accept(visitor, 0);
                return writer.toByteArray();
            }
            if (s.equals("org/apache/maven/plugin/surefire/report/DefaultReporterFactory")) {
                ClassReader reader = new ClassReader(bytes);
                ClassWriter writer = new ClassWriter(reader, 0);
                MavenSurefireVisitor visitor = new MavenSurefireVisitor(writer, Version.SUREFIRE_2_13);
                reader.accept(visitor, 0);
                return writer.toByteArray();
            }
            if (s.equals("org/apache/maven/plugin/surefire/report/FileReporterFactory")) {
                ClassReader reader = new ClassReader(bytes);
                ClassWriter writer = new ClassWriter(reader, 0);
                MavenSurefireVisitor visitor = new MavenSurefireVisitor(writer, Version.SUREFIRE_2_9);
                reader.accept(visitor, 0);
                return writer.toByteArray();
            }

            return null;
        }
    }
}

@@ -640,14 +640,8 @@
 er()
-, true
 );%0A 

Remove argument not available in Java 5.