// Path: sshd-shell-spring-boot-starter/src/main/java/sshd/shell/springboot/autoconfiguration/CommandExecutableDetails.java
// @lombok.extern.slf4j.Slf4j
// public class CommandExecutableDetails {
// 
//     private final Set<String> roles;
//     private final String command;
//     @lombok.Getter
//     private final String description;
//     @lombok.Getter
//     private final CommandExecutor commandExecutor;
// 
//     CommandExecutableDetails(SshdShellCommand command, Set<String> roles, CommandExecutor commandExecutor) {
//         this.roles = Collections.unmodifiableSet(roles);
//         this.command = command.value();
//         this.description = command.description();
//         this.commandExecutor = commandExecutor;
//     }
// 
//     public boolean matchesRole(Collection<String> userRoles) {
//         if (roles.contains("*") || userRoles.contains("*")) {
//             return true;
//         }
//         return CollectionUtils.containsAny(roles, userRoles);
//     }
// 
//     public String executeWithArg(String arg) throws InterruptedException, ShellException {
//         return commandExecutor.get(arg);
//     }
// 
//     @Override
//     public String toString() {
//         return new StringBuilder("Command: ").append(command).append(", description: ").append(description)
//                 .append(", roles: ").append(roles).toString();
//     }
// }
// 
// Path: sshd-shell-spring-boot-starter/src/main/java/sshd/shell/springboot/autoconfiguration/Constants.java
// public enum Constants {
// 
//     ;
//     public static final String USER = "__user";
//     public static final String HELP = "help";
//     public static final String USER_ROLES = "__userRoles";
//     public static final String EXECUTE = "__execute";
//     public static final String SHELL_BANNER = "__shellBanner";
// }
// 
// Path: sshd-shell-spring-boot-starter/src/main/java/sshd/shell/springboot/autoconfiguration/SshdShellProperties.java
// @ConfigurationProperties(prefix = "sshd")
// @lombok.Data
// public class SshdShellProperties {
// 
//     private final FileTransfer filetransfer = new FileTransfer();
//     private final Filesystem filesystem = new Filesystem();
//     private final Shell shell = new Shell();
// 
//     @lombok.Data
//     public static class FileTransfer {
// 
//         private boolean enabled = false;
//     }
// 
//     @lombok.Data
//     public static class Filesystem {
// 
//         private final Base base = new Base();
// 
//         @lombok.Data
//         public static class Base {
// 
//             private String dir = System.getProperty("user.home");
//         }
//     }
// 
//     @lombok.Data
//     public static class Shell {
// 
//         private int port = 8022;
//         private boolean enabled = false;
//         private String username = "admin";
//         private String password;
//         private String publicKeyFile;
//         private String host = "127.0.0.1";
//         private String hostKeyFile = "hostKey.ser";
//         private final Prompt prompt = new Prompt();
//         private final Text text = new Text();
//         private final Auth auth = new Auth();
// 
//         @lombok.Data
//         public static class Prompt {
// 
//             private ColorType color = ColorType.BLACK;
//             private String title = "app";
//         }
// 
//         @lombok.Data
//         public static class Text {
// 
//             private ColorType color = ColorType.BLACK;
//             private ColorType highlightColor = ColorType.YELLOW;
//             private String usageInfoFormat = "%n%-35s%s";
//         }
// 
//         @lombok.Data
//         public static class Auth {
// 
//             public enum AuthType {
//                 SIMPLE,
//                 AUTH_PROVIDER
//             }
// 
//             private AuthType authType = AuthType.SIMPLE;
//             private String authProviderBeanName;
//         }
//     }
// }


import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;
import org.jline.builtins.Completers;
import org.jline.builtins.Completers.TreeCompleter.Node;
import static org.jline.builtins.Completers.TreeCompleter.node;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import sshd.shell.springboot.autoconfiguration.CommandExecutableDetails;
import sshd.shell.springboot.autoconfiguration.Constants;
import sshd.shell.springboot.autoconfiguration.SshdShellProperties;
/*
 * Copyright 2017 anand.
 *
 * 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.
 */
package sshd.shell.springboot.console;


/**
 *
 * @author anand
 */
@Configuration
@ConditionalOnProperty(name = "sshd.shell.enabled", havingValue = "true")
class ConsoleConfiguration {

    @Autowired
    private SshdShellProperties properties;
    @Autowired
    private Map<String, Map<String, CommandExecutableDetails>> sshdShellCommands;
    @Autowired
    private List<BaseUserInputProcessor> userInputProcessors;

    @Bean
    TerminalProcessor terminalProcessor() {
        return new TerminalProcessor(properties.getShell(), new Completers.TreeCompleter(buildTextCompleters()),
                userInputProcessors);
    }

    private List<Node> buildTextCompleters() {
        return sshdShellCommands.entrySet().stream().map(entry -> buildTextCompleterNode(entry))
                .collect(Collectors.toList());
    }

