com.google.gerrit.server.change.FileInfoJson.java Source code

Introduction

Here is the source code for com.google.gerrit.server.change.FileInfoJson.java

Source

// Copyright (C) 2013 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.common.collect.Maps;
import com.google.gerrit.reviewdb.client.Change;
import com.google.gerrit.reviewdb.client.Patch;
import com.google.gerrit.reviewdb.client.PatchSet;
import com.google.gerrit.reviewdb.client.AccountDiffPreference.Whitespace;
import com.google.gerrit.server.patch.PatchList;
import com.google.gerrit.server.patch.PatchListCache;
import com.google.gerrit.server.patch.PatchListEntry;
import com.google.gerrit.server.patch.PatchListKey;
import com.google.gerrit.server.patch.PatchListNotAvailableException;
import com.google.inject.Inject;

import org.eclipse.jgit.lib.ObjectId;

import java.util.Map;

import javax.annotation.Nullable;

public class FileInfoJson {
    private final PatchListCache patchListCache;

    @Inject
    FileInfoJson(PatchListCache patchListCache) {
        this.patchListCache = patchListCache;
    }

    Map<String, FileInfo> toFileInfoMap(Change change, PatchSet patchSet) throws PatchListNotAvailableException {
        return toFileInfoMap(change, patchSet, null);
    }

    Map<String, FileInfo> toFileInfoMap(Change change, PatchSet patchSet, @Nullable PatchSet base)
            throws PatchListNotAvailableException {
        ObjectId a = (base == null) ? null : ObjectId.fromString(base.getRevision().get());
        ObjectId b = ObjectId.fromString(patchSet.getRevision().get());
        PatchList list = patchListCache.get(new PatchListKey(change.getProject(), a, b, Whitespace.IGNORE_NONE));

        Map<String, FileInfo> files = Maps.newTreeMap();
        for (PatchListEntry e : list.getPatches()) {
            FileInfoJson.FileInfo d = new FileInfoJson.FileInfo();
            d.status = e.getChangeType() != Patch.ChangeType.MODIFIED ? e.getChangeType().getCode() : null;
            d.oldPath = e.getOldName();
            if (e.getPatchType() == Patch.PatchType.BINARY) {
                d.binary = true;
            } else {
                d.linesInserted = e.getInsertions() > 0 ? e.getInsertions() : null;
                d.linesDeleted = e.getDeletions() > 0 ? e.getDeletions() : null;
            }

            FileInfoJson.FileInfo o = files.put(e.getNewName(), d);
            if (o != null) {
                // This should only happen on a delete-add break created by JGit
                // when the file was rewritten and too little content survived. Write
                // a single record with data from both sides.
                d.status = Patch.ChangeType.REWRITE.getCode();
                if (o.binary != null && o.binary) {
                    d.binary = true;
                }
                if (o.linesInserted != null) {
                    d.linesInserted = o.linesInserted;
                }
                if (o.linesDeleted != null) {
                    d.linesDeleted = o.linesDeleted;
                }
            }
        }
        return files;
    }

    static class FileInfo {
        Character status;
        Boolean binary;
        String oldPath;
        Integer linesInserted;
        Integer linesDeleted;
    }
}