jaxws/src/share/classes/com/sun/tools/internal/ws/processor/util/ClientProcessorEnvironment.java

/*
 * Portions Copyright 2006 Sun Microsystems, Inc.  All Rights Reserved.
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
 *
 * This code is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License version 2 only, as
 * published by the Free Software Foundation.  Sun designates this
 * particular file as subject to the "Classpath" exception as provided
 * by Sun in the LICENSE file that accompanied this code.
 *
 * This code is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
 * version 2 for more details (a copy is included in the LICENSE file that
 * accompanied this code).
 *
 * You should have received a copy of the GNU General Public License version
 * 2 along with this work; if not, write to the Free Software Foundation,
 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
 *
 * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
 * CA 95054 USA or visit www.sun.com if you need additional information or
 * have any questions.
 */

package com.sun.tools.internal.ws.processor.util;

import java.io.File;
import java.io.OutputStream;
import java.io.PrintStream;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;

import com.sun.tools.internal.ws.processor.ProcessorNotificationListener;
import com.sun.tools.internal.ws.processor.generator.Names;
import com.sun.tools.internal.ws.util.JAXWSClassFactory;
import com.sun.xml.internal.ws.util.localization.Localizable;

/**
 *
 * @author WS Development Team
 */
public class ClientProcessorEnvironment extends ProcessorEnvironmentBase
    implements ProcessorEnvironment {

    /**
     * The stream where error message are printed.
     */
    private OutputStream out;

    /**
     * A printwriter created lazily in case there are exceptions to report.
     */
    private PrintStream outprintstream;

    /**
     * listener for error/warning/info notifications
     */
    private ProcessorNotificationListener listener;

    /**
     * The classpath to use
     */
    private String classPath;

    /**
     * list of generated source files created in this environment and
     * its type
     */
    private List generatedFiles = new ArrayList();

    /**
     * The number of errors and warnings
     */
    private int nwarnings;
    private int nerrors;

    /**
     * flags
     */
    private int flags;

    private Names names;

    /**
     * Create a ClientProcessorEnvironment with the given class path,
     * stream for messages and ProcessorNotificationListener.
     */
    public ClientProcessorEnvironment(
        OutputStream out,
        String classPath,
        ProcessorNotificationListener listener) {

        this.out = out;
        this.classPath = classPath;
        this.listener = listener;
        flags = 0;

        //bug fix:4904604
        names = JAXWSClassFactory.newInstance().createNames();
    }

    /**
     * Set the environment flags
     */
    public void setFlags(int flags) {
        this.flags = flags;
    }

    /**
     * Get the environment flags
     */
    public int getFlags() {
        return flags;
    }

    /**
     * Get the ClassPath.
     */
    public String getClassPath() {
        return classPath;
    }

    /**
     * Is verbose turned on
     */
    public boolean verbose() {
        return (flags & F_VERBOSE) != 0;
    }

    /**
     * Remember info on  generated source file generated so that it
     * can be removed later, if appropriate.
     */
    public void addGeneratedFile(GeneratedFileInfo file) {
        generatedFiles.add(file);
    }

    /**
     * Return all the generated files and its types.
     */
    public Iterator getGeneratedFiles() {
        return generatedFiles.iterator();
    }

    /**
     * Delete all the generated source files made during the execution
     * of this environment (those that have been registered with the
     * "addGeneratedFile" method).
     */
    public void deleteGeneratedFiles() {
        synchronized (generatedFiles) {
            Iterator iter = generatedFiles.iterator();
            while (iter.hasNext()) {
                File file = ((GeneratedFileInfo)iter.next()).getFile();
                if (file.getName().endsWith(".java")) {
                    file.delete();
                }
            }
            generatedFiles.clear();
        }
    }

    /**
     * Release resources, if any.
     */
    public void shutdown() {
        listener = null;
        generatedFiles = null;
    }

    public void error(Localizable msg) {
        if (listener != null) {
            listener.onError(msg);
        }
        nerrors++;
    }

    public void warn(Localizable msg) {
        if (warnings()) {
            nwarnings++;
            if (listener != null) {
                listener.onWarning(msg);
            }
        }
    }

    public void info(Localizable msg) {
        if (listener != null) {
            listener.onInfo(msg);
        }
    }

    public void printStackTrace(Throwable t) {
        if (outprintstream == null) {
            outprintstream = new PrintStream(out);
        }
        t.printStackTrace(outprintstream);
    }

    public Names getNames() {
        return names;
    }

    public int getErrorCount() {
        return nerrors;
    }

    public int getWarningCount() {
        return nwarnings;
    }

    private boolean warnings() {
        return (flags & F_WARNINGS) != 0;
    }

    //bug fix:4904604
    //to called in compileTool after env is
    public void setNames(Names names) {
        this.names = names;
    }

}