towerops/lib/towerops/sites.ex

defmodule Towerops.Sites do
  @moduledoc """
  The Sites context.
  """

  import Ecto.Query

  alias Towerops.Repo
  alias Towerops.Sites.Site

  @doc """
  Returns the list of sites for an organization.
  """
  def list_organization_sites(organization_id) do
    Repo.all(
      from(s in Site, where: s.organization_id == ^organization_id, order_by: [asc: s.name], preload: [:parent_site])
    )
  end

  @doc """
  Returns the list of root sites (sites without a parent) for an organization.
  """
  def list_root_sites(organization_id) do
    Repo.all(
      from(s in Site,
        where: s.organization_id == ^organization_id,
        where: is_nil(s.parent_site_id),
        order_by: [asc: s.name]
      )
    )
  end

  @doc """
  Returns the list of child sites for a parent site.
  """
  def list_child_sites(parent_site_id) do
    Repo.all(from(s in Site, where: s.parent_site_id == ^parent_site_id, order_by: [asc: s.name]))
  end

  @doc """
  Gets a single site.
  """
  def get_site!(id) do
    Site
    |> Repo.get!(id)
    |> Repo.preload([:parent_site, :child_sites, :equipment])
  end

  @doc """
  Gets a single site belonging to an organization.
  """
  def get_organization_site!(organization_id, site_id) do
    Repo.one!(
      from(s in Site,
        where: s.id == ^site_id,
        where: s.organization_id == ^organization_id,
        preload: [:parent_site, :child_sites, :equipment]
      )
    )
  end

  @doc """
  Creates a site.
  """
  def create_site(attrs) do
    %Site{}
    |> Site.changeset(attrs)
    |> Repo.insert()
  end

  @doc """
  Updates a site.
  """
  def update_site(%Site{} = site, attrs) do
    site
    |> Site.changeset(attrs)
    |> Repo.update()
  end

  @doc """
  Deletes a site.
  """
  def delete_site(%Site{} = site) do
    Repo.delete(site)
  end

  @doc """
  Returns an `%Ecto.Changeset{}` for tracking site changes.
  """
  def change_site(%Site{} = site, attrs \\ %{}) do
    Site.changeset(site, attrs)
  end

  @doc """
  Builds a hierarchical tree structure of sites.
  """
  def build_site_tree(organization_id) do
    sites = list_organization_sites(organization_id)
    root_sites = Enum.filter(sites, &is_nil(&1.parent_site_id))

    Enum.map(root_sites, fn root ->
      build_tree_node(root, sites)
    end)
  end

  defp build_tree_node(site, all_sites) do
    children =
      all_sites
      |> Enum.filter(&(&1.parent_site_id == site.id))
      |> Enum.map(&build_tree_node(&1, all_sites))

    %{site: site, children: children}
  end
end