import java.io.*;
import java.sql.SQLException;
import java.util.Hashtable;
import java.util.Vector;

import org.w3c.dom.*;
import org.xml.sax.*;
import javax.xml.parsers.*;


public class parser {

	public PrintWriter out = null;
	public Hashtable<String, String> conversionTable;
	public Vector<String> referencesMemorizeKey;
	public Vector<String> referencesMemorizeValue;
	
	public parser()
	{
		conversionTable = new Hashtable<String, String>();
		referencesMemorizeKey = new Vector<String>();
		referencesMemorizeValue = new Vector<String>();
	}

	//Rem changer Package par ModelElement
	public void parse(String _fichier)
		throws SAXException, ParserConfigurationException, IOException {

		// Charger le document
		FileInputStream _xml_input_file = new FileInputStream(_fichier);
		
		int mode = 1;
		parse(_xml_input_file, mode);
		mode ++;
		FileInputStream _xml_input_file2 = new FileInputStream(_fichier);
		parse(_xml_input_file2, mode);

		mode ++;
		FileInputStream _xml_input_file3 = new FileInputStream(_fichier);
		parse(_xml_input_file3, mode);
		
		mode = 7;
		FileInputStream _xml_input_file4 = new FileInputStream(_fichier);
		parse(_xml_input_file4, mode);
	}
	
	public PrintWriter createFichier(int mode, Element eltClass)
	{
		String filename;
		String comment;
		
			switch (mode){
		
			case 1 : filename = "abstract.java"; 
			 comment = "//contains all the abstract classes";
			break;
	
			case 2 : filename = "plain.java"; 
			 comment = "//contains all the plain classes";
			break;
			
			case 3 : filename = "jdbc.java"; 
			 comment = "//contains the classes connected to the DB";
			break;
			
			case 4 : filename = "Abs"+eltClass.getAttribute("name") + ".java";
				
				comment = "//contains the abstract methods and attributes of "+  eltClass.getAttribute("name") ;
			 break;
			 
			case 5 : filename = eltClass.getAttribute("name") + ".java";
			
			comment = "//contains the methods and attributes of "+  eltClass.getAttribute("name") ;
			break;
		 
			case 6 : filename =  eltClass.getAttribute("name") + "Persistent.java";
				comment = "//contains the persistent methods and attributes of "+  eltClass.getAttribute("name") ;
			break;
			
			case 7 : filename = eltClass.getAttribute("name") + "table.sql";
				comment = "-- The creation of table " + eltClass.getAttribute("name");
			break;
				
			default : filename = ".error"; 
			 comment = "//errors";
			break;
						}
		
		
		try {
			out = new PrintWriter( 
			new FileOutputStream("./"+filename));
		} catch (FileNotFoundException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		}
		
		
		if (mode != 7) {
			out.println("package SaveToJDBC;");
			out.println("import java.util.Collection;");
			out.println("import java.sql.*;");
			out.println("import java.util.Vector;");
			out.println("");
			out.println(comment);
		}
		
		return out;
	}

	public void parse(InputStream _xml_input_file,int mode)
		throws SAXException, ParserConfigurationException, IOException {

		//création du fichier output
		Element bidon = null ;
		
		out = createFichier(0, bidon);
		
		//instancie le constructeur de parseurs
		DocumentBuilderFactory _factory = DocumentBuilderFactory.newInstance();

		//ignore les commentaires dans les fichiers XML parsés
		_factory.setIgnoringComments(true);
		
		// crée un parseur
		DocumentBuilder _builder = _factory.newDocumentBuilder();

		// Charge le document
		Document doc = _builder.parse(_xml_input_file);

		// Parse le document
		Element root = doc.getDocumentElement();
		
		// Liste des enfants de la root
		NodeList classes = root.getChildNodes();
		handleModel(classes, mode);
		
		out.flush();
		out.close();
		
	}
	

