KeyReader.java

package nl.minicom.gitolite.manager.models;

import java.io.BufferedReader;
import java.io.File;
import java.io.FileReader;
import java.io.FilenameFilter;
import java.io.IOException;


import com.google.common.base.Preconditions;

/**
 * This class contains a method to read all SSH keys from a specific directory
 * and register them with a provided {@link Config} object.
 * 
 * @author Michael de Jong <<a href="mailto:michaelj@minicom.nl">michaelj@minicom.nl</a>>
 */
public final class KeyReader {

	/**
	 * This method reads all SSH keys from the specified key directory, and registers them
	 * with the specified {@link Config} object.
	 * 
	 * @param config
	 * 	The {@link Config} to register the keys with.
	 * 
	 * @param keyDir
	 * 	The directory where all the SSH keys are registered.
	 * 
	 * @throws IOException
	 * 	If there were problems when reading the key directory.
	 */
	public static void readKeys(Config config, File keyDir) throws IOException {
		Preconditions.checkNotNull(config);
		Preconditions.checkNotNull(keyDir);
		Preconditions.checkArgument(keyDir.isDirectory(), "The argument 'keyDir' must be a directory!");
		
		File[] files = keyDir.listFiles(new FilenameFilter() {
			@Override
			public boolean accept(File arg0, String arg1) {
				return arg1.endsWith(".pub");
			}
		});
		
		for (File keyFile : files) {
			String fileName = keyFile.getName();
			if (!fileName.contains("@")) {
				fileName = fileName.replace(".pub", "@.pub");
			}
			
			int indexOfAt = fileName.indexOf('@');
			String userName = fileName.substring(0, indexOfAt);
			String keyName = fileName.substring(indexOfAt + 1, fileName.indexOf(".pub"));
			String content = readKeyFile(keyFile);
			
			config.ensureUserExists(userName).setKey(keyName, content);
		}
	}

	private static String readKeyFile(File keyFile) throws IOException {
		BufferedReader reader = new BufferedReader(new FileReader(keyFile));
		StringBuilder builder = new StringBuilder();
		
		try {
			String line;
			while ((line = reader.readLine()) != null) {
				if (builder.length() > 0) {
					builder.append("\n");
				}
				builder.append(line);
			}
		}
		finally {
			reader.close();
		}
		
		return builder.toString();
	}
	
	private KeyReader() {
		//Prevent instantiation.
	}
	
}