bcv-vf/src/the/bytecode/club/bytecodeviewer/gui/SearchingPane.java

package the.bytecode.club.bytecodeviewer.gui;

import java.awt.BorderLayout;
import java.awt.GridLayout;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.awt.event.ItemEvent;
import java.awt.event.ItemListener;
import java.util.regex.Pattern;
import java.util.regex.PatternSyntaxException;

import javax.swing.DefaultComboBoxModel;
import javax.swing.JButton;
import javax.swing.JCheckBox;
import javax.swing.JComboBox;
import javax.swing.JLabel;
import javax.swing.JPanel;
import javax.swing.JScrollPane;
import javax.swing.JTree;
import javax.swing.event.TreeSelectionEvent;
import javax.swing.event.TreeSelectionListener;
import javax.swing.tree.DefaultMutableTreeNode;
import javax.swing.tree.TreePath;

import org.objectweb.asm.tree.ClassNode;

import the.bytecode.club.bytecodeviewer.*;
import the.bytecode.club.bytecodeviewer.searching.*;

/***************************************************************************
 * Bytecode Viewer (BCV) - Java & Android Reverse Engineering Suite        *
 * Copyright (C) 2014 Kalen 'Konloch' Kinloch - http://bytecodeviewer.com  *
 *                                                                         *
 * This program is free software: you can redistribute it and/or modify    *
 *   it under the terms of the GNU General Public License as published by  *
 *   the Free Software Foundation, either version 3 of the License, or     *
 *   (at your option) any later version.                                   *
 *                                                                         *
 *   This program is distributed in the hope that it will be useful,       *
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
 *   GNU General Public License for more details.                          *
 *                                                                         *
 *   You should have received a copy of the GNU General Public License     *
 *   along with this program.  If not, see <http://www.gnu.org/licenses/>. *
 ***************************************************************************/

/**
 * A pane dedicating to searching the loaded files.
 * 
 * @author Konloch
 * @author WaterWolf
 * 
 */

@SuppressWarnings("rawtypes")
public class SearchingPane extends VisibleComponent {

	private static final long serialVersionUID = -1098524689236993932L;

	FileChangeNotifier fcn;

	JCheckBox exact = new JCheckBox("Exact");
	DefaultMutableTreeNode treeRoot = new DefaultMutableTreeNode("Root");
	JTree tree;

	SearchType searchType = null;
	JComboBox searchRadiusBox;

	public JButton search = new JButton("Search");
	BackgroundSearchThread t = new BackgroundSearchThread(true) {
		@Override
		public void doSearch() {
			// empty
		}

	};

