From 787cbad6055e84258c44d24639a8ea37e39ac80e Mon Sep 17 00:00:00 2001
From: Zach Daniel <zach@zachdaniel.dev>
Date: Wed, 5 Oct 2022 20:25:18 -0400
Subject: [PATCH] fix: properly set source on attributes

---
 lib/ash/resource/attribute.ex | 4 ++++
 lib/ash/resource/dsl.ex       | 2 +-
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/lib/ash/resource/attribute.ex b/lib/ash/resource/attribute.ex
index 683bdeeb..b06b0f5d 100644
--- a/lib/ash/resource/attribute.ex
+++ b/lib/ash/resource/attribute.ex
@@ -201,6 +201,10 @@ defmodule Ash.Resource.Attribute do
                               |> OptionsHelpers.set_default!(:type, :integer)
                               |> Keyword.delete(:allow_nil?)
 
+  def transform(attribute) do
+    Ash.Type.set_type_transformation(%{attribute | source: attribute.source || attribute.name})
+  end
+
   @doc false
   def attribute_schema, do: @schema
   def create_timestamp_schema, do: @create_timestamp_schema
diff --git a/lib/ash/resource/dsl.ex b/lib/ash/resource/dsl.ex
index f6a1531d..857a1b87 100644
--- a/lib/ash/resource/dsl.ex
+++ b/lib/ash/resource/dsl.ex
@@ -12,7 +12,7 @@ defmodule Ash.Resource.Dsl do
       end
       """
     ],
-    transform: {Ash.Type, :set_type_transformation, []},
+    transform: {Ash.Resource.Attribute, :transform, []},
     target: Ash.Resource.Attribute,
     args: [:name, :type],
     modules: [:type],