io.quarkiverse.roq.frontmatter.deployment.data.RoqFrontMatterDataProcessor Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of quarkus-roq-frontmatter-deployment Show documentation
There is a newer version: 1.1.1
Show newest version
package io.quarkiverse.roq.frontmatter.deployment.data;

import static io.quarkiverse.roq.frontmatter.deployment.TemplateLink.DEFAULT_PAGE_LINK_TEMPLATE;

import java.util.List;
import java.util.Map;
import java.util.Stack;
import java.util.function.Function;
import java.util.stream.Collectors;

import io.quarkiverse.roq.frontmatter.deployment.RoqFrontMatterRootUrlBuildItem;
import io.quarkiverse.roq.frontmatter.deployment.TemplateLink;
import io.quarkiverse.roq.frontmatter.deployment.publish.RoqFrontMatterPublishPageBuildItem;
import io.quarkiverse.roq.frontmatter.deployment.scan.RoqFrontMatterRawTemplateBuildItem;
import io.quarkiverse.roq.frontmatter.runtime.config.RoqSiteConfig;
import io.quarkiverse.roq.frontmatter.runtime.model.RootUrl;
import io.quarkus.deployment.annotations.BuildProducer;
import io.quarkus.deployment.annotations.BuildStep;
import io.quarkus.vertx.http.runtime.HttpBuildTimeConfig;
import io.vertx.core.json.JsonObject;

public class RoqFrontMatterDataProcessor {

    public static final String LINK_KEY = "link";
    public static final String PAGINATE_KEY = "paginate";

    @BuildStep
    void prepareData(HttpBuildTimeConfig httpConfig,
            RoqSiteConfig config,
            BuildProducer rootUrlProducer,
            BuildProducer templatesProducer,
            List roqFrontMatterTemplates) {
        if (roqFrontMatterTemplates.isEmpty()) {
            return;
        }

        final var byId = roqFrontMatterTemplates.stream()
                .collect(Collectors.toMap(RoqFrontMatterRawTemplateBuildItem::id, Function.identity()));
        final RootUrl rootUrl = new RootUrl(config.urlOptional().orElse(""), httpConfig.rootPath);
        rootUrlProducer.produce(new RoqFrontMatterRootUrlBuildItem(rootUrl));

        for (RoqFrontMatterRawTemplateBuildItem item : roqFrontMatterTemplates) {
            JsonObject data = mergeParents(item, byId);
            final String link = TemplateLink.pageLink(config.rootPath(), data.getString(LINK_KEY, DEFAULT_PAGE_LINK_TEMPLATE),
                    new TemplateLink.PageLinkData(item.info(), item.collectionId(), data));
            RoqFrontMatterTemplateBuildItem templateItem = new RoqFrontMatterTemplateBuildItem(item, rootUrl.resolve(link),
                    data);
            templatesProducer.produce(templateItem);
        }
    }

    @BuildStep
    void dispatchByType(BuildProducer pagesProducer,
            BuildProducer documentTemplatesProducer,
            BuildProducer paginatedPagesProducer,
            List templates) {
        if (templates.isEmpty()) {
            return;
        }

        for (RoqFrontMatterTemplateBuildItem item : templates) {
            if (!item.published()) {
                continue;
            }
            if (item.raw().collection() != null) {
                documentTemplatesProducer
                        .produce(new RoqFrontMatterDocumentTemplateBuildItem(item.raw(), item.url(), item.raw().collection(),
                                item.data()));
            } else {
                if (item.data().containsKey(PAGINATE_KEY)) {
                    // Pagination is created needs collections size so it's produced after
                    paginatedPagesProducer
                            .produce(new RoqFrontMatterPaginateTemplateBuildItem(item.url(), item.raw().info(), item.data(),
                                    null));
                } else {
                    pagesProducer
                            .produce(new RoqFrontMatterPublishPageBuildItem(item.url(), item.raw().info(), item.data(), null));
                }
            }

        }
    }

    public static JsonObject mergeParents(RoqFrontMatterRawTemplateBuildItem item,
            Map byId) {
        Stack fms = new Stack<>();
        String parent = item.layout();
        fms.add(item.data());
        while (parent != null) {
            if (!byId.containsKey(parent)) {
                throw new IllegalStateException("Invalid layout: " + parent + " in " + item.info().sourceFileName());
            }
            final RoqFrontMatterRawTemplateBuildItem parentItem = byId.get(parent);
            parent = parentItem.layout();
            fms.push(parentItem.data());
        }

        JsonObject merged = new JsonObject();
        while (!fms.empty()) {
            merged.mergeIn(fms.pop());
        }
        return merged;
    }
}