defmodule TreeSitterHighlight do
@moduledoc """
Elixir binding for https://docs.rs/tree-sitter-highlight/latest/tree_sitter_highlight/
"""
use Rustler, otp_app: :ex_tree_sitter_highlight, crate: "treesitterhighlight"
@doc ~S"""
Renders the given source code into html.
## Examples
iex> TreeSitterHighlight.render_html("1", :elixir)
{:ok, "\n11\n
\n
\n"}
iex> TreeSitterHighlight.render_html("1", :txt)
{:error, :unsupported_language}
"""
def render_html(_source_code, _language_atom), do: :erlang.nif_error(:nif_not_loaded)
@doc ~S"""
Returns a list of supported language atoms.
"""
def get_supported_languages(), do: :erlang.nif_error(:nif_not_loaded)
@doc ~S"""
Returns a language atom for the file, or nil if the language isn't supported.
## Examples
iex> TreeSitterHighlight.get_language_from_filename("/path/file.ex")
:elixir
iex> TreeSitterHighlight.get_language_from_filename("/path/file.txt")
nil
"""
def get_language_from_filename(_filename), do: :erlang.nif_error(:nif_not_loaded)
defp default_stylesheet(), do: "priv/default.css"
@doc ~S"""
Returns CSS content with classes for highlighting and formatting html output.
See priv/default.css for an example of how to create your own stylesheet.
"""
def get_default_css_content() do
default_stylesheet() |> File.read()
end
@doc ~S"""
Writes a complete html document with highlighted code within.
Uses the default stylesheet unless other css_content is provided.
"""
def write_highlighted_file(input_path, output_path, css_content \\ nil)
when is_binary(input_path) and is_binary(output_path) do
with {:ok, source_code} <- File.read(input_path),
{:ok, css_content} <- css_content || get_default_css_content(),
{:ok, html} <-
__MODULE__.render_html(source_code, get_language_from_filename(input_path)),
:ok <-
File.write(output_path, ~s"""
#{html}
""") do
:ok
end
end
end