Packages

rustq

0.3.0
1.0.0-rc.1 0.11.1 0.11.0 0.10.0 0.9.11 0.9.10 0.9.9 0.9.8 0.9.7 0.9.6 0.9.5 0.9.4 0.9.3 0.9.2 0.9.1 0.9.0 0.8.3 0.8.2 0.8.1 0.8.0 0.7.0 0.6.0 0.5.1 0.5.0 0.4.0 0.3.0 0.2.2 0.2.1 0.2.0 0.1.2 0.1.1 0.1.0

Rust templates and quasiquoting for Elixir

HexDocs

Current section

Files

Jump to
Readme
32 Versions
1 Dependency
1 Dependant
Files
Activity
rustq lib rustq rustler atoms.ex
Raw

lib/rustq/rustler/atoms.ex 

defmodule RustQ.Rustler.Atoms do

  @moduledoc false



  alias RustQ.Rust



  @spec build([atom() | String.t() | {atom() | String.t(), String.t()}], keyword()) ::

          Rust.Fragment.t()

  def build(atoms, opts \\ []) do

    body =

      atoms

      |> Enum.map(&atom_decl/1)

      |> Enum.map_join("\n", fn decl -> "        #{decl}" end)



    module = Keyword.get(opts, :module, :atoms)

    code = ["rustler::atoms! {\n", body, "\n}"]



    case module do

      false -> Rust.item(code)

      module -> Rust.item(["mod ", to_string(module), " {\n", indent(code, 4), "\n}"])

    end

  end



  defp indent(iodata, spaces) do

    padding = String.duplicate(" ", spaces)



    iodata

    |> IO.iodata_to_binary()

    |> String.split("\n")

    |> Enum.map_join("\n", fn

      "" -> ""

      line -> padding <> line

    end)

  end



  defp atom_decl(name) when is_atom(name), do: "#{name},"

  defp atom_decl(name) when is_binary(name), do: "#{name},"



  defp atom_decl({name, value}) when (is_atom(name) or is_binary(name)) and is_binary(value),

    do: ~s|#{name} = "#{value}",|

end