// 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.restapi.change;

import com.google.gerrit.extensions.common.Input;
import com.google.gerrit.extensions.registration.DynamicMap;
import com.google.gerrit.extensions.restapi.AcceptsPost;
import com.google.gerrit.extensions.restapi.AuthException;
import com.google.gerrit.extensions.restapi.ChildCollection;
import com.google.gerrit.extensions.restapi.IdString;
import com.google.gerrit.extensions.restapi.ResourceConflictException;
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.extensions.restapi.RestView;
import com.google.gerrit.reviewdb.client.Project;
import com.google.gerrit.server.change.ChangeEditResource;
import com.google.gerrit.server.change.ChangeResource;
import com.google.gerrit.server.edit.ChangeEditModifier;
import com.google.gerrit.server.git.GitRepositoryManager;
import com.google.gerrit.server.permissions.PermissionBackendException;
import com.google.gerrit.server.project.InvalidChangeOperationException;
import com.google.gwtorm.server.OrmException;
import com.google.inject.Inject;
import com.google.inject.Singleton;
import java.io.IOException;
import org.eclipse.jgit.lib.Repository;

@Singleton
public class RebaseChangeEdit
    implements ChildCollection<ChangeResource, ChangeEditResource.Rebase>,
        AcceptsPost<ChangeResource> {

  private final DynamicMap<RestView<ChangeEditResource.Rebase>> views;
  private final Rebase rebase;

  @Inject
  RebaseChangeEdit(DynamicMap<RestView<ChangeEditResource.Rebase>> views, Rebase rebase) {
    this.views = views;
    this.rebase = rebase;
  }

  @Override
  public DynamicMap<RestView<ChangeEditResource.Rebase>> views() {
    return views;
  }

  @Override
  public RestView<ChangeResource> list() throws ResourceNotFoundException {
    throw new ResourceNotFoundException();
  }

  @Override
  public ChangeEditResource.Rebase parse(ChangeResource parent, IdString id)
      throws ResourceNotFoundException {
    throw new ResourceNotFoundException();
  }

  @Override
  public Rebase post(ChangeResource parent) throws RestApiException {
    return rebase;
  }

  @Singleton
  public static class Rebase implements RestModifyView<ChangeResource, Input> {

    private final GitRepositoryManager repositoryManager;
    private final ChangeEditModifier editModifier;

    @Inject
    Rebase(GitRepositoryManager repositoryManager, ChangeEditModifier editModifier) {
      this.repositoryManager = repositoryManager;
      this.editModifier = editModifier;
    }

    @Override
    public Response<?> apply(ChangeResource rsrc, Input in)
        throws AuthException, ResourceConflictException, IOException, OrmException,
            PermissionBackendException {
      Project.NameKey project = rsrc.getProject();
      try (Repository repository = repositoryManager.openRepository(project)) {
        editModifier.rebaseEdit(repository, rsrc.getNotes());
      } catch (InvalidChangeOperationException e) {
        throw new ResourceConflictException(e.getMessage());
      }
      return Response.none();
    }
  }
}

@@ -1256,68 +1256,8 @@
 on;%0A
-import com.google.gerrit.extensions.restapi.RestModifyView;%0A
 impo
@@ -1722,32 +1722,199 @@
 ationException;%0A
+import com.google.gerrit.server.update.BatchUpdate;%0Aimport com.google.gerrit.server.update.RetryHelper;%0Aimport com.google.gerrit.server.update.RetryingRestModifyView;%0A
 import com.googl
@@ -3059,27 +3059,32 @@
  Rebase 
-implement
+extend
 s 
+Retrying
 RestModi
@@ -3111,21 +3111,33 @@
 e, Input
+, Response%3C?%3E
 %3E %7B%0A
-%0A
     priv
@@ -3261,16 +3261,58 @@
  Rebase(
+%0A        RetryHelper retryHelper,%0A        
 GitRepos
@@ -3342,16 +3342,24 @@
 Manager,
+%0A       
  ChangeE
@@ -3382,24 +3382,50 @@
 Modifier) %7B%0A
+      super(retryHelper);%0A
       this.r
@@ -3528,21 +3528,24 @@
 de%0A    p
-ublic
+rotected
  Respons
@@ -3554,17 +3554,65 @@
 ?%3E apply
-(
+Impl(%0A        BatchUpdate.Factory updateFactory, 
 ChangeRe

Use RetryingRestModifyView for RebaseChangeEdit.Rebase