from datetime import datetime
import click
import prettytable
from ._base import Session
from ._customer import Customer
from ._group import Group
from ._job import Job, JobPriorities
from ._staff import Staff
from ._status import Status, StatusValues


session = Session()


@click.group()
def main():
    pass


@main.command(name="customer")
@click.option("--add", is_flag=True, default=False)
def customer(add):
    if add:
        click.echo("adding new customer")
        name = click.prompt("name")
        email = click.prompt("email")
        room = click.prompt("room")
        groups = click.prompt("group id(s)").split(',')
        groups = [session.query(Group).filter(Group.id == int(g))[0] for g in groups]
        new = Customer(name=name, email=email, room=room, groups=groups)
        session.add(new)
        session.commit()
        click.echo("...done!")
    else:
        out = prettytable.PrettyTable()
        out.field_names = ["id", "name", "email", "room", "group(s)"]
        out.align = "l"
        for instance in session.query(Customer).order_by(Customer.id):
            groups = ",".join([g.name for g in instance.groups])
            out.add_row([instance.id, instance.name, instance.email, instance.room, groups])
        print(out)


@main.command(name="group")
@click.option("--add", is_flag=True, default=False)
def group(add):
    if add:
        click.echo("adding new group")
        name = click.prompt("name")
        new = Group(name=name)
        session.add(new)
        session.commit()
        click.echo("...done!")
    else:
        out = prettytable.PrettyTable()
        out.field_names = ["id", "name"]
        out.align = "l"
        for instance in session.query(Group).order_by(Group.id):
            out.add_row([instance.id, instance.name])
        print(out)


@main.command(name="job")
@click.option("--add", is_flag=True, default=False)
def job(add):
    if add:
        click.echo("adding new job")
        name = click.prompt("name")
        customers = click.prompt("customer id(s)").split(',')
        customers = [session.query(Customer).filter(Customer.id == int(g))[0] for g in customers]
        priority = click.prompt("priority", type=click.Choice([a.name for a in (JobPriorities)]))
        new = Job(name=name, customers=customers, priority=priority)
        session.add(new)
        session.commit()
        click.echo("...done!")
    else:
        out = prettytable.PrettyTable()
        out.field_names = ["id", "priority", "name", "customer(s)"]
        out.align = "l"
        for instance in session.query(Job).order_by(Job.id):
            customers = ",".join([c.name for c in instance.customers])
            out.add_row([instance.id, instance.priority.name, instance.name, customers])
        print(out)


@main.command(name="staff")
@click.option("--add", is_flag=True, default=False)
def staff(add):
    if add:
        click.echo("adding new staff member")
        name = click.prompt("name")
        new = Staff(name=name)
        session.add(new)
        session.commit()
        click.echo("...done!")
    else:
        out = prettytable.PrettyTable()
        out.field_names = ["id", "name"]
        out.align = "l"
        for instance in session.query(Staff).order_by(Staff.id):
            out.add_row([instance.id, instance.name])
        print(out)


@main.command(name="status")
@click.option("--add", is_flag=True, default=False)
def staff(add):
    if add:
        click.echo("recording status change")
        job = session.query(Job).filter(Job.id == click.prompt("job id", int))[0]
        value = click.prompt("new status", type=click.Choice([a.name for a in (StatusValues)]))
        comment = click.prompt("comment")
        new = Status(job=job, comment=comment, value=str(value), timestamp=datetime.utcnow())
        session.add(new)
        session.commit()
        click.echo("...done!")
    else:
        out = prettytable.PrettyTable()
        out.field_names = ["id", "timestamp (UTC)", "job", "value", "comment"]
        out.align = "l"
        for instance in session.query(Status).order_by(Status.id):
            out.add_row([instance.id,instance.timestamp, instance.job.name, instance.value.name, instance.comment])
        print(out)