newdistributedtestwizarditerator.java

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

                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);
        }
    }
     */
    
    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;
            FileObject curTargetFolder = targetFolder;
            while ((entry = str.getNextEntry()) != null) {
                if (!entry.isDirectory()) {
                    String fileName = entry.getName();
                    if (fileName.endsWith("html")) {
                        curTargetFolder = targetFolder;
                        fileName = name + ".html";
                    } else if (fileName.endsWith("javaclient")) {
                        curTargetFolder = targetFolder;
                        fileName = name + ".java";
                    } else if (fileName.endsWith("javaserver")) {
                        if (null == (curTargetFolder = targetFolder.getFileObject("server"))) {
                            curTargetFolder = targetFolder.createFolder("server");
                        }
                        fileName = name + "Server.java";
                    } else if (fileName.equals("server.config")) {
                        if (null == (curTargetFolder = targetFolder.getFileObject("server"))) {
                            curTargetFolder = targetFolder.createFolder("server");
                        }
                        fileName = "test.config";
                    } else if (fileName.equals("test.config")) {
                        curTargetFolder = targetFolder;
                    } else if (fileName.equals("testmebase")) {
                        curTargetFolder = targetFolder;
                        fileName = "TestMEDistributedTest.java";
                    }
                    FileObject fo = null;
                    if (fileName.equals("test.config")) {
                        if (null != curTargetFolder.getFileObject("test.config")) {
                            continue;
                        }
                    }
                    fo = FileUtil.createData(curTargetFolder, fileName);
                    if (!fileName.endsWith(".config") && !("TestMEDistributedTest.java".equals(fileName))) {
                        resultSet.add(curTargetFolder.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;
    }
    
    // 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 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
    }
}