wtf.metio.yosql.codegen.blocks.DefaultJavadoc Maven / Gradle / Ivy

Go to download
/*
 * This file is part of yosql. It is subject to the license terms in the LICENSE file found in the top-level
 * directory of this distribution and at http://creativecommons.org/publicdomain/zero/1.0/. No part of yosql,
 * including this file, may be copied, modified, propagated, or distributed except according to the terms contained
 * in the LICENSE file.
 */

package wtf.metio.yosql.codegen.blocks;

import ch.qos.cal10n.IMessageConveyor;
import com.squareup.javapoet.CodeBlock;
import de.xn__ho_hia.javapoet.TypeGuesser;
import wtf.metio.yosql.codegen.api.Javadoc;
import wtf.metio.yosql.internals.jdk.Strings;
import wtf.metio.yosql.models.immutables.FilesConfiguration;
import wtf.metio.yosql.models.immutables.SqlConfiguration;
import wtf.metio.yosql.models.immutables.SqlStatement;
import wtf.metio.yosql.models.sql.ResultRowConverter;

import java.util.List;
import java.util.Objects;

import static java.util.function.Predicate.not;

public final class DefaultJavadoc implements Javadoc {

    private final FilesConfiguration files;
    private final IMessageConveyor messages;

    public DefaultJavadoc(final FilesConfiguration files, final IMessageConveyor messages) {
        this.files = files;
        this.messages = messages;
    }

    @Override
    public CodeBlock repositoryJavadoc(final List statements) {
        final var builder = CodeBlock.builder()
                .add(messages.getMessage(Javadocs.USED_FILES))
                .add(messages.getMessage(Javadocs.LIST_START));
        final var input = files.inputBaseDirectory();
        statements.stream()
                .map(SqlStatement::getSourcePath)
                .distinct()
                .map(input::relativize)
                .forEach(path -> builder.add(messages.getMessage(Javadocs.LIST_ITEM), path));
        builder.add(messages.getMessage(Javadocs.LIST_END));
        return builder.build();
    }

    @Override
    public CodeBlock fieldJavaDoc(final SqlStatement statement) {
        final var input = files.inputBaseDirectory();
        final var builder = CodeBlock.builder()
                .add(messages.getMessage(Javadocs.USED_FILE))
                .add(messages.getMessage(Javadocs.LIST_START))
                .add(messages.getMessage(Javadocs.LIST_ITEM), input.relativize(statement.getSourcePath()));
        builder.add(messages.getMessage(Javadocs.LIST_END));
        return builder.build();
    }

    @Override
    public CodeBlock methodJavadoc(final List statements, final String configuration) {
        final var builder = CodeBlock.builder();
        statements.stream()
                .map(SqlStatement::getConfiguration)
                .map(SqlConfiguration::description)
                .filter(not(Strings::isBlank))
                .forEach(description -> builder.add(messages.getMessage(Javadocs.DESCRIPTION), description));
        if (statements.size() > 1) {
            builder.add(messages.getMessage(Javadocs.EXECUTED_STATEMENTS));
        } else {
            builder.add(messages.getMessage(Javadocs.EXECUTED_STATEMENT));
        }
        for (final var statement : statements) {
            if (Strings.isBlank(statement.getConfiguration().vendor())) {
                if (statements.size() > 1) {
                    builder.add(messages.getMessage(Javadocs.FALLBACK));
                }
            } else {
                builder.add(messages.getMessage(Javadocs.VENDOR), statement.getConfiguration().vendor());
            }
            builder.add(messages.getMessage(Javadocs.STATEMENT), statement.getRawStatement());
        }
        builder.add(messages.getMessage(Javadocs.USED_FILES_METHOD))
                .add(messages.getMessage(Javadocs.LIST_START));
        final var input = files.inputBaseDirectory();
        statements.stream()
                .map(SqlStatement::getSourcePath)
                .distinct()
                .map(input::relativize)
                .forEach(path -> builder.add(messages.getMessage(Javadocs.LIST_ITEM), path));
        builder.add(messages.getMessage(Javadocs.LIST_END));
        builder.add(messages.getMessage(Javadocs.DISABLE_WITH), configuration);
        statements.stream()
                .map(SqlStatement::getConfiguration)
                .flatMap(config -> config.resultRowConverter().stream())
                .filter(Objects::nonNull)
                .map(ResultRowConverter::resultType)
                .filter(Objects::nonNull)
                .filter(type -> !type.startsWith("java"))
                .map(type -> type.substring(0, type.contains("<") ? type.indexOf("<") : type.length()))
                .distinct()
                .map(TypeGuesser::guessTypeName)
                .filter(type -> !type.isPrimitive())
                .filter(type -> !type.isBoxedPrimitive())
                .forEach(type -> builder.add(messages.getMessage(Javadocs.SEE), type));
        return builder.build();
    }

}