/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*/
package org.apache.felix.karaf.shell.console.jline;
import java.io.InputStream;
import java.io.PrintStream;
import java.io.OutputStream;
import java.lang.reflect.Method;
import java.util.List;
import java.util.concurrent.Callable;
import org.apache.felix.karaf.shell.console.Completer;
import org.apache.felix.karaf.shell.console.completer.AggregateCompleter;
import org.osgi.framework.BundleContext;
import org.osgi.service.command.CommandProcessor;
import org.osgi.service.command.CommandSession;
import org.fusesource.jansi.AnsiConsole;
import jline.Terminal;
public class ConsoleFactory {
private BundleContext bundleContext;
private CommandProcessor commandProcessor;
private List<Completer> completers;
private TerminalFactory terminalFactory;
private Console console;
private boolean start;
public void setBundleContext(BundleContext bundleContext) {
this.bundleContext = bundleContext;
}
public synchronized void registerCommandProcessor(CommandProcessor commandProcessor) throws Exception {
this.commandProcessor = commandProcessor;
start();
}
public synchronized void unregisterCommandProcessor(CommandProcessor commandProcessor) throws Exception {
this.commandProcessor = null;
stop();
}
public void setCompleters(List<Completer> completers) {
this.completers = completers;
}
public void setTerminalFactory(TerminalFactory terminalFactory) {
this.terminalFactory = terminalFactory;
}
public void setStart(boolean start) {
this.start = start;
}
protected void start() throws Exception {
if (start) {
InputStream in = unwrap(System.in);
PrintStream out = unwrap(System.out);
PrintStream err = unwrap(System.err);
Runnable callback = new Runnable() {
public void run() {
try {
bundleContext.getBundle(0).stop();
} catch (Exception e) {
// Ignore
}
}
};
Terminal terminal = terminalFactory.getTerminal();
this.console = new Console(commandProcessor,
in,
wrap(out),
wrap(err),
terminal,
new AggregateCompleter(completers),
callback);
CommandSession session = console.getSession();
session.put("USER", "karaf");
session.put("APPLICATION", System.getProperty("karaf.name", "root"));
new Thread(console, "Karaf Shell Console Thread").start();
}
}
protected void stop() throws Exception {
if (console != null) {
console.close();
}
}
private static PrintStream wrap(PrintStream stream) {
OutputStream o = AnsiConsole.wrapOutputStream(stream);
if (o instanceof PrintStream) {
return ((PrintStream) o);
} else {
return new PrintStream(o);
}
}
private static <T> T unwrap(T stream) {
try {
Method mth = stream.getClass().getMethod("getRoot");
return (T) mth.invoke(stream);
} catch (Throwable t) {
return stream;
}
}
}