lib/git_wrapper/commands/branch.rb in git_wrapper-1.0.2 vs lib/git_wrapper/commands/branch.rb in git_wrapper-1.0.3

- old
+ new

@@ -1,73 +1,73 @@ -module GitWrapper - module Commands - class Branch < Git - - def create(name) - @mode = :create - @name = name - self - end - - def from(commit) - @commit = commit - self - end - - def remove(name) - @mode = :remove - @name = name - self - end - - def remote(remote) - @remote = remote - self - end - - def list - @mode = :list - self - end - - def current - @mode = :current - self - end - - def command - command = 'branch ' - - if @mode == :create - command += "#{@name} #{@commit.nil? ? '' : @commit}" - elsif @mode == :remove - if @remote.nil? - command += "-D #{@name}" - else - command = "push #{@remote} --delete #{@name}" - end - elsif @mode == :list || @mode == :current - command += '-a' - else - raise 'Unespecified branch mode' - end - - command - end - - def result - return result_list if @mode == :list - return result_current if @mode == :current - super - end - - def result_list - output.split("\n").map{|b| b[2..b.length]} - end - - def result_current - output.split("\n").select{|b| b.start_with?('*')}.map{|b| b[2..b.length]}.first - end - - end - end +module GitWrapper + module Commands + class Branch < Git + + def create(name) + @mode = :create + @name = name + self + end + + def from(commit) + @commit = commit + self + end + + def remove(name) + @mode = :remove + @name = name + self + end + + def remote(remote) + @remote = remote + self + end + + def list + @mode = :list + self + end + + def current + @mode = :current + self + end + + def command + command = 'branch ' + + if @mode == :create + command += "#{@name} #{@commit.nil? ? '' : @commit}" + elsif @mode == :remove + if @remote.nil? + command += "-D #{@name}" + else + command = "push #{@remote} --delete #{@name}" + end + elsif @mode == :list || @mode == :current + command += '-a' + else + raise 'Unespecified branch mode' + end + + command + end + + def result + return result_list if @mode == :list + return result_current if @mode == :current + super + end + + def result_list + output.split("\n").map{|b| b[2..b.length]} + end + + def result_current + output.split("\n").select{|b| b.start_with?('*')}.map{|b| b[2..b.length]}.first + end + + end + end end \ No newline at end of file