package blanco.plugin.dbee.wizards;

import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStream;

import org.eclipse.core.resources.IFile;
import org.eclipse.core.resources.IProject;
import org.eclipse.core.resources.IProjectNature;
import org.eclipse.core.resources.IResource;
import org.eclipse.core.runtime.CoreException;
import org.eclipse.jface.viewers.ISelection;
import org.eclipse.jface.viewers.IStructuredSelection;
import org.eclipse.jface.wizard.Wizard;
import org.eclipse.ui.INewWizard;
import org.eclipse.ui.IWorkbench;
import org.eclipse.ui.IWorkbenchPage;
import org.eclipse.ui.PartInitException;
import org.eclipse.ui.PlatformUI;
import org.eclipse.ui.ide.IDE;

import blanco.plugin.dbee.editors.BlancoDbEEEditor;

public class BlancoDbEEWizard extends Wizard implements INewWizard {
    private ISelection selection;

    public BlancoDbEEWizard() {
        super();
        setNeedsProgressMonitor(true);
        setWindowTitle("blancoDb Enterprise Edition Wizard");
    }

    public void addPages() {
        addPage(new BlancoDbEEWizardPage(selection));
        addPage(new BlancoDbEEWizardPage2(selection));
        addPage(new BlancoDbEEWizardPage3(selection));
        addPage(new BlancoDbEEWizardPage4(selection));
        addPage(new BlancoDbEEWizardPage99(selection));
    }

    public boolean performFinish() {
        // 基本的にチェックはウィザード側で終了していると仮定します。
        IProject proj = getProject(selection);
        if (proj == null) {
            return false;
        }

        // ここで ファイルを新規作成します。
        IFile file = proj.getFile(BlancoDbEEEditor.TARGET_FILENAME);
        doFinish(file);

        IWorkbenchPage page = PlatformUI.getWorkbench()
                .getActiveWorkbenchWindow().getActivePage();
        try {
            // blancoDbエディタを開きます。
            IDE.openEditor(page, file, "blanco.plugin.dbee.editors");
        } catch (PartInitException e) {
        }

        return true;
    }

    /**
     * ISelectionから IProject(プロジェクト)を取得します。
     */
    public static final IProject getProject(ISelection selection) {
        IProject proj = null;
        IStructuredSelection ssel = (IStructuredSelection) selection;
        if (ssel.isEmpty()) {
            return null;
        }
        Object obj = ssel.getFirstElement();
        if (obj instanceof IProjectNature) {
            // プロジェクトの場合には、これが入ってきます。
            IProjectNature nature = (IProjectNature) obj;
            proj = nature.getProject();
        } else if (obj instanceof IResource) {
            IResource res = (IResource) obj;
            proj = res.getProject();
        } else {
            System.out.println(obj.getClass());
            return null;
        }
        return proj;
    }

    // これを改良して組み込みます。
    private void doFinish(IFile file) {
        try {
            InputStream stream = openContentStream();
            if (file.exists()) {
                // なにもしません。
            } else {
                file.create(stream, false, null);
            }
            stream.close();
        } catch (IOException e) {
        } catch (CoreException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
    }

    private InputStream openContentStream() {
        String contents = "<?xml version=\"1.0\" encoding=\"UTF-8\"?><blanco/>";
        return new ByteArrayInputStream(contents.getBytes());
    }

    public void init(IWorkbench workbench, IStructuredSelection selection) {
        this.selection = selection;
    }
}