com.vladsch.flexmark.ext.tables.TableExtractingVisitor Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of flexmark-ext-tables Show documentation
flexmark-java extension for tables using "|" pipes with optional column spans and table caption
The newest version!
package com.vladsch.flexmark.ext.tables;

import com.vladsch.flexmark.util.ast.Node;
import com.vladsch.flexmark.util.ast.NodeVisitor;
import com.vladsch.flexmark.util.ast.VisitHandler;
import com.vladsch.flexmark.util.data.DataHolder;
import com.vladsch.flexmark.util.format.MarkdownTable;
import com.vladsch.flexmark.util.format.TableFormatOptions;
import com.vladsch.flexmark.util.html.CellAlignment;
import com.vladsch.flexmark.util.sequence.BasedSequence;
import java.util.ArrayList;
import java.util.List;

public class TableExtractingVisitor {
  private final TableFormatOptions options;

  private final NodeVisitor myVisitor =
      new NodeVisitor(
          new VisitHandler<>(TableBlock.class, this::visit),
          new VisitHandler<>(TableHead.class, this::visit),
          new VisitHandler<>(TableSeparator.class, this::visit),
          new VisitHandler<>(TableBody.class, this::visit),
          new VisitHandler<>(TableRow.class, this::visit),
          new VisitHandler<>(TableCell.class, this::visit),
          new VisitHandler<>(TableCaption.class, this::visit));

  private MarkdownTable myTable;
  private final List myTables;

  public TableExtractingVisitor(DataHolder options) {
    this.options = new TableFormatOptions(options);
    myTables = new ArrayList<>();
  }

  public MarkdownTable[] getTables(Node node) {
    myTable = null;
    myVisitor.visit(node);
    return myTables.toArray(new MarkdownTable[0]);
  }

  private void visit(TableBlock node) {
    myTable = new MarkdownTable(node.getChars(), options);
    myVisitor.visitChildren(node);
    myTables.add(myTable);

    myTable = null;
  }

  private void visit(TableHead node) {
    myTable.setSeparator(false);
    myTable.setHeader(true);
    myVisitor.visitChildren(node);
  }

  private void visit(TableSeparator node) {
    myTable.setSeparator(true);
    myVisitor.visitChildren(node);
  }

  private void visit(TableBody node) {
    myTable.setSeparator(false);
    myTable.setHeader(false);
    myVisitor.visitChildren(node);
  }

  private void visit(TableRow node) {
    myVisitor.visitChildren(node);
    if (!myTable.isSeparator()) myTable.nextRow();
  }

  private void visit(TableCaption node) {
    myTable.setCaptionWithMarkers(
        node, node.getOpeningMarker(), node.getText(), node.getClosingMarker());
  }

  private void visit(TableCell node) {
    BasedSequence text = node.getText();
    if (options.trimCellWhitespace) {
      if (text.isBlank() && !text.isEmpty()) {
        text = text.subSequence(0, 1);
      } else {
        text = text.trim();
      }
    }
    myTable.addCell(
        new com.vladsch.flexmark.util.format.TableCell(
            node,
            node.getOpeningMarker(),
            text,
            node.getClosingMarker(),
            1,
            node.getSpan(),
            node.getAlignment() == null
                ? CellAlignment.NONE
                : node.getAlignment().cellAlignment()));
  }
}