com.dattack.jtoolbox.io.console.StringConsoleReader Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of jtoolbox-core Show documentation

Utility classes that only depends on JDK

There is a newer version: 0.7

/*
 * Copyright (c) 2017, The Dattack team (http://www.dattack.com)
 *
 * 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.dattack.jtoolbox.io.console;

import com.dattack.jtoolbox.io.UnclosableInputStream;
import org.apache.commons.lang.StringUtils;
import java.io.InputStream;
import java.nio.charset.StandardCharsets;
import java.util.Scanner;
import java.util.regex.Pattern;

/**
 * A String reader from console.
 *
 * @author cvarela
 * @since 0.1
 */
public class StringConsoleReader extends AbstractConsoleReader {

    private transient String prompt;
    private transient String defaultValue;
    private transient String regex;
    private transient AnsiStyle style;

    /* default */ StringConsoleReader(final Console console, final InputStream inputStream) {
        super(console, inputStream);
    }

    @Override
    public String read() {
        while (true) {
            String result = readString();
            if (StringUtils.isBlank(result)) {
                result = defaultValue;
            }

            if (StringUtils.isNotBlank(regex) && !Pattern.compile(regex).matcher(result).matches()) {
                getConsole().error("The input not matches the regular expression: %s", regex);
            } else {
                return result;
            }
        }
    }

    private String readString() {
        try (Scanner scanner = new Scanner(new UnclosableInputStream(getInputStream()), //
                StandardCharsets.UTF_8.name())) {
            print(style, prompt);
            return scanner.nextLine();
        }
    }

    public StringConsoleReader withDefaultValue(final String value) {
        this.defaultValue = value;
        return this;
    }

    public StringConsoleReader withPrompt(final String value) {
        this.prompt = value;
        return this;
    }

    public StringConsoleReader withRegex(final String value) {
        this.regex = value;
        return this;
    }

    public StringConsoleReader withStyle(final AnsiStyle ansiStyle) {
        this.style = ansiStyle;
        return this;
    }
}