/*

  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.

  *

  * Copyright 1997-2010 Oracle and/or its affiliates. All rights reserved.

  *

  * Oracle and Java are registered trademarks of Oracle and/or its affiliates.

  * Other names may be trademarks of their respective owners.

  *

  * The contents of this file are subject to the terms of either the GNU

  * General Public License Version 2 only ("GPL") or the Common

  * Development and Distribution License("CDDL") (collectively, the

  * "License"). You may not use this file except in compliance with the

  * License. You can obtain a copy of the License at

  * http://www.netbeans.org/cddl-gplv2.html

  * or nbbuild/licenses/CDDL-GPL-2-CP. See the License for the

  * specific language governing permissions and limitations under the

  * License.  When distributing the software, include this License Header

  * Notice in each file and include the License file at

  * nbbuild/licenses/CDDL-GPL-2-CP.  Oracle designates this

  * particular file as subject to the "Classpath" exception as provided

  * by Oracle in the GPL Version 2 section of the License file that

  * accompanied this code. If applicable, add the following below the

  * License Header, with the fields enclosed by brackets [] replaced by

  * your own identifying information:

  * "Portions Copyrighted [year] [name of copyright owner]"

  * 

  * If you wish your version of this file to be governed by only the CDDL

  * or only the GPL Version 2, indicate your decision by adding

  * "[Contributor] elects to include this software in this distribution

  * under the [CDDL or GPL Version 2] license." If you do not indicate a

  * single choice of license, a recipient has the option to distribute

  * your version of this file under either the CDDL, the GPL Version 2 or

  * to extend the choice of license to its licensees as provided above.

  * However, if you add GPL Version 2 code and therefore, elected the GPL

  * Version 2 license, then the option applies only if the new code is

  * made subject to such option by the copyright holder.

  * 

  * Contributor(s):

  * 

  * Portions Copyrighted 2007 Sun Microsystems, Inc.

  */

 package org.netbeans.modules.visualweb.project.jsfloader.test;

 import java.io.File;

 import java.io.FileInputStream;

 import java.io.IOException;

 import java.io.OutputStream;

 import java.util.Arrays;

 import java.util.Collections;

 import java.util.Enumeration;

 import java.util.zip.ZipEntry;

 import java.util.zip.ZipInputStream;

 import javax.swing.event.ChangeEvent;

 import org.netbeans.api.project.Project;

 import org.netbeans.api.project.ProjectManager;

 import org.netbeans.api.project.ui.OpenProjects;

 import org.netbeans.junit.MockServices;

 import org.netbeans.modules.visualweb.project.jsfloader.JsfJavaDataLoader;

 import org.netbeans.modules.visualweb.project.jsfloader.JsfJspDataLoader;

 import org.openide.filesystems.FileLock;

 import org.openide.filesystems.FileObject;

 import org.openide.filesystems.FileUtil;

 import org.openide.loaders.DataLoader;

 import org.openide.loaders.DataLoaderPool;

 import org.openide.util.Lookup;

 import org.openide.util.LookupEvent;

 import org.openide.util.LookupListener;

 /**

  *

  * @author quynguyen

  */

 public class SetupUtils {

     public static Project setup(File workDir) throws IOException {

         File userDir = new File(workDir, "userdir");

         userDir.mkdir();

         System.getProperties().put("netbeans.user", userDir.getAbsolutePath());

         MockServices.setServices(SetupUtils.DefaultPool.class, JsfJspDataLoader.class, JsfJavaDataLoader.class);

         String zipResource = "VWJavaEE5.zip";

         String zipPath = SetupUtils.class.getResource(zipResource).getPath();

         if (zipPath == null) {

             throw new IOException("Could not load zip resource: " + zipResource);

         }

         File archiveFile = new File(zipPath);

         FileObject destFileObj = FileUtil.toFileObject(workDir);

         unZipFile(archiveFile, destFileObj);

         if (!destFileObj.isValid()) {

             throw new IOException("FileObject for project directory not valid");

         }

         FileObject testApp = destFileObj.getFileObject("VWJavaEE5");

         System.out.println("Children of VWJavaEE5:" + Arrays.toString(testApp.getChildren()));

         Project project = ProjectManager.getDefault().findProject(testApp);

         if (project == null) {

             throw new IOException("Could not load project");

         }

         OpenProjects.getDefault().open(new Project[]{project}, false);

         return project;

     }

     public static JsfJavaDataLoader getJavaLoader() {

         return JsfJavaDataLoader.findObject(JsfJavaDataLoader.class, true);

     }

     public static JsfJspDataLoader getJspLoader() {

         return JsfJspDataLoader.findObject(JsfJspDataLoader.class, true);

     }

     private static void unZipFile(File archiveFile, FileObject destDir) throws IOException {

         FileInputStream fis = new FileInputStream(archiveFile);

         try {

             ZipInputStream str = new ZipInputStream(fis);

             ZipEntry entry;

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

                 if (entry.isDirectory()) {

                     FileUtil.createFolder(destDir, entry.getName());

                 } else {

                     FileObject fo = FileUtil.createData(destDir, entry.getName());

                     FileLock lock = fo.lock();

                     try {

                         OutputStream out = fo.getOutputStream(lock);

                         try {

                             FileUtil.copy(str, out);

                         } finally {

                             out.close();

                         }

                     } finally {

                         lock.releaseLock();

                     }

                 }

             }

         } finally {

             fis.close();

         }

     }

     /**

      * Taken from DataLoaderPool.DefaultPool to override NbLoaderPool

      * 

      * Special pool for unit testing etc.

      * Finds all relevant data loaders in default lookup.

      * 

      */

     public static final class DefaultPool extends DataLoaderPool implements LookupListener {

         private final Lookup.Result<DataLoader> result;

         public DefaultPool() {

             result = Lookup.getDefault().lookupResult(DataLoader.class);

             result.addLookupListener(this);

         }

         protected Enumeration<? extends DataLoader> loaders() {

             return Collections.enumeration(result.allInstances());

         }

         public void resultChanged(LookupEvent e) {

             fireChangeEvent(new ChangeEvent(this));

         }

     }

 }