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

Introduction

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

Source

// Copyright (C) 2012 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.base.Strings;
import com.google.gerrit.common.ChangeHooks;
import com.google.gerrit.extensions.restapi.AuthException;
import com.google.gerrit.extensions.restapi.DefaultInput;
import com.google.gerrit.extensions.restapi.ResourceConflictException;
import com.google.gerrit.extensions.restapi.RestModifyView;
import com.google.gerrit.reviewdb.client.Change;
import com.google.gerrit.reviewdb.client.Change.Status;
import com.google.gerrit.reviewdb.client.ChangeMessage;
import com.google.gerrit.reviewdb.server.ReviewDb;
import com.google.gerrit.server.ApprovalsUtil;
import com.google.gerrit.server.ChangeUtil;
import com.google.gerrit.server.IdentifiedUser;
import com.google.gerrit.server.change.Restore.Input;
import com.google.gerrit.server.index.ChangeIndexer;
import com.google.gerrit.server.mail.ReplyToChangeSender;
import com.google.gerrit.server.mail.RestoredSender;
import com.google.gerrit.server.project.ChangeControl;
import com.google.gwtorm.server.AtomicUpdate;
import com.google.gwtorm.server.OrmException;
import com.google.inject.Inject;
import com.google.inject.Provider;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.util.Collections;

public class Restore implements RestModifyView<ChangeResource, Input> {
    private static final Logger log = LoggerFactory.getLogger(Restore.class);

    private final ChangeHooks hooks;
    private final RestoredSender.Factory restoredSenderFactory;
    private final Provider<ReviewDb> dbProvider;
    private final ChangeJson json;
    private final ChangeIndexer indexer;

    public static class Input {
        @DefaultInput
        public String message;
    }

    @Inject
    Restore(ChangeHooks hooks, RestoredSender.Factory restoredSenderFactory, Provider<ReviewDb> dbProvider,
            ChangeJson json, ChangeIndexer indexer) {
        this.hooks = hooks;
        this.restoredSenderFactory = restoredSenderFactory;
        this.dbProvider = dbProvider;
        this.json = json;
        this.indexer = indexer;
    }

    @Override
    public Object apply(ChangeResource req, Input input) throws Exception {
        ChangeControl control = req.getControl();
        IdentifiedUser caller = (IdentifiedUser) control.getCurrentUser();
        Change change = req.getChange();
        if (!control.canRestore()) {
            throw new AuthException("restore not permitted");
        } else if (change.getStatus() != Status.ABANDONED) {
            throw new ResourceConflictException("change is " + status(change));
        }

        ChangeMessage message;
        ReviewDb db = dbProvider.get();
        db.changes().beginTransaction(change.getId());
        try {
            change = db.changes().atomicUpdate(change.getId(), new AtomicUpdate<Change>() {
                @Override
                public Change update(Change change) {
                    if (change.getStatus() == Change.Status.ABANDONED) {
                        change.setStatus(Change.Status.NEW);
                        ChangeUtil.updated(change);
                        return change;
                    }
                    return null;
                }
            });
            if (change == null) {
                throw new ResourceConflictException(
                        "change is " + status(db.changes().get(req.getChange().getId())));
            }
            indexer.index(change);
            message = newMessage(input, caller, change);
            db.changeMessages().insert(Collections.singleton(message));
            new ApprovalsUtil(db).syncChangeStatus(change);
            db.commit();
        } finally {
            db.rollback();
        }

        try {
            ReplyToChangeSender cm = restoredSenderFactory.create(change);
            cm.setFrom(caller.getAccountId());
            cm.setChangeMessage(message);
            cm.send();
        } catch (Exception e) {
            log.error("Cannot email update for change " + change.getChangeId(), e);
        }
        hooks.doChangeRestoredHook(change, caller.getAccount(), db.patchSets().get(change.currentPatchSetId()),
                Strings.emptyToNull(input.message), dbProvider.get());
        return json.format(change);
    }

    private ChangeMessage newMessage(Input input, IdentifiedUser caller, Change change) throws OrmException {
        StringBuilder msg = new StringBuilder();
        msg.append("Restored");
        if (!Strings.nullToEmpty(input.message).trim().isEmpty()) {
            msg.append("\n\n");
            msg.append(input.message.trim());
        }

        ChangeMessage message = new ChangeMessage(
                new ChangeMessage.Key(change.getId(), ChangeUtil.messageUUID(dbProvider.get())),
                caller.getAccountId(), change.getLastUpdatedOn(), change.currentPatchSetId());
        message.setMessage(msg.toString());
        return message;
    }

    private static String status(Change change) {
        return change != null ? change.getStatus().name().toLowerCase() : "deleted";
    }
}