	public void handleAttribute(Element eltAttr, Element eltClass, int mode)
	{
		
		switch(mode){
		
		case 1 :
		if(eltAttr.hasAttribute("association"))
		{
			out.println("	public abstract Collection get"+conversionTable.get(eltAttr.getAttribute("type"))+"();");
			out.println("	public abstract void set"+conversionTable.get(eltAttr.getAttribute("type"))+"( Collection "+eltAttr.getAttribute("name")+" );");
			out.println("");
		}
		else
		{
			out.println("	public abstract "+ conversionTable.get(eltAttr.getAttribute("type"))+" get"+ eltAttr.getAttribute("name")+"();");
			out.println("	public abstract void set"+ eltAttr.getAttribute("name") + "( "+conversionTable.get(eltAttr.getAttribute("type"))+ " " +eltAttr.getAttribute("name")+" );");
			out.println("");
		}
		break;
			
		case 2 :
		
		if(eltAttr.hasAttribute("association") && eltAttr.getAttribute("upperValue").equals("-1"))
		{
			out.println("	private  Collection <"+conversionTable.get(eltAttr.getAttribute("type"))+"> " + eltAttr.getAttribute("name") + ";");
			out.println("");
			
			out.println("	public  Collection <"+ conversionTable.get(eltAttr.getAttribute("type"))+"> get"+ eltAttr.getAttribute("name")+"()");
			out.println("	{return " +eltAttr.getAttribute("name")+ ";}");
			out.println("");
			
			out.println("	public void set"+ eltAttr.getAttribute("name") + "( Collection <"+conversionTable.get(eltAttr.getAttribute("type"))+ "> " +eltAttr.getAttribute("name")+" )");
			out.println("	{this." +eltAttr.getAttribute("name")+ " = "+ eltAttr.getAttribute("name") +";}");
			out.println("");
		}
		else
		{
			out.println("	private "+ conversionTable.get(eltAttr.getAttribute("type"))+" "+ eltAttr.getAttribute("name")+";");
			out.println("");
			
			out.println("	public "+ conversionTable.get(eltAttr.getAttribute("type"))+" get"+ eltAttr.getAttribute("name")+"()");
			out.println("	{return " +eltAttr.getAttribute("name")+ ";}");
			out.println("");
			
			out.println("	public void set"+ eltAttr.getAttribute("name") + "( "+conversionTable.get(eltAttr.getAttribute("type"))+ " " +eltAttr.getAttribute("name")+" )");
			out.println("	{this." +eltAttr.getAttribute("name")+ " = "+ eltAttr.getAttribute("name") +";}");
			out.println("");
		}
		break;
		
		case 3 :
			

	
			if(eltAttr.hasAttribute("association") && eltAttr.getAttribute("upperValue").equals("-1"))
		{
			out.println("	private  Collection <"+conversionTable.get(eltAttr.getAttribute("type"))+"Persistent> " + eltAttr.getAttribute("name") + ";");
			out.println("");
			
			out.println("	public  Collection <"+ conversionTable.get(eltAttr.getAttribute("type"))+"> get"+ eltAttr.getAttribute("name")+"() throws SQLException");
			out.println("	{String sqlRequest = \"select "+conversionTable.get(eltAttr.getAttribute("type"))+".id from " +conversionTable.get(eltAttr.getAttribute("type")) + " where id"+ eltClass.getAttribute("name")+" =\"  +this.id ; ");
			
			out.println("	 	ResultSet rs =QueryUnivDB.getQueryCollection(sqlRequest); ");
			out.println("    	Vector retVector = new Vector();");
			out.println("    	while(rs.next()){");
			out.println("   		 retVector.add(new "+conversionTable.get(eltAttr.getAttribute("type"))+"Persistent(rs.getString(1)));}");
			out.println("    	return retVector;");
			
			
			out.println("	}");
			out.println("");
			
			out.println("	public void set"+ eltAttr.getAttribute("name") + "( Collection <"+conversionTable.get(eltAttr.getAttribute("type"))+ "Persistent> " +eltAttr.getAttribute("name")+" )");
			out.println("	{for ("+conversionTable.get(eltAttr.getAttribute("type"))+"Persistent e :"+ eltAttr.getAttribute("name")+") {");
			out.println("		String sqlRequest = \"update "+conversionTable.get(eltAttr.getAttribute("type"))+" set id"+ eltClass.getAttribute("name")+" =\" + this.id +\" where id= \"+e.getId() ;");
			out.println("		QueryUnivDB.setQuery(sqlRequest);");
			
			out.println("		}");
			
			out.println("	this." +eltAttr.getAttribute("name")+ " = "+ eltAttr.getAttribute("name") +";");
			out.println("	}");
			out.println("");
		}
		else
		{
			out.println("	private "+ conversionTable.get(eltAttr.getAttribute("type"))+" "+ eltAttr.getAttribute("name")+";");
			out.println("");
			
			out.println("	public "+ conversionTable.get(eltAttr.getAttribute("type"))+" get"+ eltAttr.getAttribute("name")+"()");
			out.println("	{String sqlRequest = \"select "+eltAttr.getAttribute("name")+" from " + eltClass.getAttribute("name")+ " where id =\"  +this.id ; ");
			out.println("	return ("+conversionTable.get(eltAttr.getAttribute("type")) +") QueryUnivDB.getQuery(sqlRequest);	");
			
			out.println("	}");
			out.println("");
			
			out.println("	public void set"+ eltAttr.getAttribute("name") + "( "+conversionTable.get(eltAttr.getAttribute("type"))+ " " +eltAttr.getAttribute("name")+" )");	
			out.println("	{String sqlRequest = \"update "+eltClass.getAttribute("name")+" set " + eltAttr.getAttribute("name")+ "='\"+"+ eltAttr.getAttribute("name") +"+\"' where id =\"  +this.id ; ");
			out.println("	QueryUnivDB.setQuery(sqlRequest);	");
			out.println("	this." +eltAttr.getAttribute("name")+ " = "+ eltAttr.getAttribute("name") +";");
			
			out.println("	}");
			out.println("");
		}
		break;
		
		case 7 :	
			if(!(eltAttr.hasAttribute("association")&& eltAttr.getAttribute("upperValue").equals("-1")))
				out.print(",\n\t" + eltAttr.getAttribute("name") + " " + conversionTable.get(eltAttr.getAttribute("type")));
			break;
		
		case 8 :
			if(eltAttr.hasAttribute("association") && eltAttr.getAttribute("upperValue").equals("-1"))
				{
					referencesMemorizeKey.add(eltAttr.getAttribute("type"));
					referencesMemorizeValue.add(eltClass.getAttribute("name"));
				}
			break;
		case 9 :
			for(int i = 0; i<referencesMemorizeKey.size();i++)
			{
				if(referencesMemorizeKey.contains(eltClass.getAttribute("xmi:id"))) {
					out.println(",\n\tid"+referencesMemorizeValue.elementAt(i)+" int,");
					out.print("\tCONSTRAINT fk_" + referencesMemorizeValue.elementAt(i)+" FOREIGN KEY (id" + referencesMemorizeValue.elementAt(i) + ") REFERENCES " + referencesMemorizeValue.elementAt(i)+" (id)");
				}
						
				
			}
		
			
		}
		
		
	}
	public void handlePrimitive(Element eltClass)
	{
		//conversionTable.put(eltClass.getAttribute("xmi:id"), eltClass.getAttribute("name"));
	}
	
