SymbolGatherer.java

  1. /*
  2.  *  Copyright (c) 2001-2024, Jean Tessier
  3.  *  All rights reserved.
  4.  *  
  5.  *  Redistribution and use in source and binary forms, with or without
  6.  *  modification, are permitted provided that the following conditions
  7.  *  are met:
  8.  *  
  9.  *      * Redistributions of source code must retain the above copyright
  10.  *        notice, this list of conditions and the following disclaimer.
  11.  *  
  12.  *      * Redistributions in binary form must reproduce the above copyright
  13.  *        notice, this list of conditions and the following disclaimer in the
  14.  *        documentation and/or other materials provided with the distribution.
  15.  *  
  16.  *      * Neither the name of Jean Tessier nor the names of his contributors
  17.  *        may be used to endorse or promote products derived from this software
  18.  *        without specific prior written permission.
  19.  *  
  20.  *  THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  21.  *  "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  22.  *  LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  23.  *  A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR
  24.  *  CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
  25.  *  EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
  26.  *  PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
  27.  *  PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
  28.  *  LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
  29.  *  NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
  30.  *  SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  31.  */

  32. package com.jeantessier.classreader;

  33. import java.util.*;
  34. import java.util.stream.*;

  35. public class SymbolGatherer extends VisitorBase {
  36.     private final SymbolGathererStrategy strategy;

  37.     private final Collection<Visitable> symbols = new ArrayList<>();

  38.     private InnerClass innerClass;

  39.     public SymbolGatherer(SymbolGathererStrategy strategy) {
  40.         this.strategy = strategy;
  41.     }

  42.     public Stream<Visitable> stream() {
  43.         return symbols.stream();
  44.     }

  45.     // Classfile
  46.     public void visitClassfile(Classfile classfile) {
  47.         innerClass = null;
  48.         super.visitClassfile(classfile);

  49.         if (innerClass != null) {
  50.             if (strategy.isMatching(innerClass)) {
  51.                 symbols.add(innerClass);
  52.             }
  53.         } else if (strategy.isMatching(classfile)) {
  54.             symbols.add(classfile);
  55.         }
  56.     }

  57.     // Features
  58.     public void visitField_info(Field_info entry) {
  59.         if (strategy.isMatching(entry)) {
  60.             symbols.add(entry);
  61.         }

  62.         super.visitField_info(entry);
  63.     }

  64.     public void visitMethod_info(Method_info entry) {
  65.         if (strategy.isMatching(entry)) {
  66.             symbols.add(entry);
  67.         }

  68.         super.visitMethod_info(entry);
  69.     }

  70.     public void visitLocalVariable(LocalVariable helper) {
  71.         if (strategy.isMatching(helper)) {
  72.             symbols.add(helper);
  73.         }

  74.         super.visitLocalVariable(helper);
  75.     }

  76.     public void visitInnerClass(InnerClass helper) {
  77.         if (helper.getClassfile().getRawClass() == helper.getRawInnerClassInfo()) {
  78.             this.innerClass = helper;
  79.         }

  80.         super.visitInnerClass(helper);
  81.     }
  82. }