openide/source/com/intellij/execution/impl/ConsoleState.java

package com.intellij.execution.impl;

import com.intellij.execution.process.ProcessAdapter;
import com.intellij.execution.process.ProcessEvent;
import com.intellij.execution.process.ProcessHandler;
import com.intellij.execution.ui.ConsoleViewContentType;
import com.intellij.openapi.util.Key;
import com.intellij.openapi.application.ApplicationManager;

import java.io.IOException;
import java.io.OutputStream;
import java.io.OutputStreamWriter;
import java.io.Writer;

public abstract class ConsoleState {
  public static final ConsoleState NOT_STARTED = new ConsoleState(){
    public ConsoleState attachTo(final ConsoleViewImpl console, final ProcessHandler processHandler) {
      return new RunningState(console, processHandler);
    }
  };

  public ConsoleState dispose() {
    return NOT_STARTED;
  }

  public boolean isFinished() {
    return false;
  }

  public boolean isRunning() {
    return false;
  }

  public void sendUserInput(final String input) throws IOException {}

  public abstract ConsoleState attachTo(ConsoleViewImpl console, ProcessHandler processHandler);

  private static class RunningState extends ConsoleState {
    private final ConsoleViewImpl myConsole;
    private final ProcessAdapter myProcessListener = new ProcessAdapter() {
      public void onTextAvailable(final ProcessEvent event, final Key outputType) {
        ApplicationManager.getApplication().invokeLater(new Runnable() {
          public void run() {
            myConsole.print(event.getText(), ConsoleViewContentType.getConsoleViewType(outputType));
          }
        });
      }
    };
    private final ProcessHandler myProcessHandler;
    private final Writer myUserInputWriter;

    public RunningState(final ConsoleViewImpl console, final ProcessHandler processHandler) {
      myConsole = console;
      myProcessHandler = processHandler;
      processHandler.addProcessListener(myProcessListener);
      final OutputStream processInput = myProcessHandler.getProcessInput();
      myUserInputWriter = processInput != null ? new OutputStreamWriter(processInput) : null;
    }

    public ConsoleState dispose() {
      if (myProcessHandler != null) {
        myProcessHandler.removeProcessListener(myProcessListener);
      }
      return NOT_STARTED;
    }

    public boolean isFinished() {
      return myProcessHandler == null || myProcessHandler.isProcessTerminated();
    }

    public boolean isRunning() {
      return myProcessHandler != null && !myProcessHandler.isProcessTerminated();
    }

    public void sendUserInput(final String input) throws IOException {
      if (myUserInputWriter == null)
        throw new IOException("No process input");
      myUserInputWriter.write(input);
      myUserInputWriter.flush();
    }

    public ConsoleState attachTo(final ConsoleViewImpl console, final ProcessHandler processHandler) {
      return dispose().attachTo(console, processHandler);
    }
  }
}