com.bmuschko.asciidoctorj.tabbedcode.TabbedCodeBlockDocinfoProcessor Maven / Gradle / Ivy

Go to download
package com.bmuschko.asciidoctorj.tabbedcode;

import org.asciidoctor.ast.Document;
import org.asciidoctor.extension.DocinfoProcessor;

import java.util.Map;

import static com.bmuschko.asciidoctorj.tabbedcode.FileUtils.LINE_SEPARATOR;
import static com.bmuschko.asciidoctorj.tabbedcode.FileUtils.readFileContentsFromClasspath;

public class TabbedCodeBlockDocinfoProcessor extends DocinfoProcessor {
    public static final String TABBED_CODE_CSS_FILE_PATH_ATTRIBUTE = "tabbed-code-css-path";
    public static final String TABBED_CODE_JS_FILE_PATH_ATTRIBUTE = "tabbed-code-js-path";
    public static final String DEFAULT_CSS_FILE_PATH = "/codeBlockSwitch.css";
    public static final String DEFAULT_JS_FILE_PATH = "/codeBlockSwitch.js";

    @Override
    public String process(Document document) {
        if (document.isBasebackend("html")) {
            Map attributes = document.getAttributes();
            String cssPath = getCssPath(attributes, TABBED_CODE_CSS_FILE_PATH_ATTRIBUTE, DEFAULT_CSS_FILE_PATH);
            String jsPath = getCssPath(attributes, TABBED_CODE_JS_FILE_PATH_ATTRIBUTE, DEFAULT_JS_FILE_PATH);
            String css = readFileContentsFromClasspath(cssPath);
            String js = readFileContentsFromClasspath(jsPath);
            return modifyHeadHtml(css, js);
        }

        return null;
    }

    private String getCssPath(Map attributes, String attributeKey, String defaultPath) {
        if (attributes.containsKey(attributeKey)) {
            return (String) attributes.get(attributeKey);
        }

        return defaultPath;
    }

    private String modifyHeadHtml(String css, String js) {
        StringBuilder htmlHead = new StringBuilder();
        htmlHead.append("").append(LINE_SEPARATOR);
        htmlHead.append("").append(LINE_SEPARATOR);
        htmlHead.append("").append(LINE_SEPARATOR);
        return htmlHead.toString();
    }
}