/*
* Copyright 2000-2010 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.patch;
import com.intellij.openapi.progress.ProgressManager;
import com.intellij.openapi.project.Project;
import com.intellij.openapi.util.SystemInfo;
import com.intellij.openapi.vcs.FilePath;
import com.intellij.openapi.vcs.VcsException;
import com.intellij.openapi.vcs.VcsOutgoingChangesProvider;
import com.intellij.openapi.vcs.VcsRoot;
import com.intellij.openapi.vcs.changes.BinaryContentRevision;
import com.intellij.openapi.vcs.changes.ByteBackedContentRevision;
import com.intellij.openapi.vcs.changes.Change;
import com.intellij.openapi.vcs.changes.ContentRevision;
import com.intellij.openapi.vfs.VirtualFile;
import com.intellij.util.BeforeAfter;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;
import java.nio.charset.Charset;
import java.util.*;
import static com.intellij.openapi.vcs.changes.ChangesUtil.getAfterPath;
import static com.intellij.openapi.vcs.changes.ChangesUtil.getBeforePath;
import static com.intellij.util.ObjectUtils.chooseNotNull;
import static com.intellij.vcsUtil.VcsUtil.groupByRoots;
public class IdeaTextPatchBuilder {
private IdeaTextPatchBuilder() {
}
public static List<BeforeAfter<AirContentRevision>> revisionsConvertor(final Project project, final List<Change> changes) throws VcsException {
final List<BeforeAfter<AirContentRevision>> result = new ArrayList<>(changes.size());
Map<VcsRoot, List<Change>> byRoots =
groupByRoots(project, changes, change -> chooseNotNull(getBeforePath(change), getAfterPath(change)));
for (VcsRoot root : byRoots.keySet()) {
final Collection<Change> rootChanges = byRoots.get(root);
if (root.getVcs() == null || root.getVcs().getOutgoingChangesProvider() == null) {
addConvertChanges(rootChanges, result);
continue;
}
final VcsOutgoingChangesProvider<?> provider = root.getVcs().getOutgoingChangesProvider();
final Collection<Change> basedOnLocal = provider.filterLocalChangesBasedOnLocalCommits(rootChanges, root.getPath());
rootChanges.removeAll(basedOnLocal);
addConvertChanges(rootChanges, result);
for (Change change : basedOnLocal) {
// dates are here instead of numbers
result.add(new BeforeAfter<>(convertRevision(change.getBeforeRevision(), provider),
convertRevision(change.getAfterRevision(), provider)));
}
}
return result;
}
private static void addConvertChanges(final Collection<Change> changes, final List<BeforeAfter<AirContentRevision>> result) {
for (Change change : changes) {
result.add(new BeforeAfter<>(convertRevisionToAir(change.getBeforeRevision()), convertRevisionToAir(change.getAfterRevision())));
}
}
@NotNull
public static List<FilePatch> buildPatch(final Project project, final Collection<Change> changes, final String basePath, final boolean reversePatch) throws VcsException {
final Collection<BeforeAfter<AirContentRevision>> revisions;
if (project != null) {
revisions = revisionsConvertor(project, new ArrayList<>(changes));
} else {
revisions = new ArrayList<>(changes.size());
for (Change change : changes) {
revisions.add(new BeforeAfter<>(convertRevisionToAir(change.getBeforeRevision()), convertRevisionToAir(change.getAfterRevision())));
}
}
return TextPatchBuilder.buildPatch(revisions, basePath, reversePatch, SystemInfo.isFileSystemCaseSensitive,
() -> ProgressManager.checkCanceled());
}
@Nullable
private static AirContentRevision convertRevisionToAir(final ContentRevision cr) {
return convertRevisionToAir(cr, null);
}
@Nullable
private static AirContentRevision convertRevisionToAir(final ContentRevision cr, final Long ts) {
if (cr == null) return null;
final FilePath fp = cr.getFile();
final StaticPathDescription description = new StaticPathDescription(fp.isDirectory(),
ts == null ? fp.getIOFile().lastModified() : ts, fp.getPath());
if (cr instanceof BinaryContentRevision) {
return new BinaryAirContentRevision((BinaryContentRevision)cr, description, ts);
} else {
return new TextAirContentRevision(cr, description, ts);
}
}
@Nullable
private static AirContentRevision convertRevision(@Nullable final ContentRevision cr, final VcsOutgoingChangesProvider provider) {
if (cr == null) return null;
final Date date = provider.getRevisionDate(cr.getRevisionNumber(), cr.getFile());
final Long ts = date == null ? null : date.getTime();
return convertRevisionToAir(cr, ts);
}
private static class BinaryAirContentRevision implements AirContentRevision {
@NotNull private final BinaryContentRevision myRevision;
@NotNull private final StaticPathDescription myDescription;
@Nullable private final Long myTimestamp;
public BinaryAirContentRevision(@NotNull BinaryContentRevision revision,
@NotNull StaticPathDescription description,
@Nullable Long timestamp) {
myRevision = revision;
myDescription = description;
myTimestamp = timestamp;
}
public boolean isBinary() {
return true;
}
public String getContentAsString() {
throw new IllegalStateException();
}
public byte[] getContentAsBytes() throws VcsException {
return myRevision.getBinaryContent();
}
public String getRevisionNumber() {
return myTimestamp != null ? null : myRevision.getRevisionNumber().asString();
}
@NotNull
public PathDescription getPath() {
return myDescription;
}
}
private static class TextAirContentRevision implements AirContentRevision {
@NotNull private final ContentRevision myRevision;
@NotNull private final StaticPathDescription myDescription;
@Nullable private final Long myTimestamp;
public TextAirContentRevision(@NotNull ContentRevision revision,
@NotNull StaticPathDescription description,
@Nullable Long timestamp) {
myRevision = revision;
myDescription = description;
myTimestamp = timestamp;
}
public boolean isBinary() {
return false;
}
public String getContentAsString() throws VcsException {
return myRevision.getContent();
}
public byte[] getContentAsBytes() throws VcsException {
if (myRevision instanceof ByteBackedContentRevision) {
return ((ByteBackedContentRevision)myRevision).getContentAsBytes();
}
String textContent = getContentAsString();
if (textContent == null) return null;
return textContent.getBytes(getCharset());
}
public String getRevisionNumber() {
return myTimestamp != null ? null : myRevision.getRevisionNumber().asString();
}
@NotNull
public PathDescription getPath() {
return myDescription;
}
@Override
public Charset getCharset() {
return myRevision.getFile().getCharset();
}
@Nullable
@Override
public String getLineSeparator() {
VirtualFile virtualFile = myRevision.getFile().getVirtualFile();
return virtualFile != null ? virtualFile.getDetectedLineSeparator() : null;
}
}
}