platform/platform-impl/src/com/intellij/openapi/diff/impl/mergeTool/MergeVersion.java

/*
 * Copyright 2000-2009 JetBrains s.r.o.
 *
 * 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 com.intellij.openapi.diff.impl.mergeTool;

import com.intellij.openapi.application.ApplicationManager;
import com.intellij.openapi.command.CommandProcessor;
import com.intellij.openapi.command.undo.DocumentReference;
import com.intellij.openapi.command.undo.DocumentReferenceManager;
import com.intellij.openapi.command.undo.UndoManager;
import com.intellij.openapi.diagnostic.Logger;
import com.intellij.openapi.diff.DiffBundle;
import com.intellij.openapi.editor.Document;
import com.intellij.openapi.fileEditor.FileDocumentManager;
import com.intellij.openapi.fileTypes.FileType;
import com.intellij.openapi.fileTypes.FileTypes;
import com.intellij.openapi.project.Project;
import com.intellij.openapi.project.ProjectCoreUtil;
import com.intellij.openapi.project.ex.ProjectManagerEx;
import com.intellij.openapi.util.text.StringUtil;
import com.intellij.openapi.vfs.VirtualFile;
import com.intellij.projectImport.ProjectOpenProcessor;
import org.jetbrains.annotations.Nullable;

import java.io.IOException;
import java.util.Collection;
import java.util.HashSet;
import java.util.Set;

public interface MergeVersion {
  Document createWorkingDocument(Project project);

  void applyText(String text, Project project);

  VirtualFile getFile();

  byte[] getBytes() throws IOException;

  FileType getContentType();

  void restoreOriginalContent(Project project);

  class MergeDocumentVersion implements MergeVersion {
    private static final Logger LOG = Logger.getInstance("#com.intellij.openapi.diff.impl.mergeTool.MergeVersion.MergeDocumentVersion");
    protected final Document myDocument;
    private final String myOriginalText;
    private String myTextBeforeMerge;

    public MergeDocumentVersion(Document document, String originalText) {
      LOG.assertTrue(originalText != null, "text should not be null");
      LOG.assertTrue(document != null, "document should not be null");
      LOG.assertTrue(document.isWritable(), "document should be writable");
      myDocument = document;
      myOriginalText = originalText;
    }

    public Document createWorkingDocument(final Project project) {
      //TODO[ik]: do we really need to create copy here?
      final Document workingDocument = myDocument; //DocumentUtil.createCopy(myDocument, project);
      //LOG.assertTrue(workingDocument != myDocument);
      workingDocument.setReadOnly(false);
      final DocumentReference ref = DocumentReferenceManager.getInstance().create(workingDocument);
      myTextBeforeMerge = myDocument.getText();
      ApplicationManager.getApplication().runWriteAction(new Runnable() {
        public void run() {
          setDocumentText(workingDocument, myOriginalText, DiffBundle.message("merge.init.merge.content.command.name"), project);
          if (project != null) {
            final UndoManager undoManager = UndoManager.getInstance(project);
            if (undoManager != null) { //idea.sh merge command
              undoManager.nonundoableActionPerformed(ref, false);
            }
          }
        }
      });
      return workingDocument;
    }

    public void applyText(final String text, final Project project) {
      ApplicationManager.getApplication().runWriteAction(new Runnable() {
        public void run() {
          CommandProcessor.getInstance().executeCommand(project, new Runnable() {
            @Override
            public void run() {
              doApplyText(text, project);
            }
          }, "Merge changes", null);
        }
      });
    }

    protected void doApplyText(String text, Project project) {
      setDocumentText(myDocument, text, DiffBundle.message("save.merge.result.command.name"), project);

      FileDocumentManager.getInstance().saveDocument(myDocument);
      final VirtualFile file = getFile();
      reportProjectFileChangeIfNeeded(project, file);
    }

    public static void reportProjectFileChangeIfNeeded(Project project, VirtualFile file) {
      if (file != null && ! file.isDirectory()) {
        if (ProjectCoreUtil.isProjectOrWorkspaceFile(file) || isProjectFile(file)) {
          ProjectManagerEx.getInstanceEx().saveChangedProjectFile(file, project);
        }
      }
    }

    @Nullable
    public static Runnable prepareToReportChangedProjectFiles(final Project project, final Collection<VirtualFile> files) {
      final Set<VirtualFile> vfs = new HashSet<VirtualFile>();
      for (VirtualFile vf : files) {
        if (vf != null && ! vf.isDirectory()) {
          if (ProjectCoreUtil.isProjectOrWorkspaceFile(vf) || isProjectFile(vf)) {
            vfs.add(vf);
          }
        }
      }
      return vfs.isEmpty() ? null : new Runnable() {
        @Override
        public void run() {
          ProjectManagerEx ex = ProjectManagerEx.getInstanceEx();
          for (VirtualFile vf : vfs) {
            ex.saveChangedProjectFile(vf, project);
          }
        }
      };
    }

    @Override
    public void restoreOriginalContent(final Project project) {
      ApplicationManager.getApplication().runWriteAction(new Runnable() {
        public void run() {
          doRestoreOriginalContent(project);
        }
      });
    }

    public static boolean isProjectFile(VirtualFile file) {
      final ProjectOpenProcessor importProvider = ProjectOpenProcessor.getImportProvider(file);
      return importProvider != null && importProvider.lookForProjectsInDirectory();
    }

    protected void doRestoreOriginalContent(Project project) {
      setDocumentText(myDocument, myTextBeforeMerge, "", project);
    }

    public VirtualFile getFile() {
      return FileDocumentManager.getInstance().getFile(myDocument);
    }

    public byte[] getBytes() throws IOException {
      VirtualFile file = getFile();
      if (file != null) return file.contentsToByteArray();
      return myDocument.getText().getBytes();
    }

    public FileType getContentType() {
      VirtualFile file = getFile();
      if (file == null) return FileTypes.PLAIN_TEXT;
      return file.getFileType();
    }

    private static void setDocumentText(final Document document, final String text, String name, Project project) {
      CommandProcessor.getInstance().executeCommand(project, new Runnable() {
        public void run() {
          document.replaceString(0, document.getTextLength(), StringUtil.convertLineSeparators(text));
        }
      }, name, null);
    }
  }
}