DMDirc-Client/src/dmdirc/parser/IRCParser.java

/*
 * Copyright (c) 2006-2007 Chris Smith, Shane Mc Cormack
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell 
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in 
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR 
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, 
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE 
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER 
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 */

package dmdirc.parser;

import java.io.*;
import java.net.*;
import java.util.ArrayList;

public class IRCParser implements Runnable {

	class MyInfo {
		String sNickname = "IRCParser";
		String sRealname = "Java Test IRCParser";
		String sUsername = "IRCParser";
	}

	private Socket socket = null;
	private PrintWriter out = null;
	private BufferedReader in = null;

	public MyInfo me = new MyInfo();

	private boolean HasBegan = false;
	private boolean IsFirst = true;

	// Events
	public interface IMOTDEnd { public void onMOTDEnd(IRCParser tParser); }
	public interface IDataIn { public void onDataIn(IRCParser tParser, String sData); }
	public interface IDataOut { public void onDataOut(IRCParser tParser, String sData); }
	class AllEvents {
		ArrayList<IMOTDEnd> EndOfMOTD = new ArrayList<IMOTDEnd>();
		ArrayList<IDataIn> DataIn = new ArrayList<IDataIn>();
		ArrayList<IDataOut> DataOut = new ArrayList<IDataOut>();
	}
	public AllEvents cb = new AllEvents();

	public void AddMOTDEnd(Object eMethod) { cb.EndOfMOTD.add((IMOTDEnd)eMethod); }
	public void DelMOTDEnd(Object eMethod) { 
		for (int i = 0; i < cb.EndOfMOTD.size(); i++) {
			if (eMethod.equals((Object)cb.EndOfMOTD.get(i))) { cb.EndOfMOTD.remove(i); break; }
		}
	}
	private void CallMOTDEnd() { 
		for (int i = 0; i < cb.EndOfMOTD.size(); i++) {
			// IMOTDEnd(cb.EndOfMOTD.get(i)).OnMOTDEnd(this);
			cb.EndOfMOTD.get(i).onMOTDEnd(this);
		}
	}

	public void AddDataIn(Object eMethod) { cb.DataIn.add((IDataIn)eMethod); }
	public void DelDataIn(Object eMethod) { 
		for (int i = 0; i < cb.DataIn.size(); i++) {
			if (eMethod.equals((Object)cb.DataIn.get(i))) { cb.DataIn.remove(i); break; }
		}
	}
	private void CallDataIn(String data) { 
		for (int i = 0; i < cb.DataIn.size(); i++) {
			cb.DataIn.get(i).onDataIn(this, data);
		}
	}

	public void AddDataOut(Object eMethod) { cb.DataOut.add((IDataOut)eMethod); }
	public void DelDataOut(Object eMethod) { 
		for (int i = 0; i < cb.DataOut.size(); i++) {
			if (eMethod.equals((Object)cb.DataOut.get(i))) { cb.DataOut.remove(i); break; }
		}
	}
	private void CallDataOut(String data) { 
		for (int i = 0; i < cb.DataOut.size(); i++) {
			cb.DataOut.get(i).onDataOut(this, data);
		}
	}

	// Constructor.
	IRCParser () { }

	public void connect(String sHost) throws Exception {
		try {
			connect(sHost,6667);
		} catch (Exception e) {
			throw e;
		}
	}

	public void connect(String sHost, int nPort) throws Exception {
		if (HasBegan) { return; }
		try {
			socket = new Socket(sHost,nPort);
			out = new PrintWriter(socket.getOutputStream(), true);
			in = new BufferedReader(new InputStreamReader(socket.getInputStream()));
		} catch (UnknownHostException e) {
			throw new Exception("Socket Exception");
		} catch (Exception e) {
			throw new Exception("General Exception");
		}
	}

	public void run() {
		if (HasBegan) { return; } else { HasBegan = true; }
		// :HACK: While true loops really really suck.
		while(true){
			String line = "";
			try {
				line = in.readLine(); // Blocking :/
				if (IsFirst) {
					SendString("NICK "+me.sNickname);
					SendString("USER "+me.sUsername.toLowerCase()+" * * :"+me.sRealname);
					IsFirst = false;
				}
		        ProcessLine(line);
			} catch (IOException e) {
				System.out.println("Socket read failed");
				System.exit(-1);
			}
		}
	}

	protected void finalize(){
		try{
			socket.close();
		} catch (IOException e) {
			System.out.println("Could not close socket");
			System.exit(-1);
		}
	}

	private String GetParam(String line) {
		String[] params = null;
		params = line.split(" :",2);
		return params[params.length-1];
	}

	private String[] IRCTokenise(String line) {
		String[] params = null;
		String[] tokens = null;
		params = line.split(" :",2);
		tokens = params[0].split(" ");
	
		String[] temp = new String[tokens.length+1];
		System.arraycopy(tokens, 0, temp, 0, tokens.length);
		tokens = temp;
		if (params.length == 2) { tokens[tokens.length-1] = params[1]; }

	    return tokens;
	}

	public void SendLine(String line) {SendString(line);} // This should do some checks on stuff possible, public event!
	
	// Our Method
	private void SendString(String line) {
		CallDataOut(line);
		out.printf("%s\r\n",line);
	}

	private void ProcessLine(String line) {
		String[] token = IRCTokenise(line);
		String sParam = token[token.length-1];

		int nParam;
		CallDataIn(line);

		try {nParam = Integer.parseInt(token[1]);} catch (Exception e) { nParam = -1;}

		if (token[0].equals("PING") || token[1].equals("PING")) { SendString("PONG :"+sParam); }
		else {
			if (token[0].substring(0,1).equals(":")) {
				// Post Connect
				switch (nParam) {
					case -1:
						ProcessStringParam(sParam,token);
						break;
					case 1: // 001
						break;
					case 422: // No MOTD
					case 376: // End of MOTD
						ProcessEndOfMOTD(sParam,token);
						break;
					default: // Unknown
						break;
				}
			} else {
				// Pre Connect
			}
		}
	}

	private void ProcessStringParam(String sParam,String token[]) {
		// Process a line where the parameter is a string (IE PRIVMSG, NOTICE etc - Not including PING!)
	}

	private void ProcessEndOfMOTD(String sParam,String token[]) {
		// Process EndOfMOTD
		CallMOTDEnd();
	}



	//-------------------------------------------------------------------------
	public void JoinChannel(String sChannelName) {
		SendLine("JOIN "+sChannelName);
	}
}