bear.session.Question Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of bear Show documentation
The newest version!
/*
 * Copyright (C) 2013 Andrey Chaschev.
 *
 * 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 bear.session;

import chaschev.util.Exceptions;
import com.google.common.collect.Lists;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.List;

/**
 * @author Andrey Chaschev [email protected]
 */
public class Question {
    public String question;
    public List options;
    public DynamicVariable var;

    public boolean freeInput;
    public boolean freeInputOption;

    public Question(String question, List options, DynamicVariable var) {
        this.question = question;
        this.options = options;
        this.var = var;
    }

    public void ask() {
        try {

            final BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

            final String defaultValue = var.getDefaultValue();
            final String defValInput = defaultValue == null ? "" : " [ENTER=" + defaultValue + "]";

            final String newValue;

            if (freeInput) {
                System.out.println(System.out.printf("%s", question));
                newValue = br.readLine();
            } else {

                if (options.size() == 1) {
                    if (!freeInputOption) {
                        System.out.printf("%s%n", question);

                        newValue = options.get(0);
                        System.out.printf("there is only one option, setting to '%s'%n", newValue);
                    } else {
                        System.out.printf("%s [ENTER=%s]: ", question, options.get(0));

                        String temp = br.readLine().trim();

                        newValue = temp.isEmpty() ? options.get(0) : temp;
                    }
                } else if (!options.isEmpty()) {
                    for (int i = 0; i < options.size(); i++) {
                        String s = options.get(i);
                        System.out.printf("%d) %s%n", i + 1, s);
                    }

                    if (freeInputOption) {
                        System.out.printf("%d) %s%n", options.size() + 1, "Other (enter manually)");
                    }

                    System.out.printf("Your choice%s: ", defValInput);

                    final String line = br.readLine();

                    if (line.isEmpty()) {
                        newValue = defaultValue;
                    } else {
                        final int index = Integer.parseInt(line) - 1;
                        if (index == options.size()) {
                            System.out.printf("Enter your value: ");

                            newValue = br.readLine();
                        } else {
                            newValue = options.get(index);
                        }
                    }
                } else {
                    System.out.printf("Enter value%s: ", defValInput);

                    newValue = br.readLine();
                }
            }

            var.defaultTo(newValue, true);
        } catch (IOException e) {
            throw Exceptions.runtime(e);
        }
    }

    public Question setFreeInput(boolean freeInput) {
        this.freeInput = freeInput;
        return this;
    }

    public static String freeQuestion(String question) {
        return freeQuestion(question, Variables.strVar());
    }

    public static String freeQuestion(String question, DynamicVariable strVar) {
        new Question(question, null, strVar).ask();

        return strVar.defaultValue;

    }

    public static String freeQuestionWithOption(String question, String _default) {
        return freeQuestionWithOption(question, _default, Variables.strVar());
    }

    public static String freeQuestionWithOption(String question, String _default, DynamicVariable strVar) {
        new Question(question, Lists.newArrayList(_default), strVar)
            .freeInputOption(true)
            .ask();

        return strVar.defaultValue;
    }

    private Question freeInputOption(boolean b) {
        freeInputOption = b;
        return this;
    }
}