com.google.gerrit.server.diff.DiffInfoCreator Maven / Gradle / Ivy
// Copyright (C) 2019 The Android Open Source Project
//
// 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.google.gerrit.server.diff;
import static com.google.common.base.Preconditions.checkState;
import com.google.common.collect.ImmutableList;
import com.google.common.collect.ImmutableMap;
import com.google.common.collect.Lists;
import com.google.common.collect.Maps;
import com.google.common.flogger.FluentLogger;
import com.google.gerrit.common.data.PatchScript;
import com.google.gerrit.common.data.PatchScript.DisplayMethod;
import com.google.gerrit.common.data.PatchScript.PatchScriptFileInfo;
import com.google.gerrit.entities.Patch;
import com.google.gerrit.extensions.common.ChangeType;
import com.google.gerrit.extensions.common.DiffInfo;
import com.google.gerrit.extensions.common.DiffInfo.ContentEntry;
import com.google.gerrit.extensions.common.DiffInfo.FileMeta;
import com.google.gerrit.extensions.common.DiffInfo.IntraLineStatus;
import com.google.gerrit.extensions.common.DiffWebLinkInfo;
import com.google.gerrit.extensions.common.WebLinkInfo;
import com.google.gerrit.jgit.diff.ReplaceEdit;
import com.google.gerrit.prettify.common.SparseFileContent;
import com.google.gerrit.server.change.FileContentUtil;
import com.google.gerrit.server.project.ProjectState;
import java.util.List;
import java.util.Optional;
import java.util.Set;
import org.eclipse.jgit.diff.Edit;
/** Creates and fills a new {@link DiffInfo} object based on diff between files. */
public class DiffInfoCreator {
private static final FluentLogger logger = FluentLogger.forEnclosingClass();
private static final ImmutableMap CHANGE_TYPE =
Maps.immutableEnumMap(
new ImmutableMap.Builder()
.put(Patch.ChangeType.ADDED, ChangeType.ADDED)
.put(Patch.ChangeType.MODIFIED, ChangeType.MODIFIED)
.put(Patch.ChangeType.DELETED, ChangeType.DELETED)
.put(Patch.ChangeType.RENAMED, ChangeType.RENAMED)
.put(Patch.ChangeType.COPIED, ChangeType.COPIED)
.put(Patch.ChangeType.REWRITE, ChangeType.REWRITE)
.build());
private final DiffWebLinksProvider webLinksProvider;
private final boolean intraline;
private final ProjectState state;
public DiffInfoCreator(
ProjectState state, DiffWebLinksProvider webLinksProvider, boolean intraline) {
this.webLinksProvider = webLinksProvider;
this.state = state;
this.intraline = intraline;
}
/* Returns the {@link DiffInfo} to display for end-users */
public DiffInfo create(PatchScript ps, DiffSide sideA, DiffSide sideB) {
DiffInfo result = new DiffInfo();
ImmutableList links = webLinksProvider.getDiffLinks();
result.webLinks = links.isEmpty() ? null : links;
if (ps.isBinary()) {
result.binary = true;
}
result.metaA = createFileMeta(sideA).orElse(null);
result.metaB = createFileMeta(sideB).orElse(null);
if (intraline) {
if (ps.hasIntralineTimeout()) {
result.intralineStatus = IntraLineStatus.TIMEOUT;
} else if (ps.hasIntralineFailure()) {
result.intralineStatus = IntraLineStatus.FAILURE;
} else {
result.intralineStatus = IntraLineStatus.OK;
}
logger.atFine().log("intralineStatus = %s", result.intralineStatus);
}
result.changeType = CHANGE_TYPE.get(ps.getChangeType());
logger.atFine().log("changeType = %s", result.changeType);
if (result.changeType == null) {
throw new IllegalStateException("unknown change type: " + ps.getChangeType());
}
if (ps.getPatchHeader().size() > 0) {
result.diffHeader = ps.getPatchHeader();
}
result.content = calculateDiffContentEntries(ps);
return result;
}
private static List calculateDiffContentEntries(PatchScript ps) {
ContentCollector contentCollector = new ContentCollector(ps);
Set editsDueToRebase = ps.getEditsDueToRebase();
for (Edit edit : ps.getEdits()) {
logger.atFine().log("next edit = %s", edit);
if (edit.getType() == Edit.Type.EMPTY) {
logger.atFine().log("skip empty edit");
continue;
}
contentCollector.addCommon(edit.getBeginA());
checkState(
contentCollector.nextA == edit.getBeginA(),
"nextA = %s; want %s",
contentCollector.nextA,
edit.getBeginA());
checkState(
contentCollector.nextB == edit.getBeginB(),
"nextB = %s; want %s",
contentCollector.nextB,
edit.getBeginB());
switch (edit.getType()) {
case DELETE:
case INSERT:
case REPLACE:
List internalEdit =
edit instanceof ReplaceEdit ? ((ReplaceEdit) edit).getInternalEdits() : null;
boolean dueToRebase = editsDueToRebase.contains(edit);
contentCollector.addDiff(edit.getEndA(), edit.getEndB(), internalEdit, dueToRebase);
break;
case EMPTY:
default:
throw new IllegalStateException();
}
}
contentCollector.addCommon(ps.getA().getSize());
return contentCollector.lines;
}
private Optional createFileMeta(DiffSide side) {
PatchScriptFileInfo fileInfo = side.fileInfo();
if (fileInfo.displayMethod == DisplayMethod.NONE) {
return Optional.empty();
}
FileMeta result = new FileMeta();
result.name = side.fileName();
result.contentType =
FileContentUtil.resolveContentType(
state, side.fileName(), fileInfo.mode, fileInfo.mimeType);
result.lines = fileInfo.content.getSize();
ImmutableList links = webLinksProvider.getFileWebLinks(side.type());
result.webLinks = links.isEmpty() ? null : links;
result.commitId = fileInfo.commitId;
return Optional.of(result);
}
private static class ContentCollector {
private final List lines;
private final SparseFileContent.Accessor fileA;
private final SparseFileContent.Accessor fileB;
private final boolean ignoreWS;
private int nextA;
private int nextB;
ContentCollector(PatchScript ps) {
lines = Lists.newArrayListWithExpectedSize(ps.getEdits().size() + 2);
fileA = ps.getA().createAccessor();
fileB = ps.getB().createAccessor();
ignoreWS = ps.isIgnoreWhitespace();
}
void addCommon(int end) {
logger.atFine().log("addCommon: end = %d", end);
end = Math.min(end, fileA.getSize());
logger.atFine().log("end = %d", end);
if (nextA >= end) {
logger.atFine().log("nextA >= end: nextA = %d, end = %d", nextA, end);
return;
}
while (nextA < end) {
logger.atFine().log("nextA < end: nextA = %d, end = %d", nextA, end);
if (!fileA.contains(nextA)) {
logger.atFine().log("fileA does not contain nextA: nextA = %d", nextA);
int endRegion = Math.min(end, nextA == 0 ? fileA.first() : fileA.next(nextA - 1));
int len = endRegion - nextA;
entry().skip = len;
nextA = endRegion;
nextB += len;
logger.atFine().log("setting: nextA = %d, nextB = %d", nextA, nextB);
continue;
}
ContentEntry e = null;
for (int i = nextA; i == nextA && i < end; i = fileA.next(i), nextA++, nextB++) {
if (ignoreWS && fileB.contains(nextB)) {
if (e == null || e.common == null) {
logger.atFine().log("create new common entry: nextA = %d, nextB = %d", nextA, nextB);
e = entry();
e.a = Lists.newArrayListWithCapacity(end - nextA);
e.b = Lists.newArrayListWithCapacity(end - nextA);
e.common = true;
}
e.a.add(fileA.get(nextA));
e.b.add(fileB.get(nextB));
} else {
if (e == null || e.common != null) {
logger.atFine().log(
"create new non-common entry: nextA = %d, nextB = %d", nextA, nextB);
e = entry();
e.ab = Lists.newArrayListWithCapacity(end - nextA);
}
e.ab.add(fileA.get(nextA));
}
}
}
}
void addDiff(int endA, int endB, List internalEdit, boolean dueToRebase) {
logger.atFine().log(
"addDiff: endA = %d, endB = %d, numberOfInternalEdits = %d, dueToRebase = %s",
endA, endB, internalEdit != null ? internalEdit.size() : 0, dueToRebase);
int lenA = endA - nextA;
int lenB = endB - nextB;
logger.atFine().log("lenA = %d, lenB = %d", lenA, lenB);
checkState(lenA > 0 || lenB > 0);
logger.atFine().log("create non-common entry");
ContentEntry e = entry();
if (lenA > 0) {
logger.atFine().log("lenA > 0: lenA = %d", lenA);
e.a = Lists.newArrayListWithCapacity(lenA);
for (; nextA < endA; nextA++) {
e.a.add(fileA.get(nextA));
}
}
if (lenB > 0) {
logger.atFine().log("lenB > 0: lenB = %d", lenB);
e.b = Lists.newArrayListWithCapacity(lenB);
for (; nextB < endB; nextB++) {
e.b.add(fileB.get(nextB));
}
}
if (internalEdit != null && !internalEdit.isEmpty()) {
logger.atFine().log("processing internal edits");
e.editA = Lists.newArrayListWithCapacity(internalEdit.size() * 2);
e.editB = Lists.newArrayListWithCapacity(internalEdit.size() * 2);
int lastA = 0;
int lastB = 0;
for (Edit edit : internalEdit) {
logger.atFine().log("internal edit = %s", edit);
if (edit.getBeginA() != edit.getEndA()) {
logger.atFine().log(
"edit.getBeginA() != edit.getEndA(): edit.getBeginA() = %d, edit.getEndA() = %d",
edit.getBeginA(), edit.getEndA());
e.editA.add(
ImmutableList.of(edit.getBeginA() - lastA, edit.getEndA() - edit.getBeginA()));
lastA = edit.getEndA();
logger.atFine().log("lastA = %d", lastA);
}
if (edit.getBeginB() != edit.getEndB()) {
logger.atFine().log(
"edit.getBeginB() != edit.getEndB(): edit.getBeginB() = %d, edit.getEndB() = %d",
edit.getBeginB(), edit.getEndB());
e.editB.add(
ImmutableList.of(edit.getBeginB() - lastB, edit.getEndB() - edit.getBeginB()));
lastB = edit.getEndB();
logger.atFine().log("lastB = %d", lastB);
}
}
}
e.dueToRebase = dueToRebase ? true : null;
}
private ContentEntry entry() {
ContentEntry e = new ContentEntry();
lines.add(e);
return e;
}
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy