/*******************************************************************************

 * Copyright (c) 2011, 2011 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 API and implementation

 *******************************************************************************/

package org.eclipse.compare.tests;

import java.io.BufferedReader;

import java.io.ByteArrayInputStream;

import java.io.IOException;

import java.io.InputStreamReader;

import java.lang.reflect.InvocationTargetException;

import java.util.ArrayList;

import java.util.List;

import junit.framework.TestCase;

import org.eclipse.compare.CompareConfiguration;

import org.eclipse.compare.ITypedElement;

import org.eclipse.compare.contentmergeviewer.TextMergeViewer;

import org.eclipse.compare.internal.MergeSourceViewer;

import org.eclipse.compare.structuremergeviewer.Differencer;

import org.eclipse.compare.structuremergeviewer.ICompareInput;

import org.eclipse.core.resources.IFile;

import org.eclipse.core.resources.IProject;

import org.eclipse.core.resources.IResource;

import org.eclipse.core.resources.ResourcesPlugin;

import org.eclipse.core.runtime.CoreException;

import org.eclipse.core.runtime.IProgressMonitor;

import org.eclipse.jface.dialogs.Dialog;

import org.eclipse.swt.custom.StyledText;

import org.eclipse.swt.graphics.Image;

import org.eclipse.swt.widgets.Composite;

import org.eclipse.swt.widgets.Control;

import org.eclipse.swt.widgets.Shell;

import org.eclipse.team.internal.ui.mapping.AbstractCompareInput;

import org.eclipse.team.internal.ui.mapping.CompareInputChangeNotifier;

import org.eclipse.team.internal.ui.synchronize.LocalResourceTypedElement;

import org.eclipse.team.ui.synchronize.SaveableCompareEditorInput;

import org.eclipse.ui.IWorkbenchPage;

import org.eclipse.ui.PlatformUI;

public class SaveableCompareEditorInputTest extends TestCase {

	private class FileElement implements ITypedElement {

		private IFile file;

		public IFile getFile() {

			return file;

		}

		public FileElement(IFile file) {

			super();

			this.file = file;

		}

		public String getName() {

			return file.getName();

		}

		public Image getImage() {

			return null;

		}

		public String getType() {

			return TEXT_TYPE;

		}

	}

	public class TestDiffNode extends AbstractCompareInput {

		private CompareInputChangeNotifier notifier = new CompareInputChangeNotifier() {

			private IResource getResource(ITypedElement el) {

				if (el instanceof LocalResourceTypedElement) {

					return ((LocalResourceTypedElement) el).getResource();

				}

				if (el instanceof FileElement) {

					return ((FileElement) el).getFile();

				}

				return null;

			}

			protected IResource[] getResources(ICompareInput input) {

				List resources = new ArrayList();

				if (getResource(getLeft()) != null) {

					resources.add(getResource(getLeft()));

				}

				if (getResource(getRight()) != null) {

					resources.add(getResource(getRight()));

				}

				return (IResource[]) resources.toArray(new IResource[2]);

			}

		};

		public TestDiffNode(ITypedElement left, ITypedElement right) {

			super(Differencer.CHANGE, null, left, right);

		}

		public void fireChange() {

			super.fireChange();

		}

		protected CompareInputChangeNotifier getChangeNotifier() {

			return notifier;

		}

		public boolean needsUpdate() {

			// The remote never changes

			return false;

		}

		public void update() {

			fireChange();

		}

	}

	private class TestSaveableEditorInputLocal extends

			SaveableCompareEditorInput {

		protected ITypedElement left;

		protected ITypedElement right;

		private ICompareInput input;

		public Object getCompareResult() {

			return input;

		}

		public TestSaveableEditorInputLocal(ITypedElement left,

				ITypedElement right, CompareConfiguration conf) {

			super(conf, PlatformUI.getWorkbench().getActiveWorkbenchWindow()

					.getActivePage());

			this.left = left;

			this.right = right;

		}

		protected ICompareInput prepareCompareInput(IProgressMonitor monitor)

				throws InvocationTargetException, InterruptedException {

			input = createCompareInput();

			getCompareConfiguration().setLeftEditable(true);

			getCompareConfiguration().setRightEditable(false);

			return null;

		}

		private ICompareInput createCompareInput() {

			return new TestDiffNode(left, right);

		}

		protected void fireInputChange() {

			((TestDiffNode) getCompareResult()).fireChange();

		}

		public void saveChanges(IProgressMonitor monitor) throws CoreException {

			super.saveChanges(monitor);

		}

	}

	private TextMergeViewer viewer;

	public void testDirtyFlagOnStandardLeft() throws CoreException,

			InvocationTargetException, InterruptedException,

			IllegalArgumentException, SecurityException,

			IllegalAccessException, NoSuchFieldException,

