package argos.proxy;

 import java.io.File;

 import java.lang.management.ManagementFactory;

 import java.net.MalformedURLException;

 import java.net.URL;

 import java.util.Hashtable;

 import java.util.logging.Level;

 import java.util.logging.Logger;

 import javax.management.MBeanServer;

 import javax.management.MalformedObjectNameException;

 import javax.management.ObjectName;

 import argos.config.Config;

 import argos.deploy.JarClassloader;

 public class DynamicProxyUtil {

         private static final Logger logger = Logger.getLogger(DynamicProxyUtil.class.getName());

         private static ObjectName dynamicProxyLoaderName;

         private static JarClassloader dynamicProxyLoader;

         private static long counter = 0;

         public static synchronized ObjectName instrument(String domain, String name,

                         Object object, Boolean verbose, Boolean increment, ObjectName classLoader) {

                 MBeanServer server = ManagementFactory.getPlatformMBeanServer();

                 try {

                         String mbeanName = name;

                         if(increment) {

                                 mbeanName = mbeanName + counter++;

                         }

                         ObjectName objectName = createObjectName(domain, mbeanName);

                         if(classLoader == null) {

                                 getDynamicProClassloader();

                                 server.createMBean("argos.proxy.DynamicProxy", objectName, dynamicProxyLoaderName, new Object[] {

                                                 name, object, verbose }, new String[] { String.class.getName(),

                                                 Object.class.getName(), Boolean.class.getName() });

                         }

                         else {

                                 server.createMBean("argos.proxy.DynamicProxy", objectName, classLoader, new Object[] {

                                                 name, object, verbose }, new String[] { String.class.getName(),

                                                 Object.class.getName(), Boolean.class.getName() });

                         }

                         return objectName;

                 }

                 catch(Exception e) {

                         logger.log(Level.WARNING, "Unable to instrument " + name, e);

                         return null;

                 }

         }

         public synchronized static ObjectName instrument(String name, Object object) {

                 return instrument(Config.get(Config.DEFAULT_ARGOS_DOMAIN), name, object, true, false, null);

         }

         public synchronized static ObjectName instrument(String domain, String name,

                         Object object, ObjectName classLoader) {

                 return instrument(domain, name, object, true, false, classLoader);

         }

         public synchronized static ObjectName instrument(String name, Object object,

                         boolean verbose) {

                 return instrument(Config.get(Config.DEFAULT_ARGOS_DOMAIN), name, object, verbose, false, null);

         }

         public static ObjectName createObjectName(String domain, String name)

                         throws MalformedObjectNameException {

                 Hashtable<String, String> table = new Hashtable<String, String>();

                 table.put("name", name);

                 return new ObjectName(domain, table);

         }

         public static synchronized JarClassloader getDynamicProClassloader() {

                 if(dynamicProxyLoader == null) {

                         URL[] url = new URL[1];

                         try {

                                 url[0] = new URL("file:" + (new File("DynamicProxy.jar")).getAbsolutePath());

                         }

                         catch (MalformedURLException e) {

                                 logger.log(Level.SEVERE, "Unable to create DynamicProxy classloader.", e);

                         }

                         dynamicProxyLoader = new JarClassloader(url);

                         MBeanServer server = ManagementFactory.getPlatformMBeanServer();

                         try {

                                 dynamicProxyLoaderName = new ObjectName("Classloaders:name=DynamicProxyClassLoader");

                                 server.registerMBean(dynamicProxyLoader, dynamicProxyLoaderName);

                                 logger.info("DynamicProxy's classloader registered.");

                         }

                         catch(Exception e) {

                                 logger.log(Level.SEVERE, "Unable to register DynamicProxy classloader.", e);

                         }

                 }

                 return dynamicProxyLoader;

         }

 }