package act.cli.builtin;

/*-

* #%L

* ACT Framework

* %%

* Copyright (C) 2014 - 2017 ActFramework

* %%

* Licensed 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.

* #L%

*/

import act.cli.CliCmdInfo;

import act.cli.CliContext;

import act.cli.CliDispatcher;

import act.cli.util.CommandLineParser;

import act.cli.view.CliView;

import act.handler.CliHandler;

import act.handler.CliHandlerBase;

import act.util.PropertySpec;

import org.fusesource.jansi.Ansi;

import org.osgl.util.C;

import org.osgl.util.S;

import java.util.ArrayList;

import java.util.Collection;

import java.util.List;

import java.util.SortedSet;

import static org.osgl.$.T2;

public class Help extends CliHandlerBase {

public static final Help INSTANCE = new Help();

private static int maxWidth = 0;

private PropertySpec.MetaInfo metaInfo;

private Help() {

metaInfo = new PropertySpec.MetaInfo();

metaInfo.onCli("name,shortcut,help");

}

@Override

public void handle(CliContext context) {

List<String> args = context.arguments();

String command;

if (args.size() > 0) {

command = args.get(0);

if (showHelp(command, context)) {

return;

}

}

CommandLineParser parser = context.commandLine();

boolean sys = parser.getBoolean("-s", "--system");

boolean all = parser.getBoolean("-a", "--all");

boolean tableList = parser.getBoolean("-t", "--table");

String q = parser.getString("-q", "--filter");

boolean app = true;

if (all) {

sys = true;

} else if (sys) {

app = false;

}

CliDispatcher dispatcher = context.app().cliDispatcher();

SortedSet<CliCmdInfo> sysCommands = dispatcher.commandInfoList(true, false);

SortedSet<CliCmdInfo> appCommands = dispatcher.commandInfoList(false, true);

int maxLen = -1;

if (!tableList) {

if (sys && !app) {

maxLen = calMaxCmdLen(sysCommands);

} else if (app && !sys) {

maxLen = calMaxCmdLen(appCommands);

} else {

maxLen = calMaxCmdLen(C.list(sysCommands).append(appCommands));

}

}

String fmt = maxLen < 0 ? null : "%-" + (maxLen + 4) + "s - %s";

if (sys) {

list("@|bold System commands|@", sysCommands, context, q, fmt);

}

if (app) {

if (sys) {

context.println("");

}

list("@|bold Application commands|@", appCommands, context, q, fmt);

}

}

private void list(String label, Collection<CliCmdInfo> commands, CliContext context, String q, String fmt) {

List<String> lines = new ArrayList<>();

lines.add(label.toUpperCase());

q = S.string(q).trim();

boolean hasFilter = S.notEmpty(q);

List<CliCmdInfo> toBeDisplayed = new ArrayList<>();

for (CliCmdInfo info : commands) {

String cmd = info.name;

if (hasFilter && !(cmd.toLowerCase().contains(q) || cmd.matches(q))) {

continue;

}

if (null == fmt) {

toBeDisplayed.add(info);

} else {

lines.add(S.fmt(fmt, info.nameAndShortcut(), info.help));

}

}

context.println(Ansi.ansi().render(S.join("\n", lines)).toString());

if (null == fmt) {

// display table list

context.println(CliView.TABLE.render(toBeDisplayed, metaInfo, context));

}

}

private int calMaxCmdLen(Collection<CliCmdInfo> commands) {

int max = 0;

for (CliCmdInfo info : commands) {

max = Math.max(max, info.nameAndShortcutLen());

}

return max;

}

public boolean showHelp(String command, CliContext context) {

CliDispatcher dispatcher = context.app().cliDispatcher();

CliHandler handler = dispatcher.handler(command);

if (null == handler) {

// context.println("Unrecongized command: %s", command);

return false;

}

List<String> lines = new ArrayList<>();

List<String> names = dispatcher.names(handler);

T2<String, String> commandLine = handler.commandLine();

lines.add("@|bold Usage|@: " + names.get(0));

lines.add(commandLine._2);

String summary = handler.summary();

if (S.notBlank(summary)) {

lines.add("");

lines.add(summary);

}

List<T2<String, String>> options = handler.options();

if (!options.isEmpty()) {

lines.add("");

lines.add("@|bold Options|@:");

int maxLen = 0;

for (T2<String, String> t2 : options) {

maxLen = Math.max(maxLen, t2._1.length());

}

String fmt = " %-" + (maxLen + 4) + "s %s";

for (T2<String, String> t2 : options) {

lines.add(S.fmt(fmt, t2._1, t2._2));

}

}

if (names.size() > 1) {

lines.add("");

lines.add("@|bold Aliases|@: " + S.join(", ", C.list(names).tail()));

}

List<String> shortCuts = dispatcher.shortCuts(handler);

if (shortCuts.size() > 0) {

lines.add("");

lines.add("@|bold Shortcuts|@: " + S.join(", ", shortCuts));

}

context.println(Ansi.ansi().render(S.join("\n", lines)).toString());

return true;

}

@Override

public T2<String, String> commandLine() {

return T2("help [options] [command]", "show help information");

}

@Override

public String summary() {

return "display information about a command if COMMAND is specified. " +

"Otherwise the list of help topics is printed. \n Note when " +

"COMMAND is specified, OPTIONS are ignored";

}

@Override

public List<T2<String, String>> options() {

List<T2<String, String>> retList = new ArrayList<>();

retList.add(T2("-s --system", "list system commands"));

retList.add(T2("-a --app", "list application commands"));

retList.add(T2("-t --table", "show command list in table"));

return retList;

}

private Object readResolve() {

return INSTANCE;

}

public static void updateMaxWidth(int width) {

maxWidth = Math.max(maxWidth, width);

}

}