Line data    Source code

       1             : // Copyright (C) 2016 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.common.Nullable;
      18             : import com.google.gerrit.entities.Account;
      19             : import com.google.gerrit.entities.Change;
      20             : import com.google.gerrit.extensions.common.AccountInfo;
      21             : import com.google.gerrit.extensions.common.Input;
      22             : import com.google.gerrit.extensions.restapi.Response;
      23             : import com.google.gerrit.extensions.restapi.RestApiException;
      24             : import com.google.gerrit.extensions.restapi.RestModifyView;
      25             : import com.google.gerrit.server.ChangeMessagesUtil;
      26             : import com.google.gerrit.server.IdentifiedUser;
      27             : import com.google.gerrit.server.account.AccountLoader;
      28             : import com.google.gerrit.server.account.AccountState;
      29             : import com.google.gerrit.server.change.ChangeResource;
      30             : import com.google.gerrit.server.extensions.events.AssigneeChanged;
      31             : import com.google.gerrit.server.notedb.ChangeUpdate;
      32             : import com.google.gerrit.server.permissions.ChangePermission;
      33             : import com.google.gerrit.server.permissions.PermissionBackendException;
      34             : import com.google.gerrit.server.update.BatchUpdate;
      35             : import com.google.gerrit.server.update.BatchUpdateOp;
      36             : import com.google.gerrit.server.update.ChangeContext;
      37             : import com.google.gerrit.server.update.PostUpdateContext;
      38             : import com.google.gerrit.server.update.UpdateException;
      39             : import com.google.gerrit.server.util.AccountTemplateUtil;
      40             : import com.google.gerrit.server.util.time.TimeUtil;
      41             : import com.google.inject.Inject;
      42             : import com.google.inject.Singleton;
      43             : 
      44             : @Singleton
      45             : public class DeleteAssignee implements RestModifyView<ChangeResource, Input> {
      46             :   private final BatchUpdate.Factory updateFactory;
      47             :   private final ChangeMessagesUtil cmUtil;
      48             :   private final AssigneeChanged assigneeChanged;
      49             :   private final IdentifiedUser.GenericFactory userFactory;
      50             :   private final AccountLoader.Factory accountLoaderFactory;
      51             : 
      52             :   @Inject
      53             :   DeleteAssignee(
      54             :       BatchUpdate.Factory updateFactory,
      55             :       ChangeMessagesUtil cmUtil,
      56             :       AssigneeChanged assigneeChanged,
      57             :       IdentifiedUser.GenericFactory userFactory,
      58         145 :       AccountLoader.Factory accountLoaderFactory) {
      59         145 :     this.updateFactory = updateFactory;
      60         145 :     this.cmUtil = cmUtil;
      61         145 :     this.assigneeChanged = assigneeChanged;
      62         145 :     this.userFactory = userFactory;
      63         145 :     this.accountLoaderFactory = accountLoaderFactory;
      64         145 :   }
      65             : 
      66             :   @Override
      67             :   public Response<AccountInfo> apply(ChangeResource rsrc, Input input)
      68             :       throws RestApiException, UpdateException, PermissionBackendException {
      69           2 :     rsrc.permissions().check(ChangePermission.EDIT_ASSIGNEE);
      70             : 
      71           2 :     try (BatchUpdate bu = updateFactory.create(rsrc.getProject(), rsrc.getUser(), TimeUtil.now())) {
      72           2 :       Op op = new Op();
      73           2 :       bu.addOp(rsrc.getChange().getId(), op);
      74           2 :       bu.execute();
      75           2 :       Account.Id deletedAssignee = op.getDeletedAssignee();
      76           2 :       return deletedAssignee == null
      77           2 :           ? Response.none()
      78           2 :           : Response.ok(accountLoaderFactory.create(true).fillOne(deletedAssignee));
      79             :     }
      80             :   }
      81             : 
      82           2 :   private class Op implements BatchUpdateOp {
      83             :     private Change change;
      84             :     private AccountState deletedAssignee;
      85             : 
      86             :     @Override
      87             :     public boolean updateChange(ChangeContext ctx) throws RestApiException {
      88           2 :       change = ctx.getChange();
      89           2 :       ChangeUpdate update = ctx.getUpdate(change.currentPatchSetId());
      90           2 :       Account.Id currentAssigneeId = change.getAssignee();
      91           2 :       if (currentAssigneeId == null) {
      92           2 :         return false;
      93             :       }
      94           1 :       IdentifiedUser deletedAssigneeUser = userFactory.create(currentAssigneeId);
      95           1 :       deletedAssignee = deletedAssigneeUser.state();
      96           1 :       update.removeAssignee();
      97           1 :       addMessage(ctx, deletedAssigneeUser);
      98           1 :       return true;
      99             :     }
     100             : 
     101             :     @Nullable
     102             :     public Account.Id getDeletedAssignee() {
     103           2 :       return deletedAssignee != null ? deletedAssignee.account().id() : null;
     104             :     }
     105             : 
     106             :     private void addMessage(ChangeContext ctx, IdentifiedUser deletedAssignee) {
     107           1 :       cmUtil.setChangeMessage(
     108             :           ctx,
     109             :           "Assignee deleted: "
     110           1 :               + AccountTemplateUtil.getAccountTemplate(deletedAssignee.getAccountId()),
     111             :           ChangeMessagesUtil.TAG_DELETE_ASSIGNEE);
     112           1 :     }
     113             : 
     114             :     @Override
     115             :     public void postUpdate(PostUpdateContext ctx) {
     116           2 :       assigneeChanged.fire(
     117           2 :           ctx.getChangeData(change), ctx.getAccount(), deletedAssignee, ctx.getWhen());
     118           2 :     }
     119             :   }
     120             : }