defmodule Inspex do
  defmacro __using__(opts) do
    %{module: module} = __CALLER__

    Module.put_attribute(module, :inspex_name, Keyword.get(opts, :name, inspect(module)))
    Module.put_attribute(module, :inspex_drop, Keyword.get(opts, :drop, []))
    Module.put_attribute(module, :inspex_take, Keyword.get(opts, :take, false))
    Module.put_attribute(module, :inspex_sigil, Keyword.get(opts, :sigil, "#"))
    Module.put_attribute(module, :inspex_omit_values, Keyword.get(opts, :omit_values, [nil]))
    Module.put_attribute(module, :inspex_omit_keys, Keyword.get(opts, :omit_keys, []))
    Module.put_attribute(module, :inspex_omit_pairs, Keyword.get(opts, :omit_pairs, []))
    Module.put_attribute(module, :inspex_omit_default_values, Keyword.get(opts, :omit_default_values, true))

    quote location: :keep do
      @before_compile unquote(__MODULE__)
    end
  end

  defmacro __before_compile__(%{module: module}) do
    inspex_name = Module.get_attribute(module, :inspex_name)
    inspex_drop = Module.get_attribute(module, :inspex_drop)
    inspex_take = Module.get_attribute(module, :inspex_take)
    inspex_sigil = Module.get_attribute(module, :inspex_sigil)
    inspex_omit_values = Module.get_attribute(module, :inspex_omit_values)
    inspex_omit_keys = Module.get_attribute(module, :inspex_omit_keys)
    inspex_omit_pairs = Module.get_attribute(module, :inspex_omit_pairs)
    inspex_omit_default_values = Module.get_attribute(module, :inspex_omit_default_values)

    # TODO find a better way to compose clauses

    clauses = [quote do
      _ -> false
    end]

    clauses = if length(inspex_omit_pairs) > 0 do
      [quote do
        pair when pair in unquote(inspex_omit_pairs) -> true
      end | clauses]
    end || clauses

    clauses = if length(inspex_omit_values) > 0 do
      [quote do
        {_key, value} when value in unquote(inspex_omit_values) -> true
      end | clauses]
    end || clauses

    clauses = if length(inspex_omit_keys) > 0 do
      [quote do
        {key, _value} when key in unquote(inspex_omit_keys) -> true
      end | clauses]
    end || clauses

    clauses = Enum.map(clauses, &hd/1)

    clauses = {:fn, [], clauses}

    quote location: :keep do

      def __inspex_inspect__(conf, opts) do
        content = Map.from_struct(conf)

        unquote(if inspex_drop do
          quote location: :keep do
            content = Map.drop(content, unquote(inspex_drop))
          end
        end)

        unquote(if inspex_take do
          quote location: :keep do
            content = Map.take(content, unquote(inspex_take))
          end
        end)

        content = Stream.reject(content, unquote(clauses))

        unquote(if inspex_omit_default_values do
          quote location: :keep do
            default = %__MODULE__{}
            content = Stream.reject(content, fn {key, value} -> Map.get(default, key) == value end)
          end
        end)

        content = Enum.into(content, []) |> Inspect.List.inspect(opts)

        Inspect.Algebra.concat [unquote(inspex_sigil <> inspex_name), content]
      end

      defimpl Inspect do
        defdelegate inspect(conf, opts), to: unquote(module), as: :__inspex_inspect__
      end
    end
  end
end