newdistributedtestwizarditerator.java

来自「cqME :java framework for TCK test.」· Java 代码 · 共 402 行 · 第 1/2 页

/*
 * $Id$
 *
 * Copyright 2007 Sun Microsystems, Inc. All Rights Reserved.
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License version
 * 2 only, as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 * General Public License version 2 for more details (a copy is
 * included at /legal/license.txt).
 *
 * You should have received a copy of the GNU General Public License
 * version 2 along with this work; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
 * 02110-1301 USA
 *
 * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa
 * Clara, CA 95054 or visit www.sun.com if you need additional
 * information or have any questions.
 *
 */

package com.sun.testme.file.templates;

import java.awt.Component;
import java.io.BufferedReader;
import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.OutputStream;
import java.io.PrintWriter;
import java.io.Reader;
import java.text.DateFormat;
import java.util.Calendar;
import java.util.Date;
import java.util.HashMap;
import java.util.Iterator;
import java.util.LinkedHashSet;
import java.util.Map;
import java.util.NoSuchElementException;
import java.util.Properties;
import java.util.Set;
import java.util.zip.ZipEntry;
import java.util.zip.ZipInputStream;
import javax.swing.JComponent;
import javax.swing.event.ChangeEvent;
import javax.swing.event.ChangeListener;
import org.netbeans.api.java.project.JavaProjectConstants;
import org.netbeans.api.project.Project;
import org.netbeans.api.project.ProjectUtils;
import org.netbeans.api.project.SourceGroup;
import org.netbeans.api.project.Sources;
import org.netbeans.spi.java.project.support.ui.templates.JavaTemplates;
import org.netbeans.spi.project.ui.templates.support.Templates;
import org.openide.WizardDescriptor;
import org.openide.filesystems.FileLock;
import org.openide.filesystems.FileObject;
import org.openide.filesystems.FileUtil;
import org.openide.loaders.DataFolder;

public class NewDistributedTestWizardIterator implements WizardDescriptor.InstantiatingIterator, ChangeListener {
    
    private int index;
    
    private WizardDescriptor wizard;
    private WizardDescriptor.Panel[] panels;
    
    /**
     * Initialize panels representing individual wizard's steps and sets
     * various properties for them influencing wizard appearance.
     */
    private WizardDescriptor.Panel[] getPanels() {
        if (panels == null) {
            panels = createPanels(wizard);
            String[] steps = createSteps();
            for (int i = 0; i < panels.length; i++) {
                Component c = panels[i].getComponent();
                if (steps[i] == null) {
                    // Default step name to component name of panel. Mainly
                    // useful for getting the name of the target chooser to
                    // appear in the list of steps.
                    steps[i] = c.getName();
                }
                if (c instanceof JComponent) { // assume Swing components
                    JComponent jc = (JComponent) c;
                    // Sets step number of a component
                    jc.putClientProperty("WizardPanel_contentSelectedIndex", new Integer(i));
                    // Sets steps names for a panel
                    jc.putClientProperty("WizardPanel_contentData", steps);
                    // Turn on subtitle creation on each step
                    jc.putClientProperty("WizardPanel_autoWizardStyle", Boolean.TRUE);
                    // Show steps on the left side with the image on the background
                    jc.putClientProperty("WizardPanel_contentDisplayed", Boolean.TRUE);
                    // Turn on numbering of all steps
                    jc.putClientProperty("WizardPanel_contentNumbered", Boolean.TRUE);
                }
            }
        }
        return panels;
    }
    
    private WizardDescriptor.Panel[] createPanels (WizardDescriptor wizardDescriptor) {
        Project project = Templates.getProject( wizardDescriptor );
        Sources sources = ProjectUtils.getSources(project);
        SourceGroup[] groups = sources.getSourceGroups(JavaProjectConstants.SOURCES_TYPE_JAVA);
        WizardDescriptor.Panel p = JavaTemplates.createPackageChooser( project, groups );
        p.addChangeListener(this);
        return new WizardDescriptor.Panel[] {            
            p
        };
    }

    
    private String getPackageName(WizardDescriptor wizardDescriptor, FileObject dir) {
        String resultPackage = null;
        if (dir == null) {
            return resultPackage;
        }
        
        Project project = Templates.getProject( wizardDescriptor );
        Sources sources = ProjectUtils.getSources(project);
        SourceGroup[] groups = sources.getSourceGroups(JavaProjectConstants.SOURCES_TYPE_JAVA);
        
        for (int i = 0; i < groups.length; i++) {
            if (FileUtil.isParentOf(groups[i].getRootFolder(), dir)) {
                String relPath = FileUtil.getRelativePath(groups[i].getRootFolder(), dir);
                relPath = relPath.replace('/', '.');
                resultPackage = relPath;
                break;
            }
        }
        
        return resultPackage;
    }
    
    public Set instantiate() throws IOException {
        FileObject dir = Templates.getTargetFolder( wizard );
        //Project project = Templates.getProject(wizard);
        //FileObject dir = project.getProjectDirectory().createFolder("com");
        DataFolder df = DataFolder.findFolder( dir );
        FileObject template = Templates.getTemplate( wizard );
        String target = Templates.getTargetName( wizard );
        HashMap replaceTokens = new HashMap();
        replaceTokens.put("@@CLASS_NAME@@", target);
        
        String packageName = getPackageName(wizard, dir);
        if (packageName != null) {
            replaceTokens.put("@@PACKAGE_NAME@@", packageName);
            replaceTokens.put("@@PACKAGE_PREFIX@@", packageName + ".");
            replaceTokens.put("@@PACKAGE_HEADER@@", "package " + packageName + ";");
            replaceTokens.put("@@SERVER_PACKAGE_HEADER@@", "package " + packageName + ".server;");
        } else {
            replaceTokens.put("@@PACKAGE_NAME@@", "");
            replaceTokens.put("@@PACKAGE_PREFIX@@", "");
            replaceTokens.put("@@PACKAGE_HEADER@@", "");
            replaceTokens.put("@@SERVER_PACKAGE_HEADER@@", "package server;");
        }
        addStandardTokens(replaceTokens);
        
        Project project = Templates.getProject(wizard);
        String testSuiteRoot = FileUtil.toFile(project.getProjectDirectory()).getPath();
        replaceTokens.put("@@TESTSUITE_ROOT@@", testSuiteRoot.replace('\\','/'));

        Set resultSet = unZipFile(template.getInputStream(), dir, target, replaceTokens, packageName);
        
        
        updateTestSuiteJttFile(project.getProjectDirectory(), packageName);
        return resultSet;
    }
    
    private void updateTestSuiteJttFile(FileObject projectDir, String packageName) {
        FileObject fob = projectDir.getFileObject("build/plugin.properties");
        if (fob == null) {
            return;
        }
        Properties prop = new Properties();
        try {
            InputStream in = fob.getInputStream();
            try {
                prop.load(in);
            } finally {
                in.close();
            }
            String value = prop.getProperty("testedPackagePrefix");
            value += " -in " + packageName;
            prop.put("testedPackagePrefix", value);
            FileLock lock = fob.lock();
            try {
                OutputStream out = fob.getOutputStream(lock);
                try {
                    prop.store(out, "");
                } finally {
                    out.close();
                }
            } finally {