com.aspectran.shell.console.AbstractShellConsole Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of aspectran-all Show documentation
Aspectran (All-in-One)
There is a newer version: 8.1.5
/*
 * Copyright (c) 2008-2023 The Aspectran Project
 *
 * 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 com.aspectran.shell.console;

import com.aspectran.core.context.config.ShellStyleConfig;
import com.aspectran.shell.command.ConsoleCommander;

import java.io.File;
import java.nio.charset.Charset;

/**
 * The Abstract Class for Console I/O.
 *
 * Created: 2017. 3. 4.
 */
public abstract class AbstractShellConsole implements ShellConsole {

    private final String encoding;

    private String commandPrompt = DEFAULT_PROMPT;

    private File workingDir;

    private ConsoleCommander consoleCommander;

    private String[] primaryStyle;

    private String[] secondaryStyle;

    private String[] successStyle;

    private String[] dangerStyle;

    private String[] warningStyle;

    private String[] infoStyle;

    public AbstractShellConsole(String encoding) {
        if (encoding != null) {
            this.encoding = encoding;
        } else {
            this.encoding = Charset.defaultCharset().name();
        }
    }

    @Override
    public String getEncoding() {
        return encoding;
    }

    @Override
    public File getWorkingDir() {
        return workingDir;
    }

    @Override
    public void setWorkingDir(File workingDir) {
        this.workingDir = workingDir;
    }

    @Override
    public ConsoleCommander getConsoleCommander() {
        return consoleCommander;
    }

    @Override
    public void setConsoleCommander(ConsoleCommander consoleCommander) {
        this.consoleCommander = consoleCommander;
    }

    @Override
    public String getCommandPrompt() {
        return commandPrompt;
    }

    @Override
    public void setCommandPrompt(String commandPrompt) {
        this.commandPrompt = commandPrompt;
    }

    @Override
    public String readLine() {
        return readLine(null);
    }

    @Override
    public String readPassword() {
        return readPassword(null);
    }

    protected String readMultiCommandLine(String line) {
        if (line == null) {
            return null;
        }
        if (COMMENT_DELIMITER.equals(line)) {
            String next = readRawCommandLine(COMMENT_PROMPT);
            if (next.isEmpty()) {
                return next;
            }
            readMultiCommandLine(COMMENT_DELIMITER);
        }
        String quote = searchQuote(line);
        if (quote != null || line.endsWith(MULTILINE_DELIMITER)) {
            String next = readRawCommandLine(MULTILINE_PROMPT).trim();
            if (next.startsWith(COMMENT_DELIMITER)) {
                line = readMultiCommandLine(line);
            } else if (quote != null) {
                line += System.lineSeparator() + next;
            } else {
                line = line.substring(0, line.length() - MULTILINE_DELIMITER.length()).trim() + " " + next;
            }
        }
        quote = searchQuote(line);
        if (quote != null) {
            return readMultiCommandLine(line);
        }
        if (line.endsWith(MULTILINE_DELIMITER)) {
            line = readMultiCommandLine(line);
        }
        return line;
    }

    protected String readMultiLine(String line) {
        if (line == null) {
            line = readRawLine(MULTILINE_PROMPT);
        }
        if (line.endsWith(MULTILINE_DELIMITER)) {
            line = line.substring(0, line.length() - MULTILINE_DELIMITER.length()) +
                    System.lineSeparator() + readMultiLine(null);
        }
        return line;
    }

    abstract protected String readRawCommandLine(String prompt);

    abstract protected String readRawLine(String prompt);

    private String searchQuote(String line) {
        boolean doubleQuote = false;
        boolean singleQuote = false;
        for (int i = 0; i < line.length(); i++) {
            char c = line.charAt(i);
            if (c == '"' && !singleQuote) {
                doubleQuote = !doubleQuote;
            } else if (c == '\'' && !doubleQuote) {
                singleQuote = !singleQuote;
            }
        }
        if (doubleQuote) {
            return "\"";
        } else if (singleQuote) {
            return "'";
        } else {
            return null;
        }
    }

    @Override
    public void setShellStyleConfig(ShellStyleConfig shellStyleConfig) {
        if (shellStyleConfig == null) {
            throw new IllegalArgumentException("shellStyleConfig must not be null");
        }
        primaryStyle = shellStyleConfig.getPrimaryStyle();
        secondaryStyle = shellStyleConfig.getSecondaryStyle();
        successStyle = shellStyleConfig.getSuccessStyle();
        dangerStyle = shellStyleConfig.getDangerStyle();
        warningStyle = shellStyleConfig.getWarningStyle();
        infoStyle = shellStyleConfig.getInfoStyle();
        resetStyle();
    }

    @Override
    public String[] getPrimaryStyle() {
        return primaryStyle;
    }

    @Override
    public String[] getSecondaryStyle() {
        return secondaryStyle;
    }

    @Override
    public String[] getSuccessStyle() {
        return successStyle;
    }

    @Override
    public String[] getDangerStyle() {
        return dangerStyle;
    }

    @Override
    public String[] getWarningStyle() {
        return warningStyle;
    }

    @Override
    public String[] getInfoStyle() {
        return infoStyle;
    }

    @Override
    public boolean confirmRestart() {
        if (checkReadingState()) {
            return false;
        }
        PromptStringBuilder psb = newPromptStringBuilder()
                .warningStyle()
                .append("Would you like to restart this shell [Y/n]? ");
        String yn = readLine(psb);
        return (yn == null || yn.isEmpty() || yn.equalsIgnoreCase("Y"));
    }

    @Override
    public boolean confirmQuit() {
        if (checkReadingState()) {
            return false;
        }
        PromptStringBuilder psb = newPromptStringBuilder()
                .warningStyle()
                .append("Are you sure you want to quit [Y/n]? ");
        String yn = readLine(psb);
        return (yn == null || yn.isEmpty() || yn.equalsIgnoreCase("Y"));
    }

    private boolean checkReadingState() {
        if (isReading()) {
            setStyle(getDangerStyle());
            writeAbove("Illegal State");
            resetStyle();
            return true;
        } else {
            return false;
        }
    }

}