XmlConfigurator.java

/**
 * This file is derived from Hamcrest's  {@link org.hamcrest.generator.config.XmlConfigurator} class.
 * Portions modified by Ketan Padegaonkar are provided under the EPL.
 *
 * Copyright (c) 2008 Ketan Padegaonkar and others.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *     Ketan Padegaonkar - initial API and implementation
 *******************************************************************************/
package org.eclipse.swtbot.generator;

import java.io.File;
import java.io.FileWriter;
import java.io.IOException;

import javax.xml.parsers.ParserConfigurationException;
import javax.xml.parsers.SAXParser;
import javax.xml.parsers.SAXParserFactory;

import org.xml.sax.Attributes;
import org.xml.sax.InputSource;
import org.xml.sax.SAXException;
import org.xml.sax.helpers.DefaultHandler;

public class XmlConfigurator {

	private final SugarGenerator	sugarConfiguration;
	private final SAXParserFactory	saxParserFactory;

	public XmlConfigurator(SugarGenerator sugarConfiguration, ClassLoader classLoader) {
		this.sugarConfiguration = sugarConfiguration;
		saxParserFactory = SAXParserFactory.newInstance();
		saxParserFactory.setNamespaceAware(true);
	}

	public void load(InputSource inputSource) throws ParserConfigurationException, SAXException, IOException {
		SAXParser saxParser = saxParserFactory.newSAXParser();
		saxParser.parse(inputSource, new DefaultHandler() {
			@Override
			public void startElement(String uri, String localName, String qName, Attributes attributes) throws SAXException {
				if (localName.equals("widget")) { //$NON-NLS-1$
					String className = attributes.getValue("class"); //$NON-NLS-1$
					try {
						addClass(className);
					} catch (ClassNotFoundException e) {
						throw new SAXException("Cannot find Matcher class : " + className); //$NON-NLS-1$
					}
				}
			}
		});
	}

	private void addClass(String className) throws ClassNotFoundException {
		sugarConfiguration.addFactoryMethods(new SWTBotGeneratorFactoryReader(className));
		sugarConfiguration.addImports(new SWTBotGeneratorFactoryReader(className));
	}

	public static void main(String configFile, String fullClassName, File outputDir) throws Exception {

		String fileName = fullClassName.replace('.', File.separatorChar) + ".java"; //$NON-NLS-1$
		int dotIndex = fullClassName.lastIndexOf("."); //$NON-NLS-1$
		String packageName = dotIndex == -1 ? "" : fullClassName.substring(0, dotIndex); //$NON-NLS-1$
		String shortClassName = fullClassName.substring(dotIndex + 1);

		if (!outputDir.isDirectory()) {
			System.err.println("Output directory not found : " + outputDir.getAbsolutePath()); //$NON-NLS-1$
			System.exit(-1);
		}

		File outputFile = new File(outputDir, fileName);
		outputFile.getParentFile().mkdirs();

		SugarGenerator sugarGenerator = new SugarGenerator();
		try {
			sugarGenerator.addWriter(new HamcrestFactoryWriter(packageName, shortClassName, new FileWriter(outputFile)));
			sugarGenerator.addWriter(new QuickReferenceWriter(System.out));

			XmlConfigurator xmlConfigurator = new XmlConfigurator(sugarGenerator, XmlConfigurator.class.getClassLoader());
			xmlConfigurator.load(new InputSource(configFile));
			System.out.println("Generating " + fullClassName); //$NON-NLS-1$
			sugarGenerator.generate();
		} finally {
			sugarGenerator.close();
		}
	}
}