	public void handleClass(Element eltClass,int  mode)
	{
		//on enregistre le type de la classe dans la hashtable
		String name = eltClass.getAttribute("name");
		if (mode != 7) {
			if (name.equals("int")) name = "Integer";
		}
		
		conversionTable.put(eltClass.getAttribute("xmi:id"), name);
		 
		//on crée l'appel de la classe
		
		switch (mode){
		
		case 1 :
			out.flush();
			out.close();
			out = createFichier(4, eltClass);
			out.println("public interface Abs" + eltClass.getAttribute("name") + "{"); 
		break;

		case 2 :  
			out.flush();
			out.close();
			out = createFichier(5, eltClass);
			out.println("public class " + eltClass.getAttribute("name")+"{"); 
		break;
		

		case 3 :  
			out.flush();
			out.close();
			out = createFichier(6, eltClass);
			out.println("public class " + eltClass.getAttribute("name") + "Persistent {\n"); 
					out.println("	static int nextId = 0;");
					out.println("	int id ;");
					out.println(" 	" + eltClass.getAttribute("name")+"Persistent(){");
					out.println("		id = ++nextId;");
					out.println("		String sqlRequest =\"insert into "+eltClass.getAttribute("name")+"(id) values(\"+id+\")\";");
					out.println("		QueryUnivDB.setQuery(sqlRequest);");
					out.println("	}\n");
				out.println(" 	" + eltClass.getAttribute("name")+"Persistent(String id){");
				out.println("		this.id = Integer.parseInt(id);");
				out.println("	}");
				out.println("\n	public int getId(){");
				out.println("		return this.id;");
				out.println("	}\n");
					
			
		break;
		
		case 7 :
			out.flush();
			out.close();
			out = createFichier(7, eltClass);

			out.println("drop table " + eltClass.getAttribute("name") + ";"); 
			out.println("create table " + eltClass.getAttribute("name") + " (");
			out.print("\tid int primary key");
	
			break;
			
		case 8 : break;
		
		default :out.println("*Error*"); 
		break;
				}
		

		
		NodeList attributeList = eltClass.getChildNodes();
		//on créé le corps de la classe
		
		for (int i = 0 ; i < attributeList.getLength() ; i++)
		{
			Node attr = attributeList.item(i);
			
			if (attr.getNodeType() == Node.ELEMENT_NODE)
			{
				Element eltAttr = (Element) attr;
				handleAttribute(eltAttr, eltClass, mode);
				if (mode == 7)
					handleAttribute(eltAttr, eltClass, 9);
			}
			
		}
		
		if (mode < 7)
			out.println("}");
		else if (mode ==7)
			out.println(");\n");
	}
	
