From 85b3ea9beb5934012280dc0efa475f334dd9a93a Mon Sep 17 00:00:00 2001
From: jeff
Date: Tue, 29 Jul 2014 23:12:30 -0400
Subject: added gui/cli loading of jars/mappings gui can save mappings too

---
 src/cuchaz/enigma/gui/GuiController.java | 147 +++++++++++++++++++++++++++++++
 1 file changed, 147 insertions(+)
 create mode 100644 src/cuchaz/enigma/gui/GuiController.java

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

diff --git a/src/cuchaz/enigma/gui/GuiController.java b/src/cuchaz/enigma/gui/GuiController.java
new file mode 100644
index 0000000..5df2d43
--- /dev/null
+++ b/src/cuchaz/enigma/gui/GuiController.java
@@ -0,0 +1,147 @@
+/*******************************************************************************
+ * Copyright (c) 2014 Jeff Martin.
+ * All rights reserved. This program and the accompanying materials
+ * are made available under the terms of the GNU Public License v3.0
+ * which accompanies this distribution, and is available at
+ * http://www.gnu.org/licenses/gpl.html
+ * 
+ * Contributors:
+ *     Jeff Martin - initial API and implementation
+ ******************************************************************************/
+package cuchaz.enigma.gui;
+
+import java.io.File;
+import java.io.FileInputStream;
+import java.io.FileOutputStream;
+import java.io.IOException;
+
+import cuchaz.enigma.ClassFile;
+import cuchaz.enigma.Deobfuscator;
+import cuchaz.enigma.analysis.Analyzer;
+import cuchaz.enigma.analysis.SourceIndex;
+import cuchaz.enigma.mapping.ClassEntry;
+import cuchaz.enigma.mapping.Entry;
+import cuchaz.enigma.mapping.EntryPair;
+import cuchaz.enigma.mapping.TranslationMappings;
+
+public class GuiController
+{
+	private Deobfuscator m_deobfuscator;
+	private Gui m_gui;
+	private SourceIndex m_index;
+	private ClassFile m_currentFile;
+	
+	public GuiController( Gui gui )
+	{
+		m_gui = gui;
+		m_deobfuscator = null;
+		m_index = null;
+		m_currentFile = null;
+	}
+	
+	public void openJar( File file )
+	throws IOException
+	{
+		m_deobfuscator = new Deobfuscator( file );
+		m_gui.onOpenJar( m_deobfuscator.getJarName() );
+		m_gui.setObfClasses( m_deobfuscator.getObfuscatedClasses() );
+	}
+	
+	public void closeJar( )
+	{
+		m_deobfuscator = null;
+		m_gui.onCloseJar();
+	}
+	
+	public void openMappings( File file )
+	throws IOException
+	{
+		FileInputStream in = new FileInputStream( file );
+		m_deobfuscator.setMappings( TranslationMappings.newFromStream( in ) );
+		in.close();
+		refreshOpenFiles();
+	}
+
+	public void saveMappings( File file )
+	throws IOException
+	{
+		FileOutputStream out = new FileOutputStream( file );
+		m_deobfuscator.getMappings().write( out );
+		out.close();
+	}
+
+	public void closeMappings( )
+	{
+		m_deobfuscator.setMappings( null );
+		refreshOpenFiles();
+	}
+	
+	public void deobfuscateClass( ClassFile classFile )
+	{
+		m_currentFile = classFile;
+		deobfuscate( m_currentFile );
+	}
+	
+	public EntryPair getEntryPair( int pos )
+	{
+		if( m_index == null )
+		{
+			return null;
+		}
+		
+		Entry deobfEntry = m_index.getEntry( pos );
+		if( deobfEntry == null )
+		{
+			return null;
+		}
+		return new EntryPair( m_deobfuscator.obfuscate( deobfEntry ), deobfEntry );
+	}
+	
+	public void rename( Entry obfsEntry, String newName )
+	{
+		m_deobfuscator.rename( obfsEntry, newName );
+		
+		// did we rename the current file?
+		if( obfsEntry instanceof ClassEntry )
+		{
+			ClassEntry classEntry = (ClassEntry)obfsEntry;
+			
+			// update the current file
+			if( classEntry.getName().equals( m_currentFile.getName() ) )
+			{
+				m_currentFile = new ClassFile( newName );
+			}
+		}
+		
+		refreshOpenFiles();
+	}
+	
+	private void refreshOpenFiles( )
+	{
+		if( m_currentFile != null )
+		{
+			deobfuscate( m_currentFile );
+		}
+	}
+
+	private void deobfuscate( final ClassFile classFile )
+	{
+		m_gui.setSource( "(deobfuscating...)" );
+		
+		// run the deobfuscator in a separate thread so we don't block the GUI event queue
+		new Thread( )
+		{
+			@Override
+			public void run( )
+			{
+				// deobfuscate the bytecode
+				String source = m_deobfuscator.getSource( classFile );
+				m_gui.setSource( source );
+				
+				// index the source file
+				m_index = Analyzer.analyze( classFile.getName(), source );
+				m_gui.setHighlightedTokens( m_index.tokens() );
+			}
+		}.start();
+	}
+}
-- 
cgit v1.2.3