module Git module Switcher class Menu Submenu = Struct.new(:label, :menu_items) do def to_s header = Rainbow(label).steelblue "\n\t#{header}\n\n" + menu_items.join("\n") end end MenuItem = Struct.new(:repo, :shortcut, :reference, :next) do HEAD = Rainbow(' <- HEAD ***').bold.aquamarine.freeze def head? reference.targets?(repo.head) end def to_s name = reference.name label = format('[%2s]', shortcut) format( " %s\t%s%s", label: head? ? Rainbow(label).bold : label, name: head? ? Rainbow(name).inverse : name, head: head? ? HEAD : '', ) end end def self.for(repo) new(repo) end def initialize(repo) @repo = repo remote_branches = @repo.branches.find_all(&:remote?) local_branches = @repo.branches.find_all(&:local?) shortcuts = (1..Float::INFINITY).each @submenus = [ submenu_for('REMOTE BRANCHES', remote_branches, shortcuts), submenu_for('LOCAL BRANCHES', local_branches, shortcuts), submenu_for('TAGS', @repo.tags, shortcuts), ].compact @menu_items = @submenus.map(&:menu_items).flatten @lookup = Hash[@menu_items.map(&:shortcut).zip(@menu_items)] end def [](shortcut) @lookup[shortcut] end def head_menu_item @menu_items.find(&:head?) end def default_menu_item head_menu_item&.next end def to_s @submenus.join("\n") end private def shortcuts(initial) shortcut = initial Enumerator.new do |yielder| loop do yielder << shortcut shortcut = case shortcut when Integer then shortcut + 1 when String then (shortcut.ord + 1).chr end end end end def submenu_for(label, references, shortcuts) return unless references.any? Submenu.new(label, menu_items_for(references, shortcuts)) end def menu_items_for(references, shortcuts) previous_menu_item = nil references.sort_by(&:time).map { |reference| shortcut = shortcuts.next.to_s next_menu_item = MenuItem.new(@repo, shortcut, reference, nil) previous_menu_item&.next = next_menu_item previous_menu_item = next_menu_item next_menu_item } end end end end