Imanol-Mikel Barba Sabariego / upcnews · Files

package com.upc.pbe.upcnews;

import java.net.MalformedURLException;
import java.util.ArrayList;

import android.app.Activity;
import android.widget.Toast;

public class Parser
{
	private static final String STARTWORD = "#EXTM3U";
	private String path; // CWD, completes implicit paths such as "file.m3u8"
							// appending the CWD in order to download the file
	private int fileType; // indicates if segment list or media list
	private int currentLine;
	private int currentSegment; // Cada segmento de cada calidad distinta
	private int currentList; // Indice de cada recurso distinto
	private boolean expectSegment = false;
	private boolean expectList = false;
	private Descarrega download;
	private Activity caller; // Where to send notifications and warnings
	private boolean validated; // Wether if list starts with #EXTM3U or not

	public Parser(String p, Activity caller)
	{
		this.caller = caller;
		path = p;
		currentLine = 0;
		currentSegment = 0;
		currentList = 0;
		fileType = -1;
		/*
		 * -1 indicates UNDETERMINED 0 indicates SEGMENTS 1 indicates MEDIA
		 */
		download = new Descarrega();
		validated = false;
	}

	public ArrayList<ParentList> parseFile(String file) throws ErrorException, WarningException, InfoException
	{
		ArrayList<ParentList> lists = new ArrayList<ParentList>();
		lists.add(new ParentList(""));
		lists.get(0).getLists().add(new Video(-1));
		String[] lines = file.split("\n");
		for (int i = 0; i < lines.length; i++)
		{
			if (lines[i].endsWith("\\"))
			{
				/*
				 * Merge multiline entries, which are those lines ending with
				 * backslash.
				 */
				lines[++i] = lines[i - 1].substring(0, lines[i - 1].indexOf("\\") - 1)
						+ lines[i];
			}
			try
			{
				parseLine(lines[i], lists);
			}
			catch (InfoException iE)
			{
				Toast.makeText(caller, iE.getMessage(), Toast.LENGTH_SHORT).show();
			}
			catch (WarningException wE)
			{
				Toast.makeText(caller, wE.getMessage(), Toast.LENGTH_SHORT).show();
			}
		}
		/*
		 * Esto contiene una lista de RECURSOS, que a su vez contiene una lista
		 * de CALIDADES. Si no te gusta mis VERSOS, te mando a la mierda sin
		 * SUTILIDADES.
		 * 
		 * -- Imanol, hasta las cejas de cafeína.
		 * 
		 * PD: En el último nivel de profundidad hay una lista de segmentos
		 * (debajo de calidades). Allí está el auténtico tesoro de Narnia.
		 */
		for (int i = 0; i < lists.size(); i++)
		{
			sortQuality(lists.get(i));
		}
		return lists;
	}

	private int searchID(String ID, ArrayList<ParentList> lists)
	{
		for (int i = 0; i < lists.size(); i++)
		{
			if (lists.get(i).getID().equals(ID))
			{
				return i;
			}
		}
		return -1;
	}

	private void sortQuality(ParentList ppls)
	{
		ArrayList<Video> lists = ppls.getLists();
		if (lists.get(0).getQuality() == -1)
		{
			/*
			 * That's the case of Media lists and Segment lists, they don't have
			 * distinct qualities.
			 */
			return;
		}
		// Bubblesort
		while (true)
		{
			boolean sorted = true;
			int i = 0;
			do
			{
				if (lists.get(i).getQuality() < lists.get(i + 1).getQuality())
				{
					sorted = false;
					Video aux = lists.get(i);
					lists.set(i, lists.get(i + 1));
					lists.set(i + 1, aux);
				}

			}
			while (++i != lists.size() - 1);
			if (sorted)
			{
				break;
			}
		}

	}

