/**

  * Copyright 2008 WebPhotos

  *

  * Licensed under the Apache License, Version 2.0 (the "License");

  * you may not use this file except in compliance with the License.

  * You may obtain a copy of the License at

  *

  *      http://www.apache.org/licenses/LICENSE-2.0

  *

  * Unless required by applicable law or agreed to in writing, software

  * distributed under the License is distributed on an "AS IS" BASIS,

  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

  * See the License for the specific language governing permissions and

  * limitations under the License.

  */

 package net.sf.webphotos.netbeans.project.empty;

 import java.awt.Component;

 import java.io.ByteArrayInputStream;

 import java.io.ByteArrayOutputStream;

 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 net.sf.webphotos.netbeans.project.Constants.ProjectFactory.EmptyProject;

 import org.netbeans.api.project.ProjectManager;

 import org.netbeans.api.templates.TemplateRegistration;

 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.FileObject;

 import org.openide.filesystems.FileUtil;

 import org.openide.util.Exceptions;

 import org.openide.util.NbBundle;

 import org.openide.xml.XMLUtil;

 import org.w3c.dom.Document;

 import org.w3c.dom.Element;

 import org.w3c.dom.NodeList;

 import org.xml.sax.InputSource;

 public class EmptyWebPhotosProjectWizardIterator implements WizardDescriptor.InstantiatingIterator {

     public static final String PROP_PROJECT_DIR = "projdir";

     public static final String PROP_PROJECT_NAME = "name";

     private int index;

     private WizardDescriptor.Panel[] panels;

     private WizardDescriptor wiz;

     public EmptyWebPhotosProjectWizardIterator() {

     }

     @TemplateRegistration(

             folder = EmptyProject.PROJECT_FOLDER, 

             displayName = EmptyProject.PROJECT_DISPLAY_MANE, 

             iconBase = EmptyProject.PROJECT_ICON_BASE, 

             description = EmptyProject.PROJECT_DESCRIPTION)

     public static EmptyWebPhotosProjectWizardIterator createIterator() {

         return new EmptyWebPhotosProjectWizardIterator();

     }

     private WizardDescriptor.Panel[] createPanels() {

         return new WizardDescriptor.Panel[]{

                     new EmptyWebPhotosProjectWizardPanel(),};

     }

     private String[] createSteps() {

         return new String[]{

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

                 };

     }

     @Override

     public Set<FileObject> instantiate() throws IOException {

         Set<FileObject> resultSet = new LinkedHashSet<FileObject>();

         File dirF = FileUtil.normalizeFile((File) wiz.getProperty(PROP_PROJECT_DIR));

         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<? extends FileObject> e = dir.getFolders(true);

         while (e.hasMoreElements()) {

             FileObject subfolder = e.nextElement();

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

                 resultSet.add(subfolder);

             }

         }

         File parent = dirF.getParentFile();

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

             ProjectChooser.setProjectsFolder(parent);

         }

         return resultSet;

     }

     @Override

     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(WizardDescriptor.PROP_CONTENT_SELECTED_INDEX, new Integer(i));

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

                 jc.putClientProperty(WizardDescriptor.PROP_CONTENT_DATA, steps);

             }

         }

     }

     @Override

     public void uninitialize(WizardDescriptor wiz) {

         this.wiz.putProperty(PROP_PROJECT_DIR, null);

         this.wiz.putProperty(PROP_PROJECT_NAME, null);

         this.wiz = null;

         panels = null;

     }

     @Override

     public String name() {

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

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

     }

     @Override

     public boolean hasNext() {

         return index < panels.length - 1;

     }

     @Override

     public boolean hasPrevious() {

         return index > 0;

     }

     @Override

     public void nextPanel() {

         if (!hasNext()) {

             throw new NoSuchElementException();

         }

         index++;

     }

     @Override

     public void previousPanel() {

         if (!hasPrevious()) {

             throw new NoSuchElementException();

         }

         index--;

     }

     @Override

     public WizardDescriptor.Panel current() {

         return panels[index];

     }

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

     @Override

     public final void addChangeListener(ChangeListener l) {

     }

     @Override

     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());

                     if ("nbproject/project.xml".equals(entry.getName())) {

                         // Special handling for setting name of Ant-based projects; customize as needed:

                         filterProjectXML(fo, str, projectRoot.getName());

                     } else {

                         writeFile(str, fo);

                     }

                 }

             }

         } finally {

             source.close();

         }

     }

     private static void writeFile(ZipInputStream str, FileObject fo) throws IOException {

         OutputStream out = fo.getOutputStream();

         try {

             FileUtil.copy(str, out);

         } finally {

             out.close();

         }

     }

     private static void filterProjectXML(FileObject fo, ZipInputStream str, String name) throws IOException {

         try {

             ByteArrayOutputStream baos = new ByteArrayOutputStream();

             FileUtil.copy(str, baos);

             Document doc = XMLUtil.parse(new InputSource(new ByteArrayInputStream(baos.toByteArray())), false, false, null, null);

             NodeList nl = doc.getDocumentElement().getElementsByTagName(PROP_PROJECT_NAME);

             if (nl != null) {

                 for (int i = 0; i < nl.getLength(); i++) {

                     Element el = (Element) nl.item(i);

                     if (el.getParentNode() != null && "data".equals(el.getParentNode().getNodeName())) {

                         NodeList nl2 = el.getChildNodes();

                         if (nl2.getLength() > 0) {

                             nl2.item(0).setNodeValue(name);

                         }

                         break;

                     }

                 }

             }

             OutputStream out = fo.getOutputStream();

             try {

                 XMLUtil.write(doc, out, "UTF-8");

             } finally {

                 out.close();

             }

         } catch (Exception ex) {

             Exceptions.printStackTrace(ex);

             writeFile(str, fo);

         }

     }

 }