package org.netbeans.apifest.testingtemplate;

 import java.awt.Component;

 import java.io.File;

 import java.io.IOException;

 import java.io.InputStream;

 import java.io.OutputStream;

 import java.text.MessageFormat;

 import java.util.Enumeration;

 import java.util.LinkedHashSet;

 import java.util.NoSuchElementException;

 import java.util.Set;

 import java.util.zip.ZipEntry;

 import java.util.zip.ZipInputStream;

 import javax.swing.JComponent;

 import javax.swing.event.ChangeListener;

 import org.netbeans.api.project.ProjectManager;

 import org.netbeans.spi.project.ui.support.ProjectChooser;

 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.util.NbBundle;

 public class TestingTemplateWizardIterator implements WizardDescriptor.InstantiatingIterator {

     private int index;

     private WizardDescriptor.Panel[] panels;

     private WizardDescriptor wiz;

     public TestingTemplateWizardIterator() {}

     public static TestingTemplateWizardIterator createIterator() {

         return new TestingTemplateWizardIterator();

     }

     private WizardDescriptor.Panel[] createPanels() {

         return new WizardDescriptor.Panel[] {

             new TestingTemplateWizardPanel(),

         };

     }

     private String[] createSteps() {

         return new String[] {

             NbBundle.getMessage(TestingTemplateWizardIterator.class, "LBL_CreateProjectStep")

         };

     }

     public Set/*<FileObject>*/ instantiate() throws IOException {

         Set resultSet = new LinkedHashSet();

         File dirF = FileUtil.normalizeFile((File) wiz.getProperty("projdir"));

         dirF.mkdirs();

         FileObject template = Templates.getTemplate(wiz);

         FileObject dir = FileUtil.toFileObject(dirF);

         unZipFile(template.getInputStream(), dir);

         // Always open top dir as a project:

         resultSet.add(dir);

         // Look for nested projects to open as well:

         Enumeration e = dir.getFolders(true);

         while (e.hasMoreElements()) {

             FileObject subfolder = (FileObject) e.nextElement();

             if (ProjectManager.getDefault().isProject(subfolder)) {

                 resultSet.add(subfolder);

             }

         }

         File parent = dirF.getParentFile();

         if (parent != null && parent.exists()) {

             ProjectChooser.setProjectsFolder(parent);

         }

         return resultSet;

     }

     public void initialize(WizardDescriptor wiz) {

         this.wiz = wiz;

         index = 0;

         panels = createPanels();

         // Make sure list of steps is accurate.

         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;

                 // Step #.

                 jc.putClientProperty("WizardPanel_contentSelectedIndex", new Integer(i));

                 // Step name (actually the whole list for reference).

                 jc.putClientProperty("WizardPanel_contentData", steps);

             }

         }

     }

     public void uninitialize(WizardDescriptor wiz) {

         this.wiz.putProperty("projdir",null);

         this.wiz.putProperty("name",null);

         this.wiz = null;

         panels = null;

     }

     public String name() {

         return MessageFormat.format("{0} of {1}",

             new Object[] {new Integer(index + 1), new Integer(panels.length)});

     }

     public boolean hasNext() {

         return index < panels.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--;

     }

     public WizardDescriptor.Panel current() {

         return panels[index];

     }

     // If nothing unusual changes in the middle of the wizard, simply:

     public final void addChangeListener(ChangeListener l) {}

     public final void removeChangeListener(ChangeListener l) {}

     private static void unZipFile(InputStream source, FileObject projectRoot) throws IOException {

         try {

             ZipInputStream str = new ZipInputStream(source);

             ZipEntry entry;

             while ((entry = str.getNextEntry()) != null) {

                 if (entry.isDirectory()) {

                     FileUtil.createFolder(projectRoot, entry.getName());

                 } else {

                     FileObject fo = FileUtil.createData(projectRoot, entry.getName());

                     FileLock lock = fo.lock();

                     try {

                         OutputStream out = fo.getOutputStream(lock);

                         try {

                             FileUtil.copy(str, out);

                         } finally {

                             out.close();

                         }

                     } finally {

                         lock.releaseLock();

                     }

                 }

             }

         } finally {

             source.close();

         }

     }

 }