oapi_open_ai

defmodule OpenAi.Client do
  @moduledoc """
  Module responsible for making requests to the OpenAI API.
  And transforming the response to the expected response type.
  """
  alias OpenAi.Config

  @doc """
  Makes a request to the OpenAI API
  based on the provided parameters.

  Returns the response body transformed
  to the expected response type.

  Primarily used by the operation modules.
  """
  def request(
        %{
          method: method,
          url: url,
          response: expected_response
        } = params
      ) do
    %HTTPoison.Request{
      method: method,
      url: Path.join(Config.base_url(), url),
      headers:
        [
          {"Authorization", "Bearer #{Config.api_key()}"},
          {"Content-Type", "application/json"}
        ] ++ Config.http_headers!(),
      params: params |> Map.get(:query)
    }
    |> case do
      %{method: method} = request when method in [:post, :put, :patch, :delete, :options] ->
        params
        |> Map.get(:body)
        |> case do
          body when body |> is_map or body |> is_list ->
            request |> Map.put(:body, body |> Poison.encode!())

          _ ->
            request
        end

      request ->
        request
    end
    |> Config.http_client().request()
    |> transform_to_expected_response(expected_response)
  end

  defp transform_to_expected_response(
         {:ok,
          %HTTPoison.Response{
            status_code: response_status_code,
            body: body
          }},
         [{expected_status_code, {:string, :generic}}]
       )
       when response_status_code == expected_status_code do
    {:ok, body}
  end

  defp transform_to_expected_response(
         {:ok,
          %HTTPoison.Response{
            status_code: response_status_code,
            body: body
          }},
         [{expected_status_code, expected_type}]
       )
       when response_status_code == expected_status_code do
    body
    |> Poison.decode()
    |> case do
      {:ok, decoded_body} ->
        transform_value(decoded_body, expected_type)

      {:error, _} ->
        {:error, "Failed to decode response body"}
    end
  end

  defp transform_to_expected_response(
         {:ok,
          %HTTPoison.Response{
            status_code: response_status_code,
            body: body
          }},
         _
       ) do
    {:error, "Unexpected status code: #{response_status_code}, body: #{body}"}
  end

  defp transform_to_expected_response({:error, %HTTPoison.Error{reason: reason}}, _) do
    {:error, reason}
  end

  defp transform_to_expected_response({:error, error}, _) do
    {:error, error}
  end

  defp transform_value(v, :integer) when is_integer(v), do: {:ok, v}
  defp transform_value(v, :integer), do: {:error, "Expected integer, got #{inspect(v)}"}
  defp transform_value(v, :number) when is_number(v), do: {:ok, v}
  defp transform_value(v, :number), do: {:error, "Expected number, got #{inspect(v)}"}
  defp transform_value(v, {:string, _type}) when is_binary(v), do: {:ok, v}
  defp transform_value(v, {:string, _type}), do: {:error, "Expected string, got #{inspect(v)}"}
  defp transform_value(v, {:const, const}) when const == v, do: {:ok, v}
  defp transform_value(v, {:const, const}), do: {:error, "Expected const: #{const}, got #{v}"}
  defp transform_value(v, {:enum, values}), do: in_enum(v, values)
  defp transform_value(v, :boolean) when is_boolean(v), do: {:ok, v}
  defp transform_value(v, :boolean), do: {:error, "Expected boolean, got #{inspect(v)}"}
  defp transform_value(v, :map) when is_map(v), do: {:ok, v}
  defp transform_value(v, :map), do: {:error, "Expected map, got #{inspect(v)}"}
  defp transform_value(v, {:union, types}), do: transform_value(v, collapse_union(types, v))

  defp transform_value(v, {struct, base_type}) when is_map(v),
    do: transform_struct(v, struct, base_type)

  defp transform_value(v, type) when is_list(v) and is_list(type),
    do: transform_list(v, type |> List.first())

  defp transform_value(v, {:union, types}) when is_list(v) and is_list(types),
    do: transform_list(v, types)

  defp transform_value(v, type),
    do: {:error, "Failed to transform value: #{inspect(v)} to type: #{inspect(type)}"}

  defp transform_struct(map, struct, base_type)
       when map |> is_map and struct |> is_atom and base_type |> is_atom do
    struct.__fields__(base_type)
    |> Enum.reduce({:ok, struct.__struct__(%{})}, &transform_struct_reducer(&1, &2, map))
  end

  defp transform_struct(map, struct, base_type) do
    {:error,
     """
     Some parameters are of incorrect type
     Expected map, got: #{inspect(map)}
     Expected struct(atom) got: #{inspect(struct)}
     Expected base_type(atom) got: #{inspect(base_type)}
     """}
  end

  defp transform_struct_reducer({field, type}, {:ok, acc}, map) do
    map
    |> Map.get(field |> Atom.to_string(), nil)
    |> case do
      nil ->
        # TODO check if field is nullable
        {:ok, acc}

      value ->
        value
        |> transform_value(type)
        |> case do
          {:ok, v} -> {:ok, Map.put(acc, field, v)}
          {:error, e} -> {:error, e}
        end
    end
  end

  defp transform_struct_reducer(_, {:error, e}, _) do
    {:error, e}
  end

  defp in_enum(value, enum) do
    if Enum.member?(enum, value) do
      {:ok, value}
    else
      {:error, "Expected one of #{inspect(enum)}, got #{value}"}
    end
  end

  defp transform_list(list, type) when list |> is_list do
    list
    |> Enum.reduce({:ok, []}, &transform_list_reducer(&1, &2, type))
    |> case do
      {:ok, acc} ->
        {:ok, Enum.reverse(acc)}

      {:error, e} ->
        {:error, e}
    end
  end

  defp transform_list(list, _type) do
    {:error, "Expected list, got #{inspect(list)}"}
  end

  defp transform_list_reducer(item, {:ok, acc}, type) do
    item
    |> transform_value(type)
    |> case do
      {:ok, v} -> {:ok, [v | acc]}
      {:error, e} -> {:error, e}
    end
  end

  defp transform_list_reducer(_, {:error, e}, _) do
    {:error, e}
  end

  defp collapse_union(union_types, value) do
    value_fields =
      value
      |> Map.keys()
      |> Enum.join("")

    union_types
    |> Enum.find(fn {struct, type} ->
      struct.__fields__(type)
      |> Enum.reduce("", fn {field, _}, acc -> acc <> (field |> Atom.to_string()) end)
      |> Kernel.==(value_fields)
    end)
  end
end