org.eclipse.jdt.apt.core/src/org/eclipse/jdt/apt/core/internal/env/JavaSourceFilePrintWriter.java

/*******************************************************************************
 * Copyright (c) 2005 BEA Systems, 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:
 *    mkaufman@bea.com - initial API and implementation
 *******************************************************************************/

package org.eclipse.jdt.apt.core.internal.env;

import java.io.PrintWriter;
import java.io.StringWriter;
import java.util.Collections;

import org.eclipse.core.runtime.CoreException;
import org.eclipse.jdt.apt.core.env.Phase;
import org.eclipse.jdt.apt.core.internal.AptPlugin;
import org.eclipse.jdt.apt.core.internal.generatedfile.FileGenerationResult;
import org.eclipse.jdt.apt.core.internal.generatedfile.GeneratedFileManager;
import org.eclipse.jdt.core.ICompilationUnit;

public class JavaSourceFilePrintWriter extends PrintWriter {

	private final StringWriter _sw;
    private final String _typeName;
    private final AbstractCompilationEnv _env;
	
    /**
     * @throws CoreException if type name is not valid
     */
	public JavaSourceFilePrintWriter( String typeName, StringWriter sw, AbstractCompilationEnv env ) 
    	throws CoreException
    {
        super( sw );
        _sw = sw;
        _typeName = typeName;
        _env = env;
		_env.validateTypeName(typeName);
    }
	
    public void close()
    {	
    	try {
	    	String contents = _sw.toString();
	        super.close();
	        GeneratedFileManager gfm = _env.getAptProject().getGeneratedFileManager();
	        Phase phase = _env.getPhase();
			
	        FileGenerationResult result = null;
	        if ( phase == Phase.RECONCILE )
	        {
	        	ReconcileEnv reconcileEnv = (ReconcileEnv)_env;
	        	ICompilationUnit parentCompilationUnit = reconcileEnv.getCompilationUnit();
	            result  = gfm.generateFileDuringReconcile( 
	                parentCompilationUnit, _typeName, contents );
	            // Need to call ReconcileContext.resetAst() for this to be effective;
	            // that will happen in ReconcileEnv.close().
	        }
	        else if ( phase == Phase.BUILD)	{
		        result = gfm.generateFileDuringBuild( 
						Collections.singletonList(_env.getFile()),  _typeName, contents, null /* progress monitor */ );
	        }
	        if (result != null) {
	        	_env.addGeneratedSourceFile(result.getFile(), result.isModified());
	        }
    	}
    	catch (CoreException ce) {
    		AptPlugin.log(ce, "Unable to generate type when JavaSourceFilePrintWriter was closed"); //$NON-NLS-1$
    	}
    }
}