Line data    Source code

       1             : // Copyright (C) 2015 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 static com.google.gerrit.server.project.ProjectCache.illegalState;
      18             : 
      19             : import com.google.gerrit.extensions.restapi.BinaryResult;
      20             : import com.google.gerrit.extensions.restapi.ResourceNotFoundException;
      21             : import com.google.gerrit.extensions.restapi.Response;
      22             : import com.google.gerrit.extensions.restapi.RestReadView;
      23             : import com.google.gerrit.server.change.FileContentUtil;
      24             : import com.google.gerrit.server.change.FileResource;
      25             : import com.google.gerrit.server.change.RevisionResource;
      26             : import com.google.gerrit.server.project.NoSuchChangeException;
      27             : import com.google.gerrit.server.project.ProjectCache;
      28             : import com.google.inject.Inject;
      29             : import java.io.IOException;
      30             : import org.kohsuke.args4j.Option;
      31             : 
      32             : public class DownloadContent implements RestReadView<FileResource> {
      33             :   private final FileContentUtil fileContentUtil;
      34             :   private final ProjectCache projectCache;
      35             : 
      36             :   @Option(name = "--parent")
      37             :   private Integer parent;
      38             : 
      39             :   @Inject
      40           1 :   DownloadContent(FileContentUtil fileContentUtil, ProjectCache projectCache) {
      41           1 :     this.fileContentUtil = fileContentUtil;
      42           1 :     this.projectCache = projectCache;
      43           1 :   }
      44             : 
      45             :   @Override
      46             :   public Response<BinaryResult> apply(FileResource rsrc)
      47             :       throws ResourceNotFoundException, IOException, NoSuchChangeException {
      48           1 :     String path = rsrc.getPatchKey().fileName();
      49           1 :     RevisionResource rev = rsrc.getRevision();
      50           1 :     return Response.ok(
      51           1 :         fileContentUtil.downloadContent(
      52           1 :             projectCache.get(rev.getProject()).orElseThrow(illegalState(rev.getProject())),
      53           1 :             rev.getPatchSet().commitId(),
      54             :             path,
      55             :             parent));
      56             :   }
      57             : }