package org.webslinger.instrumentation;

 import java.io.File;

 import java.io.IOException;

 import java.util.Collections;

 import org.objectweb.asm.ClassReader;

 import org.objectweb.asm.ClassWriter;

 import net.sourceforge.cobertura.coveragedata.CoverageDataFileHandler;

 import net.sourceforge.cobertura.coveragedata.ProjectData;

 import net.sourceforge.cobertura.instrument.ClassInstrumenter;

 import org.webslinger.launcher.Instrumenter;

 public final class CoberturaInstrumenter implements Instrumenter {

     protected File dataFile;

     protected ProjectData projectData;

     protected boolean forInstrumenting;

     public File getDefaultFile() throws IOException {

         return CoverageDataFileHandler.getDefaultDataFile();

     }

     public void open(File dataFile, boolean forInstrumenting) throws IOException {

         System.setProperty("net.sourceforge.cobertura.datafile", dataFile.toString());

         this.forInstrumenting = forInstrumenting;

         this.dataFile = dataFile;

         if (forInstrumenting) {

             if (dataFile.exists()) {

                 projectData = CoverageDataFileHandler.loadCoverageData(dataFile);

             } else {

                 projectData = new ProjectData();

             }

         }

     }

     public void close() throws IOException {

         if (forInstrumenting) {

             CoverageDataFileHandler.saveCoverageData(projectData, dataFile);

         }

     }

     public byte[] instrumentClass(byte[] bytes) throws IOException {

         ClassReader cr = new ClassReader(bytes);

         ClassWriter cw = new ClassWriter(true);

         ClassInstrumenter ci = new ClassInstrumenter(projectData != null ? projectData : ProjectData.getGlobalProjectData(), cw, Collections.EMPTY_LIST);

         cr.accept(ci, false);

         if (ci.isInstrumented()) return cw.toByteArray();

         return bytes;

     }

 }