package net.sf.mvnflexreports;
   
   import java.io.BufferedReader;
   import java.io.File;
   import java.io.FileInputStream;
   import java.io.InputStreamReader;
   import java.io.StringReader;
   import java.util.ArrayList;
   import java.util.List;
  import java.util.Locale;
  import java.util.Map;
  import java.util.ResourceBundle;
  import java.util.TreeMap;
  
  import javax.xml.stream.XMLInputFactory;
  import javax.xml.stream.XMLStreamConstants;
  import javax.xml.stream.XMLStreamReader;
  
  import org.apache.maven.doxia.siterenderer.Renderer;
  import org.apache.maven.project.MavenProject;
  import org.apache.maven.reporting.MavenReportException;
  
  import uk.co.badgersinfoil.metaas.ActionScriptFactory;
  import uk.co.badgersinfoil.metaas.ActionScriptParser;
  import uk.co.badgersinfoil.metaas.dom.ASClassType;
  import uk.co.badgersinfoil.metaas.dom.ASCompilationUnit;
  import uk.co.badgersinfoil.metaas.dom.ASPackage;
  import uk.co.badgersinfoil.metaas.dom.ASType;
  
  /***
   * @author Rémi Flament
   * @goal asuml-maven-report
   * @phase site
   * 
   */
  public class ActionScriptUMLMavenReport extends AbstractGraphvizMavenReport {
  
  	/***
  	 * The output directory for the final HTML report. Note that this parameter
  	 * is only evaluated if the goal is run directly from the command line or
  	 * during the default lifecycle. If the goal is run indirectly as part of a
  	 * site generation, the output directory configured in the Maven Site Plugin
  	 * is used instead.
  	 * 
  	 * @parameter expression="${project.reporting.outputDirectory}"
  	 * @required
  	 */
  	protected File outputDirectory;
  
  	/***
  	 * The fill color for classes in the graphical report
  	 * 
  	 * @parameter expression="cornflowerblue"
  	 */
  	protected String classFillColor;
  
  	/***
  	 * The fill color for interfaces in the graphical report
  	 * 
  	 * @parameter expression="firebrick1"
  	 */
  	protected String interfaceFillColor;
  
  	/***
  	 * The fill color for mxmls in the graphical report
  	 * 
  	 * @parameter expression="cadetblue3"
  	 */
  	protected String mxmlFillColor;
  
  	/***
  	 * The layout to use. This parameter will be used as the rankdir parameter.
  	 * 
  	 * @parameter expression="BT"
  	 */
  	protected String actionScriptUMLLayout;
  
  	/***
  	 * The project to analyse.
  	 * 
  	 * @parameter expression="${project}"
  	 * @required
  	 * @readonly
  	 */
  	protected MavenProject project;
  
  	/***
  	 * The directory containing source code to parse.
  	 * 
  	 * @parameter expression="${basedir}/src/main/flex"
  	 * @required
  	 */
  	private File sourceDir;
  
  	/***
  	 * Show attributes and methods in the uml report.
  	 * 
  	 * @parameter expression="true"
  	 */
 	private boolean showAttributesAndMethods;
 
 	/***
 	 * Show associations between classes in the uml report.
 	 * 
 	 * @parameter expression="true"
 	 */
 	private boolean showAssociations;
 
 	/***
 	 * Site rendering component for generating the HTML report.
 	 * 
 	 * @component
 	 */
 	private Renderer siteRenderer;
 
 	private ActionScriptParser parser = new ActionScriptFactory().newParser();
 
 	@Override
 	protected void executeReport(Locale locale) throws MavenReportException {
 
 		List<File> files = new ArrayList<File>();
 
 		ClassLoader origLoader = Thread.currentThread().getContextClassLoader();
 
 		List<ActionScriptParseError> unparseables = new ArrayList<ActionScriptParseError>();
 		Map<String, List<ASType>> asTypes = new TreeMap<String, List<ASType>>();
 
 		try {
 			getFiles(sourceDir.getAbsoluteFile(), files, ".as", ".mxml");
 			Thread.currentThread().setContextClassLoader(this.getClass().getClassLoader());
 			for (File sourceFile : files) {
 				getLog().debug("Processing flex source file " + sourceFile.getCanonicalPath());
 				try {
 					ASPackage pack = null;
 
 					if (sourceFile.getAbsolutePath().endsWith(".as")) {
 						pack = parseAsFile(sourceFile);
 					} else if (sourceFile.getAbsolutePath().endsWith(".mxml")) {
 						pack = parseMxmlFile(sourceDir.getAbsoluteFile(), sourceFile);
 					}
 
 					if (pack != null) {
 
 						List<ASType> list = asTypes.get(pack.getName() != null ? pack.getName() : "");
 
 						if (list == null) {
 							list = new ArrayList<ASType>();
 							asTypes.put(pack.getName() != null ? pack.getName() : "", list);
 						}
 
 						list.add(pack.getType());
 					}
 
 				} catch (Exception e) {
 					ActionScriptParseError actionScriptParseError = new ActionScriptParseError();
 					actionScriptParseError.setSourceFile(sourceFile);
 					actionScriptParseError.setError(e.getMessage());
 					unparseables.add(actionScriptParseError);
 					getLog().warn("Error while parsing source file " + sourceFile.getCanonicalPath(), e);
 				}
 			}
 
 			ActionScriptUMLReportGenerator generator = new ActionScriptUMLReportGenerator(getSink());
 			getLog().debug("classFillColor = " + classFillColor);
 			getLog().debug("interfaceFillColor = " + interfaceFillColor);
 			generator.setClassFillColor(classFillColor);
 			generator.setInterfaceFillColor(interfaceFillColor);
 			generator.setMxmlFillColor(mxmlFillColor);
 			generator.setShowAttributesAndMethods(showAttributesAndMethods);
 			generator.setShowAssociations(showAssociations);
 			generator.setFontName(fontName);
 			generator.setFontSize(fontSize);
 			generator.setProgramName(programName);
 			generator.setLayout(actionScriptUMLLayout);
 			generator.generate(sourceDir, asTypes, unparseables, this.getReportOutputDirectory(), getLog());
 
 		} catch (Exception e) {
 			throw new MavenReportException(e.getMessage(), e);
 		} finally {
 			Thread.currentThread().setContextClassLoader(origLoader);
 		}
 
 	}
 
 	private String readFileContent(File file) throws Exception {
 		StringBuilder content = new StringBuilder();
 
 		BufferedReader bufferedReader = new BufferedReader(new InputStreamReader(new FileInputStream(file)));
 		String line;
 		while ((line = bufferedReader.readLine()) != null) {
 			content.append(line + "\n");
 		}
 		return content.toString().replace("super.", ""); // help metaas,
 		// this
 		// doesn't change
 		// anything for us
 	}
 
 	private ASPackage parseAsFile(File sourceFile) throws Exception {
 		StringReader reader = new StringReader(readFileContent(sourceFile));
 		ASCompilationUnit unit = parser.parse(reader);
 
 		ASPackage pkg = unit.getPackage();
 
 		return pkg;
 
 	}
 
 	private ASPackage parseMxmlFile(File sourceDir, File sourceFile) throws Exception {
 
 		XMLInputFactory factory = XMLInputFactory.newInstance();
 		XMLStreamReader p = factory.createXMLStreamReader(new FileInputStream(sourceFile));
 
 		String className = sourceFile.getName().substring(0, sourceFile.getName().indexOf('.'));
 		String packageName = sourceFile.getParent().replace(sourceDir.getAbsolutePath(), "").replace(File.separatorChar, '.');
 		if (packageName.startsWith("."))
 			packageName = packageName.substring(1);
 
 		String firstElement = null;
 		while (true) {
 			int event = p.next();
 			if (event == XMLStreamConstants.END_DOCUMENT) {
 				p.close();
 				break;
 			}
 			if (event == XMLStreamConstants.START_ELEMENT) {
 
 				if (firstElement == null) {
 					firstElement = p.getLocalName();
 				}
 
 				if (p.getLocalName().equals("Script")) {
 
 					String source = "package " + packageName + " {\npublic class " + className + "	  {\n" + p.getElementText() + "\n}}";
 
 					StringReader reader = new StringReader(source);
 					ASCompilationUnit unit = parser.parse(reader);
 
 					ASPackage pkg = unit.getPackage();
 					ASType type = pkg.getType();
 
 					type.newMetaTag("MXML");
 
 					ASClassType c = (ASClassType) type;
 					c.setSuperclass(firstElement);
 
 					return pkg;
 
 				}
 			}
 		}
 
 		return null;
 
 	}
 
 	public String getOutputName() {
 		return "asuml-maven-report";
 	}
 
 	/***
 	 * @return Returns the sourceDir.
 	 */
 	public File getSourceDir() {
 		return sourceDir;
 	}
 
 	/***
 	 * @param sourceDir
 	 *            The sourceDir to set.
 	 */
 	public void setSourceDir(File sourceDir) {
 		this.sourceDir = sourceDir;
 	}
 
 	protected String getOutputDirectory() {
 		return outputDirectory.getAbsolutePath();
 
 	}
 
 	protected MavenProject getProject() {
 		return project;
 	}
 
 	public String getDescription(Locale locale) {
 		return getBundle(locale).getString("report.asuml-maven-report.name");
 
 	}
 
 	public String getName(Locale locale) {
 		return getBundle(locale).getString("report.asuml-maven-report.name");
 
 	}
 
 	private static ResourceBundle getBundle(Locale locale) {
 		return ResourceBundle.getBundle("asuml-maven-report", locale, ActionScriptUMLReportGenerator.class.getClassLoader());
 	}
 
 	public String getClassFillColor() {
 		return classFillColor;
 	}
 
 	public void setClassFillColor(String classFillColor) {
 		this.classFillColor = classFillColor;
 	}
 
 	public String getInterfaceFillColor() {
 		return interfaceFillColor;
 	}
 
 	public void setInterfaceFillColor(String interfaceFillColor) {
 		this.interfaceFillColor = interfaceFillColor;
 	}
 
 	public boolean isShowAttributesAndMethods() {
 		return showAttributesAndMethods;
 	}
 
 	public void setShowAttributesAndMethods(boolean showAttributesAndMethods) {
 		this.showAttributesAndMethods = showAttributesAndMethods;
 	}
 
 	@Override
 	protected Renderer getSiteRenderer() {
 		return siteRenderer;
 	}
 
 	public boolean isShowAssociations() {
 		return showAssociations;
 	}
 
 	public void setShowAssociations(boolean showAssociations) {
 		this.showAssociations = showAssociations;
 	}
 
 	public String getActionScriptUMLLayout() {
 		return actionScriptUMLLayout;
 	}
 
 	public void setActionScriptUMLLayout(String actionScriptUMLLayout) {
 		this.actionScriptUMLLayout = actionScriptUMLLayout;
 	}
 
 	public void setMxmlFillColor(String mxmlFillColor) {
 		this.mxmlFillColor = mxmlFillColor;
 	}
 
 }