1 files changed, 98 insertions, 0 deletions
diff --git a/org.eclipse.mylyn.bugzilla.ui/src/org/eclipse/mylyn/internal/bugzilla/ui/tasklist/NewBugzillaQueryWizard.java b/org.eclipse.mylyn.bugzilla.ui/src/org/eclipse/mylyn/internal/bugzilla/ui/tasklist/NewBugzillaQueryWizard.java
new file mode 100644
index 000000000..a4e5f0278
--- /dev/null
+++ b/org.eclipse.mylyn.bugzilla.ui/src/org/eclipse/mylyn/internal/bugzilla/ui/tasklist/NewBugzillaQueryWizard.java
@@ -0,0 +1,98 @@
+/*******************************************************************************
+ * Copyright (c) 2004 - 2006 University Of British Columbia and others.
+ * All rights reserved. This program and the accompanying materials
+ * are made available under the terms of the Eclipse Public License v1.0
+ * which accompanies this distribution, and is available at
+ * http://www.eclipse.org/legal/epl-v10.html
+ *
+ * Contributors:
+ *     University Of British Columbia - initial API and implementation
+ *******************************************************************************/
+
+package org.eclipse.mylar.internal.bugzilla.ui.tasklist;
+
+import org.eclipse.core.runtime.CoreException;
+import org.eclipse.core.runtime.IProgressMonitor;
+import org.eclipse.jface.wizard.Wizard;
+import org.eclipse.mylar.internal.bugzilla.core.BugzillaPlugin;
+import org.eclipse.mylar.internal.core.util.MylarStatusHandler;
+import org.eclipse.mylar.internal.tasklist.TaskListPreferenceConstants;
+import org.eclipse.mylar.provisional.tasklist.AbstractRepositoryConnector;
+import org.eclipse.mylar.provisional.tasklist.MylarTaskListPlugin;
+import org.eclipse.mylar.provisional.tasklist.TaskRepository;
+import org.eclipse.ui.actions.WorkspaceModifyOperation;
+
+/**
+ * @author Mik Kersten
+ * @author Brock Janiczak
+ */
+public class NewBugzillaQueryWizard extends Wizard {
+
+	private static final String TITLE = "New Bugzilla Query";
+
+	private final TaskRepository repository;
+
+	BugzillaQueryTypeWizardPage page1;
+
+	public NewBugzillaQueryWizard(TaskRepository repository) {
+		this.repository = repository;
+		setNeedsProgressMonitor(true);
+		setWindowTitle(TITLE);
+	}
+
+	@Override
+	public void addPages() {
+		page1 = new BugzillaQueryTypeWizardPage(repository);
+		page1.setWizard(this);
+		addPage(page1);
+
+	}
+
+	@Override
+	public boolean performFinish() {
+
+		AbstractBugzillaQueryPage page;
+
+		if (page1.getNextPage() != null && page1.getNextPage() instanceof AbstractBugzillaQueryPage) {
+			page = (AbstractBugzillaQueryPage) page1.getNextPage();
+		} else {
+			return false;
+		}
+
+		final BugzillaRepositoryQuery queryCategory = page.getQuery();
+
+		MylarTaskListPlugin.getTaskListManager().getTaskList().addQuery(queryCategory);
+		boolean offline = MylarTaskListPlugin.getMylarCorePrefs().getBoolean(TaskListPreferenceConstants.WORK_OFFLINE);
+		if (!offline) {
+			WorkspaceModifyOperation op = new WorkspaceModifyOperation() {
+				protected void execute(IProgressMonitor monitor) throws CoreException {
+					monitor.beginTask("Executing query", 50);
+					try {
+						AbstractRepositoryConnector client = MylarTaskListPlugin.getRepositoryManager()
+								.getRepositoryConnector(BugzillaPlugin.REPOSITORY_KIND);
+						client.synchronize(queryCategory, null);
+					} finally {
+						monitor.done();
+					}
+				}
+			};
+
+			try {
+				getContainer().run(true, false, op);
+			} catch (Exception e) {
+				MylarStatusHandler.log(e, "There was a problem executing the query refresh");
+			}
+		}
+
+		return true;
+	}
+
+	@Override
+	public boolean canFinish() {
+		if (page1.getNextPage() != null && page1.getNextPage().isPageComplete()) {
+			return true;
+		}
+		return false;
+	}
+
+}