com.vladsch.flexmark.ext.yaml.front.matter.AbstractYamlFrontMatterVisitor Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of flexmark-ext-yaml-front-matter Show documentation

flexmark-java extension for YAML front matter

The newest version!

package com.vladsch.flexmark.ext.yaml.front.matter;

import com.vladsch.flexmark.util.ast.Node;
import com.vladsch.flexmark.util.ast.NodeVisitor;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;

public class AbstractYamlFrontMatterVisitor implements YamlFrontMatterVisitor {
  private final Map> data;
  private final NodeVisitor myVisitor;

  public AbstractYamlFrontMatterVisitor() {
    // myVisitor = new NodeVisitor(
    //        new VisitHandler<>(YamlFrontMatterNode.class, this::visit),
    //        new VisitHandler<>(YamlFrontMatterBlock.class, this::visit)
    // );
    myVisitor = new NodeVisitor(YamlFrontMatterVisitorExt.VISIT_HANDLERS(this));
    data = new LinkedHashMap<>();
  }

  public void visit(Node node) {
    myVisitor.visit(node);
  }

  @Override
  public void visit(YamlFrontMatterNode node) {
    data.put(node.getKey(), node.getValues());
  }

  @Override
  public void visit(YamlFrontMatterBlock node) {
    myVisitor.visitChildren(node);
  }

  public Map> getData() {
    return data;
  }
}