/* DocuDirectory -- Directory of DocuFilesets.

  Digital Image Library servlet components

  Copyright (C) 2003 Robert Casties (robcast@mail.berlios.de)

  This program is free software; you can redistribute  it and/or modify it
  under  the terms of  the GNU General  Public License as published by the
  Free Software Foundation;  either version 2 of the  License, or (at your
  option) any later version.
   
  Please read license.txt for the full details. A copy of the GPL
  may be found at http://www.gnu.org/copyleft/lgpl.html

  You should have received a copy of the GNU General Public License
  along with this program; if not, write to the Free Software
  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA

 * Created on 25.02.2003
 */

package digilib.io;

import java.io.File;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;

import org.xml.sax.SAXException;

/**
 * @author casties
 */
public class DocuDirectory extends Directory {

	// list of files (DocuDirent)
	private ArrayList[] list = null;
	// directory object is valid (exists on disk)
	private boolean isValid = false;
	// reference of the parent DocuDirCache
	private DocuDirCache cache = null;
	// directory name (digilib canonical form)
	private String dirName = null;
	// directory metadata
	private HashMap dirMeta = null;
	// state of metadata is valid
	private boolean metaChecked = false;
	// unresolved file metadata
	private HashMap unresolvedFileMeta = null;
	// time of last access of this object (not the filesystem)
	private long objectATime = 0;
	// time the file system directory was last modified
	private long dirMTime = 0;

	/** Constructor with digilib directory path and a parent DocuDirCache.
	 * 
	 * Directory names at the given path are appended to the base directories 
	 * from the cache. The directory is checked on disk and isValid is set. 
	 * 
	 * @see readDir
	 *  
	 * @param path digilib directory path name
	 * @param cache parent DocuDirCache
	 */
	public DocuDirectory(String path, DocuDirCache cache) {
		this.dirName = path;
		this.cache = cache;
		initDir();
		checkDir();
	}

	/** Sets and checks the dir object.
	 * 
	 */
	protected void initDir() {
		String baseDirName = cache.getBaseDirNames()[0];
		// clear directory first
		list = new ArrayList[FileOps.NUM_CLASSES];
		isValid = false;
		dirMTime = 0;
		// the first directory has to exist
		dir = new File(baseDirName, dirName);
	}

	/** The digilib name of the parent directory.
	 * 
	 * Returns null if there is no parent.
	 */
	public String getParentDirName() {
		String s = null;
		int lastidx = dirName.lastIndexOf("/");
		if (lastidx > 0) {
			s = dirName.substring(0, lastidx);
		}
		return s;
	}

	/** number of DocuFiles in this directory.
	 * 
	 */
	public int size() {
		return ((list != null)&&(list[0] != null)) ? list[0].size() : 0;
	}

	/** number of files of this class in this directory.
	 * 
	 * @param fc fileClass
	 */
	public int size(int fc) {
		return ((list != null)&&(list[fc] != null)) ? list[fc].size() : 0;
	}

	/** Returns the ImageFile at the index.
	 * 
	 * @param index
	 * @return
	 */
	public ImageFileset get(int index) {
		if ((list == null) || (list[0] == null) || (index >= list[0].size())) {
			return null;
		}
		return (ImageFileset) list[0].get(index);
	}

	/** Returns the file of the class at the index.
	 * 
	 * @param index
	 * @param fc fileClass
	 * @return
	 */
	public DocuDirent get(int index, int fc) {
		if ((list == null) || (list[fc] == null) || (index >= list[fc].size())) {
			return null;
		}
		return (DocuDirent) list[fc].get(index);
	}

	/** Checks if the directory exists on the filesystem.
	 * 
	 * Sets isValid.
	 * 
	 * @return
	 */
	public boolean checkDir() {
		if (dir == null) {
			initDir();
		}
		isValid = dir.isDirectory();
		return isValid;
	}

