summaryrefslogtreecommitdiff
path: root/src/cuchaz/enigma/mapping/MappingsReader.java
blob: b0394090406ff21dc41cb5f1b67e9496884dc825 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
/*******************************************************************************
 * 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.mapping;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.Reader;
import java.util.NoSuchElementException;
import java.util.Scanner;

import cuchaz.enigma.Util;

public class MappingsReader
{
	public Mappings read( Reader in )
	throws IOException
	{
		return read( new BufferedReader( in ) );
	}
	
	public Mappings read( BufferedReader in )
	throws IOException
	{
		Mappings mappings = new Mappings();
		ClassMapping classMapping = null;
		MethodMapping methodMapping = null;
		
		int lineNumber = 0;
		String line = null;
		while( ( line = in.readLine() ) != null )
		{
			lineNumber++;
			
			// strip comments
			int commentPos = line.indexOf( '#' );
			if( commentPos >= 0 )
			{
				line = line.substring( 0, commentPos );
			}
			
			// skip blank lines
			line = line.trim();
			if( line.length() <= 0 )
			{
				continue;
			}
			
			Scanner scanner = new Scanner( line );
			try
			{
				while( scanner.hasNext() )
				{
					// read the first token
					String token = scanner.next();
					
					if( token.equalsIgnoreCase( "CLASS" ) )
					{
						classMapping = readClass( scanner );
						mappings.addClassMapping( classMapping );
						methodMapping = null;
					}
					else if( token.equalsIgnoreCase( "FIELD" ) )
					{
						if( classMapping == null )
						{
							throw new IllegalArgumentException( "Line " + lineNumber + ": Unexpected FIELD entry here!" );
						}
						classMapping.addFieldMapping( readField( scanner ) );
					}
					else if( token.equalsIgnoreCase( "METHOD" ) )
					{
						if( classMapping == null )
						{
							throw new IllegalArgumentException( "Line " + lineNumber + ": Unexpected METHOD entry here!" );
						}
						methodMapping = readMethod( scanner );
						classMapping.addMethodMapping( methodMapping );
					}
					else if( token.equalsIgnoreCase( "ARG" ) )
					{
						if( classMapping == null || methodMapping == null )
						{
							throw new IllegalArgumentException( "Line " + lineNumber + ": Unexpected ARG entry here!" );
						}
						methodMapping.addArgumentMapping( readArgument( scanner ) );
					}
				}
			}
			catch( NoSuchElementException ex )
			{
				throw new IllegalArgumentException( "Line " + lineNumber + ": malformed line!" );
			}
			finally
			{
				Util.closeQuietly( scanner );
			}
		}
		
		return mappings;
	}

	private ArgumentMapping readArgument( Scanner scanner )
	{
		return new ArgumentMapping( scanner.nextInt(), scanner.next() );
	}

	private ClassMapping readClass( Scanner scanner )
	{
		return new ClassMapping( scanner.next(), scanner.next() );
	}
	
	private FieldMapping readField( Scanner scanner )
	{
		return new FieldMapping( scanner.next(), scanner.next() );
	}
	
	private MethodMapping readMethod( Scanner scanner )
	{
		return new MethodMapping( scanner.next(), scanner.next(), scanner.next(), scanner.next() );
	}
}