gutenberg.itext.support.ITextContextBuilder Maven / Gradle / Ivy

Go to download
package gutenberg.itext.support;

import com.google.common.collect.Maps;
import gutenberg.itext.ITextContext;
import gutenberg.itext.PygmentsAdapter;
import gutenberg.itext.Sections;
import gutenberg.itext.Styles;
import gutenberg.itext.emitter.MarkdownEmitter;
import gutenberg.itext.emitter.SourceCodeDitaaExtension;
import gutenberg.itext.emitter.SourceCodeEmitter;
import gutenberg.itext.emitter.SourceCodeLaTeXExtension;
import gutenberg.itext.model.Markdown;
import gutenberg.itext.model.SourceCode;
import gutenberg.pygments.Pygments;
import gutenberg.pygments.StyleSheet;
import gutenberg.pygments.styles.FriendlyStyle;

import java.util.Map;

/**
 * @author @aloyer
 */
public class ITextContextBuilder {
    private Styles styles;
    private PygmentsAdapter pygmentsAdapter;
    private StyleSheet pygmentsStyleSheet = new FriendlyStyle();
    private Map declared = Maps.newHashMap();

    public ITextContextBuilder usingStyles(Styles styles) {
        this.styles = styles;
        return this;
    }

    public ITextContextBuilder usingPygmentsStyleSheet(StyleSheet pygmentsStyleSheet) {
        this.pygmentsStyleSheet = pygmentsStyleSheet;
        return this;
    }

    public ITextContextBuilder usingPygmentsAdapter(PygmentsAdapter pygmentsAdapter) {
        this.pygmentsAdapter = pygmentsAdapter;
        return this;
    }

    public ITextContext build() {
        ITextContext context = new ITextContext(new Sections(styles), styles);
        registerDeclared(context);
        registerEmitters(context);
        return context;
    }

    private void registerDeclared(ITextContext context) {
        for (Map.Entry entry : declared.entrySet()) {
            context.declare(entry.getKey(), entry.getValue());
        }
    }

    protected void registerEmitters(ITextContext context) {
        context.register(SourceCode.class,
                new SourceCodeEmitter(getPygmentsAdapter(),
                        new SourceCodeDitaaExtension(getPygmentsAdapter()),
                        new SourceCodeLaTeXExtension(getPygmentsAdapter())));
        context.register(Markdown.class, new MarkdownEmitter());
    }

    protected PygmentsAdapter getPygmentsAdapter() {
        if (pygmentsAdapter == null) {
            pygmentsAdapter = new PygmentsAdapter(new Pygments(), pygmentsStyleSheet, styles);
        }
        return pygmentsAdapter;
    }

    public ITextContextBuilder declare(Object key, Object value) {
        declared.put(key, value);
        return this;
    }
}