	/** Read the filesystem directory and fill this object.
	 * 
	 * Clears the List and (re)reads all files.
	 * 
	 * @return boolean the directory exists
	 */
	public boolean readDir() {
		// list of base dirs from the parent cache
		String[] baseDirNames = cache.getBaseDirNames();
		// first file extension to try for scaled directories
		String scalext = null;
		// number of base dirs
		int nb = baseDirNames.length;
		// array of base dirs
		Directory[] dirs = new Directory[nb];
		// check directory first
		checkDir();
		if (!isValid) {
			return false;
		}
		// first entry is this directory
		dirs[0] = this;
		// fill array with the remaining directories
		for (int j = 1; j < nb; j++) {
			File d = new File(baseDirNames[j], dirName);
			if (d.isDirectory()) {
				dirs[j] = new Directory(d);
				dirs[j].readDir();
			}
		}

		// read all filenames
		logger.debug("reading directory "+dir.getPath());
		/*
		 * using ReadableFileFilter is safer (we won't get directories
		 * with file extensions) but slower.
		 */
		File[] allFiles = null;
		if (cache.safeDirIndex) {
			allFiles = dir.listFiles(new FileOps.ReadableFileFilter());
		} else {
			allFiles = dir.listFiles();
		}
		logger.debug("  done");
		if (allFiles == null) {
			// not a directory
			return false;
		}
		// go through all file classes
		for (int nc = 0; nc < FileOps.NUM_CLASSES; nc++) {
			int fc = cache.getFileClasses()[nc];
			//logger.debug("filtering directory "+dir.getPath()+" for class "+fc);
			File[] fl = FileOps.listFiles(allFiles, FileOps.filterForClass(fc));
			//logger.debug("  done");
			// number of files in the directory
			int nf = fl.length;
			if (nf > 0) {
				// create new list
				list[fc] = new ArrayList(nf);
				// sort the file names alphabetically and iterate the list
				Arrays.sort(fl);
				Map hints = FileOps.newHints(FileOps.HINT_BASEDIRS, dirs);
				hints.put(FileOps.HINT_FILEEXT, scalext);
				for (int i = 0; i < nf; i++) {
					DocuDirent f = FileOps.fileForClass(fc, fl[i], hints);
					// add the file to our list
					list[fc].add(f);
					f.setParent(this);
				}
			}
		}
		dirMTime = dir.lastModified();
		// read metadata as well
		readMeta();
		return isValid;
	}

	/** Check to see if the directory has been modified and reread if necessary.
	 * 
	 * @return boolean the directory is valid
	 */
	public boolean refresh() {
		if (isValid) {
			if (dir.lastModified() > dirMTime) {
				// on-disk modification time is more recent
				readDir();
			}
			touch();
		}
		return isValid;
	}

	/** Read directory metadata. 
	 * 
	 */
	public void readMeta() {
		// check for directory metadata...
		File mf = new File(dir, "index.meta");
		if (mf.canRead()) {
			XMLMetaLoader ml = new XMLMetaLoader();
			try {
				// read directory meta file
				HashMap fileMeta = ml.loadURL(mf.getAbsolutePath());
				if (fileMeta == null) {
					return;
				}
				// meta for the directory itself is in the "" bin
				dirMeta = (HashMap) fileMeta.remove("");
				// read meta for files in this directory
				readFileMeta(fileMeta, null);
				// is there meta for other files left?
				if (fileMeta.size() > 0) {
					unresolvedFileMeta = fileMeta;
				}
			} catch (SAXException e) {
				if (cache != null) {
					cache.logger.warn("error parsing index.meta", e);
				}
			} catch (IOException e) {
				if (cache != null) {
					cache.logger.warn("error reading index.meta", e);
				}
			}
		}
		readParentMeta();
		metaChecked = true;
	}

	/** Read metadata from all known parent directories.
	 * 
	 */
	public void readParentMeta() {
		// check the parent directories for additional file meta
		Directory dd = parent;
		String path = dir.getName();
		while (dd != null) {
			if (((DocuDirectory) dd).hasUnresolvedFileMeta()) {
				readFileMeta(((DocuDirectory) dd).unresolvedFileMeta, path);
			}
			// prepend parent dir path
			path = dd.dir.getName() + "/" + path;
			// become next parent
			dd = dd.parent;
		}
	}

	/** Read metadata for the files in this directory.
	 * 
	 * Takes a HashMap with meta-information, adding the relative path
	 * before the lookup.
	 * 
	 * @param fileMeta
	 * @param relPath
	 * @param fc fileClass
	 */
	protected void readFileMeta(HashMap fileMeta, String relPath) {
		if (list == null) {
			// there are no files
			return;
		}
		String path = (relPath != null) ? (relPath + "/") : "";
		// go through all file classes
		for (int nc = 0; nc < list.length; nc++) {
			int fc = cache.getFileClasses()[nc];
			if (list[fc] == null) {
				continue;
			}
			// iterate through the list of files in this directory
			for (Iterator i = list[fc].iterator(); i.hasNext();) {
				DocuDirent f = (DocuDirent) i.next();
				// prepend path to the filename
				String fn = path + f.getName();
				// look up meta for this file and remove from dir
				HashMap meta = (HashMap) fileMeta.remove(fn);
				if (meta != null) {
					// store meta in file
					f.setFileMeta(meta);
				}
			}
		}
	}

