/*
    Wotonomy: OpenStep design patterns for pure Java applications.
    Copyright (C) 2001 Michael Powers
   
    This library is free software; you can redistribute it and/or
    modify it under the terms of the GNU Lesser General Public
    License as published by the Free Software Foundation; either
    version 2.1 of the License, or (at your option) any later version.
   
   This library is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
   Lesser General Public License for more details.
  
   You should have received a copy of the GNU Lesser General Public
   License along with this library; if not, see http://www.gnu.org
   */
  package net.wotonomy.access;
  
  import java.io.File;
  import java.io.FileInputStream;
  import java.io.FileOutputStream;
  import java.io.IOException;
  import java.io.PrintStream;
  import java.util.Enumeration;
  
  import net.wotonomy.foundation.NSArray;
  import net.wotonomy.foundation.NSDictionary;
  import net.wotonomy.foundation.NSMutableArray;
  import net.wotonomy.foundation.NSMutableDictionary;
  import net.wotonomy.foundation.NSPropertyListSerialization;
  
  /***
  * An EOModel is a set of entities and stored procedures, along with a connection
  * dictionary to connect to a database.
  *
  * @author ezamudio@nasoft.com
  * @author $Author: cgruber $
  * @version $Revision: 894 $
  */
  public class EOModel {
  
  	protected static final String IDX_NAME = "index.eomodeld";
  
  	//This array contains dictionaries with "className" and "name"
  	protected NSMutableArray _entities = new NSMutableArray();
  	protected NSMutableDictionary _entitiesByName = new NSMutableDictionary();
  	protected NSMutableDictionary _entitiesByClass = new NSMutableDictionary();
  	protected NSDictionary _connectionDictionary = NSDictionary.EmptyDictionary;
  	protected String _adaptorName = "JDBC";
  	protected NSMutableDictionary _prototypesByName = new NSMutableDictionary();
  	protected NSMutableDictionary _storedProcedures = new NSMutableDictionary();
  	protected NSMutableArray _storedProcedureNames = new NSMutableArray();
  	protected NSDictionary _userInfo = NSDictionary.EmptyDictionary;
  	protected String _name;
  	protected String _path;
  	protected NSDictionary _internalInfo = NSDictionary.EmptyDictionary;
  	protected EOModelGroup _group;
  
  	public EOModel() {
  		super();
  	}
  
  	public EOModel(NSDictionary dict, Object o) {
  		this();
  	}
  
  	public EOModel(String path) {
  		this();
  		File f = new File(path);
  		_path = f.getAbsolutePath();
  		if (!(f.exists() && f.isDirectory()))
  			throw new IllegalArgumentException("The path is invalid (" + f + ")");
  		_name = f.getName();
  		f = new File(f, "index.eomodeld");
  		if (!(f.exists() && f.isFile()))
  			throw new IllegalArgumentException("Cannot find " + f);
  		String x = null;
  		try {
  			FileInputStream in = new FileInputStream(f);
  			byte[] b = new byte[in.available()];
  			in.read(b);
  			in.close();
  			x = new String(b);
  		} catch (IOException e) {
  			throw new IllegalArgumentException("Cannot read index.eomodeld");
  		}
  		NSDictionary d = NSPropertyListSerialization.dictionaryForString(x);
  		String version = (String)d.objectForKey("EOModelVersion");
  		if (version == null || !version.startsWith("2."))
  			throw new IllegalArgumentException("Invalid eomodel version: " + version);
  		setAdaptorName((String)d.objectForKey("adaptorName"));
  		setConnectionDictionary((NSDictionary)d.objectForKey("connectionDictionary"));
  		_entities = ((NSArray)d.objectForKey("entities")).mutableClone();
  		if (d.objectForKey("storedProcedures") != null)
  			_storedProcedureNames.addObjectsFromArray((NSArray)d.objectForKey("storedProcedures"));
  		if (d.objectForKey("internalInfo") != null)
  			_internalInfo = (NSDictionary)d.objectForKey("internalInfo");
  		if (d.objectForKey("userInfo") != null)
 			_userInfo = (NSDictionary)d.objectForKey("userInfo");
 		entityNamed("EOPrototypes");
 	}
 
 	public void setConnectionDictionary(NSDictionary dict) {
 		_connectionDictionary = dict;
 	}
 	public NSDictionary connectionDictionary() {
 		return _connectionDictionary;
 	}
 
 	public void addEntity(EOEntity ent) {
 		_entitiesByName.setObjectForKey(ent, ent.name());
 		_entitiesByClass.setObjectForKey(ent, ent.className());
 	}
 
 	public void removeEntity(EOEntity ent) {
 		_entitiesByName.removeObjectForKey(ent.name());
 		_entitiesByClass.removeObjectForKey(ent.className());
 	}
 
 	public void addStoredProcedure(EOStoredProcedure proc) {
 		if (!_storedProcedureNames.containsObject(proc))
 			_storedProcedureNames.addObject(proc);
 		_storedProcedures.setObjectForKey(proc, proc.name());
 	}
 
 	public void removeStoredProcedure(EOStoredProcedure proc) {
 		_storedProcedureNames.removeObject(proc.name());
 		_storedProcedures.removeObjectForKey(proc.name());
 	}
 
 	public EOStoredProcedure storedProcedureNamed(String name) {
 		EOStoredProcedure proc = (EOStoredProcedure)_storedProcedures.objectForKey(name);
 		//if we can't find it, check if we should load it
 		if (proc == null && _path != null && _storedProcedureNames.containsObject(name)) {
 			//try to read it from file
 			File f = new File(new File(_path), name + ".storedProcedure");
 			if (!f.exists())
 				return null;
 			String sdict = null;
 			try {
 				FileInputStream fin = new FileInputStream(f);
 				byte[] b = new byte[fin.available()];
 				fin.read(b);
 				fin.close();
 				sdict = new String(b);
 			} catch (IOException ex) {
 				return null;
 			}
 			NSDictionary plist = NSPropertyListSerialization.dictionaryForString(sdict);
 			if (plist == null)
 				throw new IllegalArgumentException("File " + f + " does not contain a valid stored procedure property list.");
 			proc = new EOStoredProcedure(plist, this);
 			//add it to our collection
 			_storedProcedures.setObjectForKey(proc, proc.name());
 		}
 		return proc;
 	}
 
 	public NSArray storedProcedures() {
 		if (_storedProcedures.count() < _storedProcedureNames.count()) {
 			for (int i=0; i<_storedProcedureNames.count(); i++)
 				storedProcedureNamed((String)_storedProcedureNames.objectAtIndex(i));
 		}
 		return _storedProcedures.allValues();
 	}
 
 	public NSArray storedProcedureNames() {
 		return _storedProcedures.allKeys();
 	}
 
 	public void setAdaptorName(String value) {
 		_adaptorName = value;
 	}
 	public String adaptorName() {
 		return _adaptorName;
 	}
 
 	public NSArray entities() {
 		if (_entitiesByName.count() >= _entities.count())
 			return _entitiesByName.allValues();
 		NSMutableArray es = new NSMutableArray();
 		for (int i = 0; i < _entities.count(); i++) {
 			NSDictionary d = (NSDictionary)_entities.objectAtIndex(i);
 			es.addObject(entityNamed((String)d.objectForKey("name")));
 		}
 		return es;
 	}
 
 	public NSArray entityNames() {
 		if (_entitiesByName.count() >= _entities.count())
 			return _entitiesByName.allKeys();
 		NSMutableArray names = new NSMutableArray();
 		for (int i = 0; i < _entities.count(); i++) {
 			NSDictionary d = (NSDictionary)_entities.objectAtIndex(i);
 			names.addObject(d.objectForKey("name"));
 		}
 		return names;
 	}
 
 	public EOEntity entityNamed(String name) {
 		EOEntity e = (EOEntity)_entitiesByName.objectForKey(name);
 		if (e == null && path() != null) {
 			boolean exists = false;
 			for (int i = 0; i < _entities.count(); i++) {
 				NSDictionary d = (NSDictionary)_entities.objectAtIndex(i);
 				if (d.objectForKey("name").equals(name))
 					exists = true;
 			}
 			if (!exists)
 				return null;
 			File f = new File(new File(path()), name + ".plist");
 			if (!(f.exists() && f.isFile()))
 				throw new IllegalArgumentException("Cannot find " + name + ".plist");
 			String s = null;
 			try {
 				FileInputStream fin = new FileInputStream(f);
 				byte[] b = new byte[fin.available()];
 				fin.read(b);
 				fin.close();
 				s = new String(b);
 			} catch (IOException ex) {
 				throw new IllegalArgumentException("Cannot read " + f);
 			}
 			NSDictionary d = NSPropertyListSerialization.dictionaryForString(s);
 			if (d == null)
 				throw new IllegalArgumentException("Cannot parse dictionary for " + f);
 			e = new EOEntity(d, this);
 			_entitiesByName.setObjectForKey(e, e.name());
 		}
 		return e;
 	}
 
 	public String name() {
 		return _name;
 	}
 
 	public String path() {
 		return _path;
 	}
 
 	public void setModelGroup(EOModelGroup group) {
 		_group = group;
 	}
 	public EOModelGroup modelGroup() {
 		return _group;
 	}
 
 	public void setUserInfo(NSDictionary dict) {
 		_userInfo = dict;
 	}
 
 	public NSDictionary userInfo() {
 		return _userInfo;
 	}
 
 	public void write() {
 	}
 
 	public void writeToFile(String path) {
 		if (!path.endsWith(".eomodeld"))
 			path += ".eomodeld";
 		File f = new File(path);
 		if (f.exists()) {
 			if (f.isDirectory()) {
 				File[] kids = f.listFiles();
 				for (int i = 0; i < kids.length; i++) {
 					String fname = kids[i].getName();
 					if (kids[i].isFile() && (fname.endsWith(".plist") || fname.endsWith(".eomodeld") || fname.endsWith(".fspec") || fname.endsWith(".storedProcedure")))
 						kids[i].delete();
 				}
 			} else
 				f.delete();
 		} else
 			f.mkdirs();
 		File kid = new File(f, "index.eomodeld");
 
 		//encode the index file
 		NSMutableDictionary d = new NSMutableDictionary();
 		d.setObjectForKey(_adaptorName, "adaptorName");
 		d.setObjectForKey("2.1", "EOModelVersion");
 		d.setObjectForKey(_connectionDictionary, "connectionDictionary");
 		if (_internalInfo != null && _internalInfo.count() > 0)
 			d.setObjectForKey(_internalInfo, "internalInfo");
 		if (_userInfo != null && _userInfo.count() > 0)
 			d.setObjectForKey(_userInfo, "userInfo");
 		if (_storedProcedureNames.count() > 0)
 			d.setObjectForKey(_storedProcedureNames, "storedProcedures");
 
 		//encode the entity list
 		entities();
 		storedProcedures();
 		NSMutableArray arr = new NSMutableArray(_entitiesByName.count());
 		Enumeration enumeration = _entitiesByName.keyEnumerator();
 		NSMutableDictionary md = new NSMutableDictionary();
 		while (enumeration.hasMoreElements()) {
 			String key = (String)enumeration.nextElement();
 			EOEntity ent = (EOEntity)_entitiesByName.objectForKey(key);
 			md.removeAllObjects();
 			ent.encodeIntoPropertyList(md);
 			File plist;
 			NSDictionary fetchSpecs = (NSDictionary)md.objectForKey("fetchSpecificationDictionary");
 			if (fetchSpecs != null) {
 				if (fetchSpecs.count() > 0) {
 					plist = new File(f, key + ".fspec");
 					String ps = NSPropertyListSerialization.stringForPropertyList(md.objectForKey("fetchSpecificationDictionary"));
 					try {
 						PrintStream stream = new PrintStream(new FileOutputStream(plist));
 						stream.println(ps);
 						stream.close();
 					} catch (IOException ex) {
 					}
 				}
 				md.removeObjectForKey("fetchSpecificationDictionary");
 			}
 			plist = new File(f, key + ".plist");
 			String ps = NSPropertyListSerialization.stringForPropertyList(md);
 			try {
 				PrintStream stream = new PrintStream(new FileOutputStream(plist));
 				stream.println(ps);
 				stream.close();
 			} catch (IOException ex) {
 			}
 			//add to the entity list for the index
 			arr.addObject(new NSDictionary(
 				new Object[]{ ent.name(), ent.className() },
 				new Object[]{ "name", "className" }));
 		}
 		d.setObjectForKey(arr, "entities");
 
 		//write the index file
 		String s = NSPropertyListSerialization.stringForPropertyList(d);
 		try {
 			PrintStream stream = new PrintStream(new FileOutputStream(kid));
 			stream.println(s);
 			stream.close();
 		} catch (IOException ex) {
 			throw new RuntimeException("Cannot write index.eomodeld");
 		}
 
 		//write the stored procedures
 		for (int i = 0; i < _storedProcedureNames.count(); i++) {
 			EOStoredProcedure proc = (EOStoredProcedure)_storedProcedures.objectForKey(_storedProcedureNames.objectAtIndex(i));
 			kid = new File(f, proc.name() + ".storedProcedure");
 			d.removeAllObjects();
 			proc.encodeIntoPropertyList(d);
 			s = NSPropertyListSerialization.stringForPropertyList(d);
 			try {
 				PrintStream stream = new PrintStream(new FileOutputStream(kid));
 				stream.println(s);
 				stream.close();
 			} catch (IOException ex) {
 				throw new RuntimeException("Cannot write " + f);
 			}
 		}
 		_path = f.getAbsolutePath();
 	}
 
 }
 /*
  * $Log$
  * Revision 1.2  2006/02/16 16:47:14  cgruber
  * Move some classes in to "internal" packages and re-work imports, etc.
  *
  * Also use UnsupportedOperationExceptions where appropriate, instead of WotonomyExceptions.
  *
  * Revision 1.1  2006/02/16 13:19:57  cgruber
  * Check in all sources in eclipse-friendly maven-enabled packages.
  *
  * Revision 1.8  2005/05/11 15:21:53  cgruber
  * Change enum to enumeration, since enum is now a keyword as of Java 5.0
  *
  * A few other comments in the code.
  *
  * Revision 1.7  2003/08/13 20:46:12  chochos
  * entityNamed() only throws if the searched entity is in the entities array and not in a file.
  *
  * Revision 1.6  2003/08/12 01:45:04  chochos
  * added some code to handle prototypes
  *
  * Revision 1.5  2003/08/11 19:38:27  chochos
  * Can now read from a file and re-write to another file.
  *
  * Revision 1.4  2003/08/11 18:20:08  chochos
  * saving to a file seems to work now.
  *
  * Revision 1.3  2003/08/08 02:16:55  chochos
  * can now read stored procedures from file.
  *
  * Revision 1.2  2003/08/08 00:37:00  chochos
  * add stored procedure functionality
  *
  * Revision 1.1  2003/08/07 02:42:28  chochos
  * EOModel can read an .eomodeld file. EOModelGroup doesn't do much for now.
  *
 */