org.seedstack.maven.components.inquirer.impl.AbstractInquirer Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of seedstack-maven-plugin
The newest version!
/*
 * Copyright © 2013-2021, The SeedStack authors 
 *
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/.
 */
package org.seedstack.maven.components.inquirer.impl;

import com.fasterxml.jackson.databind.ObjectMapper;
import java.io.IOException;
import java.net.URL;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;
import org.fusesource.jansi.Ansi;
import org.seedstack.maven.components.inquirer.AnswerValidationException;
import org.seedstack.maven.components.inquirer.Condition;
import org.seedstack.maven.components.inquirer.Inquirer;
import org.seedstack.maven.components.inquirer.InquirerException;
import org.seedstack.maven.components.inquirer.Inquiry;
import org.seedstack.maven.components.inquirer.Question;
import org.seedstack.maven.components.inquirer.QuestionGroup;
import org.seedstack.maven.components.prompter.PromptException;
import org.seedstack.maven.components.prompter.Prompter;

public abstract class AbstractInquirer implements Inquirer {
    private ObjectMapper objectMapper = new ObjectMapper();

    @Override
    public Map inquire(URL inquiryURL) throws InquirerException {
        Inquiry inquiry;
        try {
            inquiry = objectMapper.readValue(inquiryURL, Inquiry.class);
        } catch (IOException e) {
            throw new InquirerException("Cannot execute inquiry at " + inquiryURL.toExternalForm(), e);
        }

        Map answers = new HashMap<>();
        for (QuestionGroup questionGroup : inquiry.getQuestionGroups()) {
            System.out.println();
            String groupLabel = questionGroup.getLabel();
            if (groupLabel != null) {
                System.out.println(Ansi.ansi().fg(Ansi.Color.YELLOW).a(groupLabel).reset());
            }
            answers.put(questionGroup.getName(), processQuestionGroup(questionGroup));
        }
        System.out.println();
        return answers;
    }

    private Map processQuestionGroup(QuestionGroup questionGroup) throws InquirerException {
        Map answers = new HashMap<>();
        for (Question question : questionGroup.getQuestions()) {
            boolean shouldAsk = true;
            for (Condition condition : question.getConditions()) {
                if (!isSatisfied(condition, answers)) {
                    shouldAsk = false;
                }
            }
            if (shouldAsk) {
                Object answer = ask(question);
                if (answer != null) {
                    answers.put(question.getName(), answer);
                }
            }
        }
        return answers;
    }

    @Override
    public Object ask(Question question) throws InquirerException {
        do {
            try {
                String message = question.getMessage();
                String defaultValue = question.getDefaultValue();
                Question.Type type = question.getType();
                switch (question.getStyle()) {
                    case CONFIRM:
                        return getPrompter().promptConfirmation(message, defaultValue);
                    case INPUT:
                        return coerce(getPrompter().promptInput(message, defaultValue), type);
                    case CHECKBOX:
                        return coerce(getPrompter().promptCheckbox(message, question.getValues()), type);
                    case LIST:
                        return coerce(getPrompter().promptList(message, question.getValues(), defaultValue), type);
                    case CHOICE:
                        return coerce(getPrompter().promptChoice(message, question.getValues()), type);
                    default:
                        return null;
                }
            } catch (AnswerValidationException e) {
                System.out.println(Ansi.ansi().fgBright(Ansi.Color.RED).a("X " + e.getMessage()).reset().toString());
            } catch (PromptException e) {
                throw new InquirerException("Unable to ask question", e);
            }

        } while (true);
    }

    private boolean isSatisfied(Condition condition, Map answers) {
        String ref = condition.getRef();
        switch (condition.getOp()) {
            case EQ:
                return answers.containsKey(ref) && answers.get(ref).equals(condition.getVal());
            case NEQ:
                return answers.containsKey(ref) && !answers.get(ref).equals(condition.getVal());
            case PRESENT:
                return answers.containsKey(ref);
            case ABSENT:
                return !answers.containsKey(ref);
            default:
                return false;
        }
    }

    private Set