z80-tools/src/nl/grauw/glass/Assembler.java

package nl.grauw.glass;

import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.io.PrintStream;
import java.util.ArrayList;
import java.util.List;

public class Assembler {
	
	private static Assembler instance;
	private final Source source;
	
	/**
	 * @param args
	 */
	public static void main(String[] args) {
		if (args.length == 0) {
			System.out.println("Usage: java -jar glass.jar [OPTION] SOURCE [OBJECT] [SYMBOL]");
			System.exit(1);
		}
		
		File sourcePath = null;
		File objectPath = null;
		File symbolPath = null;
		List<File> includePaths = new ArrayList<File>();
		for (int i = 0; i < args.length; i++) {
			if (args[i].equals("-I") && (i + 1) < args.length) {
				includePaths.add(new File(args[++i]));
			} else if (sourcePath == null) {
				sourcePath = new File(args[i]);
			} else if (objectPath == null) {
				objectPath = new File(args[i]);
			} else if (symbolPath == null) {
				symbolPath = new File(args[i]);
			} else {
				throw new AssemblyException("Too many arguments.");
			}
		}
		
		instance = new Assembler(sourcePath, includePaths);
		instance.writeObject(objectPath);
		if (symbolPath != null)
			instance.writeSymbols(symbolPath);
	}
	
	public Assembler(File sourcePath, List<File> includePaths) {
		source = new SourceBuilder(includePaths).parse(sourcePath);
	}
	
	public void writeObject(File objectPath) {
		try (OutputStream output = objectPath != null ? new FileOutputStream(objectPath) : new NullOutputStream()) {
			source.assemble(output);
		} catch (IOException e) {
			throw new RuntimeException(e);
		}
	}
	
	public void writeSymbols(File symbolPath) {
		try (PrintStream symbolOutput = new PrintStream(symbolPath)) {
			symbolOutput.print(source.getScope().serializeSymbols());
		} catch (IOException e) {
			throw new RuntimeException(e);
		}
	}
	
	public static class NullOutputStream extends OutputStream {
		public void write(int b) throws IOException {}
		public void write(byte[] b) throws IOException {}
		public void write(byte[] b, int off, int len) throws IOException {}
	}
	
}