	public void parseLine(String line, ArrayList<ParentList> lists) throws ErrorException, WarningException, InfoException
	{
		if (line.isEmpty())
		{
			currentLine++;
			return;
		}
		else
		{
			ParentList ppls = lists.get(currentList);
			Video pls = ppls.getLists().get(ppls.getLists().size() - 1);
			if (!validated)
			{
				if (line.equals(STARTWORD))
				{
					validated = true;
				}
				else
				{
					throw new ErrorException("Playlist is not valid");
				}
			}
			else
			{
				if (line.charAt(0) == '#')
				{
					if (line.substring(0, 4).equals("#EXT"))
					{
						if (expectSegment && !line.startsWith("#EXT-X-BYTERANGE"))
						{
							throw new ErrorException("Expected segment URI, got an EXT-TAG instead");
						}
						if (expectList)
						{
							throw new ErrorException("Expected List, got an EXT-TAG instead");
						}
						String[] extTag = line.split(":", 2);
						if (extTag[0].equals("#EXT-X-MEDIA-SEQUENCE"))
						{
							pls.setSequence(Integer.parseInt(extTag[1]));
						}
						else if (extTag[0].equals("#EXT-X-TARGETDURATION"))
						{
							pls.setMaxDuration(Integer.parseInt(extTag[1]));
						}
						else if (extTag[0].equals("#EXTINF"))
						{
							if (fileType == -1)
							{
								fileType = 0;
							}
							else if (fileType == 1)
							{
								throw new ErrorException("Invalid file, should contain Lists, "
										+ "but segments were found.");
							}
							String[] args = extTag[1].split(",");
							float duration = Float.parseFloat(args[0]);
							if (duration > pls.getMaxDuration())
							{
								throw new ErrorException("Segment "
										+ currentSegment + 1 + " on line "
										+ currentLine + " exceeds max duration");
							}
							Segment s = new Segment(duration);
							if (args.length == 1)
							{
								s.setName("");
							}
							else
							{
								s.setName(args[1]);
							}
							pls.getSegments().add(s);
							expectSegment = true;
						}

						else if (extTag[0].equals("#EXT-X-STREAM-INF"))
						{
							if (fileType == -1)
							{
								fileType = 1;
							}
							else if (fileType == 0)
							{
								throw new ErrorException("Invalid file, should contain Segments, "
										+ "but lists were found.");
							}
							expectList = true;
							String programID = "";
							double bandwidth = -1;
							String[] arguments = extTag[1].split(",");
							for (int i = 0; i < arguments.length; i++)
							{
								String[] argument = arguments[i].split("=");
								/*
								 * In case current argument contains several
								 * values like "val1,val2,val3" detect the "
								 * character and merge them all in one single
								 * string
								 */
								if (argument[1].startsWith("\"")
										&& !argument[1].endsWith("\""))
								{
									int j = i;
									while (true)
									{
										if (i == arguments.length - 1)
										{
											throw new ErrorException("Malformed argument on line "
													+ currentLine);
										}
										if (arguments[++i].endsWith("\""))
										{
											String end = arguments[i];
											for (int k = j + 1; k < i; k++)
											{
												argument[1] += arguments[k];
											}
											argument[1] += end;
											break;
										}
									}
								}
								if (argument[0].equals("PROGRAM-ID"))
								{
									programID = argument[1];
								}
								else if (argument[0].equals("BANDWIDTH"))
								{
									bandwidth = Double.parseDouble(argument[1]);
								}
							}
							if (programID.equals("") || bandwidth == -1)
							{
								throw new ErrorException("Playlist on line "
										+ currentLine
										+ " has missing arguments");
							}
							if (!programID.equals(ppls.getID()))
							{
								/*
								 * If Program-ID doesn't match, search it or
								 * create a new one.
								 */
								if (ppls.getID().equals(""))
								{
									/*
									 * If working with the default ParentList,
									 * then modify it, don't create a new one.
									 */
									lists.get(0).setID(programID);
								}
								else
								{
									currentList = searchID(programID, lists);
									if (currentList == -1)
									{
										/*
										 * ParentList not found, creating a new
										 * one.
										 */
										lists.add(new ParentList(programID));
										currentList = lists.size() - 1;
									}
									ppls = lists.get(currentList);
									pls = ppls.getLists().get(ppls.getLists().size() - 1);
								}
							}
							/*
							 * Now we create a new List inside the correct
							 * ParentList.
							 */
							if ((ppls.getLists().size() == 1)
									&& (pls.getQuality() == -1)
									&& pls.getSegments().isEmpty())
							{
								/*
								 * If working with the default List, then modify
								 * it, don't create a new one.
								 */
								pls.setQuality(bandwidth);
							}
							else
							{
								ppls.getLists().add(new Video(bandwidth));
							}
						}
						else if (extTag[0].equals("#EXT-X-MEDIA"))
						{
							if (fileType == -1)
							{
								fileType = 1;
							}
							else if (fileType == 0)
							{
								throw new ErrorException("Invalid file, should contain Segments, "
										+ "but lists were found.");
							}
							String Type = "";
							String Name = "";
							String GroupID = "";
							String URI = "";
							boolean Default = false;
							String[] arguments = extTag[1].split(",");
							for (int i = 0; i < arguments.length; i++)
							{
								String[] argument = arguments[i].split("=");
								/*
								 * In case current argument contains several
								 * values like "val1,val2,val3" detect the "
								 * character and merge them all in one single
								 * string
								 */
								if (argument[1].startsWith("\"")
										&& !argument[1].endsWith("\""))
								{
									int j = i;
									while (true)
									{
										if (i == arguments.length - 1)
										{
											throw new ErrorException("Malformed argument on line "
													+ currentLine);
										}
										if (arguments[++i].endsWith("\""))
										{
											String end = arguments[i];
											for (int k = j + 1; k < i; k++)
											{
												argument[1] += arguments[k];
											}
											argument[1] += end;
											break;
										}
									}
								}
								if (argument[0].equals("NAME"))
								{
									Name = argument[1].substring(1, argument[1].length() - 1);
								}
								else if (argument[0].equals("TYPE"))
								{
									Type = argument[1];
								}
								else if (argument[0].equals("GROUP-ID"))
								{
									GroupID = argument[1].substring(1, argument[1].length() - 1);
								}
								else if (argument[0].equals("DEFAULT"))
								{
									if (argument[1].equals("YES"))
									{
										Default = true;
									}
									else if (argument[1].equals("NO"))
									{
										Default = true;
									}
									else
									{
										throw new ErrorException("Invalid value for argument DEFAULT on line "
												+ currentLine);
									}
								}
								else if (argument[0].equals("URI"))
								{
									if (!argument[1].contains("http://"))
									{
										/*
										 * In case the path read is RELATIVE
										 */
										URI = path
												+ argument[1].substring(1, argument[1].length() - 1);
									}
									else
									{
										URI = argument[1].substring(1, argument[1].length() - 1);
									}
								}
							}
							if (Type.equals("") || Name.equals("")
									|| URI.equals("") || GroupID.equals(""))
							{
								throw new ErrorException("Playlist on line "
										+ currentLine
										+ " has missing arguments");
							}
							Parser p = new Parser(URI.substring(0, URI.lastIndexOf("/") + 1), caller);
							Video newList = p.parseFile(download.doInBackground(URI)).get(0).getLists().get(0);
							if (!GroupID.equals(ppls.getID()))
							{
								if (ppls.getID().equals(""))
								{
									/*
									 * If working with the default ParentList,
									 * then modify it, don't create a new one.
									 */
									ppls.setDefault(Default);
									ppls.setName(Name);
									ppls.setType(Type);
									ppls.getLists().set(0, newList);
									ppls.setID(GroupID);
								}
								else
								{

									currentList = searchID(GroupID, lists);
									if (currentList == -1)
									{
										ParentList pl = new ParentList(GroupID);
										pl.setDefault(Default);
										pl.setName(Name);
										pl.setType(Type);
										lists.add(pl);
										currentList = lists.size() - 1;
									}
									ppls = lists.get(currentList);
									ppls.getLists().add(newList);
								}
							}
							else
							{
								ppls.getLists().add(newList);
							}
						}

						else if (extTag[0].equals("#EXT-X-ENDLIST"))
						{
							if (expectSegment || expectList)
							{
								throw new ErrorException("Unexpected end of list!");
							}
						}

						else if (extTag[0].equals("#EXT-X-VERSION"))
						{
							// IGNORE
						}

						else if (extTag[0].equals("#EXT-X-PLAYLIST-TYPE"))
						{
							// IGNORE
						}

						else
						{
							currentLine++;
							throw new WarningException("Tag not implemented: "
									+ extTag[0]);
						}
					}
					else
					{
						String comment = line.substring(1);
						throw new InfoException("Comment on line "
								+ currentLine++ + " " + comment);
					}
				}
				else
				{
					if (expectSegment)
					{
						expectSegment = false;
						try
						{
							pls.getSegments().get(currentSegment).setURL(line);
						}
						catch (MalformedURLException e)
						{
							throw new ErrorException(e.getMessage());
						}
						currentSegment++;
					}
					else if (expectList)
					{
						/*
						 * Download list and merge the entries with the current
						 * list
						 */
						expectList = false;
						Parser p = new Parser(line.substring(0, line.lastIndexOf("/") + 1), caller);
						Video newList = p.parseFile(download.doInBackground(line)).get(0).getLists().get(0);
						pls.setMaxDuration(newList.getMaxDuration());
						pls.setSequence(newList.getSequence());
						for (int i = 0; i < newList.getSegments().size(); i++)
						{
							pls.getSegments().add(newList.getSegments().get(i));
						}
					}
					else
					{
						throw new ErrorException("ERROR: Unexpected string "
								+ line);
					}
				}
			}
			currentLine++;
		}
	}
}