/**
*
* 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.karaf.shell.console.help;
import java.io.ByteArrayOutputStream;
import java.io.PrintStream;
import java.lang.reflect.Field;
import java.lang.reflect.Method;
import java.util.Map;
import java.util.Set;
import java.util.SortedMap;
import java.util.TreeMap;
import jline.Terminal;
import org.apache.felix.gogo.commands.Action;
import org.apache.felix.gogo.commands.Command;
import org.apache.felix.gogo.commands.basic.AbstractCommand;
import org.apache.felix.gogo.commands.basic.DefaultActionPreparator;
import org.apache.felix.service.command.CommandSession;
import org.apache.felix.service.command.Function;
import org.apache.karaf.shell.console.HelpProvider;
import org.apache.karaf.shell.console.NameScoping;
import org.fusesource.jansi.Ansi;
import org.osgi.framework.BundleContext;
import org.osgi.framework.ServiceReference;
public class CommandListHelpProvider implements HelpProvider {
public static final String COMMANDS = ".commands";
public String getHelp(CommandSession session, String path) {
if (path.indexOf('|') > 0) {
if (path.startsWith("command-list|")) {
path = path.substring("command-list|".length());
} else {
return null;
}
}
SortedMap<String, String> commands = getCommandDescriptions(session, path);
ByteArrayOutputStream baos = new ByteArrayOutputStream();
printMethodList(session, new PrintStream(baos), commands);
return baos.toString();
}
private SortedMap<String, String> getCommandDescriptions(CommandSession session, String command) {
Set<String> names = (Set<String>) session.get(COMMANDS);
SortedMap<String,String> commands = new TreeMap<String,String>();
for (String name : names) {
if (command != null && !name.startsWith(command)) {
continue;
}
String description = null;
Function function = (Function) session.get(name);
function = unProxy(function);
if (function instanceof AbstractCommand) {
try {
Method mth = AbstractCommand.class.getDeclaredMethod("createNewAction");
mth.setAccessible(true);
Action action = (Action) mth.invoke(function);
Class<? extends Action> clazz = action.getClass();
Command ann = clazz.getAnnotation(Command.class);
description = ann.description();
} catch (Throwable e) {
}
if (name.startsWith("*:")) {
name = name.substring(2);
}
commands.put(name, description);
}
}
return commands;
}
protected void printMethodList(CommandSession session, PrintStream out, SortedMap<String, String> commands) {
Terminal term = (Terminal) session.get(".jline.terminal");
out.println(Ansi.ansi().a(Ansi.Attribute.INTENSITY_BOLD).a("COMMANDS").a(Ansi.Attribute.RESET));
int max = 0;
for (Map.Entry<String,String> entry : commands.entrySet()) {
String key = NameScoping.getCommandNameWithoutGlobalPrefix(session, entry.getKey());
max = Math.max(max,key.length());
}
int margin = 4;
String prefix1 = " ";
if (term != null && term.getWidth() - max - prefix1.length() - margin > 50) {
String prefix2 = prefix1;
for (int i = 0; i < max + margin; i++) {
prefix2 += " ";
}
for (Map.Entry<String,String> entry : commands.entrySet()) {
out.print(prefix1);
String key = NameScoping.getCommandNameWithoutGlobalPrefix(session, entry.getKey());
out.print(Ansi.ansi().a(Ansi.Attribute.INTENSITY_BOLD).a(key).a(Ansi.Attribute.RESET));
for (int i = 0; i < max - key.length() + margin; i++) {
out.print(' ');
}
if (entry.getValue() != null) {
DefaultActionPreparator.printFormatted(prefix2, entry.getValue(), term.getWidth(), out, false);
}
}
} else {
String prefix2 = prefix1 + prefix1;
for (Map.Entry<String,String> entry : commands.entrySet()) {
out.print(prefix1);
String key = NameScoping.getCommandNameWithoutGlobalPrefix(session, entry.getKey());
out.println(Ansi.ansi().a(Ansi.Attribute.INTENSITY_BOLD).a(key).a(Ansi.Attribute.RESET));
if (entry.getValue() != null) {
DefaultActionPreparator.printFormatted(prefix2, entry.getValue(),
term != null ? term.getWidth() : 80, out);
}
}
}
out.println();
}
protected Function unProxy(Function function) {
try {
if (function.getClass().getName().contains("CommandProxy")) {
Field contextField = function.getClass().getDeclaredField("context");
Field referenceField = function.getClass().getDeclaredField("reference");
contextField.setAccessible(true);
referenceField.setAccessible(true);
BundleContext context = (BundleContext) contextField.get(function);
ServiceReference reference = (ServiceReference) referenceField.get(function);
Object target = context.getService(reference);
try {
if (target instanceof Function) {
function = (Function) target;
}
} finally {
context.ungetService(reference);
}
}
} catch (Throwable t) {
}
return function;
}
}