    private Node buildTextCompleterNode(Map.Entry<String, Map<String, CommandExecutableDetails>> entry) {
        Object[] subCommands = entry.getValue().keySet().stream()
// Path: sshd-shell-spring-boot-starter/src/main/java/sshd/shell/springboot/autoconfiguration/CommandExecutableDetails.java
// @lombok.extern.slf4j.Slf4j
// public class CommandExecutableDetails {
// 
//     private final Set<String> roles;
//     private final String command;
//     @lombok.Getter
//     private final String description;
//     @lombok.Getter
//     private final CommandExecutor commandExecutor;
// 
//     CommandExecutableDetails(SshdShellCommand command, Set<String> roles, CommandExecutor commandExecutor) {
//         this.roles = Collections.unmodifiableSet(roles);
//         this.command = command.value();
//         this.description = command.description();
//         this.commandExecutor = commandExecutor;
//     }
// 
//     public boolean matchesRole(Collection<String> userRoles) {
//         if (roles.contains("*") || userRoles.contains("*")) {
//             return true;
//         }
//         return CollectionUtils.containsAny(roles, userRoles);
//     }
// 
//     public String executeWithArg(String arg) throws InterruptedException, ShellException {
//         return commandExecutor.get(arg);
//     }
// 
//     @Override
//     public String toString() {
//         return new StringBuilder("Command: ").append(command).append(", description: ").append(description)
//                 .append(", roles: ").append(roles).toString();
//     }
// }
// 
// Path: sshd-shell-spring-boot-starter/src/main/java/sshd/shell/springboot/autoconfiguration/Constants.java
// public enum Constants {
// 
//     ;
//     public static final String USER = "__user";
//     public static final String HELP = "help";
//     public static final String USER_ROLES = "__userRoles";
//     public static final String EXECUTE = "__execute";
//     public static final String SHELL_BANNER = "__shellBanner";
// }
// 
// Path: sshd-shell-spring-boot-starter/src/main/java/sshd/shell/springboot/autoconfiguration/SshdShellProperties.java
// @ConfigurationProperties(prefix = "sshd")
// @lombok.Data
// public class SshdShellProperties {
// 
//     private final FileTransfer filetransfer = new FileTransfer();
//     private final Filesystem filesystem = new Filesystem();
//     private final Shell shell = new Shell();
// 
//     @lombok.Data
//     public static class FileTransfer {
// 
//         private boolean enabled = false;
//     }
// 
//     @lombok.Data
//     public static class Filesystem {
// 
//         private final Base base = new Base();
// 
//         @lombok.Data
//         public static class Base {
// 
//             private String dir = System.getProperty("user.home");
//         }
//     }
// 
//     @lombok.Data
//     public static class Shell {
// 
//         private int port = 8022;
//         private boolean enabled = false;
//         private String username = "admin";
//         private String password;
//         private String publicKeyFile;
//         private String host = "127.0.0.1";
//         private String hostKeyFile = "hostKey.ser";
//         private final Prompt prompt = new Prompt();
//         private final Text text = new Text();
//         private final Auth auth = new Auth();
// 
//         @lombok.Data
//         public static class Prompt {
// 
//             private ColorType color = ColorType.BLACK;
//             private String title = "app";
//         }
// 
//         @lombok.Data
//         public static class Text {
// 
//             private ColorType color = ColorType.BLACK;
//             private ColorType highlightColor = ColorType.YELLOW;
//             private String usageInfoFormat = "%n%-35s%s";
//         }
// 
//         @lombok.Data
//         public static class Auth {
// 
//             public enum AuthType {
//                 SIMPLE,
//                 AUTH_PROVIDER
//             }
// 
//             private AuthType authType = AuthType.SIMPLE;
//             private String authProviderBeanName;
//         }
//     }
// }


// Path: sshd-shell-spring-boot-starter/src/main/java/sshd/shell/springboot/console/ConsoleConfiguration.java
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;
import org.jline.builtins.Completers;
import org.jline.builtins.Completers.TreeCompleter.Node;
import static org.jline.builtins.Completers.TreeCompleter.node;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import sshd.shell.springboot.autoconfiguration.CommandExecutableDetails;
import sshd.shell.springboot.autoconfiguration.Constants;
import sshd.shell.springboot.autoconfiguration.SshdShellProperties;

/*
 * Copyright 2017 anand.
 *
 * 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.
 */
package sshd.shell.springboot.console;


/**
 *
 * @author anand
 */
@Configuration
@ConditionalOnProperty(name = "sshd.shell.enabled", havingValue = "true")
class ConsoleConfiguration {

    @Autowired
    private SshdShellProperties properties;
    @Autowired
    private Map<String, Map<String, CommandExecutableDetails>> sshdShellCommands;
    @Autowired
    private List<BaseUserInputProcessor> userInputProcessors;

    @Bean
    TerminalProcessor terminalProcessor() {
        return new TerminalProcessor(properties.getShell(), new Completers.TreeCompleter(buildTextCompleters()),
                userInputProcessors);
    }

    private List<Node> buildTextCompleters() {
        return sshdShellCommands.entrySet().stream().map(entry -> buildTextCompleterNode(entry))
                .collect(Collectors.toList());
    }

    private Node buildTextCompleterNode(Map.Entry<String, Map<String, CommandExecutableDetails>> entry) {
        Object[] subCommands = entry.getValue().keySet().stream()
                .filter(s -> !s.equals(Constants.EXECUTE))