From 76002ca549ccdec733fd490d634c5bbf0749e183 Mon Sep 17 00:00:00 2001
From: Junil Kim <logyourself@gmail.com>
Date: Wed, 2 Nov 2016 11:11:48 +0900
Subject: [PATCH] support shallow clone a repo on a specific branch. (#91)

ex) git clone <git-url> --depth 1 --branch <name>
---
 README.md              |  2 +-
 src/index.coffee       | 16 +++++++++++-----
 test/index.test.coffee | 15 +++++++++++++++
 3 files changed, 27 insertions(+), 6 deletions(-)

diff --git a/README.md b/README.md
index 9b33729..26628c3 100644
--- a/README.md
+++ b/README.md
@@ -42,7 +42,7 @@ Clone a repository:
 
     git = require 'gift'
 
-    git.clone "git@host:path/to/remote/repo.git", "path/to/local/clone/repo", depth, (err, _repo) ->
+    git.clone "git@host:path/to/remote/repo.git", "path/to/local/clone/repo", depth, branch, (err, _repo) ->
       repo = _repo
       # => #<Repo>
 
diff --git a/src/index.coffee b/src/index.coffee
index 201dc28..816f375 100644
--- a/src/index.coffee
+++ b/src/index.coffee
@@ -35,14 +35,20 @@ Git.init = (path, bare, callback) ->
 # depth      - The specified number of revisions of shallow clone
 # callback   - Receives `(err, repo)`.
 #
-Git.clone = (repository, path, depth = 0, callback) ->
+Git.clone = (repository, path, depth = 0, branch = null, callback) ->
+  if typeof branch is 'function'
+    callback = branch
+    branch = null
   if typeof depth is 'function'
     callback = depth
     depth = 0
-  if depth is 0 or typeof depth isnt 'number'
-    bash = "git clone \"#{repository}\" \"#{path}\""
-  else
-    bash = "git clone \"#{repository}\" \"#{path}\" --depth \"#{depth}\""
+  bash = "git clone \"#{repository}\" \"#{path}\""
+
+  if branch isnt null and typeof branch is 'string'
+    bash += " --branch \"#{branch}\""
+  if depth isnt 0 and typeof depth is 'number'
+    bash += " --depth \"#{depth}\""
+
   exec bash, (err, stdout, stderr) ->
     return callback err if err
     return callback err, (new Repo path, false, { maxBuffer: Git.maxBuffer })
diff --git a/test/index.test.coffee b/test/index.test.coffee
index 711007c..6fb62e1 100644
--- a/test/index.test.coffee
+++ b/test/index.test.coffee
@@ -71,3 +71,18 @@ describe "git", ->
         done()
     after (done) ->
       exec "rm -rf #{newRepositoryDir}", done
+
+  describe "clone() with depth and branch", ->
+    @timeout 30000
+    repo = null
+    newRepositoryDir = "#{__dirname}/fixtures/clone_depth_branch"
+    before (done) ->
+      git.clone "https://github.com/notatestuser/gift.git", newRepositoryDir, 1, "develop", (err, _repo) ->
+        repo = _repo
+        done err
+    it "clone a repository", (done) ->
+      repo.should.be.an.instanceof Repo
+      repo.branch "develop", (err, head) ->
+        done err
+    after (done) ->
+      exec "rm -rf #{newRepositoryDir}", done