feat: query descendants of a query

Author: fahchen

lib/query/descendants.ex

@@ -7,6 +7,23 @@ defmodule Hierarch.Query.Descendants do
 
     * `:with_self` - when true to include itself. Defaults to false.
   """
+  def query(%{from: %{source: {_table_name, schema}}} = queryable) do
+    path_column = schema.__hierarch__(:path_column)
+
+    [pk_column] = schema.__schema__(:primary_key)
+
+    uuids_query =
+      from t in queryable,
+      select: %{__ancestry__: fragment("text2ltree(replace(ltree2text(?), '-', '_')) || replace(text(?), '-', '_')", field(t, ^path_column), field(t, ^pk_column))}
+
+    uuids_array_query =
+      from t in subquery(uuids_query),
+      select: %{__ltrees__: fragment("array_agg(?)", t.__ancestry__)}
+
+    schema
+    |> join(:cross, [], a in subquery(uuids_array_query))
+    |> where([t, a], fragment("? @> ?", a.__ltrees__, field(t, ^path_column)))
+  end
   def query(%schema{} = struct, opts \\ []) do
     with_self = Keyword.get(opts, :with_self, false)
 

test/hierarch/descendants_test.exs

@@ -59,4 +59,31 @@ defmodule Hierarch.DescendantsTest do
       assert_match descendants, [astronomy, astrophysics, cosmology]
     end
   end
+
+  describe "query/1" do
+    import Ecto.Query
+
+    test "returns their descendants", catelogs do
+      science = Map.get(catelogs, "Top.Science")
+      hobbies   = Map.get(catelogs, "Top.Hobbies")
+
+      astronomy = Map.get(catelogs, "Top.Science.Astronomy")
+      astrophysics = Map.get(catelogs, "Top.Science.Astronomy.Astrophysics")
+      cosmology    = Map.get(catelogs, "Top.Science.Astronomy.Cosmology")
+
+      amateurs_astronomy = Map.get(catelogs, "Top.Hobbies.Amateurs_Astronomy")
+
+      query = from(
+        c in Catelog,
+        where: c.id in ^[science.id, hobbies.id]
+      )
+
+      descendants =
+        query
+        |> Hierarch.Query.Descendants.query()
+        |> Repo.all
+
+      assert_match descendants, [astronomy, astrophysics, cosmology, amateurs_astronomy]
+    end
+  end
 end