defmodule Eai.Tool.ResetSession do
  @behaviour Eai.Tool

  @impl true
  def schema do
    %{
      type: "function",
      function: %{
        name: "reset_session",
        description: """
        Force-kill a stuck or unresponsive PTY session.
        Always call list_pty_sessions first to identify the correct pty_session_id.
        After reset, the next execute_script will automatically create a fresh session.
        """,
        parameters: %{
          type: "object",
          properties: %{
            pty_session_id: %{type: "string", description: "PTY session ID to reset."}
          },
          required: ["pty_session_id"]
        }
      }
    }
  end

  @impl true
  def execute(args, pty_session_id, _chat_session_id) do
    target = Map.get(args, "pty_session_id", pty_session_id)
    Eai.PTY.force_reset(target)

    %{
      status: "ok",
      message: "Session #{target} killed. Next execute_script creates fresh session."
    }
    |> Eai.Utils.sanitize_value()
    |> Jason.encode!()
  end
end