| // Copyright (C) 2014 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 static com.google.common.base.Preconditions.checkNotNull; |
| |
| import com.google.gerrit.common.TimeUtil; |
| import com.google.gerrit.common.data.LabelTypes; |
| import com.google.gerrit.extensions.api.changes.DeleteVoteInput; |
| import com.google.gerrit.extensions.api.changes.NotifyHandling; |
| import com.google.gerrit.extensions.restapi.AuthException; |
| import com.google.gerrit.extensions.restapi.BadRequestException; |
| import com.google.gerrit.extensions.restapi.MethodNotAllowedException; |
| import com.google.gerrit.extensions.restapi.ResourceNotFoundException; |
| import com.google.gerrit.extensions.restapi.Response; |
| import com.google.gerrit.extensions.restapi.RestApiException; |
| import com.google.gerrit.extensions.restapi.RestModifyView; |
| import com.google.gerrit.reviewdb.client.Account; |
| import com.google.gerrit.reviewdb.client.Change; |
| import com.google.gerrit.reviewdb.client.ChangeMessage; |
| import com.google.gerrit.reviewdb.client.LabelId; |
| import com.google.gerrit.reviewdb.client.PatchSet; |
| import com.google.gerrit.reviewdb.client.PatchSetApproval; |
| import com.google.gerrit.reviewdb.server.ReviewDb; |
| import com.google.gerrit.server.ApprovalsUtil; |
| import com.google.gerrit.server.ChangeMessagesUtil; |
| import com.google.gerrit.server.IdentifiedUser; |
| import com.google.gerrit.server.PatchSetUtil; |
| import com.google.gerrit.server.extensions.events.VoteDeleted; |
| import com.google.gerrit.server.mail.send.DeleteVoteSender; |
| import com.google.gerrit.server.mail.send.ReplyToChangeSender; |
| import com.google.gerrit.server.project.ChangeControl; |
| import com.google.gerrit.server.update.BatchUpdate; |
| import com.google.gerrit.server.update.BatchUpdateOp; |
| import com.google.gerrit.server.update.ChangeContext; |
| import com.google.gerrit.server.update.Context; |
| import com.google.gerrit.server.update.UpdateException; |
| import com.google.gerrit.server.util.LabelVote; |
| import com.google.gwtorm.server.OrmException; |
| import com.google.inject.Inject; |
| import com.google.inject.Provider; |
| import com.google.inject.Singleton; |
| import java.util.Collections; |
| import java.util.HashMap; |
| import java.util.Map; |
| import org.slf4j.Logger; |
| import org.slf4j.LoggerFactory; |
| |
| @Singleton |
| public class DeleteVote implements RestModifyView<VoteResource, DeleteVoteInput> { |
| private static final Logger log = LoggerFactory.getLogger(DeleteVote.class); |
| |
| private final Provider<ReviewDb> db; |
| private final BatchUpdate.Factory batchUpdateFactory; |
| private final ApprovalsUtil approvalsUtil; |
| private final PatchSetUtil psUtil; |
| private final ChangeMessagesUtil cmUtil; |
| private final IdentifiedUser.GenericFactory userFactory; |
| private final VoteDeleted voteDeleted; |
| private final DeleteVoteSender.Factory deleteVoteSenderFactory; |
| private final NotifyUtil notifyUtil; |
| |
| @Inject |
| DeleteVote( |
| Provider<ReviewDb> db, |
| BatchUpdate.Factory batchUpdateFactory, |
| ApprovalsUtil approvalsUtil, |
| PatchSetUtil psUtil, |
| ChangeMessagesUtil cmUtil, |
| IdentifiedUser.GenericFactory userFactory, |
| VoteDeleted voteDeleted, |
| DeleteVoteSender.Factory deleteVoteSenderFactory, |
| NotifyUtil notifyUtil) { |
| this.db = db; |
| this.batchUpdateFactory = batchUpdateFactory; |
| this.approvalsUtil = approvalsUtil; |
| this.psUtil = psUtil; |
| this.cmUtil = cmUtil; |
| this.userFactory = userFactory; |
| this.voteDeleted = voteDeleted; |
| this.deleteVoteSenderFactory = deleteVoteSenderFactory; |
| this.notifyUtil = notifyUtil; |
| } |
| |
| @Override |
| public Response<?> apply(VoteResource rsrc, DeleteVoteInput input) |
| throws RestApiException, UpdateException { |
| if (input == null) { |
| input = new DeleteVoteInput(); |
| } |
| if (input.label != null && !rsrc.getLabel().equals(input.label)) { |
| throw new BadRequestException("label must match URL"); |
| } |
| if (input.notify == null) { |
| input.notify = NotifyHandling.ALL; |
| } |
| ReviewerResource r = rsrc.getReviewer(); |
| Change change = r.getChange(); |
| |
| if (r.getRevisionResource() != null && !r.getRevisionResource().isCurrent()) { |
| throw new MethodNotAllowedException("Cannot delete vote on non-current patch set"); |
| } |
| |
| try (BatchUpdate bu = |
| batchUpdateFactory.create( |
| db.get(), change.getProject(), r.getControl().getUser(), TimeUtil.nowTs())) { |
| bu.addOp(change.getId(), new Op(r.getReviewerUser().getAccountId(), rsrc.getLabel(), input)); |
| bu.execute(); |
| } |
| |
| return Response.none(); |
| } |
| |
| private class Op implements BatchUpdateOp { |
| private final Account.Id accountId; |
| private final String label; |
| private final DeleteVoteInput input; |
| private ChangeMessage changeMessage; |
| private Change change; |
| private PatchSet ps; |
| private Map<String, Short> newApprovals = new HashMap<>(); |
| private Map<String, Short> oldApprovals = new HashMap<>(); |
| |
| private Op(Account.Id accountId, String label, DeleteVoteInput input) { |
| this.accountId = accountId; |
| this.label = label; |
| this.input = input; |
| } |
| |
| @Override |
| public boolean updateChange(ChangeContext ctx) |
| throws OrmException, AuthException, ResourceNotFoundException { |
| ChangeControl ctl = ctx.getControl(); |
| change = ctl.getChange(); |
| PatchSet.Id psId = change.currentPatchSetId(); |
| ps = psUtil.current(db.get(), ctl.getNotes()); |
| |
| boolean found = false; |
| LabelTypes labelTypes = ctx.getControl().getLabelTypes(); |
| |
| for (PatchSetApproval a : approvalsUtil.byPatchSetUser(ctx.getDb(), ctl, psId, accountId)) { |
| if (labelTypes.byLabel(a.getLabelId()) == null) { |
| continue; // Ignore undefined labels. |
| } else if (!a.getLabel().equals(label)) { |
| // Populate map for non-matching labels, needed by VoteDeleted. |
| newApprovals.put(a.getLabel(), a.getValue()); |
| continue; |
| } else if (!ctl.canRemoveReviewer(a)) { |
| throw new AuthException("delete vote not permitted"); |
| } |
| // Set the approval to 0 if vote is being removed. |
| newApprovals.put(a.getLabel(), (short) 0); |
| found = true; |
| |
| // Set old value, as required by VoteDeleted. |
| oldApprovals.put(a.getLabel(), a.getValue()); |
| break; |
| } |
| if (!found) { |
| throw new ResourceNotFoundException(); |
| } |
| |
| ctx.getUpdate(psId).removeApprovalFor(accountId, label); |
| ctx.getDb().patchSetApprovals().upsert(Collections.singleton(deletedApproval(ctx))); |
| |
| StringBuilder msg = new StringBuilder(); |
| msg.append("Removed "); |
| LabelVote.appendTo(msg, label, checkNotNull(oldApprovals.get(label))); |
| msg.append(" by ").append(userFactory.create(accountId).getNameEmail()).append("\n"); |
| changeMessage = |
| ChangeMessagesUtil.newMessage(ctx, msg.toString(), ChangeMessagesUtil.TAG_DELETE_VOTE); |
| cmUtil.addChangeMessage(ctx.getDb(), ctx.getUpdate(psId), changeMessage); |
| |
| return true; |
| } |
| |
| private PatchSetApproval deletedApproval(ChangeContext ctx) { |
| // Set the effective user to the account we're trying to remove, and don't |
| // set the real user; this preserves the calling user as the NoteDb |
| // committer. |
| return new PatchSetApproval( |
| new PatchSetApproval.Key(ps.getId(), accountId, new LabelId(label)), |
| (short) 0, |
| ctx.getWhen()); |
| } |
| |
| @Override |
| public void postUpdate(Context ctx) { |
| if (changeMessage == null) { |
| return; |
| } |
| |
| IdentifiedUser user = ctx.getIdentifiedUser(); |
| if (NotifyUtil.shouldNotify(input.notify, input.notifyDetails)) { |
| try { |
| ReplyToChangeSender cm = deleteVoteSenderFactory.create(ctx.getProject(), change.getId()); |
| cm.setFrom(user.getAccountId()); |
| cm.setChangeMessage(changeMessage.getMessage(), ctx.getWhen()); |
| cm.setNotify(input.notify); |
| cm.setAccountsToNotify(notifyUtil.resolveAccounts(input.notifyDetails)); |
| cm.send(); |
| } catch (Exception e) { |
| log.error("Cannot email update for change " + change.getId(), e); |
| } |
| } |
| |
| voteDeleted.fire( |
| change, |
| ps, |
| newApprovals, |
| oldApprovals, |
| input.notify, |
| changeMessage.getMessage(), |
| user.getAccount(), |
| ctx.getWhen()); |
| } |
| } |
| } |