JLineAgentInputReader.java

/*
 * RHQ Management Platform
 * Copyright (C) 2005-2009 Red Hat, Inc.
 * All rights reserved.
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation version 2 of the License.
 *
 * This program 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 for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
 */
package org.rhq.enterprise.agent;

import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.Map;

import jline.ConsoleReader;
import jline.SimpleCompletor;

import org.rhq.enterprise.agent.i18n.AgentI18NResourceKeys;
import org.rhq.enterprise.agent.promptcmd.AgentPromptCommand;

public class JLineAgentInputReader implements AgentInputReader {

    private final ConsoleReader jline;
    private final AgentMain agent;
    private final boolean consoleInput;

    public JLineAgentInputReader(AgentMain agent) throws IOException {
        this.jline = new ConsoleReader();
        this.agent = agent;
        this.addCompletor();
        this.consoleInput = true;
    }

    public JLineAgentInputReader(AgentMain agent, FileInputStream fis) throws IOException {
        this.jline = new ConsoleReader(fis, agent.getOut());
        this.agent = agent;
        this.addCompletor();
        this.consoleInput = false;
    }

    public void addCompletor() {
        Map<String, Class<? extends AgentPromptCommand>> cmds = agent.getPromptCommands();
        if (cmds != null) {
            String[] cmdArray = cmds.keySet().toArray(new String[0]);
            jline.addCompletor(new SimpleCompletor(cmdArray));
        }
    }

    public boolean isConsole() {
        return this.consoleInput;
    }

    public String readLine() throws IOException {
        return this.jline.readLine();
    }

    public String readLineNoEcho() throws IOException {
        String input = null;

        while (true) {
            // get the answer the first time - the '*' is echoed back to the user
            input = jline.readLine(Character.valueOf('*'));

            if (!isConsole()) {
                break; // not reading input from a console, so we don't ask for confirmation
            }

            // get the answer a second time
            this.agent.getOut().write(this.agent.getI18NMsg().getMsg(AgentI18NResourceKeys.PROMPT_CONFIRM));
            this.agent.getOut().println();
            String confirmation = jline.readLine(Character.valueOf('*'));
            this.agent.getOut().println();

            // make sure the first and second answers match; otherwise, ask again
            if (input.equals(confirmation)) {
                break;
            }

            this.agent.getOut().write(this.agent.getI18NMsg().getMsg(AgentI18NResourceKeys.PROMPT_CONFIRM_FAILED));
            this.agent.getOut().println();
        }

        return input;
    }

    public void close() throws IOException {
        InputStream input = jline.getInput();
        if (input != null) {
            input.close();
        }
    }
}