package com.itextpdf.samples.sandbox.pdfhtml.headertagging;

import com.itextpdf.html2pdf.attach.ITagWorker;
import com.itextpdf.html2pdf.attach.ProcessorContext;
import com.itextpdf.html2pdf.attach.impl.DefaultTagWorkerFactory;
import com.itextpdf.styledxmlparser.node.IElementNode;

public class AccessibilityTagWorkerFactory extends DefaultTagWorkerFactory {

    @Override
    public ITagWorker getCustomTagWorker(IElementNode tag, ProcessorContext context) {
        switch (tag.name()) {
            case "h1":
                return new CustomHTagWorker(tag, context, 1);
            case "h2":
                return new CustomHTagWorker(tag, context, 2);
            case "h3":
                return new CustomHTagWorker(tag, context, 3);
            case "h4":
                return new CustomHTagWorker(tag, context, 4);
            case "h5":
                return new CustomHTagWorker(tag, context, 5);
            case "h6":
                return new CustomHTagWorker(tag, context, 6);
            case "th":
                return new CustomThTagWorker(tag, context);
            default:
                return null;
        }
    }
}