/*
 * Copyright (c) 2001, 2018, Oracle and/or its affiliates. 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.  Oracle designates this
 * particular file as subject to the "Classpath" exception as provided
 * by Oracle 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 Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
 * or visit www.oracle.com if you need additional information or have any
 * questions.
 */

/*
 * This source code is provided to illustrate the usage of a given feature
 * or technique and has been deliberately simplified. Additional steps
 * required for a production-quality application, such as security checks,
 * input validation and proper error handling, might not be present in
 * this sample code.
 */

package com.sun.tools.example.debug.tty;

import java.io.ByteArrayOutputStream;
import java.io.PrintWriter;
import java.util.*;
import java.text.MessageFormat;
/**
 * Internationalization (i18n) convenience methods for jdb.
 *
 * All program output should flow through these methods, and this is
 * the only class that should be printing directly or otherwise
 * accessing System.[out,err].
 *
 * @bug 4348376
 * @author Tim Bell
 */
public class MessageOutput {
    /**
     * The resource bundle containing localizable message content.
     * This is loaded by TTY.main() at start-up
     */
    static ResourceBundle textResources;

    /** If set then used to buffer the output before writing it to System.out */
    private final static ThreadLocal<BufferedOutput> bufferedOutput = new ThreadLocal<>();

    /** Default writer to System.out */
    private final static PrintWriter systemOutWriter = new PrintWriter(System.out, true);

    /** Our message formatter.  Allocated once, used many times */
    private static MessageFormat messageFormat;

    /**
     * Fatal shutdown notification.  This is sent to System.err
     * instead of System.out
     */
    static void fatalError(String messageKey) {
        System.err.println();
        System.err.println(format("Fatal error"));
        System.err.println(format(messageKey));
        Env.shutdown();
    }

    /**
     * "Format" a string by doing a simple key lookup.
     */
    static String format(String key) {
        return (textResources.getString(key));
    }

    /**
     * Fetch and format a message with one string argument.
     * This is the most common usage.
     */
    static String format(String key, String argument) {
        return format(key, new Object [] {argument});
    }

    /**
     * Fetch a string by key lookup and format in the arguments.
     */
    static synchronized String format(String key, Object [] arguments) {
        if (messageFormat == null) {
            messageFormat = new MessageFormat (textResources.getString(key));
        } else {
            messageFormat.applyPattern (textResources.getString(key));
        }
        return (messageFormat.format (arguments));
    }

    /**
     * Print directly to System.out.
     * Every rule has a few exceptions.
     * The exceptions to "must use the MessageOutput formatters" are:
     *     VMConnection.dumpStream()
     *     TTY.monitorCommand()
     *     TTY.TTY() (for the '!!' command only)
     *     Commands.java (multiple locations)
     * These are the only sites that should be calling this
     * method.
     */
    static void printDirectln(String line) {
        getWriter().println(line);
    }
    static void printDirect(String line) {
        getWriter().print(line);
    }
    static void printDirect(char c) {
        getWriter().print(c);
    }

    /**
     * Print a newline.
     * Use this instead of '\n'
     */
    static void println() {
        getWriter().println();
    }

    /**
     * Format and print a simple string.
     */
    static void print(String key) {
        getWriter().print(format(key));
    }
    /**
     * Format and print a simple string.
     */
    static void println(String key) {
        getWriter().println(format(key));
    }


    /**
     * Fetch, format and print a message with one string argument.
     * This is the most common usage.
     */
    static void print(String key, String argument) {
        getWriter().print(format(key, argument));
    }
    static void println(String key, String argument) {
        getWriter().println(format(key, argument));
    }

    /**
     * Fetch, format and print a message with an arbitrary
     * number of message arguments.
     */
    static void println(String key, Object [] arguments) {
        getWriter().println(format(key, arguments));
    }

    /**
     * Print a newline, followed by the string.
     */
    static void lnprint(String key) {
        getWriter().println();
        getWriter().print(textResources.getString(key));
    }

    static void lnprint(String key, String argument) {
        getWriter().println();
        getWriter().print(format(key, argument));
    }

    static void lnprint(String key, Object [] arguments) {
        getWriter().println();
        getWriter().print(format(key, arguments));
    }

    /**
     * Print an exception message with a stack trace.
     */
    static void printException(String key, Exception e) {
        if (key != null) {
            try {
                println(key);
            } catch (MissingResourceException mex) {
                printDirectln(key);
            }
        }
        getWriter().flush();
        e.printStackTrace(getWriter());
    }

    static void printPrompt() {
        printPrompt(false);
    }

    static void printPrompt(boolean simple) {
        ThreadInfo threadInfo = ThreadInfo.getCurrentThreadInfo();
        if (simple || threadInfo == null) {
            getWriter().print
                (MessageOutput.format("jdb prompt with no current thread"));
        } else {
            getWriter().print
                (MessageOutput.format("jdb prompt thread name and current stack frame",
                                      new Object [] {
                                          threadInfo.getThread().name(),
                                          Integer.valueOf(threadInfo.getCurrentFrameIndex() + 1)}));
        }
        getWriter().flush();
    }

    /** Starts buffering the output to the internal buffer */
    static void startBuffering() {
        assert (bufferedOutput.get() == null);
        bufferedOutput.set(new BufferedOutput());
    }

    /** Stops buffering the output and writes the buffer to System.out */
    static void stopBuffering() {
        BufferedOutput bo = bufferedOutput.get();
        assert (bo != null);
        bufferedOutput.set(null);
        bo.close();
        systemOutWriter.print(bo.toString());
        systemOutWriter.flush();
    }

    /**
     * If bufferedOutput is set returns the writer the writes to the internal buffer, otherwise
     * returns writer that writes to System.out
     */
    private static PrintWriter getWriter() {
        return bufferedOutput.get() != null? bufferedOutput.get().getPrintWriter() : systemOutWriter;
    }

    /** Provides a writer that buffers the output to the string */
    private static class BufferedOutput {
        private final ByteArrayOutputStream baos;
        private final PrintWriter pw;

        BufferedOutput() {
            baos = new ByteArrayOutputStream();
            pw = new PrintWriter(baos);
        }

        PrintWriter getPrintWriter() {
            return pw;
        }

        void close() {
            pw.flush();
            pw.close();
        }

        @Override
        public String toString() {
            return baos.toString();
        }

    }
}