mirror of
https://github.com/ash-project/ash_state_machine.git
synced 2024-09-20 13:23:52 +12:00
60 lines
1.6 KiB
Elixir
60 lines
1.6 KiB
Elixir
defmodule AshStateMachine.Transformers.FillInTransitionDefaults do
|
|
use Spark.Dsl.Transformer
|
|
alias Spark.Dsl.Transformer
|
|
@moduledoc false
|
|
|
|
def transform(dsl_state) do
|
|
initial_states =
|
|
case AshStateMachine.Info.state_machine_initial_states(dsl_state) do
|
|
{:ok, value} -> List.wrap(value)
|
|
_ -> []
|
|
end
|
|
|
|
initial_states =
|
|
case initial_states do
|
|
[] ->
|
|
case AshStateMachine.Info.state_machine_default_initial_state(dsl_state) do
|
|
{:ok, value} when not is_nil(value) ->
|
|
[value]
|
|
|
|
_ ->
|
|
initial_states
|
|
end
|
|
|
|
_ ->
|
|
initial_states
|
|
end
|
|
|
|
transitions =
|
|
dsl_state
|
|
|> AshStateMachine.Info.state_machine_transitions()
|
|
|
|
all_states =
|
|
transitions
|
|
|> Enum.flat_map(fn transition ->
|
|
List.wrap(transition.from) ++ List.wrap(transition.to)
|
|
end)
|
|
|> Enum.concat(List.wrap(initial_states))
|
|
|> Enum.uniq()
|
|
|
|
dsl_state =
|
|
case AshStateMachine.Info.state_machine_initial_states(dsl_state) do
|
|
{:ok, value} when not is_nil(value) and value != [] ->
|
|
dsl_state
|
|
|
|
_ ->
|
|
Transformer.set_option(dsl_state, [:state_machine], :initial_states, all_states)
|
|
end
|
|
|
|
{:ok,
|
|
transitions
|
|
|> Enum.reduce(dsl_state, fn transition, dsl_state ->
|
|
Transformer.replace_entity(dsl_state, [:state_machine], %{
|
|
transition
|
|
| from: transition.from || all_states,
|
|
to: transition.to || all_states
|
|
})
|
|
end)
|
|
|> Transformer.persist(:all_state_machine_states, all_states)}
|
|
end
|
|
end
|