org.python.util.JLineConsole Maven / Gradle / Ivy

package org.python.util;

import java.io.File;
import java.io.IOException;
import jline.ConsoleReader;
import jline.Terminal;
import org.python.core.Py;
import org.python.core.PyObject;

/**
 * This class uses JLine to provide
 * readline like functionality to its console without requiring native readline
 * support.
 */
public class JLineConsole extends InteractiveConsole {

    public JLineConsole() {
        this(null);
    }

    public JLineConsole(PyObject locals) {
        this(locals, CONSOLE_FILENAME);
        try {
            File historyFile = new File(System.getProperty("user.home"),
                                        ".jline-jython.history");
            reader.getHistory().setHistoryFile(historyFile);
        } catch(IOException e) {
            // oh well, no history from file 
        }
    }

    public JLineConsole(PyObject locals, String filename) {
        super(locals, filename, true);
        Terminal.setupTerminal();
        try {
            reader = new ConsoleReader();
        } catch(IOException e) {
            throw new RuntimeException(e);
        }
    }

    public String raw_input(PyObject prompt) {
        String line = null;
        try {
            line = reader.readLine(prompt.toString());
        } catch(IOException io) {
            throw Py.IOError(io);
        }
        if(line == null) {
            throw Py.EOFError("Ctrl-D exit");
        }
        return line.endsWith("\n") ? line.substring(0, line.length() - 1) : line;
    }

    protected ConsoleReader reader;
}