Ash extension attempt

lib/geo/geography/country.ex

@@ -2,10 +2,10 @@ defmodule Geo.Geography.Country do
   use Ash.Resource,
     otp_app: :geo,
     domain: Geo.Geography,
-    data_layer: AshPostgres.DataLayer
+    data_layer: AshPostgres.DataLayer,
+    extensions: [Geo.Resources.Extensions.Id]
 
   # === Attributes ===
-  use Geo.Resources.Attributes.Id
   use Geo.Resources.Attributes.Name, allow_nil?: false, unique?: true
   use Geo.Resources.Attributes.Slug, allow_nil?: false, unique?: true
   use Geo.Resources.Attributes.Timestamps

lib/geo/resources/attributes/id.ex

@@ -1,4 +1,4 @@
-defmodule Geo.Resources.Attributes.Id do
+defmodule Geo.Resources.Attributes.Id.Defunct do
   @moduledoc """
   """
 

lib/geo/resources/extensions/id.ex

@@ -0,0 +1,33 @@
+defmodule Geo.Resources.Extensions.Id do
+  @moduledoc """
+  An Ash extension that adds a UUID v7 primary key attribute to a resource.
+  """
+
+  use Spark.Dsl.Extension,
+    transformers: [
+      __MODULE__.Transformer
+    ]
+
+  defmodule Transformer do
+    @moduledoc false
+    use Spark.Dsl.Transformer
+
+    def before?(Ash.Resource.Transformers.BelongsToAttribute), do: true
+    def before?(_), do: false
+
+    def transform(dsl_state) do
+      case Spark.Dsl.Transformer.build_entity(
+             Ash.Resource.Dsl,
+             [:attributes],
+             :uuid_v7_primary_key,
+             [:id]
+           ) do
+        {:ok, attribute} ->
+          {:ok, Spark.Dsl.Transformer.add_entity(dsl_state, [:attributes], attribute)}
+
+        {:error, error} ->
+          {:error, error}
+      end
+    end
+  end
+end