Current section

Files

Jump to
pow lib pow extension ecto schema migration.ex
Raw

lib/pow/extension/ecto/schema/migration.ex

defmodule Pow.Extension.Ecto.Schema.Migration do
@moduledoc """
Generates schema migration content for extensions.
"""
alias Pow.{Config,
Ecto.Schema.Migration,
Extension.Ecto.Schema}
@template """
defmodule <%= inspect schema.repo %>.Migrations.<%= schema.migration_name %> do
use Ecto.Migration
def change do
alter table(:<%= schema.table %>) do
<%= for {k, v} <- schema.attrs do %> add <%= inspect k %>, <%= inspect v %><%= schema.migration_defaults[k] %>
<% end %><%= for {_, i, _, s} <- schema.assocs do %> add <%= if(String.ends_with?(inspect(i), "_id"), do: inspect(i), else: inspect(i) <> "_id") %>, references(<%= inspect(s) %>, on_delete: :nothing<%= if schema.binary_id do %>, type: :binary_id<% end %>)
<% end %>
end
<%= for index <- schema.indexes do %>
<%= index %><% end %>
end
end
"""
@doc """
Generates migration file content.
"""
@spec gen(atom(), atom(), Config.t()) :: binary()
def gen(extension, context_base, config \\ []) do
context_base
|> parse_options(extension, config)
|> migration_file(extension)
end
@doc """
Generates a migration module name.
"""
@spec name(atom(), binary()) :: binary()
def name(extension, table) do
"Add#{normalize_extension_name(extension)}To#{Macro.camelize(table)}"
end
defp parse_options(context_base, extension, config) do
repo = Config.get(config, :repo, Module.concat([context_base, "Repo"]))
table = Config.get(config, :table, "users")
config = Config.put(config, :extensions, [extension])
attrs = Schema.attrs(config)
indexes = Schema.indexes(config)
migration_name = name(extension, table)
Migration.schema(context_base, repo, table, migration_name, attrs, indexes, config)
end
defp migration_file(schema, extension) do
EEx.eval_string(unquote(@template), schema: schema, extension: normalize_extension_name(extension))
end
defp normalize_extension_name(extension) do
extension
|> Module.split()
|> List.last()
end
end