From d71c0af8ed298bfb4b35b4e3e61b5678bc1c7d9f Mon Sep 17 00:00:00 2001
From: jeff
Date: Wed, 6 Aug 2014 00:22:25 -0400
Subject: added simple class inheritance browsing

---
 src/cuchaz/enigma/gui/Gui.java | 61 +++++++++++++++++++++++++++++++++++-------
 1 file changed, 52 insertions(+), 9 deletions(-)

(limited to 'src/cuchaz/enigma/gui/Gui.java')

diff --git a/src/cuchaz/enigma/gui/Gui.java b/src/cuchaz/enigma/gui/Gui.java
index f9afb64..2002a4d 100644
--- a/src/cuchaz/enigma/gui/Gui.java
+++ b/src/cuchaz/enigma/gui/Gui.java
@@ -49,12 +49,18 @@ import javax.swing.JScrollPane;
 import javax.swing.JSplitPane;
 import javax.swing.JTabbedPane;
 import javax.swing.JTextField;
+import javax.swing.JTree;
 import javax.swing.ListSelectionModel;
 import javax.swing.WindowConstants;
 import javax.swing.event.CaretEvent;
 import javax.swing.event.CaretListener;
 import javax.swing.text.BadLocationException;
 import javax.swing.text.Highlighter;
+import javax.swing.tree.DefaultTreeModel;
+import javax.swing.tree.TreeNode;
+import javax.swing.tree.TreePath;
+
+import com.beust.jcommander.internal.Lists;
 
 import jsyntaxpane.DefaultSyntaxKit;
 import jsyntaxpane.SyntaxDocument;
@@ -100,7 +106,7 @@ public class Gui
 	private JPanel m_infoPanel;
 	private BoxHighlightPainter m_obfuscatedHighlightPainter;
 	private BoxHighlightPainter m_deobfuscatedHighlightPainter;
-	private JPanel m_inheritancePanel;
+	private JTree m_inheritanceTree;
 	
 	// dynamic menu items
 	private JMenuItem m_closeJarMenu;
@@ -136,6 +142,7 @@ public class Gui
 		m_obfClasses.setCellRenderer( new ObfuscatedClassListCellRenderer() );
 		m_obfClasses.addMouseListener( new MouseAdapter()
 		{
+			@Override
 			public void mouseClicked( MouseEvent event )
 			{
 				if( event.getClickCount() == 2 )
@@ -161,6 +168,7 @@ public class Gui
 		m_deobfClasses.setCellRenderer( new DeobfuscatedClassListCellRenderer() );
 		m_deobfClasses.addMouseListener( new MouseAdapter()
 		{
+			@Override
 			public void mouseClicked( MouseEvent event )
 			{
 				if( event.getClickCount() == 2 )
@@ -252,7 +260,26 @@ public class Gui
 		}
 		
 		// init inheritance panel
-		m_inheritancePanel = new JPanel();
+		m_inheritanceTree = new JTree();
+		m_inheritanceTree.setModel( null );
+		m_inheritanceTree.addMouseListener( new MouseAdapter( )
+		{
+			@Override
+			public void mouseClicked( MouseEvent event )
+			{
+				if( event.getClickCount() == 2 )
+				{
+					ClassInheritanceTreeNode node = (ClassInheritanceTreeNode)m_inheritanceTree.getSelectionPath().getLastPathComponent();
+					if( node != null )
+					{
+						m_controller.deobfuscateClass( new ClassFile( node.getClassName() ) );
+					}
+				}
+			}
+		} );
+		JPanel inheritancePanel = new JPanel();
+		inheritancePanel.setLayout( new BorderLayout() );
+		inheritancePanel.add( new JScrollPane( m_inheritanceTree ) );
 		
 		// layout controls
 		JSplitPane splitLeft = new JSplitPane( JSplitPane.VERTICAL_SPLIT, true, obfPanel, deobfPanel );
@@ -263,7 +290,7 @@ public class Gui
 		centerPanel.add( sourceScroller, BorderLayout.CENTER );
 		JTabbedPane tabbedPane = new JTabbedPane();
 		tabbedPane.setPreferredSize( new Dimension( 200, 0 ) );
-		tabbedPane.addTab( "Inheritance", m_inheritancePanel );
+		tabbedPane.addTab( "Inheritance", inheritancePanel );
 		JSplitPane splitRight = new JSplitPane( JSplitPane.HORIZONTAL_SPLIT, true, centerPanel, tabbedPane );
 		splitRight.setResizeWeight( 1 ); // let the left side take all the slack
 		splitRight.resetToPreferredSizes();
@@ -748,12 +775,28 @@ public class Gui
 	
 	private void showInheritance( )
 	{
-		m_inheritancePanel.removeAll();
-		
-		// TEMP
-		m_inheritancePanel.add( new JLabel( m_selectedEntryPair.obf.getName() ) );
-		m_inheritancePanel.add( new JLabel( m_selectedEntryPair.deobf.getName() ) );
-		
+		// get the current class
+		if( m_selectedEntryPair.obf instanceof ClassEntry )
+		{
+			ClassInheritanceTreeNode classNode = m_controller.getClassInheritance( (ClassEntry)m_selectedEntryPair.obf );
+			
+			// build the path from the root to the class node
+			List<TreeNode> nodes = Lists.newArrayList();
+			TreeNode node = classNode;
+			do
+			{
+				nodes.add( node );
+				node = node.getParent();
+			}
+			while( node != null );
+			Collections.reverse( nodes );
+			TreePath path = new TreePath( nodes.toArray() );
+			
+			// show the tree at the root
+			m_inheritanceTree.setModel( new DefaultTreeModel( (TreeNode)path.getPathComponent( 0 ) ) );
+			m_inheritanceTree.expandPath( path );
+			m_inheritanceTree.setSelectionRow( m_inheritanceTree.getRowForPath( path ) );
+		}
 		redraw();
 	}
 	
-- 
cgit v1.2.3