xarango

defmodule Xarango.Traversal do

  defstruct [:sort, :direction, :minDepth, :startVertex, :visitor, :itemOrder,
    :strategy, :filter, :init, :maxIterations, :maxDepth, :uniqueness, :graphName, :expander, :edgeCollection]

  import Xarango.Client
  use Xarango.URI, prefix: "traversal"

  def traverse(traversal, database\\nil, options\\[]) do
    url("", database, options)
    |> post(traversal)
    |> Xarango.TraversalResult.to_result
  end

end

defmodule Xarango.TraversalResult do

  defstruct [:vertices, :paths]

  def to_result(data) do
    visited = get_in(data, [:result, :visited])
    result = visited
      |> Map.put(:vertices, Enum.map(visited[:vertices], &Xarango.Vertex.to_vertex(&1)))
      |> Map.put(:paths, Enum.map(visited[:paths], &Xarango.Path.to_path(&1)))
    struct(__MODULE__, result)
  end

  def edges(result) do
    result.paths
    |> Enum.flat_map(fn %{edges: edges} -> edges end)
    |> MapSet.new
    |> MapSet.to_list
  end

end

defmodule Xarango.Path do

  defstruct [:vertices, :edges]

  def to_path(data) do
    path = data
      |> Map.put(:vertices, Enum.map(data[:vertices], &Xarango.Vertex.to_vertex(&1)))
      |> Map.put(:edges, Enum.map(data[:edges], &Xarango.Edge.to_edge(&1)))
    struct(__MODULE__, path)
  end

end