001: /*
002: * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
003: *
004: * Copyright 1997-2007 Sun Microsystems, Inc. All rights reserved.
005: *
006: * The contents of this file are subject to the terms of either the GNU
007: * General Public License Version 2 only ("GPL") or the Common
008: * Development and Distribution License("CDDL") (collectively, the
009: * "License"). You may not use this file except in compliance with the
010: * License. You can obtain a copy of the License at
011: * http://www.netbeans.org/cddl-gplv2.html
012: * or nbbuild/licenses/CDDL-GPL-2-CP. See the License for the
013: * specific language governing permissions and limitations under the
014: * License. When distributing the software, include this License Header
015: * Notice in each file and include the License file at
016: * nbbuild/licenses/CDDL-GPL-2-CP. Sun designates this
017: * particular file as subject to the "Classpath" exception as provided
018: * by Sun in the GPL Version 2 section of the License file that
019: * accompanied this code. If applicable, add the following below the
020: * License Header, with the fields enclosed by brackets [] replaced by
021: * your own identifying information:
022: * "Portions Copyrighted [year] [name of copyright owner]"
023: *
024: * If you wish your version of this file to be governed by only the CDDL
025: * or only the GPL Version 2, indicate your decision by adding
026: * "[Contributor] elects to include this software in this distribution
027: * under the [CDDL or GPL Version 2] license." If you do not indicate a
028: * single choice of license, a recipient has the option to distribute
029: * your version of this file under either the CDDL, the GPL Version 2 or
030: * to extend the choice of license to its licensees as provided above.
031: * However, if you add GPL Version 2 code and therefore, elected the GPL
032: * Version 2 license, then the option applies only if the new code is
033: * made subject to such option by the copyright holder.
034: *
035: * Contributor(s):
036: *
037: * Portions Copyrighted 2007 Sun Microsystems, Inc.
038: */
039:
040: package org.netbeans.modules.visualweb.project.jsfloader.test;
041:
042: import java.io.File;
043: import java.io.FileInputStream;
044: import java.io.IOException;
045: import java.io.OutputStream;
046: import java.util.Arrays;
047: import java.util.Collections;
048: import java.util.Enumeration;
049: import java.util.zip.ZipEntry;
050: import java.util.zip.ZipInputStream;
051: import javax.swing.event.ChangeEvent;
052: import org.netbeans.api.project.Project;
053: import org.netbeans.api.project.ProjectManager;
054: import org.netbeans.api.project.ui.OpenProjects;
055: import org.netbeans.modules.visualweb.project.jsfloader.JsfJavaDataLoader;
056: import org.netbeans.modules.visualweb.project.jsfloader.JsfJspDataLoader;
057: import org.openide.filesystems.FileLock;
058: import org.openide.filesystems.FileObject;
059: import org.openide.filesystems.FileUtil;
060: import org.openide.loaders.DataLoader;
061: import org.openide.loaders.DataLoaderPool;
062: import org.openide.util.Lookup;
063: import org.openide.util.LookupEvent;
064: import org.openide.util.LookupListener;
065: import org.openide.util.test.MockLookup;
066:
067: /**
068: *
069: * @author quynguyen
070: */
071: public class SetupUtils {
072: private static JsfJspDataLoader jspLoader = null;
073: private static JsfJavaDataLoader javaLoader = null;
074:
075: public static Project setup(File workDir) throws IOException {
076: File userDir = new File(workDir, "userdir");
077: userDir.mkdir();
078: System.getProperties().put("netbeans.user",
079: userDir.getAbsolutePath());
080:
081: MockLookup.init();
082: DataLoaderPool pool = new SetupUtils.DefaultPool();
083: MockLookup.setInstances(pool);
084:
085: if (jspLoader == null) {
086: jspLoader = new JsfJspDataLoader();
087: }
088:
089: if (javaLoader == null) {
090: javaLoader = new JsfJavaDataLoader();
091: }
092:
093: MockLookup.setInstances(pool, jspLoader, javaLoader);
094:
095: String zipResource = "VWJavaEE5.zip";
096: String zipPath = SetupUtils.class.getResource(zipResource)
097: .getPath();
098: if (zipPath == null) {
099: throw new IOException("Could not load zip resource: "
100: + zipResource);
101: }
102:
103: File archiveFile = new File(zipPath);
104:
105: FileObject destFileObj = FileUtil.toFileObject(workDir);
106: unZipFile(archiveFile, destFileObj);
107:
108: if (!destFileObj.isValid()) {
109: throw new IOException(
110: "FileObject for project directory not valid");
111: }
112:
113: FileObject testApp = destFileObj.getFileObject("VWJavaEE5");
114: System.out.println("Children of VWJavaEE5:"
115: + Arrays.toString(testApp.getChildren()));
116:
117: Project project = ProjectManager.getDefault().findProject(
118: testApp);
119:
120: if (project == null) {
121: throw new IOException("Could not load project");
122: }
123:
124: OpenProjects.getDefault()
125: .open(new Project[] { project }, false);
126: return project;
127: }
128:
129: public static JsfJavaDataLoader getJavaLoader() {
130: return javaLoader;
131: }
132:
133: public static JsfJspDataLoader getJspLoader() {
134: return jspLoader;
135: }
136:
137: private static void unZipFile(File archiveFile, FileObject destDir)
138: throws IOException {
139: FileInputStream fis = new FileInputStream(archiveFile);
140: try {
141: ZipInputStream str = new ZipInputStream(fis);
142: ZipEntry entry;
143: while ((entry = str.getNextEntry()) != null) {
144: if (entry.isDirectory()) {
145: FileUtil.createFolder(destDir, entry.getName());
146: } else {
147: FileObject fo = FileUtil.createData(destDir, entry
148: .getName());
149: FileLock lock = fo.lock();
150: try {
151: OutputStream out = fo.getOutputStream(lock);
152: try {
153: FileUtil.copy(str, out);
154: } finally {
155: out.close();
156: }
157: } finally {
158: lock.releaseLock();
159: }
160: }
161: }
162: } finally {
163: fis.close();
164: }
165: }
166:
167: /**
168: * Taken from DataLoaderPool.DefaultPool to override NbLoaderPool
169: *
170: * Special pool for unit testing etc.
171: * Finds all relevant data loaders in default lookup.
172: *
173: */
174: public static final class DefaultPool extends DataLoaderPool
175: implements LookupListener {
176:
177: private final Lookup.Result<DataLoader> result;
178:
179: public DefaultPool() {
180: result = Lookup.getDefault().lookupResult(DataLoader.class);
181: result.addLookupListener(this );
182: }
183:
184: protected Enumeration<? extends DataLoader> loaders() {
185: return Collections.enumeration(result.allInstances());
186: }
187:
188: public void resultChanged(LookupEvent e) {
189: fireChangeEvent(new ChangeEvent(this));
190: }
191:
192: }
193: }
|