/*

 Copyright (c) 2006, University of Tromsø

 All rights reserved.

 Redistribution and use in source and binary forms, with or without 

 modification, are permitted provided that the following conditions are met:

  * Redistributions of source code must retain the above copyright notice, this list 

    of conditions and the following disclaimer.

  * Redistributions in binary form must reproduce the above copyright notice, this 

    list of conditions and the following disclaimer in the documentation and/or other 

    materials provided with the distribution.

  * Neither the name of the University of Tromsø nor the names of its contributors may 

    be used to endorse or promote products derived from this software without specific 

    prior written permission.

 THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY 

 EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES 

 OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT 

 SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, 

 INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED 

 TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR 

 BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN 

 CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN 

 ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH 

 DAMAGE.

 */

 package argos.util;

 import java.io.BufferedReader;

 import java.io.File;

 import java.io.FileInputStream;

 import java.io.FileOutputStream;

 import java.io.IOException;

 import java.io.InputStreamReader;

 import java.io.PrintWriter;

 import java.io.Serializable;

 import java.net.MalformedURLException;

 import java.util.ArrayList;

 import java.util.List;

 import java.util.zip.ZipEntry;

 import java.util.zip.ZipInputStream;

 import javax.management.InstanceNotFoundException;

 import javax.management.MBeanServerConnection;

 import javax.management.MalformedObjectNameException;

 import javax.management.NotificationListener;

 import javax.management.ObjectName;

 import javax.management.remote.JMXConnector;

 import javax.management.remote.JMXConnectorFactory;

 import javax.management.remote.JMXServiceURL;

 public class Util {

         public static final int BUFFER_SIZE = 10 * 1024;

         public static List<File> extractFromJar(File jar, String fromFolder, String toFolder, boolean removeFirstDir) 

                          throws IOException {

                 List<File> extracted = new ArrayList<File>();

                 byte[] buf = new byte[BUFFER_SIZE];

                 FileInputStream tempIn = null;

                 ZipInputStream stream = null;

                 tempIn = new FileInputStream(jar);

                 stream = new ZipInputStream(tempIn);

                 ZipEntry entry;

                 while((entry = stream.getNextEntry()) != null) {

                         if(entry.getName().startsWith(fromFolder)) {

                                 String tempFileName = entry.getName();

                                 if(removeFirstDir) {

                                         tempFileName = tempFileName.substring(tempFileName.indexOf('/'));

                                 }

                                 File file = new File(toFolder + "/" + tempFileName);

                                 if(entry.isDirectory()) {

                                         file.mkdir();

                                 }

                                 else {

                                         FileOutputStream out = new FileOutputStream(file, false);

                                         int n = 0;

                                         while((n = stream.read(buf, 0, 1024)) > -1) {

                                                 out.write(buf, 0, n);

                                         }

                                         out.close();

                                         extracted.add(file);

                                 }

                         }

                 }

                 stream.close();

                 tempIn.close();

                 return extracted;

         }

         public static void copyFile(File inFile, File outFile) throws IOException {

                 byte[] buf = new byte[BUFFER_SIZE];

                 FileInputStream in = new FileInputStream(inFile);

                 FileOutputStream out = new FileOutputStream(outFile);

                 int n = 0;

                 while((n = in.read(buf, 0, 1024)) > -1) {

                         out.write(buf, 0, n);

                 }

                 out.close();

                 in.close();

         }

         public static void cleanDir(File file) {

                 if(file.isDirectory() && !file.getName().equals("CVS")) {

                         for(File f : file.listFiles()) {

                                 cleanDir(f);

                         }

                         delete(file);

                 }

                 else {

                         delete(file);

                 }

         }

         private static void delete(File file) {

                 if(!file.getName().equals(".cvsignore") && !file.delete()) {

                         file.deleteOnExit();

                 }

         }

         public static JMXConnector listenTo(String who, String url, NotificationListener me, 

                         Serializable handback) throws MalformedObjectNameException, MalformedURLException,

                         InstanceNotFoundException, IOException {

                 int rounds = 10;

                 for(int i = 0; i < rounds; i++) {

                         if(i > 0) {

                                 try {

                                         Thread.sleep(5000);

                                 }

                                 catch (InterruptedException ignore) {}

                         }

                         try {

                                 JMXConnector connector = JMXConnectorFactory.connect(new JMXServiceURL(url));

                                 MBeanServerConnection server = connector.getMBeanServerConnection();

                                 server.addNotificationListener(new ObjectName(who), me, 

                                                 null, handback);

                                 return connector;

                         }

                         catch(InstanceNotFoundException e) {

                                 if(i == rounds - 1) {

                                         throw e;

                                 }

                         }

                         catch(IOException e) {

                                 if(i == rounds - 1) {

                                         throw e;

                                 }

                         }

                 }

                 return null;

         }

         public static void unlistenTo(String who, String url, NotificationListener me) throws Exception,

                         InstanceNotFoundException, IOException {

                 int rounds = 10;

                 for(int i = 0; i < rounds; i++) {

                         if(i > 0) {

                                 try {

                                         Thread.sleep(5000);

                                 }

                                 catch (InterruptedException ignore) {}

                         }

                         try {

                                 JMXConnector connector = JMXConnectorFactory.connect(new JMXServiceURL(url));

                                 MBeanServerConnection server = connector.getMBeanServerConnection();

                                 server.removeNotificationListener(new ObjectName(who), me);

                                 return;

                         }

                         catch(Exception e) {

                                 if(i == rounds - 1) {

                                         throw e;

                                 }

                         }

                 }                

         }

     public static void writeResourceToFile(String resource, String outFilename) throws IOException {

         BufferedReader reader = new BufferedReader(new InputStreamReader(Util.class.getResourceAsStream(resource)));

         PrintWriter writer = new PrintWriter(outFilename);

         String line = null;

         while((line = reader.readLine()) != null) {

             writer.println(line);

         }

         writer.flush();

         writer.close();

         reader.close();

     }

 }