mirror of
https://github.com/ash-project/ash_postgres.git
synced 2024-09-17 03:53:14 +12:00
113 lines
3 KiB
Elixir
113 lines
3 KiB
Elixir
defmodule AshPostgres.Igniter do
|
|
@moduledoc "Codemods and utilities for working with AshPostgres & Igniter"
|
|
|
|
@doc false
|
|
def default_repo_contents(otp_app) do
|
|
"""
|
|
use AshPostgres.Repo, otp_app: #{inspect(otp_app)}
|
|
|
|
def min_pg_version do
|
|
# Adjust this according to your postgres version
|
|
%Version{major: 16, minor: 0, patch: 0}
|
|
end
|
|
|
|
def installed_extensions do
|
|
# Add extensions here, and the migration generator will install them.
|
|
["ash-functions"]
|
|
end
|
|
"""
|
|
end
|
|
|
|
def table(igniter, resource) do
|
|
igniter
|
|
|> Spark.Igniter.get_option(resource, [:postgres, :table])
|
|
|> case do
|
|
{igniter, {:ok, value}} when is_binary(value) or is_nil(value) ->
|
|
{:ok, igniter, value}
|
|
|
|
{igniter, _} ->
|
|
{:error, igniter}
|
|
end
|
|
end
|
|
|
|
def repo(igniter, resource) do
|
|
igniter
|
|
|> Spark.Igniter.get_option(resource, [:postgres, :repo])
|
|
|> case do
|
|
{igniter, {:ok, value}} when is_atom(value) ->
|
|
{:ok, igniter, value}
|
|
|
|
{igniter, _} ->
|
|
{:error, igniter}
|
|
end
|
|
end
|
|
|
|
def add_postgres_extension(igniter, repo_name, extension) do
|
|
Igniter.Code.Module.find_and_update_module!(igniter, repo_name, fn zipper ->
|
|
case Igniter.Code.Function.move_to_def(zipper, :installed_extensions, 0) do
|
|
{:ok, zipper} ->
|
|
case Igniter.Code.List.append_new_to_list(zipper, extension) do
|
|
{:ok, zipper} ->
|
|
{:ok, zipper}
|
|
|
|
_ ->
|
|
{:warning,
|
|
"Could not add installed extension #{inspect(extension)} to #{inspect(repo_name)}.installed_extensions/0"}
|
|
end
|
|
|
|
_ ->
|
|
zipper = Sourceror.Zipper.rightmost(zipper)
|
|
|
|
code = """
|
|
def installed_extensions do
|
|
[#{inspect(extension)}]
|
|
end
|
|
"""
|
|
|
|
{:ok, Igniter.Code.Common.add_code(zipper, code)}
|
|
end
|
|
end)
|
|
end
|
|
|
|
def select_repo(igniter, opts \\ []) do
|
|
label = Keyword.get(opts, :label, "Which repo should be used?")
|
|
generate = Keyword.get(opts, :generate?, false)
|
|
|
|
case list_repos(igniter) do
|
|
{igniter, []} ->
|
|
if generate do
|
|
repo = Igniter.Code.Module.module_name(igniter, "Repo")
|
|
otp_app = Igniter.Project.Application.app_name(igniter)
|
|
|
|
igniter =
|
|
Igniter.Code.Module.create_module(igniter, repo, default_repo_contents(otp_app))
|
|
|
|
{igniter, repo}
|
|
else
|
|
{igniter, nil}
|
|
end
|
|
|
|
{igniter, [repo]} ->
|
|
{igniter, repo}
|
|
|
|
{igniter, repos} ->
|
|
{igniter, Owl.IO.select(repos, label: label, render_as: &inspect/1)}
|
|
end
|
|
end
|
|
|
|
def list_repos(igniter) do
|
|
Igniter.Code.Module.find_all_matching_modules(igniter, fn _mod, zipper ->
|
|
move_to_repo_use(zipper) != :error
|
|
end)
|
|
end
|
|
|
|
defp move_to_repo_use(zipper) do
|
|
Igniter.Code.Function.move_to_function_call(zipper, :use, 2, fn zipper ->
|
|
Igniter.Code.Function.argument_equals?(
|
|
zipper,
|
|
0,
|
|
AshPostgres.Repo
|
|
)
|
|
end)
|
|
end
|
|
end
|