From 72fada1bec4e17d7746213d8b8b33b85c31e6e87 Mon Sep 17 00:00:00 2001
From: Zach Daniel <zach@zachdaniel.dev>
Date: Mon, 14 Aug 2023 16:18:45 -0700
Subject: [PATCH] improvement: find all installed extensions by default

---
 lib/mix/tasks/helpers.ex | 41 ++++++++++++++++++++++++++++++++--------
 1 file changed, 33 insertions(+), 8 deletions(-)

diff --git a/lib/mix/tasks/helpers.ex b/lib/mix/tasks/helpers.ex
index 9ccf344a..f3b76e46 100644
--- a/lib/mix/tasks/helpers.ex
+++ b/lib/mix/tasks/helpers.ex
@@ -6,21 +6,37 @@ defmodule Ash.Mix.Tasks.Helpers do
   @doc """
   Gets all extensions in use by the current project's apis and resources
   """
-  def extensions!(argv \\ []) do
-    apis = Ash.Mix.Tasks.Helpers.apis!(argv)
+  def extensions!(argv, opts \\ []) do
+    if opts[:in_use?] do
+      apis = Ash.Mix.Tasks.Helpers.apis!(argv)
 
-    resource_extensions =
-      apis
-      |> Enum.flat_map(&Ash.Api.Info.resources/1)
-      |> all_extensions()
+      resource_extensions =
+        apis
+        |> Enum.flat_map(&Ash.Api.Info.resources/1)
+        |> all_extensions()
 
-    Enum.uniq(all_extensions(apis) ++ resource_extensions)
+      Enum.uniq(all_extensions(apis) ++ resource_extensions)
+    else
+      Application.loaded_applications()
+      |> Enum.map(&elem(&1, 0))
+      |> Enum.flat_map(&elem(:application.get_key(&1, :modules), 1))
+      |> Stream.chunk_every(100)
+      # This takes a while, but it doesn't when we split up the work
+      |> Task.async_stream(fn modules ->
+        modules
+        |> Enum.filter(&Spark.implements_behaviour?(&1, Spark.Dsl.Extension))
+        |> Enum.uniq()
+      end)
+      # we're assuming no failures
+      |> Stream.flat_map(&elem(&1, 1))
+      |> Enum.uniq()
+    end
   end
 
   @doc """
   Get all apis for the current project and ensure they are compiled.
   """
-  def apis!(argv \\ []) do
+  def apis!(argv) do
     {opts, _} = OptionParser.parse!(argv, strict: [apis: :string])
 
     apis =
@@ -84,4 +100,13 @@ defmodule Ash.Mix.Tasks.Helpers do
         Mix.raise("Could not load #{inspect(api)}, error: #{inspect(error)}. ")
     end
   end
+
+  defp all_loaded do
+    :code.all_loaded()
+    |> Enum.filter(fn
+      {mod, _} when is_atom(mod) -> true
+      _ -> false
+    end)
+    |> Enum.map(&elem(&1, 0))
+  end
 end