DependencyExtractor.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.dependencyfinder.cli;

  33. import java.util.*;

  34. import com.jeantessier.classreader.*;
  35. import com.jeantessier.commandline.*;
  36. import com.jeantessier.dependency.*;

  37. public class DependencyExtractor extends DirectoryExplorerCommand {
  38.     protected void populateCommandLineSwitches() {
  39.         super.populateCommandLineSwitches();
  40.         populateCommandLineSwitchesForXMLOutput(com.jeantessier.dependency.XMLPrinter.DEFAULT_ENCODING, com.jeantessier.dependency.XMLPrinter.DEFAULT_DTD_PREFIX, com.jeantessier.dependency.XMLPrinter.DEFAULT_INDENT_TEXT);
  41.         populateCommandLineSwitchesForFiltering();

  42.         getCommandLine().addToggleSwitch("maximize");
  43.         getCommandLine().addToggleSwitch("minimize");

  44.         getCommandLine().addToggleSwitch("xml");
  45.     }

  46.     protected Collection<CommandLineException> parseCommandLine(String[] args) {
  47.         Collection<CommandLineException> exceptions = super.parseCommandLine(args);

  48.         exceptions.addAll(validateCommandLineForFiltering());

  49.         if (getCommandLine().getToggleSwitch("maximize") && getCommandLine().getToggleSwitch("minimize")) {
  50.             exceptions.add(new CommandLineException("Only one of -maximize or -minimize is allowed"));
  51.         }

  52.         return exceptions;
  53.     }

  54.     protected void doProcessing() throws Exception {
  55.         SelectionCriteria filterCriteria = getFilterCriteria();

  56.         NodeFactory factory = new NodeFactory();
  57.         CodeDependencyCollector collector = new CodeDependencyCollector(factory, filterCriteria);

  58.         ClassfileLoader loader = new TransientClassfileLoader();
  59.         loader.addLoadListener(new LoadListenerVisitorAdapter(collector));
  60.         loader.addLoadListener(getVerboseListener());
  61.         loader.load(getCommandLine().getParameters());

  62.         if (getCommandLine().getToggleSwitch("minimize")) {
  63.             LinkMinimizer minimizer = new LinkMinimizer();
  64.             minimizer.traverseNodes(factory.getPackages().values());
  65.         } else if (getCommandLine().getToggleSwitch("maximize")) {
  66.             LinkMaximizer maximizer = new LinkMaximizer();
  67.             maximizer.traverseNodes(factory.getPackages().values());
  68.         }

  69.         getVerboseListener().print("Printing the graph ...");

  70.         com.jeantessier.dependency.Printer printer;
  71.         if (getCommandLine().getToggleSwitch("xml")) {
  72.             printer = new com.jeantessier.dependency.XMLPrinter(getOut(), getCommandLine().getSingleSwitch("encoding"), getCommandLine().getSingleSwitch("dtd-prefix"));
  73.         } else {
  74.             printer = new com.jeantessier.dependency.TextPrinter(getOut());
  75.         }

  76.         if (getCommandLine().isPresent("indent-text")) {
  77.             printer.setIndentText(getCommandLine().getSingleSwitch("indent-text"));
  78.         }

  79.         printer.traverseNodes(factory.getPackages().values());
  80.     }

  81.     public static void main(String[] args) throws Exception {
  82.         new DependencyExtractor().run(args);
  83.     }
  84. }