Skip to content

Commit f5ef77c

Browse files
committed
Improve Ecto repo detection
1 parent 61dcdee commit f5ef77c

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

lib/mix/tasks/error_tracker.install.ex

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -65,29 +65,29 @@ if Code.ensure_loaded?(Igniter) do
6565
@impl Igniter.Mix.Task
6666
def igniter(igniter) do
6767
app_name = Igniter.Project.Application.app_name(igniter)
68-
repo_module = Igniter.Project.Module.module_name(igniter, "Repo")
68+
{igniter, repo} = Igniter.Libs.Ecto.select_repo(igniter)
6969
{igniter, router} = Igniter.Libs.Phoenix.select_router(igniter)
7070

7171
igniter
72-
|> configure(app_name, repo_module)
73-
|> set_up_database(repo_module)
72+
|> configure(app_name, repo)
73+
|> set_up_database(repo)
7474
|> set_up_web_ui(app_name, router)
7575
end
7676

77-
defp configure(igniter, app_name, repo_module) do
77+
defp configure(igniter, app_name, repo) do
7878
igniter
79-
|> Igniter.Project.Config.configure_new("config.exs", :error_tracker, [:repo], repo_module)
79+
|> Igniter.Project.Config.configure_new("config.exs", :error_tracker, [:repo], repo)
8080
|> Igniter.Project.Config.configure_new("config.exs", :error_tracker, [:otp_app], app_name)
8181
|> Igniter.Project.Config.configure_new("config.exs", :error_tracker, [:enabled], true)
8282
end
8383

84-
defp set_up_database(igniter, repo_module) do
84+
defp set_up_database(igniter, repo) do
8585
migration_body = """
8686
def up, do: ErrorTracker.Migration.up()
8787
def down, do: ErrorTracker.Migration.down(version: 1)
8888
"""
8989

90-
Igniter.Libs.Ecto.gen_migration(igniter, repo_module, "add_error_tracker",
90+
Igniter.Libs.Ecto.gen_migration(igniter, repo, "add_error_tracker",
9191
body: migration_body,
9292
on_exists: :skip
9393
)

0 commit comments

Comments
 (0)