View Javadoc
   package net.sf.plb4jedit.filelocator;
   import java.io.*;
   import java.util.*;
   import org.gjt.sp.util.Log;
   
   public class FileLocator {
     private final String DATABASENAME=".FileLocator.map";
     private String storeName;
     private HashMap store; // map of Maps (key,MapforThisKey)
    
    public FileLocator() throws Exception {
      storeName = System.getProperty("user.home") + File.separator + DATABASENAME; 
      loadMap();
    }
    
    public FileLocator(String storeName) throws Exception {
      this.storeName = storeName; 
      loadMap();
    }
    
    public void insertMap(String mapName, String path, String extension) throws Exception {
      // try to append to an existing map, not overwrite!
      Map m = (Map)store.get(mapName);
      if (m == null) {
       m = new HashMap();
      }
      store.put(mapName,m);
      try {
        File f = new File(path);
        if (f != null && f.isDirectory()) {
          processDirectory(f,m,extension);
        }
      } catch (Exception e) {
        e.printStackTrace();
      }
      // store map in users home directory
      FileOutputStream ostream = new FileOutputStream(storeName);
      ObjectOutputStream p = new ObjectOutputStream(ostream);
      p.writeObject(store);
      p.flush();
      ostream.flush();
      ostream.close();
      debug("stored " +storeName);
    }
    
    private void loadMap() throws Exception {
      File f = new File(storeName);
      debug("storeName "+storeName);
      if (f != null && f.exists()) {
        FileInputStream istream = new FileInputStream(f);
        ObjectInputStream p = new ObjectInputStream(istream);
        store = (HashMap)p.readObject();
        istream.close();
        debug("map loaded");
      } else {
        debug("no map loaded");
        store = new HashMap();      
      }
    }
    
    public void reload() throws Exception {
  		store.clear();
      loadMap();
    }
    
    public void delete() {
      store = null;
      try {
        File f = new File(System.getProperty("user.home") + File.separator + DATABASENAME);
        if (f != null && f.exists()) {
          f.delete();
          loadMap();
          System.out.println(f.getAbsolutePath() + " deleted.");
        } else {
          System.out.println(f.getAbsolutePath() + " not deleted.");
        }
      } catch (Exception ex) {
        ex.printStackTrace();
        System.out.println("delete failed.");
      }
    }
    
    public void delete(String mapName) {
      if (store.containsKey(mapName)) {
        store.remove(mapName);
      }
    }
    
    public String[] locate(String mapName, String source) throws Exception {
      debug("locate("+mapName+","+source+")" + " size of store "+ store.size());
      Map m = (Map)store.get(mapName);
      if (m != null) {
        debug("found map");
        Object o = m.get(source);
        if (o == null) {
          throw new IllegalStateException("didn't find '" + source + "' in map " + mapName);
        }
        if (o instanceof String) {
          String[] s = new String[1];
         s[0] = (String)o;
         return s;
       } else if (o instanceof Set) {
         Set set = (Set)o;
         String[] s = new String[set.size()];
         int i = 0;
         for (Iterator it = set.iterator(); it.hasNext(); ) {
           s[i++] = (String) it.next();
         }
         return s;
       } else {
         return null;
       }
     } else {
       throw new IllegalStateException("no map defined for " + mapName);
     }
   }
   
   public String[] locatePartially(String mapName, String source) throws Exception {
     debug("locatePartially("+mapName+","+source+")" + " size of store "+ store.size());
     String[] s = null;
     ArrayList a = new ArrayList();
     Map m = (Map)store.get(mapName);
     if (m == null) {
       throw new IllegalStateException("no map defined for " + mapName);
     }
     for (Iterator it = m.keySet().iterator(); it.hasNext();) {
       String key = (String) it.next();
       if (key.indexOf(source) > -1) {
         Object o = m.get(key);
         if (o instanceof String) {
           a.add(o);
         } else if (o instanceof Set) {
           Set set = (Set)o;
           for (Iterator it2 = set.iterator(); it2.hasNext(); ) {
             a.add(it2.next());
           }
         }
       }
     }
     if (a.size() > 0) {
       a.trimToSize();
       s = new String[a.size()];
       s = (String[]) a.toArray(s);
     }
     return s;
   }
   
   public String dump() throws Exception {
     StringBuffer out = new StringBuffer("File: " + storeName + "\n");
     if (store == null || store.size() == 0) {
       out.append("database empty\n");
       return out.toString();
     }
     
     for (Iterator it = store.keySet().iterator(); it.hasNext();) {
       String mapName = (String) it.next();
       out.append("Map " + mapName + "\n");
       if (mapName == null || mapName.length() == 0) continue;
       Map m = (Map)store.get(mapName);
       if (m != null) {
         for (Iterator it2 = m.keySet().iterator(); it2.hasNext(); ) {
           String key = (String) it2.next();
           String[] s = locate(mapName,key);
           for (int i=0; i<s.length;i++) {
             out.append(key+"\t\t"+s[i]+ "\n");
           }
         }
       }
     }
     return out.toString();
   }
   
 	public String[] getMaps() {
 		Set keys = store.keySet();
 		if (keys.size() == 0) {
 			return null;
 		}
 		String[] mapNames = new String[keys.size()];
 		int i = 0;
 		for (Iterator it = keys.iterator(); it.hasNext();) {
 			mapNames[i++] = (String) it.next(); 
 		}
 		return mapNames;
 	}
 	
   private void processDirectory(File d, Map m, String extension) throws Exception {
     if (d == null || !d.isDirectory() || !d.exists() || !d.canRead()) return;
     File[] files = d.listFiles();
     if (files == null || files.length == 0) return;
 
 		for (int i = 0; i < files.length; i++) {
 			try {
 				if (files[i].isDirectory()) {
 					processDirectory(files[i], m, extension);
 				} else {
 					String path = files[i].getCanonicalPath();
 					String fileName = files[i].getName();
 					if (extension != null && extension.length() > 0 && fileName.endsWith(extension)) {
 						String key = getKey(fileName,extension);
 						if (!m.containsKey(key)) {
 							m.put(key,path);
 							debug("added "+key+" in " + path);
 						} else {
 							Object entry = m.get(key);
 							if (entry instanceof String) {
 								Set set = new HashSet(2);
 								set.add(entry);
 								set.add(path);
 								m.put(key,set);
 							} else if (entry instanceof Set) {
 								((Set)entry).add(path);
 							}
 						}
 					}
 				}
 			} catch (Exception e) {
 				System.err.println("Catched exception. Trying to proceed");
 				e.printStackTrace();
 				//return;
 			}
 		}	
   }
   
   private String getKey(String fileName,String extension) {
     return fileName.substring(0,fileName.indexOf(extension));
   }
   
   private void debug(String s) {
     if (false) {
 			Log.log(Log.DEBUG,this,s);
     }
   }
 
   public static void main(String[] args) throws Exception {
     if (args.length < 1) {
       System.out.println("usage FileLocator mode modespecific");
       System.out.println("  modes: updatedb | locate | partial | delete | dump");
       System.exit(0);
     }
     FileLocator fl = new FileLocator();
     if (args[0].equals("updatedb")) {
       if (args.length == 4) {
         fl.insertMap(args[1],args[2],args[3]);
       } else {
         System.out.println("FileLocator updatedb mapName path extension");
       }
     } else if (args[0].equals("locate")) {
       if (args.length == 3) {
         String[] s = fl.locate(args[1],args[2]);
         if (s != null) {
           for (int i = 0; i < s.length ; i++) {
             System.out.println(s[i]);
           }
         } else {
           System.out.println("Nothing found");
         }
       } else {
         System.out.println("FileLocator locate mapName file");
       }
     } else if (args[0].equals("delete")) {
       fl.delete();
     } else if (args[0].equals("dump")) {
       System.out.println(fl.dump());
     } else if (args[0].equals("partial")) {
       if (args.length == 3) {
         String[] s = fl.locatePartially(args[1],args[2]);
         if (s != null) {
           for (int i = 0; i < s.length ; i++) {
             System.out.println(s[i]);
           }
         } else {
           System.out.println("Nothing found");
         }
       } else {
         System.out.println("FileLocator partial mapName file");
       }
     } else {
       System.out.println("unknown mode");
     }
   }
 }