from alot.commands import Command, registerCommand

MODE = 'bufferlist'


@registerCommand(MODE, 'select', help='focus selected buffer')
class BufferFocusCommand(Command):
    def apply(self, ui):
        selected = ui.current_buffer.get_selected_buffer()
        ui.buffer_focus(selected)


@registerCommand(MODE, 'close', help='close focussed buffer')
class BufferCloseCommand(Command):
    def apply(self, ui):
        bufferlist = ui.current_buffer
        selected = bufferlist.get_selected_buffer()
        ui.buffer_close(selected)
        if bufferlist is not selected:
            bufferlist.rebuild()
        ui.update()