newtestwizarditerator.java

来自「cqME :java framework for TCK test.」· Java 代码 · 共 383 行

/*
 * $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 NewTestWizardIterator 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 + ";");
        } else {
            replaceTokens.put("@@PACKAGE_NAME@@", "");
            replaceTokens.put("@@PACKAGE_PREFIX@@", "");
            replaceTokens.put("@@PACKAGE_HEADER@@", "");
        }
        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 {
                lock.releaseLock();
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
    
    private void addStandardTokens(Map tokens) {
        Calendar calendar = Calendar.getInstance();
        Date currentDate = calendar.getTime();
        tokens.put("@@DATE@@", DateFormat.getDateInstance().format(currentDate));
        tokens.put("@@TIME@@", DateFormat.getTimeInstance().format(currentDate));
    }
    
    public void initialize(WizardDescriptor wizard) {
        this.wizard = wizard;
    }
    
    public void uninitialize(WizardDescriptor wizard) {
        panels = null;
    }
    
    public WizardDescriptor.Panel current() {
        return getPanels()[index];
    }
    
    public String name() {
        return index + 1 + ". from " + getPanels().length;
    }
    
    public boolean hasNext() {
        return index < getPanels().length - 1;
    }
    
    public boolean hasPrevious() {
        return index > 0;
    }
    
    public void nextPanel() {
        if (!hasNext()) {
            throw new NoSuchElementException();
        }
        index++;
    }
    
    public void previousPanel() {
        if (!hasPrevious()) {
            throw new NoSuchElementException();
        }
        index--;
    }
    
    // If nothing unusual changes in the middle of the wizard, simply:
    public void addChangeListener(ChangeListener l) {}
    public void removeChangeListener(ChangeListener l) {}
    
    // If something changes dynamically (besides moving between panels), e.g.
    // the number of panels changes in response to user input, then uncomment
    // the following and call when needed: fireChangeEvent();
    /*
    private Set<ChangeListener> listeners = new HashSet<ChangeListener>(1);
    public final void addChangeListener(ChangeListener l) {
        synchronized (listeners) {
            listeners.add(l);
        }
    }
    public final void removeChangeListener(ChangeListener l) {
        synchronized (listeners) {
            listeners.remove(l);
        }
    }
    protected final void fireChangeEvent() {
        Iterator<ChangeListener> it;
        synchronized (listeners) {
            it = new HashSet<ChangeListener>(listeners).iterator();
        }
        ChangeEvent ev = new ChangeEvent(this);
        while (it.hasNext()) {
            it.next().stateChanged(ev);
        }
    }
     */
    
    // You could safely ignore this method. Is is here to keep steps which were
    // there before this wizard was instantiated. It should be better handled
    // by NetBeans Wizard API itself rather than needed to be implemented by a
    // client code.
    private String[] createSteps() {
        String[] beforeSteps = null;
        Object prop = wizard.getProperty("WizardPanel_contentData");
        if (prop != null && prop instanceof String[]) {
            beforeSteps = (String[]) prop;
        }
        
        if (beforeSteps == null) {
            beforeSteps = new String[0];
        }
        
        String[] res = new String[(beforeSteps.length - 1) + panels.length];
        for (int i = 0; i < res.length; i++) {
            if (i < (beforeSteps.length - 1)) {
                res[i] = beforeSteps[i];
            } else {
                res[i] = panels[i - beforeSteps.length + 1].getComponent().getName();
            }
        }
        return res;
    }

    private static Set unZipFile(InputStream source, FileObject targetFolder, String name, Map tokens, String packageName) throws IOException {
        Set resultSet = new LinkedHashSet();
        try {
            ZipInputStream str = new ZipInputStream(source);
            ZipEntry entry;
            while ((entry = str.getNextEntry()) != null) {
                if (!entry.isDirectory()) {
                    String fileName = entry.getName();
                    if (fileName.endsWith("html")) {
                        fileName = name + ".html";
                    } else if (fileName.endsWith("java")) {
                        fileName = name + ".java";
                    }
                    FileObject fo = null;
                    if (fileName.equals("test.config")) {
                        if (null != targetFolder.getFileObject("test.config")) {
                            continue;
                        }
                    }
                    fo = FileUtil.createData(targetFolder, fileName);
                    if (!"test.config".equals(fileName)) {
                        resultSet.add(targetFolder.getFileObject(fileName));
                    }
                    
                    FileLock lock = fo.lock();
                    
                    try {
                        OutputStream out = fo.getOutputStream(lock);
                        try {
                            copyAndSubstituteTokens(str, out, tokens);
                        } finally {
                            out.close();
                        }
                    } finally {
                        lock.releaseLock();
                    }
                }
            }
        } finally {
            source.close();
        }
        return resultSet;
    }
    
    private static void copyAndSubstituteTokens(InputStream is, OutputStream os, Map tokens) throws IOException {
        PrintWriter pw = new PrintWriter(os);
        try {
            Reader r = new InputStreamReader(is);
            BufferedReader br = new BufferedReader(r);
            String line;
            while ((line = br.readLine()) != null) {
                pw.println(tokens == null ? line : replaceTokens(tokens, line));
            }
        } finally {
            pw.close();
        }
    }
    
    private static String replaceTokens(Map tokens, String line) {
        for (Iterator it = tokens.entrySet().iterator(); it.hasNext(); ) {
            Map.Entry entry = (Map.Entry) it.next();
            line = line.replaceAll((String) entry.getKey(), (String) entry.getValue());
        }
        return line;
    }

    public void stateChanged(ChangeEvent e) {
//        FileObject dir = Templates.getTargetFolder( wizard );
//        String packageName = getPackageName(wizard, dir);
//        System.out.println("packageName=" + packageName);
//        wizard.putProperty("WizardPanel_errorMessage", "AAA"); // NOI18N
    }
}