com.ruleoftech.markdown.page.generator.plugin.FlexmarkLinkResolver Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of markdown-page-generator-plugin Show documentation

Markdown to HTML Page Generator Maven Plugin

There is a newer version: 2.4.2

Show newest version

package com.ruleoftech.markdown.page.generator.plugin;

import com.vladsch.flexmark.util.ast.Node;
import com.vladsch.flexmark.ext.wikilink.internal.WikiLinkLinkResolver;
import com.vladsch.flexmark.html.IndependentLinkResolverFactory;
import com.vladsch.flexmark.html.LinkResolver;
import com.vladsch.flexmark.html.LinkResolverFactory;
import com.vladsch.flexmark.html.renderer.*;
import com.vladsch.flexmark.util.data.DataHolder;

import java.util.HashSet;
import java.util.Set;

public class FlexmarkLinkResolver implements LinkResolver {
    final String[] inputFileExtensions;

    public FlexmarkLinkResolver(LinkResolverContext context) {
        DataHolder options = context.getOptions();
        this.inputFileExtensions = options.get(PageGeneratorExtension.INPUT_FILE_EXTENSIONS).trim().split("\\s*,\\s*");
    }

    @Override
    public ResolvedLink resolveLink(Node node, LinkResolverContext context, ResolvedLink link) {
        ResolvedLink result = link;

        for (String inputFileExtension : inputFileExtensions) {
            if (link.getLinkType() == LinkType.LINK) {
                String url = link.getUrl();
                if (!url.startsWith("http://") && !url.startsWith("https://")) {
                    if (url.endsWith("." + inputFileExtension)) {
                        url = url.substring(0, url.length() - inputFileExtension.length()) + "html";
                        result = result.withStatus(LinkStatus.VALID).withUrl(url);
                        return result;
                    } else if (url.contains("." + inputFileExtension + "#")) {
                        url = url.replace("." + inputFileExtension + "#", ".html#");
                        result = result.withStatus(LinkStatus.VALID).withUrl(url);
                        return result;
                    }
                }
            }
        }

        return result;
    }

    public static class Factory extends IndependentLinkResolverFactory {
        @Override
        public Set> getBeforeDependents() {
            Set> set = new HashSet>();
            set.add(WikiLinkLinkResolver.Factory.class);
            return set;
        }

        @Override
        public LinkResolver apply(LinkResolverContext context) {
            return new FlexmarkLinkResolver(context);
        }
    }
}