class Git::Branches

def [](symbol)
  @branches[symbol.to_s]
end

def each(&block)
  @branches.values.each(&block)
end

def initialize(base)
  @branches = {}
  
  @base = base
        
  @base.lib.branches_all.each do |b|
    @branches[b[0]] = Git::Branch.new(@base, b[0])
  end
end

def local
  self.select { |b| !b.remote }
end

def remote
  self.select { |b| b.remote }
end

def size
  @branches.size
end    

def to_s
  out = ''
  @branches.each do |k, b|
    out << (b.current ? '* ' : '  ') << b.to_s << "\n"
  end
  out
end