defmodule Genie.Openai.Thread do
  use Ecto.Schema
  import Ecto.Changeset

  alias Genie.Error
  alias Genie.Openai.Thread
  import Genie.Openai.Api

  require Logger

  @url "/threads"

  @primary_key false
  embedded_schema do
    field(:id, :string)
    field(:object, :string)
    field(:created_at, :integer)
    field(:metadata, :map, default: %{})
  end

  @create_fields ~w(id object created_at metadata)a

  def new(%{} = attrs \\ %{}) do
    %Thread{}
    |> cast(attrs, @create_fields)
    |> apply_action(:insert)
  end

  def new!(attrs \\ %{}) do
    case new(attrs) do
      {:ok, thread} ->
        thread

      {:error, changeset} ->
        raise Error, changeset
    end
  end

  def create(attrs \\ %{}) do
    prepare(@url)
    |> Req.post(body: Jason.encode!(attrs))
    |> handle_response()
  end

  def retrieve(thread_id) do
    prepare(@url <> "/#{thread_id}")
    |> Req.get()
    |> handle_response()
  end

  def delete(thread_id) do
    prepare(@url <> "/#{thread_id}")
    |> Req.delete()
    |> handle_response(:delete)
  end

  defp handle_response(res, method \\ nil) do
    res
    |> case do
      {:ok, %Req.Response{body: data}} ->
        case method do
          :delete -> %{id: data["id"], object: data["object"], deleted: data["deleted"]}
          _ -> Thread.new!(data)
        end

      {:error, %Mint.TransportError{reason: :timeout}} ->
        {:error, "Request timed out"}

      other ->
        Logger.error("Unexpected and unhandled API response! #{inspect(other)}")
        other
    end
  end
end