//Author: ^-^Veerle^-^
import java.sql.ResultSet;
import java.io.BufferedWriter;
import java.io.FileWriter;
import java.io.IOException;
import javax.swing.JOptionPane;
import java.util.Calendar;
import java.text.DecimalFormat;

public class databaseOutput
{
	
	public static void TXT (ResultSet rsList, String strFile) throws Exception
	{
		
		strFile = removeExtension(strFile, ".txt");
		
		try
		{
	
			BufferedWriter createDocument = new BufferedWriter(new FileWriter(strFile + ".txt"));
			String strLine = "";
			int counter = 1;
			
			strLine = "Records from table: " + strFile;
			writeLine(strLine, createDocument);
			strLine = getTimeStamp();
			writeLine(strLine, createDocument);
			createDocument.newLine();
			
			while (rsList.next())
			{
				
				strLine = "(" + counter + ") " + rsList.getString(1) + " [" + rsList.getInt(2) + "]";
				writeLine(strLine, createDocument);
				counter++;
				
			}
			
			createDocument.close();
			JOptionPane.showMessageDialog(null, "Records written successfully to\n" + System.getProperty("user.dir") + "\\" + strFile + ".txt", "Save", JOptionPane.INFORMATION_MESSAGE);
			
		}
		catch (IOException ex)
		{
			throw new Exception("Error writing file " + System.getProperty("user.dir") + "\\" + strFile + ".txt\nPlease check file permissions");
		}
		
	}
	
	public static void XML (ResultSet rsList, String strFile) throws Exception
	{
		
		strFile = removeExtension(strFile, ".xml");
		
		try
		{
			
			BufferedWriter createDocument = new BufferedWriter(new FileWriter(strFile + ".xml"));
			String strLine = "";
			int counter = 1;
			
			strLine = "<?xml version=\"1.0\" encoding=\"ISO-8859-1\"?>";
			writeLine(strLine, createDocument);
			createDocument.newLine();
			strLine = "<!DOCTYPE conversions[";
			writeLine(strLine, createDocument);
			createDocument.newLine();
			strLine = "\t<!ELEMENT " + strFile + " (title, year)>";
			writeLine(strLine, createDocument);
			strLine = "\t\t<!ELEMENT title (#PCDATA)>";
			writeLine(strLine, createDocument);
			strLine = "\t\t<!ELEMENT year (#PCDATA)>";
			writeLine(strLine, createDocument);
			createDocument.newLine();
			strLine = "]>";
			writeLine(strLine, createDocument);
			createDocument.newLine();
			
			strLine = "<!-- " + getTimeStamp() + " -->";
			writeLine(strLine, createDocument);
			createDocument.newLine();
			strLine = "<" + strFile + ">";
			writeLine(strLine, createDocument);
			createDocument.newLine();
			
			while (rsList.next())
			{
				
				strLine = "\t<index>" + counter + "</index>";
				writeLine(strLine, createDocument);
				strLine = "\t<title>" + rsList.getString(1) + "</title>";
				writeLine(strLine, createDocument);
				strLine = "\t<year>" + rsList.getInt(2) + "</year>";
				writeLine(strLine, createDocument);
				createDocument.newLine();
				counter++;
				
			}
			
			strLine = "</" + strFile + ">";
			writeLine(strLine, createDocument);
			
			createDocument.close();
			JOptionPane.showMessageDialog(null, "Records written successfully to\n" + System.getProperty("user.dir") + "\\" + strFile + ".xml", "Save", JOptionPane.INFORMATION_MESSAGE);
			
		}
		catch (IOException ex)
		{
			throw new Exception("Error writing file " + System.getProperty("user.dir") + "\\" + strFile + ".xml\nPlease check file permissions");
		}			
		
	}
	
	private static void writeLine (String strLine, BufferedWriter createDocument) throws IOException
	{
		
		createDocument.write(strLine, 0, strLine.length());
		createDocument.newLine();
		
	}
	
	private static String removeExtension (String strFile, String strExt)
	{
		
		if (strFile.toLowerCase().endsWith(strExt.toLowerCase()))
		{
			strFile = strFile.substring(0, strFile.length() - 4);
		}
		
		return strFile;
		
	}
	
	private static String getTimeStamp ()
	{
		
		Calendar calNow = Calendar.getInstance();
		DecimalFormat decFormat = new DecimalFormat("00");
		return "Last Updated " + decFormat.format(calNow.get(Calendar.DAY_OF_MONTH)) + "\\" + decFormat.format((calNow.get(Calendar.MONTH) + 1)) + "\\" + calNow.get(Calendar.YEAR) + " at " + decFormat.format(calNow.get(Calendar.HOUR_OF_DAY)) + ":" + decFormat.format(calNow.get(Calendar.MINUTE));
			
	}
		
		

}