git-dd 0.2.2 → 0.2.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: f01e84c0cc0fb17ab8335ecd63b9f27a380c2daf
-  data.tar.gz: 0fda2df25f06cbcb8dc43ef2bb4d7c875fab22d2
+  metadata.gz: 97dc4a649420571ccf91bc81296d361f7d5cf391
+  data.tar.gz: aa5ab72e486ce86dc61918d3eb3971c0c910527c
 SHA512:
-  metadata.gz: 8488bf1e911fe6e4526c4cd5f22ef41fdfeaf962c911711b8a7fde99901924e1019e34c7b4af80c71e857c1d50e9573d7428803deb8bb8ca79aee5a66c03068b
-  data.tar.gz: 8775c1cd9dab58b06d785a06c32fb019767a9ca771ccb64757fe82eaf1b5990d8f61f0b8ad0237104406d53a4df2c594e6dd9cfb1873a06b10478df2e6bd6955
+  metadata.gz: 9dd51a8d265e0649b8d01f5ce17ce0b88a69436dbfda4bb984a59266eb07f0c3e434d8ae07c9128d4e626aa22d581d61ae48e79eb8ccbcbada90f6418d8df69f
+  data.tar.gz: b65485880763ff0e485e964a93c47ff307c6cddf6c7d36e9e8e12b789fb2c5c7b3bd8b396e7541ab4d0911e97950a795e0faddc66da23f130912e0b77ca9bfda

data/lib/const.rb

@@ -0,0 +1,4 @@
+# encoding: utf-8
+
+NO_BRANCH_SELECTED = "No branch selected"
+ONLY_ONE_BRANCH = "You only have one branch"

data/lib/git-dd.rb

@@ -2,16 +2,24 @@
 
 require 'tty-prompt'
 require 'rainbow/ext/string'
+require 'const.rb'
 
 class GitDD
-  def run
+  attr_accessor :prompt
+
+  def run(test_prompt = nil)
     branch_names = `git branch`
     return if $?.exitstatus != 0
 
     branch_names = branch_names.split("\n")
 
     branches_with_more_info = `git branch -vv`
-    branches_with_more_info = branches_with_more_info.split("\n")
+    begin
+      branches_with_more_info = branches_with_more_info.split("\n")
+    rescue
+      branches_with_more_info = `git branch`
+      branches_with_more_info = branches_with_more_info.split("\n")
+    end
 
     return if branch_names.size != branches_with_more_info.size
 
@@ -19,15 +27,18 @@ class GitDD
     branch_names.each_with_index { |b, i| branches[b] = branches_with_more_info[i] }
 
     if branches.size == 1
-      puts "You only have one branch."
-      return
+      return print(ONLY_ONE_BRANCH)
     end
 
     branches = branches.select { |k, v| k != current_branch_with_mark && !k.include?("* (HEAD detached at") }
 
     puts "Current branch is: #{current_branch.color(:green)}"
 
-    prompt = TTY::Prompt.new(interrupt: :exit)
+    if test_prompt
+      prompt = test_prompt
+    else
+      prompt = TTY::Prompt.new(interrupt: :exit)
+    end
 
     prompt.on(:keypress) do |event|
       if event.value == 'q'
@@ -40,8 +51,7 @@ class GitDD
     end
 
     if branches_to_delete.size == 0
-      puts "No branch selected"
-      return
+      return print(NO_BRANCH_SELECTED)
     end
 
     branches_to_delete.each { |b| puts b.color(:red) }
@@ -53,6 +63,8 @@ class GitDD
         puts `git branch -D #{b}`.chomp.color(:yellow)
       end
     end
+
+    return branches_to_delete
   end
 
   private
@@ -64,4 +76,9 @@ class GitDD
   def current_branch
     @current_branch ||= `git rev-parse --abbrev-ref HEAD`.chomp
   end
+
+  def print(return_message)
+    puts return_message
+    return_message
+  end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: git-dd
 version: !ruby/object:Gem::Version
-  version: 0.2.2
+  version: 0.2.3
 platform: ruby
 authors:
 - Weiqing Chu
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2017-07-16 00:00:00.000000000 Z
+date: 2017-07-19 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: tty-prompt
@@ -38,6 +38,34 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.6'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.6'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 description: 
 email: cwq1913@gmail.com
 executables:
@@ -46,6 +74,7 @@ extensions: []
 extra_rdoc_files: []
 files:
 - bin/git-dd
+- lib/const.rb
 - lib/git-dd.rb
 homepage: https://github.com/Sanster/git-dd
 licenses: