ash_uuid

defmodule AshUUID.Macros do
  @moduledoc false

  defmacro uuid_attribute(name, opts \\ []) do
    computed_opts = AshUUID.Config.get_config(opts)

    default_prefix =
      __CALLER__.module
      |> Module.split()
      |> List.last()
      |> Macro.underscore()
      |> String.replace("_", "-")

    prefix = Keyword.get(opts, :prefix, default_prefix)

    constraints = [
      prefix: prefix,
      version: computed_opts.version,
      encoded?: computed_opts.encoded?,
      prefixed?: computed_opts.prefixed?,
      migration_default?: !Keyword.get(opts, :allow_nil?, false) && computed_opts.migration_default?,
      strict?: computed_opts.strict?
    ]

    default =
      quote do
        fn -> AshUUID.UUID.generate(unquote(constraints)) end
      end

    field_opts =
      opts
      |> Keyword.delete(:version)
      |> Keyword.delete(:encoded?)
      |> Keyword.delete(:prefixed?)
      |> Keyword.delete(:migration_default?)
      |> Keyword.delete(:strict?)
      |> Keyword.delete(:prefix)
      |> Keyword.put_new(:primary_key?, true)
      |> Keyword.put_new(:allow_nil?, false)
      |> Keyword.put_new(:default, default)
      |> Keyword.put_new(:writable?, false)
      |> Keyword.update(:constraints, constraints, fn kw ->
        kw
        |> Keyword.put(:prefix, prefix)
        |> Keyword.put(:version, computed_opts.version)
        |> Keyword.put(:encoded?, computed_opts.encoded?)
        |> Keyword.put(:prefixed?, computed_opts.prefixed?)
        |> Keyword.put(:migration_default?, !Keyword.get(opts, :allow_nil?, false) && computed_opts.migration_default?)
        |> Keyword.put(:strict?, computed_opts.strict?)
      end)

    quote do
      attribute unquote(name), AshUUID.UUID, unquote(field_opts)
    end
  end

  defmacro uuid_argument(name, opts \\ []) do
    computed_opts = AshUUID.Config.get_config(opts)

    default_prefix =
      __CALLER__.module
      |> Module.split()
      |> List.last()
      |> Macro.underscore()
      |> String.replace("_", "-")

    prefix = Keyword.get(opts, :prefix, default_prefix)

    constraints = [
      prefix: prefix,
      version: computed_opts.version,
      encoded?: computed_opts.encoded?,
      prefixed?: computed_opts.prefixed?,
      migration_default?: !Keyword.get(opts, :allow_nil?, false) && computed_opts.migration_default?,
      strict?: computed_opts.strict?
    ]

    default =
      quote do
        fn -> AshUUID.UUID.generate(unquote(constraints)) end
      end

    argument_opts =
      opts
      |> Keyword.delete(:version)
      |> Keyword.delete(:encoded?)
      |> Keyword.delete(:prefixed?)
      |> Keyword.delete(:migration_default?)
      |> Keyword.delete(:strict?)
      |> Keyword.delete(:prefix)
      |> Keyword.put_new(:allow_nil?, false)
      |> Keyword.put_new(:default, default)
      |> Keyword.update(:constraints, constraints, fn kw ->
        kw
        |> Keyword.put(:prefix, prefix)
        |> Keyword.put(:version, computed_opts.version)
        |> Keyword.put(:encoded?, computed_opts.encoded?)
        |> Keyword.put(:prefixed?, computed_opts.prefixed?)
        |> Keyword.put(:migration_default?, !Keyword.get(opts, :allow_nil?, false) && computed_opts.migration_default?)
        |> Keyword.put(:strict?, computed_opts.strict?)
      end)

    quote do
      argument unquote(name), AshUUID.UUID, unquote(argument_opts)
    end
  end
end