Minecraft-SSHD/Minecraft-SSHD-Bukkit/src/main/java/com/ryanmichela/sshd/FlushyStreamHandler.java

package com.ryanmichela.sshd;

import jline.console.ConsoleReader;
import org.apache.sshd.common.SshException;

import java.io.IOException;
import java.io.OutputStream;
import java.util.logging.*;

/**
 * Copyright 2013 Ryan Michela
 */
public class FlushyStreamHandler extends StreamHandler 
{
    private ConsoleReader reader;

    public FlushyStreamHandler(OutputStream out, Formatter formatter, ConsoleReader reader) 
    {
        super(out, formatter);
        this.reader = reader;
        setLevel(Level.INFO);
    }

    @Override
    public synchronized void publish(LogRecord record) 
    {
        record.setMessage(record.getMessage().replace("\n", "\n\r"));
        super.publish(record);
        flush();
    }

    @Override
    public synchronized void flush() 
    {
        try 
        {
            reader.print(ConsoleReader.RESET_LINE + "");
            reader.flush();
            super.flush();
            try 
            {
                reader.drawLine();
            } 
            catch (Throwable ex) 
            {
                reader.getCursorBuffer().clear();
            }

            reader.flush();
            super.flush();
        }
        catch (SshException ex)
        {
            // do nothing
        } 
        catch (IOException ex) 
        {
            Logger.getLogger(FlushyStreamHandler.class.getName()).log(Level.SEVERE, null, ex);
        }
    }
}