/*
    * Jour - java profiler and monitoring library
    *
    * Copyright (C) 2004 Jour team
    *
    * This library is free software; you can redistribute it and/or
    * modify it under the terms of the GNU Library General Public
    * License as published by the Free Software Foundation; either
    * version 2 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
   * Library General Public License for more details.
   *
   * You should have received a copy of the GNU Library General Public
   * License along with this library; if not, write to the
   * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
   * Boston, MA  02111-1307, USA.
   */
  package net.sf.jour.rt.agent;
  
  import java.util.HashMap;
  import java.util.*;
  
  import java.io.*;
  import java.net.*;
  import java.io.PrintStream;
  import java.io.FileOutputStream;
  import java.io.BufferedReader;
  
  import org.apache.log4j.Logger;
  
  /***
   * TODO Add docs
   * Created on 02.12.2004
   *
   * Contributing Author(s):
   *
   *   Misha Lifschitz <mishalifschitz at users.sourceforge.net> (Inital implementation)
   *   Vlad Skarzhevskyy <vlads at users.sourceforge.net> (Inital implementation)
   *
   * @author vlads
   * @version $Revision: 1.6 $ ($Author: vlads $) $Date: 2004/12/09 19:11:35 $
   */
  
  public class InstrumentationMap {
  
  	protected final Logger log = Logger.getLogger(InstrumentationMap.class);
  	
      private static InstrumentationMap instance = null;
  
      public final static String separator = "|";
      
  	public final static String DEFULAT_FILE_NAME = "jourInstrumentationMap.txt";
  
  	/*
  	 * TODO use BigMap to contain more than int numbers.
  	 */
      private HashMap map;
      private HashMap mapID;
      private long maxID = 0;
  	private boolean updated;
  
      public InstrumentationMap() {
          map = new HashMap();
  		mapID = new HashMap();
  		updated = false;
      }
  
      // Static Initializer
  	static {
  		instance();
  	}
  
  	/*
  	 * Returns an instance
  	 */
  	public static InstrumentationMap instance() {
  		// Create a default instance if none was provided
  		if (instance == null) {
  			synchronized (InstrumentationMap.class) {
  				if (instance == null) {
  					instance = new InstrumentationMap();
  				}
  			}
  		}
  		return instance;
  	}
  
  	public boolean isUpdated() {
  		return this.updated;
  	}
  	
  	public long size() {
  	    return map.size();
  	}
  
  	public long getID(String className, String methodName, String methodSignature) {
 	    return getID(className + separator + methodName + separator + methodSignature);
 	}
 
     long getID(String name) {
         if (map.containsKey(name)) {
 			Long l = (Long)map.get(name);
 			return l.longValue();
 		} else {
 			Long l = getNewID(0);
 			addID(l, name);
 			return l.longValue();
 		}
     }
 
     public Long getNewID(long base) {
 		// Map starts from 1
 		long l = maxID;
 		if (maxID == 0) {
 			l = mapID.size();
 		}
 		if (maxID == 0) {
 			l = 1;
 		}
 		if (base != 0) {
 			l = base + 1;
 		}
 		while (mapID.containsKey(new Long(l))) {
 			l ++;
 		}
 		maxID = l + 1;
 		return new Long(l);
 	}
 
 	public void addID(Long l, String name) {
 		addID(l, name, true);
 	}
 
 	private void addID(Long l, String name, boolean stateChange) {
 			map.put(name, l);
 			mapID.put(l, name);
 			if (stateChange) {
 				updated = true;
 			}
 	}
 		
 	public String getName(long cid) {
 		if (mapID.containsKey(new Long(cid))) {
 			String name = (String)mapID.get(new Long(cid));
 			return name;
 		} else {
 			return "?#" + cid;
 		}
 	}
 
 	public boolean save(String filename) {
 		try {
 			//Iterator itr = map.keySet().iterator();
 		    Vector sortedMap = new Vector(map.keySet());
 		    Collections.sort(sortedMap);
 		    Iterator itr = sortedMap.iterator();
 
 			PrintStream out =
 				new PrintStream(new FileOutputStream(filename, false));
 
 			while (itr.hasNext()) {
 				String name = (String) itr.next();
 				Long id = (Long) map.get(name);
 				out.println(id + ";" + name);
 			}
 
 			out.close();
 			return true;
 		} catch (Exception e) {
 			log.error("Error saving map", e);
 			return false;
 		}
 
 	}
 
 	public boolean load(File file) {
 	    try {
             return load(new URL("file:" + file.getAbsolutePath()));
         } catch (MalformedURLException e) {
             log.error("Error", e);
             return false;
         }
 	}
 	
 	public boolean load(URL url) {
 		try {
 			if (url == null) {
 				log.warn("fails to load map, file does not exists");
 				return false;
 			}
 			BufferedReader in = new BufferedReader(new InputStreamReader(url.openStream()));
 			while (true) {
 				String l = in.readLine();
 			    if (l == null) break;
 				l = l.trim();
 				if ((l.length() == 0) || (l.startsWith("#"))) {
 					continue;
 				}
 
                 int idx = l.indexOf(";");
                 if (idx >= 0) {
                     String id = l.substring(0, idx);
                     String name = l.substring(idx + 1);
                     addID(new Long(id), name, false);
                 }
 			}
 			in.close();
 			return true;
 		} catch (Exception e) {
 			log.error("Error loading map", e);
 			return false;
 		}
 	}
 }