#!/usr/bin/env python

from ponster import ind as i
from ponster import device_info as d
import click


@click.group()
def main():
    """ponster - Command-line utility for managing PON whiteboxes"""
    pass


@main.command()
@click.argument('olt')
@click.option('--broker', '-b', help='Kafka broker in the form of HOST:PORT')
def ind(olt, broker):
    """Stream indications from olt."""
    i.Indications(broker, olt).start()


@main.command()
@click.argument('olt')
@click.option('--broker', '-b', help='Kafka broker in the form of HOST:PORT')
def devinfo(olt, broker):
    """Get device info from olt."""
    d.DeviceInfo(olt).start()


if __name__ == "__main__":
    main()
