1 /* 2 * Portions of this software was developed by employees of the National Institute 3 * of Standards and Technology (NIST), an agency of the Federal Government and is 4 * being made available as a public service. Pursuant to title 17 United States 5 * Code Section 105, works of NIST employees are not subject to copyright 6 * protection in the United States. This software may be subject to foreign 7 * copyright. Permission in the United States and in foreign countries, to the 8 * extent that NIST may hold copyright, to use, copy, modify, create derivative 9 * works, and distribute this software and its documentation without fee is hereby 10 * granted on a non-exclusive basis, provided that this notice and disclaimer 11 * of warranty appears in all copies. 12 * 13 * THE SOFTWARE IS PROVIDED 'AS IS' WITHOUT ANY WARRANTY OF ANY KIND, EITHER 14 * EXPRESSED, IMPLIED, OR STATUTORY, INCLUDING, BUT NOT LIMITED TO, ANY WARRANTY 15 * THAT THE SOFTWARE WILL CONFORM TO SPECIFICATIONS, ANY IMPLIED WARRANTIES OF 16 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE, AND FREEDOM FROM 17 * INFRINGEMENT, AND ANY WARRANTY THAT THE DOCUMENTATION WILL CONFORM TO THE 18 * SOFTWARE, OR ANY WARRANTY THAT THE SOFTWARE WILL BE ERROR FREE. IN NO EVENT 19 * SHALL NIST BE LIABLE FOR ANY DAMAGES, INCLUDING, BUT NOT LIMITED TO, DIRECT, 20 * INDIRECT, SPECIAL OR CONSEQUENTIAL DAMAGES, ARISING OUT OF, RESULTING FROM, 21 * OR IN ANY WAY CONNECTED WITH THIS SOFTWARE, WHETHER OR NOT BASED UPON WARRANTY, 22 * CONTRACT, TORT, OR OTHERWISE, WHETHER OR NOT INJURY WAS SUSTAINED BY PERSONS OR 23 * PROPERTY OR OTHERWISE, AND WHETHER OR NOT LOSS WAS SUSTAINED FROM, OR AROSE OUT 24 * OF THE RESULTS OF, OR USE OF, THE SOFTWARE OR SERVICES PROVIDED HEREUNDER. 25 */ 26 27 package gov.nist.secauto.metaschema.core.datatype.markup.flexmark; 28 29 import com.vladsch.flexmark.util.ast.Node; 30 import com.vladsch.flexmark.util.ast.NodeVisitorBase; 31 32 import gov.nist.secauto.metaschema.core.datatype.markup.IMarkupString; 33 import gov.nist.secauto.metaschema.core.datatype.markup.flexmark.InsertAnchorExtension.InsertAnchorNode; 34 35 import java.util.LinkedList; 36 import java.util.List; 37 import java.util.function.Predicate; 38 39 import edu.umd.cs.findbugs.annotations.NonNull; 40 41 public class InsertVisitor 42 extends NodeVisitorBase { 43 @NonNull 44 private final List<InsertAnchorNode> inserts 45 = new LinkedList<>(); 46 @NonNull 47 private final Predicate<InsertAnchorNode> filter; 48 49 public InsertVisitor( 50 @NonNull Predicate<InsertAnchorNode> filter) { 51 this.filter = filter; 52 } 53 54 public InsertVisitor processNode(@NonNull IMarkupString<?> markup) { 55 visit(markup.getDocument()); 56 return this; 57 } 58 59 @Override 60 protected void visit(Node node) { 61 if (node instanceof InsertAnchorNode) { 62 InsertAnchorNode insert = (InsertAnchorNode) node; 63 if (filter.test(insert)) { 64 inserts.add((InsertAnchorNode) node); 65 } 66 } else { 67 visitChildren(node); 68 } 69 } 70 71 @NonNull 72 public List<InsertAnchorNode> getInserts() { 73 return inserts; 74 } 75 }