|
| 1 | +defmodule TypedStructor.GuideCase do |
| 2 | + @moduledoc false |
| 3 | + |
| 4 | + use ExUnit.CaseTemplate |
| 5 | + |
| 6 | + using opts do |
| 7 | + guide = Keyword.fetch!(opts, :guide) |
| 8 | + |
| 9 | + file = Path.expand([__DIR__, "../../../", "guides/plugins/", guide]) |
| 10 | + |
| 11 | + ast = Code.string_to_quoted!(extract_code(file)) |
| 12 | + |
| 13 | + quote do |
| 14 | + Code.compiler_options(debug_info: true) |
| 15 | + |
| 16 | + {:module, _module_name, bytecode, _submodule} = unquote(ast) |
| 17 | + |
| 18 | + ExUnit.Case.register_module_attribute(__MODULE__, :bytecode) |
| 19 | + @bytecode bytecode |
| 20 | + |
| 21 | + import unquote(__MODULE__), only: [types: 1] |
| 22 | + end |
| 23 | + end |
| 24 | + |
| 25 | + def types(bytecode) when is_binary(bytecode) do |
| 26 | + TypedStructor.TypeCase.types(bytecode) <> "\n" |
| 27 | + end |
| 28 | + |
| 29 | + defp extract_code(file) do |
| 30 | + content = |
| 31 | + File.read!(file) |
| 32 | + |
| 33 | + content |
| 34 | + |> String.split("\n") |
| 35 | + |> Enum.with_index(1) |
| 36 | + |> extract_code(%{in_code_block?: false, codes: []}) |
| 37 | + |> case do |
| 38 | + %{codes: [implementation, usage | _resut]} -> |
| 39 | + """ |
| 40 | + #{implementation |> Enum.reverse() |> Enum.join("\n")} |
| 41 | + #{usage |> Enum.reverse() |> Enum.join("\n")} |
| 42 | + """ |
| 43 | + |
| 44 | + ctx -> |
| 45 | + raise ArgumentError, |
| 46 | + """ |
| 47 | + Cannot find implementation and usage in file #{inspect(file)} |
| 48 | + context: #{inspect(ctx)} |
| 49 | + #{content} |
| 50 | + """ |
| 51 | + end |
| 52 | + end |
| 53 | + |
| 54 | + @start "```elixir" |
| 55 | + @stop "```" |
| 56 | + |
| 57 | + defp extract_code([], %{in_code_block?: false} = ctx) do |
| 58 | + %{ctx | codes: Enum.reverse(ctx.codes)} |
| 59 | + end |
| 60 | + |
| 61 | + defp extract_code([{@start, _index} | rest], %{in_code_block?: false} = ctx) do |
| 62 | + extract_code(rest, %{ctx | in_code_block?: true, codes: [[] | ctx.codes]}) |
| 63 | + end |
| 64 | + |
| 65 | + defp extract_code([{@stop, _index} | rest], %{in_code_block?: true} = ctx) do |
| 66 | + extract_code(rest, %{ctx | in_code_block?: false}) |
| 67 | + end |
| 68 | + |
| 69 | + defp extract_code([{text, _index} | rest], %{in_code_block?: true} = ctx) do |
| 70 | + [current | codes] = ctx.codes |
| 71 | + extract_code(rest, %{ctx | codes: [[text | current] | codes]}) |
| 72 | + end |
| 73 | + |
| 74 | + defp extract_code([_text | rest], %{in_code_block?: false} = ctx) do |
| 75 | + extract_code(rest, ctx) |
| 76 | + end |
| 77 | + |
| 78 | + defp extract_code([{text, index} | _rest], ctx) do |
| 79 | + raise ArgumentError, |
| 80 | + """ |
| 81 | + Unexpected text at line #{index}: #{inspect(text)} |
| 82 | + context: #{inspect(ctx)} |
| 83 | + """ |
| 84 | + end |
| 85 | + |
| 86 | + defp extract_code([], ctx) do |
| 87 | + raise ArgumentError, |
| 88 | + """ |
| 89 | + Code block not closed |
| 90 | + context: #{inspect(ctx)} |
| 91 | + """ |
| 92 | + end |
| 93 | +end |
0 commit comments