Line data    Source code

       1             : // Copyright (C) 2012 The Android Open Source Project
       2             : //
       3             : // Licensed under the Apache License, Version 2.0 (the "License");
       4             : // you may not use this file except in compliance with the License.
       5             : // You may obtain a copy of the License at
       6             : //
       7             : // http://www.apache.org/licenses/LICENSE-2.0
       8             : //
       9             : // Unless required by applicable law or agreed to in writing, software
      10             : // distributed under the License is distributed on an "AS IS" BASIS,
      11             : // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
      12             : // See the License for the specific language governing permissions and
      13             : // limitations under the License.
      14             : 
      15             : package com.google.gerrit.server.restapi.change;
      16             : 
      17             : import com.google.gerrit.entities.Change;
      18             : import com.google.gerrit.extensions.api.changes.DeleteReviewerInput;
      19             : import com.google.gerrit.extensions.api.changes.NotifyHandling;
      20             : import com.google.gerrit.extensions.restapi.Response;
      21             : import com.google.gerrit.extensions.restapi.RestApiException;
      22             : import com.google.gerrit.extensions.restapi.RestModifyView;
      23             : import com.google.gerrit.server.change.DeleteReviewerByEmailOp;
      24             : import com.google.gerrit.server.change.DeleteReviewerOp;
      25             : import com.google.gerrit.server.change.NotifyResolver;
      26             : import com.google.gerrit.server.change.ReviewerResource;
      27             : import com.google.gerrit.server.update.BatchUpdate;
      28             : import com.google.gerrit.server.update.BatchUpdateOp;
      29             : import com.google.gerrit.server.update.UpdateException;
      30             : import com.google.gerrit.server.util.time.TimeUtil;
      31             : import com.google.inject.Inject;
      32             : import com.google.inject.Singleton;
      33             : 
      34             : @Singleton
      35             : public class DeleteReviewer implements RestModifyView<ReviewerResource, DeleteReviewerInput> {
      36             :   private final BatchUpdate.Factory updateFactory;
      37             :   private final DeleteReviewerOp.Factory deleteReviewerOpFactory;
      38             :   private final DeleteReviewerByEmailOp.Factory deleteReviewerByEmailOpFactory;
      39             : 
      40             :   @Inject
      41             :   DeleteReviewer(
      42             :       BatchUpdate.Factory updateFactory,
      43             :       DeleteReviewerOp.Factory deleteReviewerOpFactory,
      44         142 :       DeleteReviewerByEmailOp.Factory deleteReviewerByEmailOpFactory) {
      45         142 :     this.updateFactory = updateFactory;
      46         142 :     this.deleteReviewerOpFactory = deleteReviewerOpFactory;
      47         142 :     this.deleteReviewerByEmailOpFactory = deleteReviewerByEmailOpFactory;
      48         142 :   }
      49             : 
      50             :   @Override
      51             :   public Response<Object> apply(ReviewerResource rsrc, DeleteReviewerInput input)
      52             :       throws RestApiException, UpdateException {
      53          15 :     if (input == null) {
      54           2 :       input = new DeleteReviewerInput();
      55             :     }
      56             : 
      57          15 :     try (BatchUpdate bu =
      58          15 :         updateFactory.create(
      59          15 :             rsrc.getChangeResource().getProject(),
      60          15 :             rsrc.getChangeResource().getUser(),
      61          15 :             TimeUtil.now())) {
      62          15 :       bu.setNotify(getNotify(rsrc.getChange(), input));
      63             :       BatchUpdateOp op;
      64          15 :       if (rsrc.isByEmail()) {
      65           7 :         op = deleteReviewerByEmailOpFactory.create(rsrc.getReviewerByEmail());
      66             :       } else {
      67          15 :         op = deleteReviewerOpFactory.create(rsrc.getReviewerUser().getAccount(), input);
      68             :       }
      69          15 :       bu.addOp(rsrc.getChange().getId(), op);
      70          15 :       bu.execute();
      71             :     }
      72          15 :     return Response.none();
      73             :   }
      74             : 
      75             :   private static NotifyResolver.Result getNotify(Change change, DeleteReviewerInput input) {
      76          15 :     NotifyHandling notifyHandling = input.notify;
      77          15 :     if (notifyHandling == null) {
      78          15 :       notifyHandling = change.isWorkInProgress() ? NotifyHandling.NONE : NotifyHandling.ALL;
      79             :     }
      80          15 :     return NotifyResolver.Result.create(notifyHandling);
      81             :   }
      82             : }