GitBucket/src/main/scala/util/Directory.scala

package util

import java.io.File
import util.ControlUtil._

/**
 * Provides directories used by GitBucket.
 */
object Directory {

  val GitBucketHome = (scala.util.Properties.envOrNone("GITBUCKET_HOME") match {
    case Some(env) => new File(env)
    case None => new File(System.getProperty("user.home"), "gitbucket")
  }).getAbsolutePath

  val GitBucketConf = new File(GitBucketHome, "gitbucket.conf")
  
  val RepositoryHome = s"${GitBucketHome}/repositories"

  val DatabaseHome = s"${GitBucketHome}/data"
  
  /**
   * Repository names of the specified user.
   */
  def getRepositories(owner: String): List[String] =
    defining(new File(s"${RepositoryHome}/${owner}")){ dir =>
      if(dir.exists){
        dir.listFiles.filter { file =>
          file.isDirectory && !file.getName.endsWith(".wiki.git")
        }.map(_.getName.replaceFirst("\\.git$", "")).toList
      } else {
        Nil
      }
    }

  /**
   * Substance directory of the repository.
   */
  def getRepositoryDir(owner: String, repository: String): File =
    new File(s"${RepositoryHome}/${owner}/${repository}.git")

  /**
   * Directory for uploaded files by the specified user.
   */
  def getUserUploadDir(userName: String): File = new File(s"${GitBucketHome}/data/${userName}/files")

  /**
   * Root of temporary directories for the specified repository.
   */
  def getTemporaryDir(owner: String, repository: String): File =
    new File(s"${GitBucketHome}/tmp/${owner}/${repository}")

  /**
   * Temporary directory which is used to create an archive to download repository contents.
   */
  def getDownloadWorkDir(owner: String, repository: String, sessionId: String): File = 
    new File(getTemporaryDir(owner, repository), s"download/${sessionId}")
  
  /**
   * Substance directory of the wiki repository.
   */
  def getWikiRepositoryDir(owner: String, repository: String): File =
    new File(s"${RepositoryHome}/${owner}/${repository}.wiki.git")

}