diff options
Diffstat (limited to 'java/com/google/gerrit/server/restapi/change/SetWorkInProgress.java')
-rw-r--r-- | java/com/google/gerrit/server/restapi/change/SetWorkInProgress.java | 109 |
1 files changed, 109 insertions, 0 deletions
diff --git a/java/com/google/gerrit/server/restapi/change/SetWorkInProgress.java b/java/com/google/gerrit/server/restapi/change/SetWorkInProgress.java new file mode 100644 index 0000000000..da5d8bbe9e --- /dev/null +++ b/java/com/google/gerrit/server/restapi/change/SetWorkInProgress.java @@ -0,0 +1,109 @@ +// Copyright (C) 2017 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.restapi.change; + +import static com.google.gerrit.extensions.conditions.BooleanCondition.and; +import static com.google.gerrit.extensions.conditions.BooleanCondition.or; + +import com.google.gerrit.extensions.restapi.ResourceConflictException; +import com.google.gerrit.extensions.restapi.Response; +import com.google.gerrit.extensions.restapi.RestApiException; +import com.google.gerrit.extensions.webui.UiAction; +import com.google.gerrit.reviewdb.client.Change; +import com.google.gerrit.reviewdb.client.Change.Status; +import com.google.gerrit.reviewdb.server.ReviewDb; +import com.google.gerrit.server.ChangeUtil; +import com.google.gerrit.server.CurrentUser; +import com.google.gerrit.server.change.ChangeResource; +import com.google.gerrit.server.change.WorkInProgressOp; +import com.google.gerrit.server.change.WorkInProgressOp.Input; +import com.google.gerrit.server.permissions.GlobalPermission; +import com.google.gerrit.server.permissions.PermissionBackend; +import com.google.gerrit.server.permissions.PermissionBackendException; +import com.google.gerrit.server.permissions.ProjectPermission; +import com.google.gerrit.server.update.BatchUpdate; +import com.google.gerrit.server.update.RetryHelper; +import com.google.gerrit.server.update.RetryingRestModifyView; +import com.google.gerrit.server.update.UpdateException; +import com.google.gerrit.server.util.time.TimeUtil; +import com.google.inject.Inject; +import com.google.inject.Provider; +import com.google.inject.Singleton; + +@Singleton +public class SetWorkInProgress extends RetryingRestModifyView<ChangeResource, Input, Response<?>> + implements UiAction<ChangeResource> { + private final WorkInProgressOp.Factory opFactory; + private final Provider<ReviewDb> db; + private final PermissionBackend permissionBackend; + private final Provider<CurrentUser> user; + + @Inject + SetWorkInProgress( + WorkInProgressOp.Factory opFactory, + RetryHelper retryHelper, + Provider<ReviewDb> db, + PermissionBackend permissionBackend, + Provider<CurrentUser> user) { + super(retryHelper); + this.opFactory = opFactory; + this.db = db; + this.permissionBackend = permissionBackend; + this.user = user; + } + + @Override + protected Response<?> applyImpl( + BatchUpdate.Factory updateFactory, ChangeResource rsrc, Input input) + throws RestApiException, UpdateException, PermissionBackendException { + WorkInProgressOp.checkPermissions(permissionBackend, user.get(), rsrc.getChange()); + + Change change = rsrc.getChange(); + if (change.getStatus() != Status.NEW) { + throw new ResourceConflictException("change is " + ChangeUtil.status(change)); + } + + if (change.isWorkInProgress()) { + throw new ResourceConflictException("change is already work in progress"); + } + + try (BatchUpdate bu = + updateFactory.create(db.get(), rsrc.getProject(), rsrc.getUser(), TimeUtil.nowTs())) { + bu.addOp(rsrc.getChange().getId(), opFactory.create(true, input)); + bu.execute(); + return Response.ok(""); + } + } + + @Override + public Description getDescription(ChangeResource rsrc) { + return new Description() + .setLabel("WIP") + .setTitle("Set Work In Progress") + .setVisible( + and( + rsrc.getChange().getStatus() == Status.NEW && !rsrc.getChange().isWorkInProgress(), + or( + rsrc.isUserOwner(), + or( + permissionBackend + .currentUser() + .testCond(GlobalPermission.ADMINISTRATE_SERVER), + permissionBackend + .currentUser() + .project(rsrc.getProject()) + .testCond(ProjectPermission.WRITE_CONFIG))))); + } +} |