	protected void notifyChildMeta(HashMap childmeta) {
		List children = cache.getChildren(this.dirName, true);
		if (children.size() > 0) {
			for (Iterator i = children.iterator(); i.hasNext();) {
				// TODO: finish this!
				//((DocuDirectory) i.next()).readFileMeta()
			}
		}
	}

	/** Update access time.
	 * 
	 * @return long time of last access.
	 */
	public long touch() {
		long t = objectATime;
		objectATime = System.currentTimeMillis();
		return t;
	}

	/** Searches for the file with the name <code>fn</code>.
	 * 
	 * Searches the directory for the file with the name <code>fn</code> and returns 
	 * its index. Returns -1 if the file cannot be found. 
	 *  
	 * @param fn filename
	 * @param fc file class
	 * @return int index of file <code>fn</code>
	 */
	public int indexOf(String fn) {
		int fc = FileOps.classForFilename(fn);
		return indexOf(fn, fc);
	}
	
	/** Searches for the file with the name <code>fn</code> and class fc.
	 * 
	 * Searches the directory for the file with the name <code>fn</code> and returns 
	 * its index. Returns -1 if the file cannot be found. 
	 *  
	 * @param fn filename
	 * @return int index of file <code>fn</code>
	 */
	public int indexOf(String fn, int fc) {
		if (!isRead()) {
			// read directory now
			if (!readDir()) {
				return -1;
			}
		}
		// empty directory?
		if (list[fc] == null) {
			return -1;
		}
		// linear search -> worst performance
		int n = list[fc].size();
		for (int i = 0; i < n; i++) {
			DocuDirent fs = (DocuDirent) list[fc].get(i);
			if (fs.getName().equals(fn)) {
				// filename matches
				return i;
			}
		}
		// try again without extension
		for (int i = 0; i < n; i++) {
			DocuDirent fs = (DocuDirent) list[fc].get(i);
			if (FileOps.basename(fs.getName()).equals(FileOps.basename(fn))) {
				// basename matches
				return i;
			}
		}
		return -1;
	}

	/** Finds the DocuDirent with the name <code>fn</code>.
	 * 
	 * Searches the directory for the DocuDirent with the name <code>fn</code> and returns 
	 * it. Returns null if the file cannot be found. 
	 *  
	 * @param fn filename
	 * @return DocuDirent
	 */
	public DocuDirent find(String fn) {
		int fc = FileOps.classForFilename(fn);
		int i = indexOf(fn, fc);
		if (i >= 0) {
			return (DocuDirent) list[0].get(i);
		}
		return null;
	}

	/** Finds the DocuDirent with the name <code>fn</code> and class <code>fc</code>.
	 * 
	 * Searches the directory for the DocuDirent with the name <code>fn</code> and returns 
	 * it. Returns null if the file cannot be found. 
	 *  
	 * @param fn filename
	 * @return DocuDirent
	 */
	public DocuDirent find(String fn, int fc) {
		int i = indexOf(fn, fc);
		if (i >= 0) {
			return (DocuDirent) list[fc].get(i);
		}
		return null;
	}

	/**
	 * @return String
	 */
	public String getDirName() {
		return dirName;
	}

	/** The directory is valid (exists on disk).
	 * 
	 * @return boolean
	 */
	public boolean isValid() {
		return isValid;
	}

	/** The directory has been read from disk.
	 * 
	 * @return
	 */
	public boolean isRead() {
		return (dirMTime != 0);
	}

	/**
	 * @return long
	 */
	public long getAccessTime() {
		return objectATime;
	}

	/**
	 * @return Hashtable
	 */
	public HashMap getDirMeta() {
		return dirMeta;
	}
	
	/**
	 * Checks metadata
	 *  
	 */
	public void checkMeta() {
		if (metaChecked) {
			return;
		} else {
			readMeta();
		}
	}

	/**
	 * @return long
	 */
	public long getDirMTime() {
		return dirMTime;
	}

	/**
	 * Sets the dirMeta.
	 * @param dirMeta The dirMeta to set
	 */
	public void setDirMeta(HashMap dirMeta) {
		this.dirMeta = dirMeta;
	}

	public boolean hasUnresolvedFileMeta() {
		return (this.unresolvedFileMeta != null);
	}

}