fix: add use Absinthe.Phoenix.Socket in endpoints in installer

Author: zachdaniel

lib/igniter.ex

@@ -260,7 +260,13 @@ if Code.ensure_loaded?(Igniter) do
 
       igniter =
         Enum.reduce(endpoints_that_need_parser, igniter, fn endpoint, igniter ->
-          Igniter.Project.Module.find_and_update_module!(igniter, endpoint, fn zipper ->
+          igniter
+          |> Igniter.Project.Module.find_and_update_module!(endpoint, fn zipper ->
+            with {:ok, zipper} <- Igniter.Code.Module.move_to_use(zipper, Phoenix.Endpoint) do
+              {:ok, Igniter.Code.Common.add_code(zipper, "use Absinthe.Phoenix.Endpoint")}
+            end
+          end)
+          |> Igniter.Project.Module.find_and_update_module!(endpoint, fn zipper ->
             case Igniter.Code.Function.move_to_function_call_in_current_scope(
                    zipper,
                    :plug,

test/mix/tasks/ash_graphql_install_test.exs

@@ -106,25 +106,16 @@ defmodule Mix.Tasks.AshGraphqlInstallTest do
   end
 
   test "adds the socket and parser to the endpoint", %{igniter: igniter} do
-    assert_has_patch(igniter, "lib/test_web/endpoint.ex", ~S'''
-          ...|
-     17 17   |  )
-     18 18   |
-        19 + |  socket("/ws/gql", TestWeb.GraphqlSocket,
-        20 + |    websocket: true,
-        21 + |    longpoll: true
-        22 + |  )
-        23 + |
-     19 24   |  # Serve at "/" the static files from "priv/static" directory.
-     20 25   |  #
-          ...|
-     46 51   |
-     47 52   |  plug(Plug.Parsers,
-     48    - |    parsers: [:urlencoded, :multipart, :json],
-        53 + |    parsers: [:urlencoded, :multipart, :json, Absinthe.Plug.Parser],
-     49 54   |    pass: ["*/*"],
-     50 55   |    json_decoder: Phoenix.json_library()
-
+    igniter
+    |> assert_has_patch("lib/test_web/endpoint.ex", ~S'''
+    + |  socket("/ws/gql", TestWeb.GraphqlSocket,
+    + |    websocket: true,
+    + |    longpoll: true
+    + |  )
+    + |
+    ''')
+    |> assert_has_patch("lib/test_web/endpoint.ex", ~S'''
+    + |  use Absinthe.Phoenix.Endpoint
     ''')
   end