feat: add repo callback to disable atomic actions and error expressions (#464)

Author: barnabasJ

lib/data_layer.ex

@@ -644,8 +644,11 @@ defmodule AshPostgres.DataLayer do
   def can?(_, :transact), do: true
   def can?(_, :composite_primary_key), do: true
 
-  def can?(_resource, {:atomic, :update}), do: true
-  def can?(_resource, {:atomic, :upsert}), do: true
+  def can?(resource, {:atomic, :update}),
+    do: not AshPostgres.DataLayer.Info.repo(resource, :mutate).disable_atomic_actions?()
+
+  def can?(resource, {:atomic, :upsert}),
+    do: not AshPostgres.DataLayer.Info.repo(resource, :mutate).disable_atomic_actions?()
 
   def can?(_, :upsert), do: true
   def can?(_, :changeset_filter), do: true
@@ -709,10 +712,13 @@ defmodule AshPostgres.DataLayer do
   def can?(_, {:aggregate_relationship, _}), do: true
 
   def can?(_, :timeout), do: true
-  def can?(_, :expr_error), do: true
+
+  def can?(resource, :expr_error),
+    do: not AshPostgres.DataLayer.Info.repo(resource, :mutate).disable_expr_error?()
 
   def can?(resource, {:filter_expr, %Ash.Query.Function.Error{}}) do
-    "ash-functions" in AshPostgres.DataLayer.Info.repo(resource, :read).installed_extensions() &&
+    not AshPostgres.DataLayer.Info.repo(resource, :mutate).disable_expr_error?() &&
+      "ash-functions" in AshPostgres.DataLayer.Info.repo(resource, :read).installed_extensions() &&
       "ash-functions" in AshPostgres.DataLayer.Info.repo(resource, :mutate).installed_extensions()
   end
 

lib/repo.ex

@@ -104,6 +104,12 @@ defmodule AshPostgres.Repo do
   @doc "Should the repo should be dropped by `mix ash_postgres.drop`?"
   @callback drop?() :: boolean
 
+  @doc "Disable atomic actions for this repo"
+  @callback disable_atomic_actions?() :: boolean
+
+  @doc "Disable expression errors for this repo"
+  @callback disable_expr_error?() :: boolean
+
   defmacro __using__(opts) do
     quote bind_quoted: [opts: opts] do
       if Keyword.get(opts, :define_ecto_repo?, true) do
@@ -133,6 +139,8 @@ defmodule AshPostgres.Repo do
       def override_migration_type(type), do: type
       def create?, do: true
       def drop?, do: true
+      def disable_atomic_actions?, do: false
+      def disable_expr_error?, do: false
 
       # default to false in 4.0
       def prefer_transaction?, do: true
@@ -288,7 +296,9 @@ defmodule AshPostgres.Repo do
                      default_prefix: 0,
                      override_migration_type: 1,
                      create?: 0,
-                     drop?: 0
+                     drop?: 0,
+                     disable_atomic_actions?: 0,
+                     disable_expr_error?: 0
 
       # We do this switch because `!@warn_on_missing_ash_functions` in the function body triggers
       # a dialyzer error