//------------------------------------------------------------------------------
// Copyright (c) 2005, 2007 IBM Corporation 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:
// IBM Corporation - initial implementation
//------------------------------------------------------------------------------
//------------------------------------------------------------------------------
//Copyright (c) 2005, 2007 IBM Corporation 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:
//IBM Corporation - initial implementation
//------------------------------------------------------------------------------
package org.eclipse.epf.diagram.ui.service;

import org.eclipse.epf.diagram.core.part.AbstractDiagramEditor;
import org.eclipse.epf.diagram.core.part.DiagramEditorInput;
import org.eclipse.epf.diagram.core.part.DiagramEditorInputProxy;
import org.eclipse.epf.diagram.core.part.util.DiagramEditorUtil;
import org.eclipse.epf.diagram.core.providers.SharedResourceDiagramDocumentProvider;
import org.eclipse.epf.diagram.ui.DiagramUIPlugin;
import org.eclipse.epf.library.edit.util.Suppression;
import org.eclipse.gmf.runtime.diagram.ui.resources.editor.document.IDocumentProvider;
import org.eclipse.ui.IEditorInput;
import org.eclipse.ui.IEditorPart;
import org.eclipse.ui.IEditorReference;
import org.eclipse.ui.IFileEditorInput;
import org.eclipse.ui.PlatformUI;

/**
 * @author Phong Nguyen Le
 * @since 1.2
 */
public class DiagramEditorStatePreserver {
	private IEditorReference diagramEditorRef;
	private boolean dirty;

	public DiagramEditorStatePreserver(Object activityOrWrapper, Suppression suppression, int diagramType) {
		IEditorInput input = new DiagramEditorInputProxy(
				new DiagramEditorInput(activityOrWrapper, suppression, diagramType), 
				DiagramEditorHelper.getDiagramPreferencesHint(diagramType));
		try {
			IEditorReference[] editorRefs = PlatformUI.getWorkbench()
					.getActiveWorkbenchWindow().getActivePage()
					.getEditorReferences();
			String editorId = DiagramEditorUtil.getEditorId(diagramType);
			for (int i = 0; i < editorRefs.length; i++) {
				IEditorReference editorRef = editorRefs[i];
				if(input.equals(editorRef.getEditorInput()) && editorRef.equals(editorId)) {
					IEditorPart editor = editorRef.getEditor(false);
					if(editor instanceof AbstractDiagramEditor) {
						diagramEditorRef = editorRef;		
						break;
					}
				}
			}
			if(diagramEditorRef != null) {
				dirty = diagramEditorRef.getEditor(false).isDirty();
			}
		} catch (Exception e) {
			//
		}
	}
	
	public void restore() {
		try {
			if (diagramEditorRef != null && !dirty) {
				AbstractDiagramEditor editor = (AbstractDiagramEditor) diagramEditorRef
						.getEditor(false);
				if (editor != null && editor.isDirty()) {
					IDocumentProvider docProvider = ((AbstractDiagramEditor) editor)
							.getDocumentProvider();

					// this will clear dirty flag of the editor
					//
					if (docProvider instanceof SharedResourceDiagramDocumentProvider) {
						((SharedResourceDiagramDocumentProvider) docProvider)
								.markDocumentAsSaved((IFileEditorInput) diagramEditorRef
										.getEditorInput());
					}

				}
			}
		} catch (Exception e) {
			DiagramUIPlugin.getDefault().getLogger().logError(e);
		}
	}
}