defmodule Mix.Tasks.Rivet.New.Migration do
  import Rivet.Utils.Cli
  import Rivet.Utils.Cli.Print
  alias Rivet.Migration

  # ignore-coveralls-start
  def run(optcfg, opts, [model, label]) do
    with {:ok, cfg} <- Mix.Tasks.Rivet.New.get_config(optcfg, opts),
         :ok <- Migration.Manage.add_migration(model, label, cfg) do
      :ok
    else
      {:error, reason} -> die(reason)
    end
  end

  def run(optcfg, _, _) do
    syntax(optcfg, "migration {model} {label}")
  end

  # ignore-coveralls-end
end