001    /*
002     * The Apache Software License, Version 1.1
003     *
004     * Copyright (c) 2000-2002 The Apache Software Foundation.  All rights
005     * reserved.
006     *
007     * Redistribution and use in source and binary forms, with or without
008     * modification, are permitted provided that the following conditions
009     * are met:
010     *
011     * 1. Redistributions of source code must retain the above copyright
012     *    notice, this list of conditions and the following disclaimer.
013     *
014     * 2. Redistributions in binary form must reproduce the above copyright
015     *    notice, this list of conditions and the following disclaimer in
016     *    the documentation and/or other materials provided with the
017     *    distribution.
018     *
019     * 3. The end-user documentation included with the redistribution, if
020     *    any, must include the following acknowlegement:
021     *       "This product includes software developed by the
022     *        Apache Software Foundation (http://www.apache.org/)."
023     *    Alternately, this acknowlegement may appear in the software itself,
024     *    if and wherever such third-party acknowlegements normally appear.
025     *
026     * 4. The names "Ant" and "Apache Software
027     *    Foundation" must not be used to endorse or promote products derived
028     *    from this software without prior written permission. For written
029     *    permission, please contact apache@apache.org.
030     *
031     * 5. Products derived from this software may not be called "Apache"
032     *    nor may "Apache" appear in their names without prior written
033     *    permission of the Apache Group.
034     *
035     * THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESSED OR IMPLIED
036     * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
037     * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
038     * DISCLAIMED.  IN NO EVENT SHALL THE APACHE SOFTWARE FOUNDATION OR
039     * ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
040     * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
041     * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
042     * USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
043     * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
044     * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
045     * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
046     * SUCH DAMAGE.
047     * ====================================================================
048     *
049     * This software consists of voluntary contributions made by many
050     * individuals on behalf of the Apache Software Foundation.  For more
051     * information on the Apache Software Foundation, please see
052     * <http://www.apache.org/>.
053     */
054    package com.jcoverage.ant;
055    
056    import java.io.File;
057    
058    import java.net.URLClassLoader;
059    import java.net.URL;
060    
061    import java.util.HashSet;
062    import java.util.Iterator;
063    import java.util.Set;
064    
065    import org.apache.tools.ant.AntClassLoader;
066    import org.apache.tools.ant.BuildException;
067    import org.apache.tools.ant.taskdefs.Java;
068    import org.apache.tools.ant.taskdefs.MatchingTask;
069    import org.apache.tools.ant.types.Path;
070    import org.apache.tools.ant.types.Reference;
071    
072    public class CheckTask extends MatchingTask {
073      final Set regexes=new HashSet();
074    
075      private Java java=null;
076    
077      String branchCoverageRate=null;
078      String lineCoverageRate=null;
079    
080    
081      protected Java getJava() {
082        if(java==null) {
083          java=(Java)getProject().createTask("java");
084          java.setTaskName(getTaskName());
085          java.setClassname("com.jcoverage.tool.coveragecheck.Main");
086          java.setFork(true);
087          java.setDir(getProject().getBaseDir());
088    
089          if(getClass().getClassLoader() instanceof AntClassLoader) {
090            createClasspath().setPath(((AntClassLoader)getClass().getClassLoader()).getClasspath());
091          } else if(getClass().getClassLoader() instanceof URLClassLoader) {
092            URL[] earls=((URLClassLoader)getClass().getClassLoader()).getURLs();
093            for(int i=0;i<earls.length;i++) {
094              createClasspath().setPath(earls[i].getFile());
095            }
096          }
097        }
098        return java;
099      }
100    
101      public Path createClasspath() {
102        return getJava().createClasspath().createPath();
103      }
104    
105      public void setClasspath(Path classpath) {
106        createClasspath().append(classpath);
107      }
108    
109      public void setClasspathRef(Reference r) {
110        createClasspath().setRefid(r);
111      }
112    
113      public void setBranch(String branchCoverageRate) {
114        this.branchCoverageRate=branchCoverageRate;
115      }
116    
117      public void setLine(String lineCoverageRate) {
118        this.lineCoverageRate=lineCoverageRate;
119      }
120    
121      public Regex createRegex() {
122        Regex regex=new Regex();
123        regexes.add(regex);
124        return regex;
125      }
126    
127      public void execute() throws BuildException {
128        if(lineCoverageRate!=null) {
129          getJava().createArg().setValue("-l");
130          getJava().createArg().setValue(lineCoverageRate);
131        }
132    
133        if(branchCoverageRate!=null) {
134          getJava().createArg().setValue("-b");
135          getJava().createArg().setValue(branchCoverageRate);
136        }
137    
138        Iterator i=regexes.iterator();
139        while(i.hasNext()) {
140          getJava().createArg().setValue("-r");
141          getJava().createArg().setValue(i.next().toString());
142        }
143    
144        if(getJava().executeJava()!=0) {
145          throw new BuildException();
146        }
147      }
148    
149    }