Download snippet repository as zip or tar.gz archive file

Author: takezoe

src/main/scala/app/GistController.scala

@@ -3,7 +3,9 @@ package app
 import java.io.File
 import jp.sf.amateras.scalatra.forms._
 import model.{GistUser, Gist, Account}
-import service.{AccountService, GistService}
+import org.apache.commons.io.FileUtils
+import org.scalatra.BadRequest
+import service.{RepositoryService, AccountService, GistService}
 import util.Directory._
 import util._
 import util.ControlUtil._
@@ -221,6 +223,37 @@ trait GistControllerBase extends ControllerBase {
     } else NotFound
   }
 
+  get("/gist/:userName/:repoName/download/*"){
+    val format = multiParams("splat").head match {
+      case name if name.endsWith(".zip")    => "zip"
+      case name if name.endsWith(".tar.gz") => "tar.gz"
+    }
+
+    val userName = params("userName")
+    val repoName = params("repoName")
+
+    val workDir = getDownloadWorkDir(userName, repoName, session.getId)
+    if(workDir.exists) {
+      FileUtils.deleteDirectory(workDir)
+    }
+    workDir.mkdirs
+
+    using(Git.open(new File(GistRepoDir, userName + "/" + repoName))){ git =>
+      val revCommit = JGitUtil.getRevCommitFromId(git, git.getRepository.resolve("master"))
+
+      contentType = "application/octet-stream"
+      response.setHeader("Content-Disposition", s"attachment; filename=${repoName}.${format}")
+      response.setBufferSize(1024 * 1024);
+
+      git.archive
+        .setFormat(format)
+        .setTree(revCommit.getTree)
+        .setOutputStream(response.getOutputStream)
+        .call()
+    }
+  }
+
+
   get("/gist/:userName"){
     _gist(params("userName"))
   }

src/main/twirl/gist/menu.scala.html

@@ -38,4 +38,25 @@
       "Revision")
     <li style="height: 12px"><div class="gradient pull-left" style="height: 12px"></div></li>
   </ul>
+  @*
+  <div class="small">
+    <strong id="repository-url-proto">HTTP</strong> <span class="mute">clone URL</span>
+  </div>
+  @helper.html.copy("repository-url-copy", "http://localhost:8080/"){
+    <input type="text" value="http://localhost:8080/" id="repository-url" readonly>
+  }
+  @if(settings.ssh && loginAccount.isDefined){
+    <div class="small">
+      <span class="mute">You can clone <a href="javascript:void(0);" id="repository-url-http">HTTP</a> or <a href="javascript:void(0);" id="repository-url-ssh">SSH</a>.</span>
+    </div>
+  }
+  *@
+  <div style="margin-top: 10px;">
+    <a href="@path/gist/@{gist.userName}/@{gist.repositoryName}/download/@{gist.repositoryName}.zip"
+       class="btn btn-small" style="width: 147px;"><i class="icon-download-alt"></i>Download ZIP</a>
+  </div>
+  <div style="margin-top: 10px;">
+    <a href="@path/gist/@{gist.userName}/@{gist.repositoryName}/download/@{gist.repositoryName}.tar.gz"
+       class="btn btn-small" style="width: 147px;"><i class="icon-download-alt"></i>Download TAR.GZ</a>
+  </div>
 </div>