	@SuppressWarnings("unchecked")
	public SearchingPane(final FileChangeNotifier fcn) {
		super("Search");

		this.fcn = fcn;

		final JPanel optionPanel = new JPanel(new BorderLayout());

		final JPanel searchRadiusOpt = new JPanel(new BorderLayout());

		final JPanel searchOpts = new JPanel(new GridLayout(2, 1));

		searchRadiusOpt.add(new JLabel("Search from "), BorderLayout.WEST);

		DefaultComboBoxModel model = new DefaultComboBoxModel();
		for (final SearchRadius st : SearchRadius.values()) {
			model.addElement(st);
		}

		searchRadiusBox = new JComboBox(model);

		searchRadiusOpt.add(searchRadiusBox, BorderLayout.CENTER);

		searchOpts.add(searchRadiusOpt);

		model = new DefaultComboBoxModel();
		for (final SearchType st : SearchType.values()) {
			model.addElement(st);
		}

		final JComboBox typeBox = new JComboBox(model);
		final JPanel searchOptPanel = new JPanel();

		final ItemListener il = new ItemListener() {
			@Override
			public void itemStateChanged(final ItemEvent arg0) {
				searchOptPanel.removeAll();
				searchType = (SearchType) typeBox.getSelectedItem();
				searchOptPanel.add(searchType.details.getPanel());

				searchOptPanel.revalidate();
				searchOptPanel.repaint();
			}
		};

		typeBox.addItemListener(il);

		typeBox.setSelectedItem(SearchType.LDC);
		il.itemStateChanged(null);

		searchOpts.add(typeBox);

		optionPanel.add(searchOpts, BorderLayout.NORTH);

		JPanel p2 = new JPanel();
		p2.setLayout(new BorderLayout());
		p2.add(searchOptPanel, BorderLayout.NORTH);
		p2.add(exact, BorderLayout.SOUTH);

		optionPanel.add(p2, BorderLayout.CENTER);

		search.addActionListener(new ActionListener() {
			@Override
			public void actionPerformed(final ActionEvent arg0) {
				treeRoot.removeAllChildren();
				searchType = (SearchType) typeBox.getSelectedItem();
				final SearchRadius radius = (SearchRadius) searchRadiusBox
						.getSelectedItem();
				final SearchResultNotifier srn = new SearchResultNotifier() {
					@Override
					public void notifyOfResult(String debug) {
						treeRoot.add(new DefaultMutableTreeNode(debug));
					}
				};
				if (radius == SearchRadius.All_Classes) {
					if (t.finished) {
						t = new BackgroundSearchThread() {
							@Override
							public void doSearch() {

								try {
									Pattern.compile(RegexInsnFinder.processRegex(RegexSearch.searchText.getText()), Pattern.MULTILINE);
								} catch (PatternSyntaxException ex) {
									BytecodeViewer.showMessage("You have an error in your regex syntax.");
								}
								
								for (ClassNode cln : BytecodeViewer.getLoadedClasses())
									searchType.details.search(cln, srn, exact.isSelected());

								MainViewerGUI.getComponent(SearchingPane.class).search.setEnabled(true);
								MainViewerGUI.getComponent(SearchingPane.class).search.setText("Search");
								tree.expandPath(new TreePath(tree.getModel().getRoot()));
								tree.updateUI();
							}

						};
						MainViewerGUI.getComponent(SearchingPane.class).search
								.setEnabled(false);
						MainViewerGUI.getComponent(SearchingPane.class).search
								.setText("Searching, please wait..");
						t.start();
					} else { // this should really never be called.
						BytecodeViewer
								.showMessage("You currently have a search performing in the background, please wait for that to finish.");
					}
				} else if (radius == SearchRadius.Current_Class) {
					final Viewer cv = MainViewerGUI.getComponent(WorkPane.class).getCurrentViewer();
					if (cv != null) {
						searchType.details.search(cv.cn, srn,
								exact.isSelected());
					}
				}
			}
		});

		optionPanel.add(search, BorderLayout.SOUTH);

		this.tree = new JTree(treeRoot);

		getContentPane().setLayout(new BorderLayout());

		getContentPane().add(new JScrollPane(optionPanel), BorderLayout.NORTH);
		getContentPane().add(new JScrollPane(tree), BorderLayout.CENTER);

		this.tree.addTreeSelectionListener(new TreeSelectionListener() {
			@Override
			public void valueChanged(final TreeSelectionEvent arg0) {
				String path = arg0.getPath().toString();

				String className = path.split(", ")[1].split("\\.")[0];
				final ClassNode fN = BytecodeViewer.getClassNode(className);
				if (fN != null) {
					MainViewerGUI.getComponent(FileNavigationPane.class)
							.openClassFileToWorkSpace(className+".class", fN);
				}

				System.out.println(className);
			}
		});

		this.setVisible(true);

	}

	public enum SearchType {
		LDC(new LDCSearch()), Regex(new RegexSearch()), MethodCall(
				new MethodCallSearch()), FieldCall(new FieldCallSearch());

		public final SearchTypeDetails details;

		SearchType(final SearchTypeDetails details) {
			this.details = details;
		}
	}

	public enum SearchRadius {
		All_Classes, Current_Class;
	}

	public void resetWorkspace() {
		treeRoot.removeAllChildren();
		tree.updateUI();
	}

	@Override
	public void openFile(String name, byte[] contents) {
		// TODO Auto-generated method stub
		
	}

}