org.thiesen.ant.git.ExtractGitInfo.java Source code

Introduction

Here is the source code for org.thiesen.ant.git.ExtractGitInfo.java

Source

/*
 * $ Id $
 * (c) Copyright 2009 Marcus Thiesen (marcus@thiesen.org)
 *
 *  This file is part of gitant.
 *
 *  gitant is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation, either version 3 of the License, or
 *  (at your option) any later version.
 *
 *  gitant is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with gitant.  If not, see <http://www.gnu.org/licenses/>.
 *
 */
package org.thiesen.ant.git;

import java.io.File;
import java.io.IOException;
import java.net.URL;
import java.util.Calendar;
import java.util.Enumeration;
import java.util.jar.Attributes;
import java.util.jar.Manifest;

import org.apache.commons.lang.StringUtils;
import org.apache.commons.lang.time.DateFormatUtils;
import org.apache.commons.lang.time.StopWatch;
import org.apache.tools.ant.BuildException;
import org.apache.tools.ant.Project;
import org.apache.tools.ant.Task;
import org.eclipse.jgit.errors.StopWalkException;

public class ExtractGitInfo extends Task {

    private static final String STATIC_PREFIX = "git.";

    private File _baseDir;
    private String _propertyPrefix;
    private boolean _displayInfo;

    public File getBaseDir() {
        return _baseDir;
    }

    @Override
    public void execute() throws BuildException {
        if (getBaseDir() == null) {
            throw new BuildException("baseDir property must be set.");
        }

        if (!getBaseDir().exists()) {
            throw new BuildException("Base dir " + getBaseDir().getAbsolutePath() + " does not exist!");
        }

        try {
            final StopWatch watch = new StopWatch();
            watch.start();
            final GitInfo info = GitInfoExtractor.extractInfo(getBaseDir());
            watch.stop();

            log("This is GitAnt " + loadVersion());
            log("Using " + loadJGitVersion());
            log("Data collection took " + watch);

            if (isDisplayInfo()) {
                log(info.getDisplayString(), Project.MSG_INFO);
            }

            final Project currentProject = getProject();
            if (currentProject != null) {
                exportProperties(info, currentProject);
            }

        } catch (final IOException e) {
            throw new BuildException(e);
        }
    }

    private void exportProperties(final GitInfo info, final Project currentProject) {
        currentProject.setProperty(prefixName("branch"), info.getCurrentBranch());
        currentProject.setProperty(prefixName("workingcopy.dirty"), String.valueOf(info.isWorkingCopyDirty()));
        currentProject.setProperty(prefixName("commit"), info.getLastCommit());
        currentProject.setProperty(prefixName("commit.short"), info.getLastCommitShort());
        currentProject.setProperty(prefixName("commit.date"),
                DateFormatUtils.format(info.getLastCommitDate(), "EEE, dd MMM yyyy HH:mm:ss Z"));
        currentProject.setProperty(prefixName("tag"), info.getLastTagName());
        currentProject.setProperty(prefixName("tag.hash"), info.getLastTagHash());
        currentProject.setProperty(prefixName("tag.dirty"), String.valueOf(info.isLastTagDirty()));
        currentProject.setProperty(prefixName("tag.author.name"), info.getLastTagAuthorName());
        currentProject.setProperty(prefixName("tag.author.email"), info.getLastTagAuthorEmail());
        currentProject.setProperty(prefixName("dirty"),
                String.valueOf(info.isWorkingCopyDirty() || info.isLastTagDirty()));
        currentProject.setProperty(prefixName("version"), info.getVersionPostfix());
    }

    private String loadVersion() {
        try {
            final Enumeration<URL> resources = getClass().getClassLoader().getResources("META-INF/MANIFEST.MF");
            while (resources.hasMoreElements()) {

                final Manifest manifest = new Manifest(resources.nextElement().openStream());

                final Attributes mainAttributes = manifest.getMainAttributes();

                if ("gitant".equalsIgnoreCase(mainAttributes.getValue("Project-Name"))) {
                    return mainAttributes.getValue("Project-Version");
                }

            }

        } catch (final IOException E) {
            // do nothing
        }
        return "unknown version";

    }

    private String loadJGitVersion() {
        try {
            final Enumeration<URL> resources = getClass().getClassLoader().getResources("META-INF/MANIFEST.MF");
            while (resources.hasMoreElements()) {

                final Manifest manifest = new Manifest(resources.nextElement().openStream());

                final Attributes mainAttributes = manifest.getMainAttributes();

                if ("org.eclipse.jgit".equalsIgnoreCase(mainAttributes.getValue("Bundle-SymbolicName"))) {
                    return mainAttributes.getValue("Implementation-Title") + " "
                            + mainAttributes.getValue("Implementation-Version");
                }

            }

        } catch (final IOException E) {
            // do nothing
        }
        return "unknown version";

    }

    private String prefixName(final String string) {
        final String propertyPrefix = getPropertyPrefix();

        if (StringUtils.isNotBlank(propertyPrefix)) {
            return STATIC_PREFIX + propertyPrefix + "." + string;
        }

        return STATIC_PREFIX + string;

    }

    public void setBaseDir(final File baseDir) {
        _baseDir = baseDir;
    }

    public static void main(final String... args) {
        final ExtractGitInfo vf = new ExtractGitInfo();
        vf.setBaseDir(new File(args[0]));
        vf.setDisplayInfo(true);

        vf.execute();
    }

    public void setPropertyPrefix(final String propertyPrefix) {
        _propertyPrefix = propertyPrefix;
    }

    public String getPropertyPrefix() {
        return _propertyPrefix;
    }

    public void setDisplayInfo(final boolean displayInfo) {
        _displayInfo = displayInfo;
    }

    public boolean isDisplayInfo() {
        return _displayInfo;
    }

}