from signalwire import AgentBase
from signalwire.core.function_result import FunctionResult

class ServiceDesk(AgentBase):
    def __init__(self):
        super().__init__(name="service-desk", route="/service-desk")
        self.prompt_add_section("Role", "You book service appointments.")

        ctx = self.define_contexts().add_context("default")

        ctx.add_step("book") \
            .set_text("Find an open time that works and book it.") \
            .set_functions(["book_slot"]) \
            .set_valid_steps([])

        ctx.add_step("wrap_up") \
            .set_text("Confirm the booking and ask if they need anything else.") \
            .set_functions(["transfer_to_staff"])

        self.define_tool(
            name="book_slot",
            description="Book an open slot and text the caller a confirmation",
            parameters={"slot": {"type": "string"}},
            handler=self.book_slot,
        )

    def book_slot(self, args, raw_data):
        caller = raw_data["caller_id_num"]
        booking = schedule(args["slot"], caller)
        return FunctionResult(f"Booked for {booking.time}.") \
            .send_sms(to_number=caller, from_number=DESK_NUMBER,
                      body=f"Confirmed for {booking.time}.") \
            .swml_change_step("wrap_up")  # SMS and step change, one result