com.github.cukedoctor.extension.CukedoctorStyleExtension Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of cukedoctor-extension Show documentation

There is a newer version: 3.9.0

package com.github.cukedoctor.extension;

import static com.github.cukedoctor.extension.CukedoctorExtensionRegistry.*;

import java.util.Map;

import org.asciidoctor.ast.Document;
import org.asciidoctor.extension.Postprocessor;
import org.jsoup.Jsoup;
import org.jsoup.nodes.Element;

/**
 * Created by pestano on 20/07/15.
 * extends html document styles
 */
public class CukedoctorStyleExtension extends Postprocessor{

    public CukedoctorStyleExtension(Map config) {
        super(config);
    }

    @Override
    public String process(Document document, String output) {
        if(document.basebackend("html") ){
            org.jsoup.nodes.Document doc = Jsoup.parse(output, "UTF-8");

            Element contentElement = doc.getElementById("footer");
                if(System.getProperty(STYLE_DISABLE_EXT_KEY) == null) {
                        addStyleClass(contentElement);
                }

            return doc.html();

        }else{
            return output;
        }
    }

        private void addStyleClass(Element contentElement) {
             String styleClass = "