Current section

Files

Jump to
maru lib mix tasks routes.ex
Raw

lib/mix/tasks/routes.ex

defmodule Mix.Tasks.Maru.Routes do
use Mix.Task
def run(args) do
unless System.get_env("MIX_ENV") do
Mix.env(:dev)
end
Mix.Task.run "compile", args
cond do
module = List.first(args) ->
Module.concat("Elixir", module) |> generate_module([])
[] != (servers = Maru.Config.servers) ->
for {module, _} <- servers do
generate_module(module, [])
end
Code.ensure_loaded?(Phoenix) ->
phoenix_module = Module.concat(Mix.Phoenix.base(), "Router")
for %{
__struct__: Phoenix.Router.Route,
kind: :forward,
path: path,
plug: module,
} <- phoenix_module.__routes__ do
try do
prefix = path |> String.split("/", trim: true)
generate_module(module, prefix)
rescue
UndefinedFunctionError -> nil
end
end
true -> nil
end
end
defp generate_module(module, prefix) do
module
|> Maru.Builder.Routers.generate
|> Map.values
|> List.flatten
|> Enum.map(&generate_endpoint(&1, prefix))
|> Enum.map(&IO.puts/1)
end
defp generate_endpoint(ep, prefix) do
format_version(ep.version) <>
format_method(ep.method) <>
format_path(prefix ++ ep.path) <>
" " <> (ep.desc || "")
end
defp format_version(nil), do: "nil" |> format_version
defp format_version(version) do
version |> String.ljust(5)
end
defp format_method({:_, [], nil}), do: ":match" |> format_method
defp format_method(method) do
method |> String.ljust(7)
end
defp format_path([]), do: "/"
defp format_path(p) do
func = fn
i when is_atom(i) -> ":#{i}"
i -> i
end
["" | p] |> Enum.map(func) |> Enum.join("/")
end
end