ScreenPojosRecursiveAjGenerator.java

/*******************************************************************************
 * Copyright (c) 2012 OpenLegacy Inc.
 * 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:
 *     OpenLegacy Inc. - initial API and implementation
 *******************************************************************************/
package org.openlegacy.designtime.mains;

import freemarker.template.TemplateException;

import org.openlegacy.designtime.terminal.generators.ScreenPojosAjGenerator;
import org.openlegacy.utils.FileCommand;
import org.openlegacy.utils.FileCommandExecuter;
import org.springframework.context.ApplicationContext;
import org.springframework.context.support.ClassPathXmlApplicationContext;
import org.springframework.stereotype.Component;

import japa.parser.ParseException;

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

import javax.inject.Inject;

@Component
public class ScreenPojosRecursiveAjGenerator {

	@Inject
	private ApplicationContext applicationContext;

	public static void main(String[] args) throws ParseException, IOException, TemplateException {
		if (args.length == 0) {
			System.out.println("Usage:\njava ScreenEntityRecursiveAjGenerator java-source-folder");
			return;
		}
		String root = args[0];

		ApplicationContext applicationContext = new ClassPathXmlApplicationContext(
				"classpath:/openlegacy-basic-designtime-context.xml");

		ScreenPojosRecursiveAjGenerator generator = applicationContext.getBean(ScreenPojosRecursiveAjGenerator.class);
		generator.generateAll(new File(root));
	}

	public void generateAll(File root) throws IOException, TemplateException {
		FileCommandExecuter.execute(root, applicationContext.getBean(AspectGeneratorCommand.class));
	}

	@Component
	public static class AspectGeneratorCommand implements FileCommand {

		@Inject
		private ScreenPojosAjGenerator screenPojosAjGenerator;

		public boolean accept(File file) {
			if (file.getName().endsWith("aj")) {
				return false;
			}
			if (file.isDirectory()) {
				return true;
			}
			if (file.getName().endsWith("java")) {
				return true;
			}

			return false;
		}

		public void doCommand(File file) {
			try {
				screenPojosAjGenerator.generate(file);
			} catch (Exception e) {
				throw (new RuntimeException(e));
			}

		}

	}

}