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.dataconnectivity.test.utils;
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.openide.filesystems.FileLock;
056: import org.openide.filesystems.FileObject;
057: import org.openide.filesystems.FileUtil;
058: import org.openide.loaders.DataLoader;
059: import org.openide.loaders.DataLoaderPool;
060: import org.openide.util.Lookup;
061: import org.openide.util.LookupEvent;
062: import org.openide.util.LookupListener;
063: import org.openide.loaders.DataLoader;
064: import org.openide.loaders.DataLoaderPool;
065: import org.openide.util.Lookup;
066: import org.openide.util.LookupEvent;
067: import org.openide.util.LookupListener;
068: import org.openide.util.test.MockLookup;
069:
070: /**
071: *
072: * @author JohnBaker
073: */
074: public class SetupProject {
075: private static String RESOURCEPATH = ".."
076: + System.getProperty("file.separator") + "resources"
077: + System.getProperty("file.separator");
078:
079: public static Project setup(File workDir) throws IOException {
080: assert (workDir != null);
081: File userDir = new File(workDir, "userdir");
082: userDir.mkdir();
083: System.getProperties().put("netbeans.user",
084: userDir.getAbsolutePath());
085: System.out.println(userDir.getAbsolutePath());
086: // String zipResource = RESOURCEPATH + "VWJavaEE5.zip";
087: String zipResource = "VWJavaEE5.zip";
088: System.out.println("resource = " + zipResource);
089:
090: String zipPath = SetupProject.class.getResource(zipResource)
091: .getPath();
092: if (zipPath == null) {
093: throw new IOException("Could not load zip resource: "
094: + RESOURCEPATH + zipResource);
095: }
096:
097: File archiveFile = new File(zipPath);
098:
099: FileObject destFileObj = FileUtil.toFileObject(workDir);
100: unZipFile(archiveFile, destFileObj);
101:
102: if (!destFileObj.isValid()) {
103: throw new IOException(
104: "FileObject for project directory not valid");
105: }
106:
107: MockLookup.init();
108: DataLoaderPool pool = new SetupProject.DefaultPool();
109: MockLookup.setInstances(pool);
110: FileObject testApp = destFileObj.getFileObject("VWJavaEE5");
111: System.out.println("Children of VWJavaEE5:"
112: + Arrays.toString(testApp.getChildren()));
113:
114: Project project = ProjectManager.getDefault().findProject(
115: testApp);
116:
117: if (project == null) {
118: throw new IOException("Could not load project");
119: }
120:
121: OpenProjects.getDefault()
122: .open(new Project[] { project }, false);
123: return project;
124: }
125:
126: private static void unZipFile(File archiveFile, FileObject destDir)
127: throws IOException {
128: FileInputStream fis = new FileInputStream(archiveFile);
129: try {
130: ZipInputStream str = new ZipInputStream(fis);
131: ZipEntry entry;
132: while ((entry = str.getNextEntry()) != null) {
133: if (entry.isDirectory()) {
134: FileUtil.createFolder(destDir, entry.getName());
135: } else {
136: FileObject fo = FileUtil.createData(destDir, entry
137: .getName());
138: FileLock lock = fo.lock();
139: try {
140: OutputStream out = fo.getOutputStream(lock);
141: try {
142: FileUtil.copy(str, out);
143: } finally {
144: out.close();
145: }
146: } finally {
147: lock.releaseLock();
148: }
149: }
150: }
151: } finally {
152: fis.close();
153: }
154: }
155:
156: /**
157: * Taken from DataLoaderPool.DefaultPool to override NbLoaderPool
158: *
159: * Special pool for unit testing etc.
160: * Finds all relevant data loaders in default lookup.
161: *
162: */
163: public static final class DefaultPool extends DataLoaderPool
164: implements LookupListener {
165:
166: private final Lookup.Result<DataLoader> result;
167:
168: public DefaultPool() {
169: result = Lookup.getDefault().lookupResult(DataLoader.class);
170: result.addLookupListener(this );
171: }
172:
173: protected Enumeration<? extends DataLoader> loaders() {
174: return Collections.enumeration(result.allInstances());
175: }
176:
177: public void resultChanged(LookupEvent e) {
178: fireChangeEvent(new ChangeEvent(this));
179: }
180:
181: }
182:
183: }
|