ash_postgres/lib/mix/helpers.ex
2021-02-22 23:53:18 -05:00

83 lines
1.8 KiB
Elixir

defmodule AshPostgres.MixHelpers do
@moduledoc false
def apis(opts, args) do
apps =
if apps_paths = Mix.Project.apps_paths() do
apps_paths |> Map.keys() |> Enum.sort()
else
[Mix.Project.config()[:app]]
end
configured_apis = Enum.flat_map(apps, &Application.get_env(&1, :ash_apis, []))
apis =
if opts[:apis] && opts[:apis] != "" do
opts[:apis]
|> Kernel.||("")
|> String.split(",")
|> Enum.flat_map(fn
"" ->
[]
api ->
[Module.concat([api])]
end)
else
configured_apis
end
Enum.map(apis, &ensure_compiled(&1, args))
end
def repos(opts, args) do
opts
|> apis(args)
|> Enum.flat_map(&Ash.Api.resources/1)
|> Enum.filter(&(Ash.DataLayer.data_layer(&1) == AshPostgres.DataLayer))
|> Enum.map(&AshPostgres.repo(&1))
|> Enum.uniq()
end
def delete_flag(args, arg) do
case Enum.split_while(args, &(&1 != arg)) do
{left, [_ | rest]} ->
left ++ rest
_ ->
args
end
end
def delete_arg(args, arg) do
case Enum.split_while(args, &(&1 != arg)) do
{left, [_, _ | rest]} ->
left ++ rest
_ ->
args
end
end
defp ensure_compiled(api, args) do
if Code.ensure_loaded?(Mix.Tasks.App.Config) do
Mix.Task.run("app.config", args)
else
Mix.Task.run("loadpaths", args)
"--no-compile" not in args && Mix.Task.run("compile", args)
end
case Code.ensure_compiled(api) do
{:module, _} ->
api
|> Ash.Api.resources()
|> Enum.each(&Code.ensure_compiled/1)
# TODO: We shouldn't need to make sure that the resources are compiled
api
{:error, error} ->
Mix.raise("Could not load #{inspect(api)}, error: #{inspect(error)}. ")
end
end
end