001/* 002 * Portions of this software was developed by employees of the National Institute 003 * of Standards and Technology (NIST), an agency of the Federal Government and is 004 * being made available as a public service. Pursuant to title 17 United States 005 * Code Section 105, works of NIST employees are not subject to copyright 006 * protection in the United States. This software may be subject to foreign 007 * copyright. Permission in the United States and in foreign countries, to the 008 * extent that NIST may hold copyright, to use, copy, modify, create derivative 009 * works, and distribute this software and its documentation without fee is hereby 010 * granted on a non-exclusive basis, provided that this notice and disclaimer 011 * of warranty appears in all copies. 012 * 013 * THE SOFTWARE IS PROVIDED 'AS IS' WITHOUT ANY WARRANTY OF ANY KIND, EITHER 014 * EXPRESSED, IMPLIED, OR STATUTORY, INCLUDING, BUT NOT LIMITED TO, ANY WARRANTY 015 * THAT THE SOFTWARE WILL CONFORM TO SPECIFICATIONS, ANY IMPLIED WARRANTIES OF 016 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE, AND FREEDOM FROM 017 * INFRINGEMENT, AND ANY WARRANTY THAT THE DOCUMENTATION WILL CONFORM TO THE 018 * SOFTWARE, OR ANY WARRANTY THAT THE SOFTWARE WILL BE ERROR FREE. IN NO EVENT 019 * SHALL NIST BE LIABLE FOR ANY DAMAGES, INCLUDING, BUT NOT LIMITED TO, DIRECT, 020 * INDIRECT, SPECIAL OR CONSEQUENTIAL DAMAGES, ARISING OUT OF, RESULTING FROM, 021 * OR IN ANY WAY CONNECTED WITH THIS SOFTWARE, WHETHER OR NOT BASED UPON WARRANTY, 022 * CONTRACT, TORT, OR OTHERWISE, WHETHER OR NOT INJURY WAS SUSTAINED BY PERSONS OR 023 * PROPERTY OR OTHERWISE, AND WHETHER OR NOT LOSS WAS SUSTAINED FROM, OR AROSE OUT 024 * OF THE RESULTS OF, OR USE OF, THE SOFTWARE OR SERVICES PROVIDED HEREUNDER. 025 */ 026 027package gov.nist.secauto.metaschema.core.datatype.markup.flexmark; 028 029import com.vladsch.flexmark.util.ast.Node; 030import com.vladsch.flexmark.util.ast.NodeVisitorBase; 031 032import gov.nist.secauto.metaschema.core.datatype.markup.IMarkupString; 033import gov.nist.secauto.metaschema.core.datatype.markup.flexmark.InsertAnchorExtension.InsertAnchorNode; 034 035import java.util.LinkedList; 036import java.util.List; 037import java.util.function.Predicate; 038 039import edu.umd.cs.findbugs.annotations.NonNull; 040 041public class InsertVisitor 042 extends NodeVisitorBase { 043 @NonNull 044 private final List<InsertAnchorNode> inserts 045 = new LinkedList<>(); 046 @NonNull 047 private final Predicate<InsertAnchorNode> filter; 048 049 public InsertVisitor( 050 @NonNull Predicate<InsertAnchorNode> filter) { 051 this.filter = filter; 052 } 053 054 public InsertVisitor processNode(@NonNull IMarkupString<?> markup) { 055 visit(markup.getDocument()); 056 return this; 057 } 058 059 @Override 060 protected void visit(Node node) { 061 if (node instanceof InsertAnchorNode) { 062 InsertAnchorNode insert = (InsertAnchorNode) node; 063 if (filter.test(insert)) { 064 inserts.add((InsertAnchorNode) node); 065 } 066 } else { 067 visitChildren(node); 068 } 069 } 070 071 @NonNull 072 public List<InsertAnchorNode> getInserts() { 073 return inserts; 074 } 075}