ru.curs.celesta.plugin.maven.AbstractGenCursorsMojo Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of celesta-maven-plugin Show documentation
The newest version!
package ru.curs.celesta.plugin.maven;

import ru.curs.celesta.score.Grain;
import ru.curs.celesta.score.GrainElement;
import ru.curs.celesta.score.GrainPart;
import ru.curs.celesta.score.MaterializedView;
import ru.curs.celesta.score.ParameterizedView;
import ru.curs.celesta.score.ReadOnlyTable;
import ru.curs.celesta.score.Score;
import ru.curs.celesta.score.SequenceElement;
import ru.curs.celesta.score.Table;
import ru.curs.celesta.score.View;
import ru.curs.celesta.score.io.FileResource;
import ru.curs.celesta.score.io.Resource;

import java.io.File;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collection;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.function.Consumer;
import java.util.function.Supplier;

abstract class AbstractGenCursorsMojo extends AbstractCelestaMojo {

    Supplier> getScorePaths;
    String generatedSourcesDirName;
    Consumer addCompileSourceRoot;

    @Override
    public void execute() {
        this.getLog().info("celesta project is " + project);
        getScorePaths.get().forEach(this::processScore);
        addSourceRoot(getSourceRoot());
    }

    private void processScore(ScoreProperties properties) {
        final String scorePath = properties.getPath();
        Score score = initScore(scorePath);
        score.getGrains().values()
                .stream()
                .filter(this::isAllowGrain)
                .forEach(g -> generateCursors(g, scorePath));
    }

    private void generateCursors(Grain g, String scorePath) {

        final boolean isSysSchema = g.getName().equals(g.getScore().getSysSchemaName());

        Map> partsToElements = new HashMap<>();

        List elements = new ArrayList<>();
        elements.addAll(g.getElements(SequenceElement.class).values());
        elements.addAll(g.getElements(Table.class).values());
        elements.addAll(g.getElements(ReadOnlyTable.class).values());
        elements.addAll(g.getElements(View.class).values());
        elements.addAll(g.getElements(MaterializedView.class).values());
        elements.addAll(g.getElements(ParameterizedView.class).values());

        elements.forEach(
                ge -> partsToElements.computeIfAbsent(ge.getGrainPart(), gp -> new ArrayList<>())
                        .add(ge)
        );

        CursorGenerator generator = new CursorGenerator(getSourceRoot(), isSnakeToCamel());
        partsToElements.forEach((key, value) -> {
            final String sp;
            if (isSysSchema) {
                sp = "";
            } else {
                final Resource grainPartSource = key.getSource();
                final String scoreRelativeOrAbsolutePath = Arrays.stream(scorePath.split(File.pathSeparator))
                        .filter(path -> new FileResource(new File(path)).contains(grainPartSource))
                        .findFirst().get();
                File scoreDir = new File(scoreRelativeOrAbsolutePath);
                sp = scoreDir.getAbsolutePath();
            }
            value.forEach(
                    ge -> generator.generateCursor(ge, sp)
            );
        });

    }

    private File getSourceRoot() {
        return new File(project.getBuild().getDirectory()
                        + File.separator + generatedSourcesDirName + File.separator + "celesta");
    }

    private void addSourceRoot(File directory) {
        if (this.project != null) {
            this.getLog().info("Adding compile source root for cursors: " + directory);
            addCompileSourceRoot.accept(directory.getAbsolutePath());
        }
    }

}