defmodule BubbleLib.MapUtil do
  @moduledoc """
  Map utility functions
  """

  alias BubbleLib.MapUtil.AutoMap.ETS

  def normalize(value) do
    value
    |> deep_keyword_to_map()
    |> enum_materialize()
    |> deatomize()
    |> drop_nils()
  end

  def deatomize(%{__struct__: _} = struct) do
    struct
    |> Map.keys()
    |> Enum.reduce(struct, fn k, s -> Map.put(s, k, deatomize(Map.get(struct, k))) end)
  end

  def deatomize(%{} = map) do
    map
    |> Enum.map(fn
      {%{__struct__: _} = k, v} -> {k, deatomize(v)}
      {k, v} -> {to_string(k), deatomize(v)}
    end)
    |> Map.new()
  end

  def deatomize(list) when is_list(list) do
    list
    |> Enum.map(&deatomize/1)
  end

  def deatomize(value) do
    value
  end

  def drop_nils(%{__struct__: _} = struct) do
    struct
    |> Map.delete(:__struct__)
    |> drop_nils()
  end

  def drop_nils(%{} = map) do
    map
    |> Enum.filter(fn {_, v} -> v != nil end)
    |> Enum.map(fn {k, v} -> {k, drop_nils(v)} end)
    |> Enum.into(%{})
  end

  def drop_nils(list) when is_list(list) do
    list
    |> Enum.map(&drop_nils/1)
  end

  def drop_nils(value) do
    value
  end

  def deep_keyword_to_map(%{__struct__: _} = value) do
    value
  end

  def deep_keyword_to_map(value) when is_map(value) do
    Enum.map(value, fn {k, v} -> {k, deep_keyword_to_map(v)} end)
    |> Enum.into(%{})
  end

  def deep_keyword_to_map([{_, _} | _] = value) do
    Enum.map(value, fn {k, v} -> {k, deep_keyword_to_map(v)} end)
    |> Enum.into(%{})
  end

  def deep_keyword_to_map(value) when is_list(value) do
    Enum.map(value, fn v -> deep_keyword_to_map(v) end)
  end

  def deep_keyword_to_map(value), do: value

  def enum_materialize(%ETS{} = value) do
    Enum.to_list(value)
  end

  def enum_materialize(%{__struct__: _} = value) do
    value
  end

  def enum_materialize(%{} = map) do
    map
    |> Enum.map(fn {k, v} -> {k, enum_materialize(v)} end)
    |> Map.new()
  end

  def enum_materialize(list) when is_list(list) do
    list
    |> Enum.map(&enum_materialize/1)
  end

  def enum_materialize(value) do
    value
  end
end