			NoSuchMethodException, IOException {

		IProject project = ResourcesPlugin.getWorkspace().getRoot()

				.getProject("Project_" + System.currentTimeMillis());

		project.create(null);

		project.open(null);

		String fileContents1 = "FileContents";

		String appendFileContents = "_append";

		String fileContents2 = "FileContents2";

		IFile file1 = project.getFile("File1_" + System.currentTimeMillis()

				+ ".txt");

		file1.create(new ByteArrayInputStream(fileContents1.getBytes()), true,

				null);

		IFile file2 = project.getFile("File2_" + System.currentTimeMillis()

				+ ".txt");

		file2.create(new ByteArrayInputStream(fileContents2.getBytes()), true,

				null);

		LocalResourceTypedElement el1 = (LocalResourceTypedElement) SaveableCompareEditorInput

				.createFileElement(file1);

		ITypedElement el2 = new FileElement(file2);

		IWorkbenchPage page = PlatformUI.getWorkbench()

				.getActiveWorkbenchWindow().getActivePage();

		CompareConfiguration conf = new CompareConfiguration();

		conf.setLeftEditable(true);

		TestSaveableEditorInputLocal compareEditorInput = new TestSaveableEditorInputLocal(

				el1, el2, conf);

		final TestSaveableEditorInputLocal[] editorInputs = { compareEditorInput };

		compareEditorInput.prepareCompareInput(null);

		Shell shell = PlatformUI.getWorkbench().getActiveWorkbenchWindow()

				.getShell();

		Dialog dialog = new Dialog(shell) {

			protected Control createDialogArea(Composite parent) {

				Composite composite = (Composite) super

						.createDialogArea(parent);

				viewer = (TextMergeViewer) editorInputs[0].findContentViewer(

						null, editorInputs[0].input, composite);

				viewer.setInput(editorInputs[0].getCompareResult());

				return composite;

			}

		};

		dialog.setBlockOnOpen(false);

		dialog.open();

		MergeSourceViewer left = (MergeSourceViewer) ReflectionUtils.getField(

				viewer, "fLeft");

		StyledText leftText = left.getSourceViewer().getTextWidget();

		// modify the left side of editor

		leftText.append(appendFileContents);

		assertTrue(compareEditorInput.isDirty());

		// save editor

		viewer.flush(null);

		assertFalse(compareEditorInput.isDirty());

		dialog.close();

		// check weather file was saved

		BufferedReader reader = new BufferedReader(new InputStreamReader(

				file1.getContents()));

		String line = reader.readLine();

		assertEquals(fileContents1 + appendFileContents, line);

	}

	public void testDirtyFlagOnCustomLeft() throws CoreException,

			InvocationTargetException, InterruptedException,

			IllegalArgumentException, SecurityException,

			IllegalAccessException, NoSuchFieldException,

			NoSuchMethodException, IOException {

		IProject project = ResourcesPlugin.getWorkspace().getRoot()

				.getProject("Project_" + System.currentTimeMillis());

		project.create(null);

		project.open(null);

		String fileContents1 = "FileContents";

		String appendFileContents = "_append";

		String fileContents2 = "FileContents2";

		IFile file1 = project.getFile("File1_" + System.currentTimeMillis()

				+ ".txt");

		file1.create(new ByteArrayInputStream(fileContents1.getBytes()), true,

				null);

		IFile file2 = project.getFile("File2_" + System.currentTimeMillis()

				+ ".txt");

		file2.create(new ByteArrayInputStream(fileContents2.getBytes()), true,

				null);

		ITypedElement el1 = new FileElement(file1);

		ITypedElement el2 = new FileElement(file2);

		IWorkbenchPage page = PlatformUI.getWorkbench()

				.getActiveWorkbenchWindow().getActivePage();

		CompareConfiguration conf = new CompareConfiguration();

		conf.setLeftEditable(true);

		TestSaveableEditorInputLocal compareEditorInput = new TestSaveableEditorInputLocal(

				el1, el2, conf);

		final TestSaveableEditorInputLocal[] editorInputs = { compareEditorInput };

		compareEditorInput.prepareCompareInput(null);

		Shell shell = PlatformUI.getWorkbench().getActiveWorkbenchWindow()

				.getShell();

		Dialog dialog = new Dialog(shell) {

			protected Control createDialogArea(Composite parent) {

				Composite composite = (Composite) super

						.createDialogArea(parent);

				viewer = (TextMergeViewer) editorInputs[0].findContentViewer(

						null, editorInputs[0].input, composite);

				viewer.setInput(editorInputs[0].getCompareResult());

				return composite;

			}

		};

		dialog.setBlockOnOpen(false);

		dialog.open();

		MergeSourceViewer left = (MergeSourceViewer) ReflectionUtils.getField(

				viewer, "fLeft");

		StyledText leftText = left.getSourceViewer().getTextWidget();

		leftText.append(appendFileContents);

		assertTrue(compareEditorInput.isDirty());

		viewer.flush(null);

		assertFalse(compareEditorInput.isDirty());

		dialog.close();

		/*

		 * not checking if changes were saved because in this case saving is not

		 * handled

		 */

	}

}