	public void handleModel(NodeList classes, int mode)
	{
		//1er passage pour definir les types
		for (int i = 0 ; i < classes.getLength() ; i++)
		{
			Node classe = classes.item(i);
			
			if (classe.getNodeType() == Node.ELEMENT_NODE)
			{
				Element eltClass = (Element) classe;
				
				// on remplit la hashtable	
				
				String name = eltClass.getAttribute("name");
				
				if (mode != 7) {
					if (name.equals("int")) name = "Integer";
					if (name.equals("double")) name = "Double";
					if (name.equals("long")) name = "Long";
					if (name.equals("float")) name = "Float";
					if (name.equals("short")) name = "Short";
					if (name.equals("byte")) name = "Byte";
					if (name.equals("char")) name = "Char";
					if (name.equals("boolean")) name = "Boolean";
				}
				else {
					if (name.equals("String")) name = "varchar(250)";
					if (name.equals("double")) name = "number";
					if (name.equals("long")) name = "number";
					if (name.equals("float")) name = "number";
					if (name.equals("short")) name = "number";
					if (name.equals("byte")) name = "varchar(1)";
					if (name.equals("char")) name = "varchar(1)";
				}
				
				conversionTable.put(eltClass.getAttribute("xmi:id"), name);
			}
		}
		
		//deuxieme passage pour remplir les classes
		for (int i = 0 ; i < classes.getLength() ; i++)
		{
			Node classe = classes.item(i);
			
			if (classe.getNodeType() == Node.ELEMENT_NODE)
			{
				Element eltClass = (Element) classe;
				
				if(eltClass.getAttribute("xmi:type").equals("uml:PrimitiveType"))
				{
					handlePrimitive(eltClass);
				}
				else
				{
					if (mode == 7)
					{	
						handleClass(eltClass, 8);
					}
					
					
					
					
					handleClass(eltClass, mode);
					
				}
				
			}
		}
		
	}
		
}