| // Copyright (C) 2021 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.change; |
| |
| import com.google.gerrit.common.Nullable; |
| import com.google.gerrit.entities.Change; |
| import com.google.gerrit.entities.Patch; |
| import com.google.gerrit.entities.PatchSet; |
| import com.google.gerrit.entities.Project; |
| import com.google.gerrit.extensions.common.FileInfo; |
| import com.google.gerrit.extensions.restapi.ResourceConflictException; |
| import com.google.gerrit.server.patch.DiffNotAvailableException; |
| import com.google.gerrit.server.patch.DiffOperations; |
| import com.google.gerrit.server.patch.FilePathAdapter; |
| import com.google.gerrit.server.patch.PatchListNotAvailableException; |
| import com.google.gerrit.server.patch.filediff.FileDiffOutput; |
| import com.google.inject.Inject; |
| import java.util.HashMap; |
| import java.util.Map; |
| import org.eclipse.jgit.errors.NoMergeBaseException; |
| import org.eclipse.jgit.lib.ObjectId; |
| |
| /** Implementation of {@link FileInfoJson} using the new diff cache {@link DiffOperations}. */ |
| public class FileInfoJsonNewImpl implements FileInfoJson { |
| private final DiffOperations diffs; |
| |
| @Inject |
| FileInfoJsonNewImpl(DiffOperations diffOperations) { |
| this.diffs = diffOperations; |
| } |
| |
| @Override |
| public Map<String, FileInfo> getFileInfoMap( |
| Change change, ObjectId objectId, @Nullable PatchSet base) |
| throws ResourceConflictException, PatchListNotAvailableException { |
| try { |
| if (base == null) { |
| return asFileInfo( |
| diffs.listModifiedFilesAgainstParent(change.getProject(), objectId, null)); |
| } |
| return asFileInfo(diffs.listModifiedFiles(change.getProject(), base.commitId(), objectId)); |
| } catch (DiffNotAvailableException e) { |
| convertException(e); |
| return null; // unreachable. handleAndThrow will throw an exception anyway |
| } |
| } |
| |
| @Override |
| public Map<String, FileInfo> getFileInfoMap( |
| Project.NameKey project, ObjectId objectId, int parent) |
| throws ResourceConflictException, PatchListNotAvailableException { |
| try { |
| Map<String, FileDiffOutput> modifiedFiles = |
| diffs.listModifiedFilesAgainstParent(project, objectId, parent + 1); |
| return asFileInfo(modifiedFiles); |
| } catch (DiffNotAvailableException e) { |
| convertException(e); |
| return null; // unreachable. handleAndThrow will throw an exception anyway |
| } |
| } |
| |
| private void convertException(DiffNotAvailableException e) |
| throws ResourceConflictException, PatchListNotAvailableException { |
| Throwable cause = e.getCause(); |
| if (cause != null && !(cause instanceof NoMergeBaseException)) { |
| cause = cause.getCause(); |
| } |
| if (cause instanceof NoMergeBaseException) { |
| throw new ResourceConflictException( |
| String.format("Cannot create auto merge commit: %s", e.getMessage()), e); |
| } |
| throw new PatchListNotAvailableException(e); |
| } |
| |
| private Map<String, FileInfo> asFileInfo(Map<String, FileDiffOutput> fileDiffs) { |
| Map<String, FileInfo> result = new HashMap<>(); |
| for (String path : fileDiffs.keySet()) { |
| FileDiffOutput fileDiff = fileDiffs.get(path); |
| FileInfo fileInfo = new FileInfo(); |
| fileInfo.status = |
| fileDiff.changeType() != Patch.ChangeType.MODIFIED |
| ? fileDiff.changeType().getCode() |
| : null; |
| fileInfo.oldPath = FilePathAdapter.getOldPath(fileDiff.oldPath(), fileDiff.changeType()); |
| fileInfo.sizeDelta = fileDiff.sizeDelta(); |
| fileInfo.size = fileDiff.size(); |
| if (fileDiff.patchType().get() == Patch.PatchType.BINARY) { |
| fileInfo.binary = true; |
| } else { |
| fileInfo.linesInserted = fileDiff.insertions() > 0 ? fileDiff.insertions() : null; |
| fileInfo.linesDeleted = fileDiff.deletions() > 0 ? fileDiff.deletions() : null; |
| } |
| result.put(path, fileInfo); |
| } |
| return result; |
| } |
| } |