Subversion Repositories XServices

/*

 *   Copyright 2013 Brian Rosenberger (Brutex Network)

 *

 *   Licensed 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 net.brutex.svn;

import java.io.BufferedReader;

import java.io.File;

import java.io.IOException;

import java.io.InputStreamReader;

import java.util.ArrayList;

import java.util.Date;

import java.util.List;

import java.util.Map;

import java.util.StringTokenizer;

import net.brutex.svn.SVNCommitInfo.ChangeType;

import org.apache.log4j.Logger;

/* Executes the svnlook utility

 *

 * @author Brian Rosenberger bru(at)brutex.de

 * @since 0.1

 */

public class SVNLookExecutor {

	private Logger logger = Logger.getLogger(SVNLookExecutor.class);

	private final File svnlook;

	private final String repos;

	private String TXN = null;

	private String rev = null;

	private String locale = "de_DE.UTF-8";

	private String encoding = "UTF-8";

	/**

	 * Instantiates a new sVN look executor.

	 *

	 * @param svnlook the svnlook

	 * @param repos the repos

	 */

	public SVNLookExecutor(File svnlook, String repos) {

		if(! svnlook.exists() ) throw new IllegalArgumentException( String.format("The svnlook executable at '%s' does not exist.", svnlook.toString()));

		if(! svnlook.isFile() ) throw new IllegalArgumentException( String.format("The svnlook utility at'%s' is not a file.", svnlook.toString()));

		logger.debug(String.format("Instantiating '%s' with svnlook at '%s'.", this.getClass().getCanonicalName(), svnlook.toString()));

		this.svnlook = svnlook;

		logger.debug(String.format("Working against svn repository at '%s'.", repos));

		this.repos = repos;

	}

	/**

	 * Execute svn look.

	 *

	 * @param command the command

	 * @return the string

	 */

	private String executeSVNLook(SVNLookCommand command) {

		StringBuilder sb = new StringBuilder();

		StringBuilder sberr = new StringBuilder();

		//This throws an IllegalArgumentException when neither TXN nor REV is valid

		String[] params;

		try {

			params = getTargetParam();

		} catch (IllegalArgumentException e) {

			logger.error(e.getMessage());

			throw e;

		}

		try {

			List<String> cmdline =  new ArrayList<String>(5);

			cmdline.add(svnlook.toString());

			cmdline.add(command.getValue());

			cmdline.add(params[0]);

			cmdline.add(params[1]);

			cmdline.add(repos);

			ProcessBuilder pf = new ProcessBuilder(cmdline);

			logger.debug(String.format("Executing svnlook with commandline '%s'.", pf.command()));

			Map<String, String> env = pf.environment();

			env.put("LANG", locale);

			Process svnprocess = pf.start();

			BufferedReader stdin = new BufferedReader(new InputStreamReader(svnprocess.getInputStream(), encoding));

			BufferedReader stderr = new BufferedReader(new InputStreamReader(svnprocess.getErrorStream(), encoding));

			String s;

			 while( (s = stdin.readLine()) != null ) {

				 sb.append(s + '\n');

			 }

			 while( (s = stderr.readLine()) != null ) {

				 sberr.append(s + '\n');

			 }

			 stdin.close(); stderr.close();

		} catch (IOException e) {

			logger.error( String.format( "Error calling the svnlook utility: '%s'", e.getMessage()));

			e.printStackTrace();

		}

		String error = sberr.toString();

		if( error.length()>0 ) {

			error = "Failed to call svnlook. The STDERR was '"+error+"'";

			logger.error(error);

			throw new IllegalArgumentException(error);

		}

		String output = sb.toString().trim();

		logger.debug(String.format("Svnlook output was '%s'", output));

		return output;

	}

	/*

	 * Returns either -t TXN oder -rev REVISION

	 *

	 * @returns

	 */

	private String[] getTargetParam() {

		String[] result = new String[2];

		if( (TXN==null || TXN.length()<=0) && (rev==null || rev.length()<=0) ) throw new IllegalArgumentException("Either TXN or revision must be provided.");

		if( TXN!=null && TXN.length()>0 && rev!=null && rev.length()>0 ) throw new IllegalArgumentException("Both, TXN and revision are given. Don't know what to use.");

		if( TXN!=null && TXN.length()>0) {

			result[0] = "-t";

			result[1] = TXN;

			return result;

		}

		if( rev!=null && rev.length()>0) {

			result[0] = "-r";

			result[1] = rev;

			return result;

		}

		throw new IllegalArgumentException("Either TXN or revision must be provided.");

	}

	public void setTXN(String TXN) {

		if(TXN==null || TXN.length()<=0) throw new IllegalArgumentException("TXN cannot be null or empty.");

		this.TXN = TXN;

	}

	public void setRev(String rev) {

		if(rev==null || rev.length()<=0) throw new IllegalArgumentException("Revision cannot be null or empty.");

		this.rev = rev;

	}

	public SVNCommitInfo getCommitInfo() {

		String author = executeSVNLook(SVNLookCommand.AUTHOR);

		String logmessage = executeSVNLook(SVNLookCommand.LOG);

		String files = executeSVNLook(SVNLookCommand.CHANGED);

		SVNCommitInfo result = new SVNCommitInfo(author,

												new Date(),

												logmessage);

		files += "\n";

		StringTokenizer tokenizer = new StringTokenizer(files, "\n");

		String s;

		while( tokenizer.hasMoreTokens()) {

			s=tokenizer.nextToken();

			logger.debug(String.format("Tokenizing file list. Token '%s'.", s));

			if(s.startsWith("A")) { result.addFileInfo(ChangeType.ADDED, s.substring(1).trim()); continue; }

			if(s.startsWith("D")) { result.addFileInfo(ChangeType.DELETED, s.substring(1).trim()); continue; }

			if(s.startsWith("UU")) { result.addFileInfo(ChangeType.BOTHUPDATE, s.substring(2).trim()); continue; }

			if(s.startsWith("_U")) { result.addFileInfo(ChangeType.METAUPDATE, s.substring(2).trim()); continue; }

			if(s.startsWith("U")) { result.addFileInfo(ChangeType.UPDATED, s.substring(1).trim()); continue; }

		}

		result.setTxn(TXN);

		result.setRev(rev);

		return result;

	}

	public void setEncoding(String encoding) {

		this.encoding = encoding;

	}

	 /**

	  * Set the locale that will be pushed into

	  * 'LANG' environment variable.

	  *

	  * @param locale i.e. de_DE.UTF-8

	  */

	public void setLocale(String locale) {

		this.locale = locale;

	}

}