src/java/org/apache/ivy/ant/IvyExtractFromSources.java

/*
 *  Licensed to the Apache Software Foundation (ASF) under one or more
 *  contributor license agreements.  See the NOTICE file distributed with
 *  this work for additional information regarding copyright ownership.
 *  The ASF licenses this file to You under the Apache License, Version 2.0
 *  (the "License"); you may not use this file except in compliance with
 *  the License.  You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 *
 */
package org.apache.ivy.ant;

import java.io.File;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.PrintWriter;
import java.io.StringWriter;
import java.io.Writer;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.Set;

import org.apache.ivy.core.module.id.ModuleRevisionId;
import org.apache.tools.ant.BuildException;
import org.apache.tools.ant.Project;
import org.apache.tools.ant.Task;
import org.apache.tools.ant.filters.LineContainsRegExp;
import org.apache.tools.ant.filters.TokenFilter;
import org.apache.tools.ant.taskdefs.Concat;
import org.apache.tools.ant.types.FileSet;
import org.apache.tools.ant.types.FilterChain;
import org.apache.tools.ant.types.RegularExpression;

/**
 * Extracts imports from a set of java sources and generate corresponding ivy file
 */
public class IvyExtractFromSources extends Task {
    public static class Ignore {
        private String packageName;

        public String getPackage() {
            return packageName;
        }

        public void setPackage(String package1) {
            packageName = package1;
        }
    }

    private String organisation;

    private String module;

    private String revision;

    private String status;

    private List ignoredPackaged = new ArrayList(); // List (String package)

    private Map mapping = new HashMap(); // Map (String package -> ModuleRevisionId)

    private Concat concat = new Concat();

    private File to;

    public void addConfiguredIgnore(Ignore ignore) {
        ignoredPackaged.add(ignore.getPackage());
    }

    public File getTo() {
        return to;
    }

    public void setTo(File to) {
        this.to = to;
    }

    public String getModule() {
        return module;
    }

    public void setModule(String module) {
        this.module = module;
    }

    public String getOrganisation() {
        return organisation;
    }

    public void setOrganisation(String organisation) {
        this.organisation = organisation;
    }

    public String getRevision() {
        return revision;
    }

    public void setRevision(String revision) {
        this.revision = revision;
    }

    public String getStatus() {
        return status;
    }

    public void setStatus(String status) {
        this.status = status;
    }

    public void addConfiguredMapping(PackageMapping mapping) {
        this.mapping.put(mapping.getPackage(), mapping.getModuleRevisionId());
    }

    public void addFileSet(FileSet fileSet) {
        concat.addFileset(fileSet);
    }

    public void execute() throws BuildException {
        configureConcat();
        Writer out = new StringWriter();
        concat.setWriter(out);
        concat.execute();
        Set importsSet = new HashSet(Arrays.asList(out.toString().split("\n")));
        Set dependencies = new HashSet();
        for (Iterator iter = importsSet.iterator(); iter.hasNext();) {
            String pack = ((String) iter.next()).trim();
            ModuleRevisionId mrid = getMapping(pack);
            if (mrid != null) {
                dependencies.add(mrid);
            }
        }
        try {
            PrintWriter writer = new PrintWriter(new FileOutputStream(to));
            writer.println("<ivy-module version=\"1.0\">");
            writer.println("\t<info organisation=\"" + organisation + "\"");
            writer.println("\t       module=\"" + module + "\"");
            if (revision != null) {
                writer.println("\t       revision=\"" + revision + "\"");
            }
            if (status != null) {
                writer.println("\t       status=\"" + status + "\"");
            } else {
                writer.println("\t       status=\"integration\"");
            }
            writer.println("\t/>");
            if (!dependencies.isEmpty()) {
                writer.println("\t<dependencies>");
                for (Iterator iter = dependencies.iterator(); iter.hasNext();) {
                    ModuleRevisionId mrid = (ModuleRevisionId) iter.next();
                    writer.println("\t\t<dependency org=\"" + mrid.getOrganisation() + "\" name=\""
                            + mrid.getName() + "\" rev=\"" + mrid.getRevision() + "\"/>");
                }
                writer.println("\t</dependencies>");
            }
            writer.println("</ivy-module>");
            writer.close();
            log(dependencies.size() + " dependencies put in " + to);
        } catch (FileNotFoundException e) {
            throw new BuildException("impossible to create file " + to + ": " + e, e);
        }
    }

    /**
     * @param pack
     * @return
     */
    private ModuleRevisionId getMapping(String pack) {
        String askedPack = pack;
        ModuleRevisionId ret = null;
        while (ret == null && pack.length() > 0) {
            if (ignoredPackaged.contains(pack)) {
                return null;
            }
            ret = (ModuleRevisionId) mapping.get(pack);
            int lastDotIndex = pack.lastIndexOf('.');
            if (lastDotIndex != -1) {
                pack = pack.substring(0, lastDotIndex);
            } else {
                break;
            }
        }
        if (ret == null) {
            log("no mapping found for " + askedPack, Project.MSG_VERBOSE);
        }
        return ret;
    }

    private void configureConcat() {
        concat.setProject(getProject());
        concat.setTaskName(getTaskName());
        FilterChain filterChain = new FilterChain();
        LineContainsRegExp lcre = new LineContainsRegExp();
        RegularExpression regexp = new RegularExpression();
        regexp.setPattern("^import .+;");
        lcre.addConfiguredRegexp(regexp);
        filterChain.add(lcre);
        TokenFilter tf = new TokenFilter();
        TokenFilter.ReplaceRegex rre = new TokenFilter.ReplaceRegex();
        rre.setPattern("import (.+);.*");
        rre.setReplace("\\1");
        tf.add(rre);
        filterChain.add(tf);
        concat.addFilterChain(filterChain);
    }
}